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

comparison artifacts/src/main/java/org/dive4elements/river/exports/ComputedDischargeCurveExporter.java @ 6305:40c77429319d

Update PDF/CSV export for discharge curves - Unit is now dynamic, also adds gauge and datum - Chose output unit depending on weather we are at a gauge or not

author	Andre Heinecke <aheinecke@intevation.de>
date	Wed, 12 Jun 2013 16:10:56 +0200
parents	a0078e5e3b39
children	cb045ccba9b2

comparison

equal deleted inserted replaced

-:152e1ba8e5bd
+:40c77429319d
 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.GaugesFactory;
+import org.dive4elements.river.artifacts.model.WQ;
 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.model.River;
 import org.dive4elements.river.utils.RiverUtils;
 import org.dive4elements.river.utils.Formatter;
 /**
 public static final String DEFAULT_CSV_W_HEADER  = "W [NN + m]";
 public static final String DEFAULT_CSV_Q_HEADER  = "Q [m\u00b3/s]";
 public static final String PDF_HEADER_MODE = "export.computed.discharge.pdf.mode";
+public static final String PDF_HEADER_CALC_MODE = "export.computed.discharge.pdf.calc.mode";
 public static final String JASPER_FILE = "export.computed.discharge.pdf.file";
 protected List<WQKms> data;
+protected String wUnit;
+protected String riverUnit;
+protected String gaugeName;
+protected double gaugeDatum;
+protected boolean isCalculated;
 public void init(Document request, OutputStream out, CallContext context) {
 logger.debug("ComputedDischargeCurveExporter.init");
 @Override
 protected void addData(Object d) {
 if (d instanceof CalculationResult) {
 d = ((CalculationResult)d).getData();
 }
+WQKms referenceWQ = null; // used for gauge / unit observations
 if (d instanceof WQKms[]){
 data.addAll(Arrays.asList((WQKms [])d));
-return;
+// If there is a unit mix in this list
+// we are screwed anyway.
+referenceWQ = ((WQKms[])d)[0];
 }
 else if (d instanceof WQKms) {
 data.add((WQKms)d);
+referenceWQ = (WQKms)d;
+} else {
+logger.warn("Can't add data for export. Unkown data type " +
+d.getClass().getName());
 return;
 }
-logger.warn("Can't add data for csv export. Unkown data type " +
+logger.debug("Andre 1");
-d.getClass().getName());
+if (referenceWQ != null) {
-}
+wUnit = referenceWQ.getWUnit();
+D4EArtifact arti = (D4EArtifact)master;
+River river = RiverUtils.getRiver(arti);
+riverUnit = river.getWstUnit().getName();
+RangeAccess rangeAccess = new RangeAccess(arti);
+double[] kms = rangeAccess.getKmRange();
+Gauge gauge = river.determineGaugeByPosition(kms[0]);
+logger.debug("Kms 0: " + kms[0]);
+if (Math.abs(kms[0] - gauge.getStation().doubleValue()) < 1e-4) {
+logger.debug("At Gauge");
+gaugeName = gauge.getName();
+gaugeDatum = gauge.getDatum().doubleValue();
+// Assumption is that values at a gauge that are in the meter
+// reference system are calculated.
+isCalculated = wUnit.equals("m");
+// Now convert the data to cm because we are at gauge
+List<WQKms> newData = new ArrayList<WQKms>();
+for (WQKms d2: data) {
+newData.add(new WQKms(d2.getKms(),
+WQ.getFixedWQforExportAtGauge((WQ)d2, gauge.getDatum())));
+}
+data = newData; // All hail the garbage collector
+wUnit = "cm";
+logger.debug("Setting gauge / datum to: " + gaugeName + " / " + gaugeDatum);
+} else {
+gaugeName = "";
+gaugeDatum = Double.NaN;
+}
+}
+}
 protected void writeCSVData(CSVWriter writer) {
 logger.info("ComputedDischargeCurveExporter.writeData");
 writeCSVHeader(writer);
 protected void writeCSVHeader(CSVWriter writer) {
 logger.debug("ComputedDischargeCurveExporter.writeCSVHeader");
-String unit = RiverUtils.getRiver((D4EArtifact) master).getWstUnit().getName();
 RangeAccess access = new RangeAccess((D4EArtifact)master);
 double[] km = access.getLocations();
 String header =
-msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER, new Object[] { unit });
+msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER, new Object[] { wUnit });
-if (km != null) {
-List<Gauge>gauges = GaugesFactory.getGauges(access.getRiver());
-for (Gauge g: gauges) {
-if (Math.abs(g.getStation().doubleValue() - km[0]) < 0.0001d) {
-header = "W [cm]";
-break;
-}
-}
-}
 writer.writeNext(new String[] {
 header,
 msg(CSV_Q_HEADER, DEFAULT_CSV_Q_HEADER)
 });
 }
 * Returns the number formatter for W values.
 *
 * @return the number formatter for W values.
 */
 protected NumberFormat getWFormatter() {
+if (wUnit.equals("cm")) {
+return Formatter.getFormatter(context, 0, 0);
+}
 return Formatter.getComputedDischargeW(context);
 }
 /**
 protected void addMetaData(WKmsJRDataSource source) {
 CallMeta meta = context.getMeta();
 D4EArtifact flys = (D4EArtifact) master;
+source.addMetaData("gauge", gaugeName);
+if (!Double.isNaN(gaugeDatum)) {
+NumberFormat mf = Formatter.getMeterFormat(context);
+source.addMetaData("datum", mf.format(gaugeDatum) + " " + riverUnit);
+} else {
+source.addMetaData("datum", "");
+}
 source.addMetaData ("river", RiverUtils.getRivername(flys));
 Locale locale = Resources.getLocale(meta);
 DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, locale);
 source.addMetaData("date", df.format(new Date()));
+source.addMetaData("wUnit", wUnit.equals("m") ? riverUnit : "cm");
 RangeAccess rangeAccess = new RangeAccess(flys);
 double[] kms = rangeAccess.getKmRange();
 source.addMetaData("range", String.valueOf(kms[0]));
-source.addMetaData("calculation", Resources.getMsg(
+if (isCalculated) {
-locale,
+source.addMetaData("calculation", Resources.getMsg(
-PDF_HEADER_MODE,
+locale,
-"Computed Discharge"));
+PDF_HEADER_CALC_MODE,
+"Computed Discharge"));
+} else {
+source.addMetaData("calculation", Resources.getMsg(
+locale,
+PDF_HEADER_MODE,
+"Discharge"));
+}
 }
 protected void addWQData(WKmsJRDataSource source) {
 NumberFormat wf  = getWFormatter();
 NumberFormat qf  = getQFormatter();

Mercurial > dive4elements > river

comparison artifacts/src/main/java/org/dive4elements/river/exports/ComputedDischargeCurveExporter.java @ 6305:40c77429319d