ingo@3072: package de.intevation.flys.artifacts.model.sq;
ingo@3072: 
sascha@3222: import de.intevation.flys.artifacts.access.SQRelationAccess;
ingo@3072: 
sascha@3304: import de.intevation.flys.artifacts.math.fitting.Function;
sascha@3304: import de.intevation.flys.artifacts.math.fitting.FunctionFactory;
sascha@3304: 
ingo@3072: import de.intevation.flys.artifacts.model.Calculation;
ingo@3072: import de.intevation.flys.artifacts.model.CalculationResult;
sascha@3222: import de.intevation.flys.artifacts.model.DateRange;
sascha@3304: import de.intevation.flys.artifacts.model.Parameters;
ingo@3072: 
sascha@3289: import de.intevation.flys.backend.SedDBSessionHolder;
sascha@3289: 
sascha@3304: import java.util.List;
sascha@3304: 
sascha@3222: import org.apache.log4j.Logger;
ingo@3072: 
ingo@3072: public class SQRelationCalculation extends Calculation {
ingo@3072: 
sascha@3222:     private static final Logger log =
ingo@3072:         Logger.getLogger(SQRelationCalculation.class);
ingo@3072: 
sascha@3304:     public static final String SQ_FUNCTION_NAME = "sq-pow";
sascha@3304: 
ingo@3426:     protected String    river;
ingo@3426:     protected double    location;
ingo@3426:     protected DateRange period;
ingo@3426:     protected double    outliers;
sascha@3305: 
sascha@3222:     public SQRelationCalculation() {
ingo@3079:     }
ingo@3079: 
sascha@3222:     public SQRelationCalculation(SQRelationAccess access) {
ingo@3079: 
ingo@3426:         String    river    = access.getRiver();
ingo@3426:         Double    location = access.getLocation();
ingo@3426:         DateRange period  = access.getPeriod();
ingo@3426:         Double    outliers = access.getOutliers();
ingo@3101: 
sascha@3390:         //river = "Rhein";
sascha@3390: 
sascha@3222:         if (river == null) {
sascha@3222:             // TODO: i18n
sascha@3222:             addProblem("sq.missing.river");
sascha@3222:         }
ingo@3101: 
sascha@3222:         if (location == null) {
sascha@3222:             // TODO: i18n
sascha@3222:             addProblem("sq.missing.location");
sascha@3222:         }
ingo@3101: 
ingo@3426:         if (period == null) {
sascha@3222:             // TODO: i18n
sascha@3222:             addProblem("sq.missing.periods");
sascha@3222:         }
sascha@3222: 
sascha@3222:         if (outliers == null) {
sascha@3222:             // TODO: i18n
sascha@3222:             addProblem("sq.missing.outliers");
sascha@3222:         }
sascha@3222: 
sascha@3222:         if (!hasProblems()) {
sascha@3222:             this.river    = river;
sascha@3222:             this.location = location;
ingo@3426:             this.period   = period;
sascha@3222:             this.outliers = outliers;
sascha@3222:         }
ingo@3079:     }
ingo@3079: 
ingo@3105: 
sascha@3222:     public CalculationResult calculate() {
sascha@3222:         log.debug("SQRelationCalculation.calculate");
ingo@3079: 
sascha@3222:         if (hasProblems()) {
sascha@3222:             return new CalculationResult(this);
ingo@3118:         }
ingo@3079: 
sascha@3289:         SedDBSessionHolder.acquire();
sascha@3289:         try {
sascha@3289:             return internalCalculate();
sascha@3289:         }
sascha@3289:         finally {
sascha@3289:             SedDBSessionHolder.release();
sascha@3289:         }
sascha@3289:     }
sascha@3289: 
sascha@3289:     protected CalculationResult internalCalculate() {
sascha@3289: 
sascha@3289:         boolean debug = log.isDebugEnabled();
sascha@3289: 
sascha@3304:         Function function = FunctionFactory
sascha@3304:             .getInstance()
sascha@3304:             .getFunction(SQ_FUNCTION_NAME);
sascha@3304: 
sascha@3304:         if (function == null) {
sascha@3304:             log.error("No '" + SQ_FUNCTION_NAME + "' function found.");
sascha@3304:             // TODO: i18n
sascha@3304:             addProblem("sq.missing.sq.function");
sascha@3304:         }
sascha@3304: 
sascha@3222: 
sascha@3304:         String [] parameterNames = function.getParameterNames();
sascha@3304: 
sascha@3310:         Fitting fitting = new Fitting(function, outliers);
sascha@3304: 
sascha@3310:             Measurements measurements =
sascha@3310:                 MeasurementFactory.getMeasurements(river, location, period);
sascha@3310: 
ingo@3426:         SQFractionResult [] fractionResults =
ingo@3426:             new SQFractionResult[SQResult.NUMBER_FRACTIONS];
sascha@3304: 
ingo@3426:         for (int i = 0; i < fractionResults.length; ++i) {
ingo@3426:             List<SQ> sqs = measurements.getSQs(i);
sascha@3310: 
ingo@3426:             SQFractionResult fractionResult;
sascha@3310: 
ingo@3426:             if (!fitting.fit(sqs)) {
ingo@3426:                 // TODO: i18n
ingo@3426:                 addProblem("sq.fitting.failed." + i);
ingo@3426:                 fractionResult = new SQFractionResult();
ingo@3426:             }
ingo@3426:             else {
ingo@3426:                 Parameters parameters = createParameters(parameterNames);
ingo@3426:                 int row = parameters.newRow();
ingo@3426:                 double [] coeffs = fitting.getParameters();
ingo@3426:                 for (int j = 0; j < parameterNames.length; ++j) {
ingo@3426:                     parameters.set(row, parameterNames[j], coeffs[j]);
sascha@3310:                 }
ingo@3426:                 parameters.set(row, "chi_sqr", fitting.getChiSqr());
ingo@3426:                 parameters.set(row, "std_dev", fitting.getStandardDeviation());
sascha@3310: 
ingo@3426:                 fractionResult = new SQFractionResult(
ingo@3426:                     parameters,
ingo@3426:                     fitting.getRemaining(),
ingo@3426:                     fitting.getOutliers());
sascha@3310:             }
ingo@3426:             fitting.reset();
ingo@3426:             fractionResults[i] = fractionResult;
sascha@3297:         }
sascha@3289: 
sascha@3310:         return new CalculationResult(
ingo@3426:             new SQResult[] { new SQResult(location, fractionResults) },
sascha@3310:             this);
ingo@3072:     }
sascha@3304: 
sascha@3304:     public static final Parameters createParameters(String [] names) {
sascha@3304: 
sascha@3304:         String [] columns = new String[names.length + 2];
sascha@3304:         columns[0] = "chi_sqr";
sascha@3304:         columns[1] = "std_dev";
sascha@3304:         System.arraycopy(names, 0, columns, 2, names.length);
sascha@3304:         return new Parameters(columns);
sascha@3304:     }
ingo@3072: }
ingo@3072: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :