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

comparison flys-artifacts/src/main/java/de/intevation/flys/artifacts/WINFOArtifact.java @ 451:73bc64c4a7b0

Use new logic to calculate "W für ungleichwertige Abfluesse". Not working yet. flys-artifacts/trunk@1946 c6561f87-3c4e-4783-a992-168aeb5c3f6f

author	Sascha L. Teichmann <sascha.teichmann@intevation.de>
date	Wed, 18 May 2011 17:37:06 +0000
parents	c8bb38115290
children	343f248e4c8c

comparison

equal deleted inserted replaced

-:c8bb38115290
+:73bc64c4a7b0
 import de.intevation.flys.artifacts.model.MainValuesFactory;
 import de.intevation.flys.artifacts.model.WQCKms;
 import de.intevation.flys.artifacts.model.WQDay;
 import de.intevation.flys.artifacts.model.WQKms;
 import de.intevation.flys.artifacts.model.WstValueTable;
+import de.intevation.flys.artifacts.model.WstValueTable.QPosition;
 import de.intevation.flys.artifacts.model.WstValueTableFactory;
+import de.intevation.flys.artifacts.math.LinearRemap;
+import gnu.trove.TDoubleArrayList;
 /**
 * The default WINFO artifact.
 *
 * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
 }
 return wqkms;
 }
 /**
 * Returns the data computed by the discharge longitudinal section
 * computation.
 *
 * @return an array of WQKms object - one object for each given Q value.
 */
