teichmann@5863: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde teichmann@5863: * Software engineering by Intevation GmbH teichmann@5863: * teichmann@5994: * This file is Free Software under the GNU AGPL (>=v3) teichmann@5863: * and comes with ABSOLUTELY NO WARRANTY! Check out the teichmann@5994: * documentation coming with Dive4Elements River for details. teichmann@5863: */ teichmann@5863: teichmann@5831: package org.dive4elements.river.artifacts.model.sq; ingo@3072: teichmann@5831: import org.dive4elements.river.artifacts.access.SQRelationAccess; sascha@3304: teichmann@5831: import org.dive4elements.river.artifacts.math.fitting.Function; teichmann@5831: import org.dive4elements.river.artifacts.math.fitting.FunctionFactory; ingo@3072: teichmann@5831: import org.dive4elements.river.artifacts.model.Calculation; teichmann@5831: import org.dive4elements.river.artifacts.model.CalculationResult; teichmann@5831: import org.dive4elements.river.artifacts.model.DateRange; teichmann@5831: import org.dive4elements.river.artifacts.model.Parameters; teichmann@5831: teichmann@5831: import org.dive4elements.river.backend.SedDBSessionHolder; sascha@3289: sascha@3552: import java.util.ArrayList; 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; rrenkert@5396: private String method; 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(); sascha@3552: DateRange period = access.getPeriod(); ingo@3426: Double outliers = access.getOutliers(); rrenkert@5396: String method = access.getOutlierMethod(); 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: rrenkert@5396: if (method == null) { rrenkert@5396: //TODO: i18n rrenkert@5396: addProblem("sq.missing.method"); rrenkert@5396: } rrenkert@5396: sascha@3222: if (!hasProblems()) { sascha@3222: this.river = river; sascha@3222: this.location = location; ingo@3426: this.period = period; sascha@3222: this.outliers = outliers; rrenkert@5396: this.method = method; 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@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: teichmann@6780: SQ.View sqView = SQ.SQ_VIEW; teichmann@6780: SQ.Factory sqFactory = SQ.SQ_FACTORY; teichmann@6780: sascha@3552: Measurements measurements = teichmann@6780: MeasurementFactory.getMeasurements( teichmann@6780: river, location, period, sqFactory); 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 sqs = measurements.getSQs(i); sascha@3310: ingo@3426: SQFractionResult fractionResult; sascha@3310: sascha@3552: List iterations = teichmann@6780: doFitting(function, sqs, sqView); sascha@3552: sascha@3552: if (iterations == null) { ingo@3426: // TODO: i18n ingo@3426: addProblem("sq.fitting.failed." + i); ingo@3426: fractionResult = new SQFractionResult(); ingo@3426: } ingo@3426: else { sascha@3552: fractionResult = new SQFractionResult( sascha@3552: sqs.toArray(new SQ[sqs.size()]), sascha@3552: iterations); sascha@3552: } sascha@3310: 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@3552: protected List doFitting( sascha@3552: final Function function, teichmann@6780: List sqs, teichmann@6780: SQ.View sqView sascha@3552: ) { sascha@3552: final List iterations = sascha@3552: new ArrayList(); sascha@3304: teichmann@6780: boolean success = new Fitting(function, outliers, sqView).fit( sascha@3552: sqs, rrenkert@5396: method, sascha@3552: new Fitting.Callback() { sascha@3552: @Override sascha@3552: public void afterIteration( sascha@3552: double [] coeffs, sascha@3552: SQ [] measurements, sascha@3552: SQ [] outliers, sascha@3552: double standardDeviation, sascha@3552: double chiSqr sascha@3552: ) { sascha@3552: Parameters parameters = createParameters( sascha@3552: function.getParameterNames(), sascha@3552: coeffs, sascha@3552: standardDeviation, sascha@3552: chiSqr); sascha@3552: iterations.add(new SQFractionResult.Iteration( sascha@3552: parameters, sascha@3552: measurements, sascha@3552: outliers)); sascha@3552: } sascha@3552: }); sascha@3552: sascha@3552: return success ? iterations : null; sascha@3552: } sascha@3552: sascha@3552: public static final Parameters createParameters( sascha@3552: String [] names, sascha@3552: double [] values, sascha@3552: double standardDeviation, sascha@3552: double chiSqr sascha@3552: ) { 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@3552: Parameters parameters = new Parameters(columns); sascha@3552: int row = parameters.newRow(); sascha@3552: parameters.set(row, names, values); sascha@3552: parameters.set(row, "chi_sqr", chiSqr); sascha@3552: parameters.set(row, "std_dev", standardDeviation); sascha@3552: return parameters; sascha@3304: } ingo@3072: } ingo@3072: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :