diff artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flood_duration/FloodDurationCalculation.java @ 9266:465347d12990

Station specific calculation of flood duration curve and infrastructure annotations
author mschaefer
date Wed, 18 Jul 2018 12:20:01 +0200
parents c2a0028bfa9f
children 83ebeb620b5a
line wrap: on
line diff
--- a/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flood_duration/FloodDurationCalculation.java	Tue Jul 17 19:48:28 2018 +0200
+++ b/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flood_duration/FloodDurationCalculation.java	Wed Jul 18 12:20:01 2018 +0200
@@ -9,11 +9,14 @@
  */
 package org.dive4elements.river.artifacts.sinfo.flood_duration;
 
+import java.util.List;
+
 import org.apache.commons.lang.math.DoubleRange;
 import org.dive4elements.artifacts.CallContext;
 import org.dive4elements.river.artifacts.WINFOArtifact;
 import org.dive4elements.river.artifacts.model.Calculation;
 import org.dive4elements.river.artifacts.model.CalculationResult;
+import org.dive4elements.river.artifacts.model.WQDay;
 import org.dive4elements.river.artifacts.resources.Resources;
 import org.dive4elements.river.artifacts.sinfo.SINFOArtifact;
 import org.dive4elements.river.artifacts.sinfo.common.RiverInfoProvider;
@@ -21,6 +24,7 @@
 import org.dive4elements.river.artifacts.sinfo.tkhstate.WinfoArtifactWrapper;
 import org.dive4elements.river.artifacts.sinfo.util.CalculationUtils;
 import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
+import org.dive4elements.river.jfree.StickyAxisAnnotation;
 import org.dive4elements.river.model.River;
 
 /**
@@ -74,4 +78,40 @@
         final FloodDurationCalculator calculator = new FloodDurationCalculator(this.context, riverInfoProvider);
         calculator.execute(problems, label, calcRange, riverside, winfo, results);
     }
+
+    /**
+     * Calculates the flood duration curve of a station
+     */
+    public WQDay calcStationCurve(final double station, final SINFOArtifact sinfo) {
+
+        /* access input data */
+        final FloodDurationAccess access = new FloodDurationAccess(sinfo);
+        final River river = access.getRiver();
+        final DoubleRange calcRange = access.getRange();
+        final RiverInfoProvider infoProvider = RiverInfoProvider.forRange(this.context, river, calcRange);
+
+        final Calculation problems = new Calculation();
+
+        // Calculate the selected main values, if any
+        /* misuse winfo-artifact to calculate waterlevels in the same way */
+        final WINFOArtifact winfo = new WinfoArtifactWrapper(sinfo);
+        winfo.addStringData("ld_locations", Double.toString(station));
+
+        final FloodDurationCalculator calculator = new FloodDurationCalculator(this.context, infoProvider);
+        if (!Double.isNaN(station))
+            return calculator.calcWQDays(problems, station, winfo);
+        else
+            return calculator.calcWQDays(problems, calcRange.getMinimumFloat(), winfo);
+    }
+
+    /**
+     * Calculates the annotations of the infrastructure(s) of a station for a flood duration calculation
+     */
+    public List<StickyAxisAnnotation> calcInfrastructureAnnotations(final double station, final FloodDurationCalculationResult result) {
+
+        final Calculation problems = new Calculation();
+
+        final FloodDurationCalculator calculator = new FloodDurationCalculator(this.context, null);
+        return calculator.calcInfrastructureAnnotations(problems, station, result);
+    }
 }
\ No newline at end of file

http://dive4elements.wald.intevation.org