dive4elements/river: artifacts/src/main/java/org/dive4elements/river/artifacts/model/sq/SQRelationCalculation.java comparison

comparison artifacts/src/main/java/org/dive4elements/river/artifacts/model/sq/SQRelationCalculation.java @ 6796:978ab716a15e

flys/issue1347: Added missing calcutions. TODO: Bring them into the generated outs.

author	Sascha L. Teichmann <teichmann@intevation.de>
date	Fri, 09 Aug 2013 18:24:55 +0200
parents	51eb6491c537
children	e237a83fd87d

comparison

equal deleted inserted replaced

-:0747ca95ad6e
+:978ab716a15e
 public static final boolean NON_LINEAR_FITTING =
 Boolean.getBoolean("minfo.sq.calcution.non.linear.fitting");
 public static final String SQ_POW_FUNCTION_NAME = "sq-pow";
 public static final String SQ_LIN_FUNCTION_NAME = "linear";
+public static final String [] EXTRA_PARAMETERS = {
+"chi_sqr",
+"std_dev",
+"max_q",
+"c_perguson",
+"c_duan",
+"r2"
+};
 protected String    river;
 protected double    location;
 protected DateRange period;
 protected double    outliers;
 Function         function;
 SQ.View          sqView;
 SQ.Factory       sqFactory;
 ParameterCreator pc;
 if (NON_LINEAR_FITTING) {
 log.debug("Use non linear fitting.");
 sqView    = SQ.SQ_VIEW;
 sqFactory = SQ.SQ_FACTORY;
 function  = powFunction;
 pc = new ParameterCreator(
 powFunction.getParameterNames(),
-powFunction.getParameterNames());
+powFunction.getParameterNames(),
+powFunction,
+sqView);
 }
 else {
 log.debug("Use linear fitting.");
 sqView    = LogSQ.LOG_SQ_VIEW;
 sqFactory = LogSQ.LOG_SQ_FACTORY;
 addProblem("sq.missing.sq.function");
 return new CalculationResult(new SQResult[0], this);
 }
 pc = new LinearParameterCreator(
 powFunction.getParameterNames(),
-function.getParameterNames());
+function.getParameterNames(),
+function,
+sqView);
 }
 Measurements measurements =
 MeasurementFactory.getMeasurements(
 river, location, period, sqFactory);
 double    chiSqr
 ) {
 Parameters parameters = pc.createParameters(
 coeffs,
 standardDeviation,
-chiSqr);
+chiSqr,
+measurements);
 iterations.add(new SQFractionResult.Iteration(
 parameters,
 measurements,
 outliers));
 }
 public static class ParameterCreator {
 protected String [] origNames;
 protected String [] proxyNames;
-public ParameterCreator(String [] origNames, String [] proxyNames) {
+protected Function  function;
+protected SQ.View   view;
+public ParameterCreator(
+String [] origNames,
+String [] proxyNames,
+Function  function,
+SQ.View   view
+) {
 this.origNames  = origNames;
 this.proxyNames = proxyNames;
+this.view       = view;
 }
 protected double [] transformCoeffs(double [] coeffs) {
 return coeffs;
 }
+private static double maxQ(SQ [] sqs) {
+double max = -Double.MAX_VALUE;
+for (SQ sq: sqs) {
+double q = sq.getQ(); // Don't use view here!
+if (q > max) {
+max = q;
+}
+}
+return Math.max(0d, max);
+}
+private double cPerguson(
+org.dive4elements.river.artifacts.math.Function instance,
+SQ [] sqs
+) {
+double sqrSum = 0d;
+for (SQ sq: sqs) {
+double s = view.getS(sq);
+double q = view.getQ(sq);
+double diffS = s - instance.value(q);
+sqrSum += diffS*diffS;
+}
+return Math.exp(0.5d * sqrSum/(sqs.length-2));
+}
+private double cDuan(
+org.dive4elements.river.artifacts.math.Function instance,
+SQ [] sqs
+) {
+double sum = 0d;
+for (SQ sq: sqs) {
+double s = view.getS(sq);
+double q = view.getQ(sq);
+double diffS = s - instance.value(q);
+sum += Math.exp(diffS);
+}
+return sum / sqs.length;
+}
+private double r2(
+org.dive4elements.river.artifacts.math.Function instance,
+SQ [] sqs
+) {
+double xm = 0;
+double ym = 0;
+for (SQ sq: sqs) {
+double s  = view.getS(sq);
+double q  = view.getQ(sq);
+double fs = instance.value(q);
+xm += s;
+ym += fs;
+}
+xm /= sqs.length;
+ym /= sqs.length;
+double mixXY = 0d;
+double sumX = 0d;
+double sumY = 0d;
+for (SQ sq: sqs) {
+double s  = view.getS(sq);
+double q  = view.getQ(sq);
+double fs = instance.value(q);
+double xDiff = xm - s;
+double yDiff = ym - fs;
+mixXY += xDiff*yDiff;
+sumX += xDiff*xDiff;
+sumY *= yDiff*yDiff;
+}
+double r = mixXY/Math.sqrt(sumX*sumY);
+return r*r;
+}
 public Parameters createParameters(
 double [] coeffs,
 double    standardDeviation,
-double    chiSqr
+double    chiSqr,
-) {
+SQ []     measurements
-String [] columns = new String[origNames.length + 2];
+) {
-columns[0] = "chi_sqr";
+String [] columns = StringUtils.join(EXTRA_PARAMETERS, origNames);
-columns[1] = "std_dev";
-System.arraycopy(origNames, 0, columns, 2, origNames.length);
 Parameters parameters = new Parameters(columns);
 int row = parameters.newRow();
 parameters.set(row, origNames, transformCoeffs(coeffs));
 parameters.set(row, "chi_sqr", chiSqr);
 parameters.set(row, "std_dev", standardDeviation);
+parameters.set(row, "max_q", maxQ(measurements));
+// We need to instantiate the function to calculate
+// the remaining values.
+org.dive4elements.river.artifacts.math.Function f =
+function.instantiate(coeffs);
+parameters.set(row, "c_perguson", cPerguson(f, measurements));
+parameters.set(row, "c_duan", cDuan(f, measurements));
+parameters.set(row, "r2", r2(f, measurements));
 return parameters;
 }
 }
 /** We need to transform the coeffs back to the original function. */
 public static class LinearParameterCreator extends ParameterCreator {
 public LinearParameterCreator(
 String [] origNames,
-String [] proxyNames
+String [] proxyNames,
-) {
+Function  function,
-super(origNames, proxyNames);
+SQ.View   view
+) {
+super(origNames, proxyNames, function, view);
 }
 @Override
 protected double [] transformCoeffs(double [] coeffs) {

Mercurial > dive4elements > river

comparison artifacts/src/main/java/org/dive4elements/river/artifacts/model/sq/SQRelationCalculation.java @ 6796:978ab716a15e