-public WQKms[] getDischargeLongitudinalSectionData() {
+public WQKms [] getDischargeLongitudinalSectionData() {
 logger.debug("WINFOArtifact.getDischargeLongitudinalSectionData");
 River river = getRiver();
 if (river == null) {
 logger.error("No river selected.");
+return new WQKms[0];
 }
 WstValueTable wst = WstValueTableFactory.getTable(river);
 if (wst == null) {
-logger.error("No Wst found for selected river.");
+logger.error("No wst found for selected river.");
-}
+return new WQKms[0];
+}
-double[][] dist = getSplittedDistance();
-int        num  = dist != null ? dist.length : 0;
+double [][] segments = getSplittedDistance();
-double[][] allQs = new double[num][];
+if (segments.length < 1) {
-double[][] allWs = new double[num][];
+logger.warn("no segments given");
-WQKms[][]  wqkms = new WQKms[num][];
+return new WQKms[0];
+}
-boolean qSel = true;
+if (segments.length == 1) {
-for (int i = 0; i < num; i++) {
+// fall back to normal "Wasserstand/Wasserspiegellage" calculation
-double[] kms = getKms(dist[i]);
+double [] qs = toQs(segments[0]);
-if (kms == null) {
+if (qs == null) {
-// XXX maybe we should cancel this operation here.
+logger.warn("no qs given");
+return new WQKms[0];
+}
+if (qs.length == 1) {
+double [] kms = getKms(segments[0]);
+return computeWaterlevelData(kms, qs, wst);
+}
+}
+// more than one segment
+double [] boundKms;
+if (segments.length == 2) {
+boundKms = new double [] { segments[0][0], segments[1][1] };
+}
+else {
+TDoubleArrayList bounds = new TDoubleArrayList();
+bounds.add(segments[0][0]);
+for (int i = 1; i < segments.length-1; ++i) {
+double [] segment = segments[i];
+Gauge gauge = river.determineGauge(segment[0], segment[1]);
+if (gauge == null) {
+logger.warn("no gauge found between " +
+segment[0] + " and " + segment[1]);
+bounds.add(0.5*(segment[0] + segment[1]));
+}
+else {
+bounds.add(gauge.getStation().doubleValue());
+}
+}
+bounds.add(segments[segments.length-1][1]);
+boundKms = bounds.toNativeArray();
+}
+if (logger.isDebugEnabled()) {
+logger.debug("bound kms: " + joinDoubles(boundKms));
+}
+double [][] iqs = null;
+for (int i = 0; i < segments.length; ++i) {
+double [] iqsi = toQs(segments[i]);
+if (iqsi == null) {
+logger.warn("iqsi == null");
+return new WQKms[0];
+}
+if (iqs == null) {
+iqs = new double[iqsi.length][boundKms.length];
+}
+else if (iqs.length != iqsi.length) {
+logger.warn("iqsi.logger != iqs.length: "
++ iqsi.length + " " + iqsi.length);
+return new WQKms[0];
+}
+if (logger.isDebugEnabled()) {
+logger.debug("segments qs[ " + i + "]: " + joinDoubles(iqsi));
+}
+for (int j = 0; j < iqs.length; ++j) {
+iqs[j][i] = iqsi[j];
+}
+}
+if (logger.isDebugEnabled()) {
+for (int i = 0; i < iqs.length; ++i) {
+logger.debug("iqs[" + i + "]: " + joinDoubles(iqs[i]));
+}
+}
+double [] boundWs = new double[boundKms.length];
+double [] boundQs = new double[boundKms.length];
+// XXX: Is there some state missing?
+double [] okms = getExplodedValues(
+boundKms[0], boundKms[boundKms.length-1], DEFAULT_KM_STEPS);
+ArrayList<WQKms> results = new ArrayList<WQKms>();
+for (int i = 0; i < iqs.length; ++i) {
+double [] iqsi = iqs[i];
+QPosition qPosition = wst.interpolate(
+iqsi[0], 0, boundKms, boundWs, boundQs);
+if (qPosition == null) {
+logger.warn("interpolation failed for " + iqsi[i]);
 continue;
 }
-double[] qs = getQs(dist[i]);
+LinearRemap remap = new LinearRemap();
-allQs[i]    = qs;
+for (int j = 1; j < boundKms.length; ++j) {
-if (qs == null) {
+remap.add(
-logger.debug("Determine Q values based on a set of W values.");
+boundKms[j-1], boundKms[j],
-qSel = false;
+boundQs[j-1],  iqsi[j-1],
+boundQs[j],    iqsi[j]);
-allWs[i] = getWs(dist[i]);
+}
-qs       = getQsForWs(allWs[i]);
-}
+double [] oqs = new double[okms.length];
+double [] ows = new double[okms.length];
-wqkms[i] = computeWaterlevelData(kms, qs, wst);
-}
+wst.interpolate(okms, ows, oqs, qPosition, remap);
-WQKms[] merged    = WQKms.merge(wqkms);
+BackJumpCorrector bjc = new BackJumpCorrector();
-int     numMerged = merged.length;
+if (bjc.doCorrection(okms, ows)) {
-WQKms[] computed  = new WQKms[numMerged];
+logger.debug("Discharge longitudinal section has backjumps.");
+results.add(new WQCKms(okms, oqs, ows, bjc.getCorrected()));
-for (int i = 0; i < numMerged; i++) {
+}
-computed[i] = computeDischargeLongitudinalSectionData(merged[i]);
+else {
+results.add(new WQKms(okms, oqs, ows));
-setDischargeLongitudinalSectionNames(
+}
-computed[i],
+}
-qSel ? allQs : allWs,
+// TODO: set names
-i,
-qSel ? "Q" : "W");
+return results.toArray(new WQKms[results.size()]);
 }
-// TODO Introduce a caching mechanism here!
+protected static String joinDoubles(double [] x) {
+if (x == null) {
-return computed;
+return "null";
+}
+StringBuilder sb = new StringBuilder();
+for (int i = 0; i < x.length; ++i) {
+if (i > 0) sb.append(", ");
+sb.append(x[i]);
+}
+return sb.toString();
+}
+protected double [] toQs(double [] range) {
+double [] qs = getQs(range);
+if (qs == null) {
+logger.debug("Determine Q values based on a set of W values.");
+double [] ws = getWs(range);
+qs = getQsForWs(ws);
+}
+return qs;
 }
 /**
 * Sets the name for discharge longitudinal section curves where each WQKms

Mercurial > dive4elements > river

comparison flys-artifacts/src/main/java/de/intevation/flys/artifacts/WINFOArtifact.java @ 451:73bc64c4a7b0