dive4elements/river: flys-artifacts/src/main/java/de/intevation/flys/artifacts/model/WstValueTable.java comparison

comparison flys-artifacts/src/main/java/de/intevation/flys/artifacts/model/WstValueTable.java @ 339:4509ba8fae68

Implementation of "Abflusskurve/Abflusstafel" calculation. flys-artifacts/trunk@1738 c6561f87-3c4e-4783-a992-168aeb5c3f6f

author	Sascha L. Teichmann <sascha.teichmann@intevation.de>
date	Wed, 20 Apr 2011 14:26:51 +0000
parents	64cfbd631f29
children	79401797f4e1

comparison

equal deleted inserted replaced

-:cf84f0f926e9
+:4509ba8fae68
 import java.util.Comparator;
 import java.util.List;
 import java.util.Collections;
 import java.util.Iterator;
+import org.apache.log4j.Logger;
+import org.apache.commons.math.analysis.interpolation.SplineInterpolator;
+import org.apache.commons.math.analysis.polynomials.PolynomialSplineFunction;
+import org.apache.commons.math.ArgumentOutsideDomainException;
 import org.hibernate.Session;
 import org.hibernate.Query;
 import org.hibernate.SQLQuery;
 import org.hibernate.type.StandardBasicTypes;
 public class WstValueTable
 implements   Serializable
 {
+private static Logger log = Logger.getLogger(WstValueTable.class);
 // TODO: put this into a property file
 public static final String SQL_POS_WQ =
 "SELECT position, w, q, column_pos" +
 "    FROM wst_value_table"          +
 "    WHERE wst_id = :wst_id";
+public static final double DEFAULT_STEP_WIDTH = 0.01;
 public static class Column
 implements          Serializable
 {
 protected String name;
 double d = km - other.km;
 if (d < 0.0) return -1;
 if (d > 0.0) return +1;
 return 0;
 }
+public double [][] cloneWQs() {
+return new double [][] {
+(double [])ws.clone(),
+(double [])ws.clone() };
+}
+public double [][] interpolateWQ(Row other, double km, double stepWidth) {
+int W1 =       ascendingWs();
+int W2 = other.ascendingWs();
+int W = Math.min(W1, W2);
+if (W < 1) {
+return new double[2][0];
+}
+double factor = factor(km, this.km, other.km);
+double minW = weight(factor, ws[0],   other.ws[0]);
+double maxW = weight(factor, ws[W-1], other.ws[W-1]);
+if (minW > maxW) {
+double t = minW;
+minW = maxW;
+maxW = t;
+}
+double [] x = new double[W];
+double [] y = new double[W];
+for (int i = 0; i < W; ++i) {
+x[i] = weight(factor, ws[i], other.ws[i]);
+y[i] = weight(factor, qs[i], other.qs[i]);
+}
+SplineInterpolator interpolator = new SplineInterpolator();
+PolynomialSplineFunction spline = interpolator.interpolate(x, y);
+double [] outWs =
+new double[(int)Math.ceil((maxW - minW)/stepWidth)];
+double [] outQs =
+new double[outWs.length];
+try {
+double w = minW;
+for (int i = 0; i < outWs.length; ++i, w += stepWidth) {
+outQs[i] = spline.value(outWs[i] = w);
+}
+}
+catch (ArgumentOutsideDomainException aode) {
+log.error("Spline interpolation failed.", aode);
+}
+return new double [][] { outWs, outQs };
+}
+public double [][] interpolateWQ(double stepWidth) {
+int W = ascendingWs(); // ignore back jumps
+if (W < 1) {
+return new double[2][0];
+}
+double [] x = new double[W];
+double [] y = new double[W];
+for (int i = 0; i < W; ++i) {
+x[i] = ws[i];
+y[i] = qs[i];
+}
+SplineInterpolator interpolator = new SplineInterpolator();
+PolynomialSplineFunction spline = interpolator.interpolate(x, y);
+double minW = ws[0];
+double maxW = ws[W-1];
+double [] outWs =
+new double[(int)Math.ceil((maxW - minW)/stepWidth)];
+double [] outQs =
+new double[outWs.length];
+try {
+double w = minW;
+for (int i = 0; i < outWs.length; ++i, w += stepWidth) {
+outQs[i] = spline.value(outWs[i] = w);
+}
+}
+catch (ArgumentOutsideDomainException aode) {
+log.error("Spline interpolation failed.", aode);
+}
+return new double [][] { outWs, outQs };
+}
+public int ascendingWs() {
+if (ws.length < 2) {
+return ws.length;
+}
+int idx = 1;
+for (; idx < ws.length; ++idx) {
+if (Double.isNaN(ws[idx]) || ws[idx] < ws[idx-1]) {
+return idx;
+}
+}
+return idx;
+}
+public double [][] weightWQs(Row other, double km) {
+int W = Math.min(ws.length, other.ws.length);
+double factor = factor(km, this.km, other.km);
+double [] outWs = new double[W];
+double [] outQs = new double[W];
+for (int i = 0; i < W; ++i) {
+outWs[i] = weight(factor, ws[i], other.ws[i]);
+outQs[i] = weight(factor, qs[i], other.qs[i]);
+}
+return new double [][] { outWs, outQs };
+}
 }
 // class Row
 protected List<Row> rows;
 km, qs, ws);
 }
 }
 return ws;
+}
+public double [][] interpolateWQ(double km) {
+return interpolateWQ(km, DEFAULT_STEP_WIDTH, true);
+}
+public double [][] interpolateWQ(
+double  km,
+double  stepWidth,
+boolean checkAscending
+) {
+int rowIndex = Collections.binarySearch(rows, new Row(km));
+if (rowIndex >= 0) { // direct row match
+Row row = rows.get(rowIndex);
+return checkAscending
+? row.interpolateWQ(stepWidth)
+: row.cloneWQs();
+}
+rowIndex = -rowIndex -1;
+if (rowIndex < 1 || rowIndex >= rows.size()) {
+// do not extrapolate
+return new double[2][0];
+}
+Row r1 = rows.get(rowIndex-1);
+Row r2 = rows.get(rowIndex);
+return checkAscending
+? r1.interpolateWQ(r2, km, stepWidth)
+: r1.weightWQs(r2, km);
 }
 public static WstValueTable getTable(River river) {
 return getTable(river, 0);
 }

Mercurial > dive4elements > river

comparison flys-artifacts/src/main/java/de/intevation/flys/artifacts/model/WstValueTable.java @ 339:4509ba8fae68