dive4elements/river: artifacts/src/main/java/org/dive4elements/river/exports/ComputedDischargeCurveExporter.java comparison

Use 'W [cm]' as CSV header if discharge table is at a gauge.

author	Raimund Renkert <rrenkert@intevation.de>
date	Fri, 10 May 2013 17:14:10 +0200
parents	59ff03ff48f1
children	2ec6fae23bba

comparison

equal deleted inserted replaced

-:f33d717aae5c
+:b740f0c4179e
 import org.dive4elements.river.artifacts.D4EArtifact;
 import org.dive4elements.river.artifacts.access.RangeAccess;
 import org.dive4elements.river.artifacts.model.CalculationResult;
+import org.dive4elements.river.artifacts.model.GaugeFinder;
+import org.dive4elements.river.artifacts.model.GaugeFinderFactory;
+import org.dive4elements.river.artifacts.model.GaugesFactory;
 import org.dive4elements.river.artifacts.model.WQKms;
 import org.dive4elements.river.artifacts.model.WKmsJRDataSource;
 import org.dive4elements.river.artifacts.resources.Resources;
+import org.dive4elements.river.model.Gauge;
 import org.dive4elements.river.utils.RiverUtils;
 import org.dive4elements.river.utils.Formatter;
 /**
 protected void writeCSVHeader(CSVWriter writer) {
 logger.debug("ComputedDischargeCurveExporter.writeCSVHeader");
 String unit = RiverUtils.getRiver((D4EArtifact) master).getWstUnit().getName();
+RangeAccess access = new RangeAccess((D4EArtifact)master, context);
+double[] km = access.getLocations();
+String header =
+msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER, new Object[] { unit });
+if (km != null) {
+List<Gauge>gauges = GaugesFactory.getGauges(access.getRiver());
+for (Gauge g : gauges) {
+if (g.getStation().doubleValue() == km[0]) {
+header = "W [cm]";
+}
+}
+}
 writer.writeNext(new String[] {
-msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER, new Object[] { unit }),
+header,
 msg(CSV_Q_HEADER, DEFAULT_CSV_Q_HEADER)
 });
 }

Mercurial > dive4elements > river