diff artifacts/src/main/java/org/dive4elements/river/artifacts/model/GaugeDischargeCurveFacet.java @ 5838:5aa05a7a34b7

Rename modules to more fitting names.
author Sascha L. Teichmann <teichmann@intevation.de>
date Thu, 25 Apr 2013 15:23:37 +0200
parents flys-artifacts/src/main/java/org/dive4elements/river/artifacts/model/GaugeDischargeCurveFacet.java@bd047b71ab37
children 4897a58c8746
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/artifacts/src/main/java/org/dive4elements/river/artifacts/model/GaugeDischargeCurveFacet.java	Thu Apr 25 15:23:37 2013 +0200
@@ -0,0 +1,82 @@
+package org.dive4elements.river.artifacts.model;
+
+import java.util.Arrays;
+import java.util.Map;
+
+import org.dive4elements.artifacts.Artifact;
+import org.dive4elements.artifacts.CallContext;
+
+import org.dive4elements.artifactdatabase.state.DefaultFacet;
+import org.dive4elements.artifactdatabase.state.Facet;
+
+import org.dive4elements.river.artifacts.FLYSArtifact;
+
+import org.dive4elements.river.model.Gauge;
+
+import org.dive4elements.river.utils.FLYSUtils;
+
+import org.apache.log4j.Logger;
+
+/**
+ * A Facet that returns discharge curve data at a gauge
+ *
+ * @author <a href="mailto:bjoern.ricks@intevation.de">Björn Ricks</a>
+ */
+public class GaugeDischargeCurveFacet
+extends      DefaultFacet
+implements FacetTypes
+{
+    private static final Logger log =
+        Logger.getLogger(GaugeDischargeCurveFacet.class);
+
+    public GaugeDischargeCurveFacet(String name, String description) {
+        super(0, name, description);
+    }
+
+    @Override
+    public Object getData(Artifact art, CallContext context) {
+        return getWQKms(art, context);
+    }
+
+    protected WQKms getWQKms(Artifact art, CallContext context) {
+        if (!(art instanceof FLYSArtifact)) {
+            log.warn("Invalid artifact type");
+            return null;
+        }
+
+        FLYSArtifact flys = (FLYSArtifact)art;
+
+        String river = flys.getDataAsString("river");
+
+        Gauge gauge = FLYSUtils.getReferenceGauge(flys);
+
+        if (river == null || gauge == null) {
+            log.warn("Unknown river or gauge");
+            return null;
+        }
+
+        String name = gauge.getName();
+
+        DischargeTables dt = new DischargeTables(river, name);
+
+        Map<String, double [][]> map = dt.getValues(
+                DischargeTables.MASTER_SCALE);
+
+        double [][] values = map.get(name);
+        if (values == null) {
+            return null;
+        }
+        double [] kms = new double[values[0].length];
+        Arrays.fill(kms, gauge.getStation().doubleValue());
+        return new WQKms(kms, values[0], values[1], name);
+    }
+
+    @Override
+    public Facet deepCopy() {
+        GaugeDischargeCurveFacet copy = new GaugeDischargeCurveFacet(
+                this.name,
+                this.description);
+        copy.set(this);
+        return copy;
+    }
+}

http://dive4elements.wald.intevation.org