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

comparison artifacts/src/main/java/org/dive4elements/river/exports/WaterlevelExporter.java @ 9486:ce13a2f07290

pdf exports added for fixation+bundu (waterlevel)

author	gernotbelger
date	Mon, 17 Sep 2018 19:07:57 +0200
parents	dd05a5eef210
children	aa23225fd85f

comparison

equal deleted inserted replaced

-:dd05a5eef210
+:ce13a2f07290
 import java.util.regex.Pattern;
 import org.apache.log4j.Logger;
 import org.dive4elements.artifacts.Artifact;
 import org.dive4elements.artifacts.CallMeta;
-import org.dive4elements.artifacts.common.utils.Config;
 import org.dive4elements.river.artifacts.AbstractFixBunduArtifact;
 import org.dive4elements.river.artifacts.D4EArtifact;
 import org.dive4elements.river.artifacts.StaticWQKmsArtifact;
 import org.dive4elements.river.artifacts.WINFOArtifact;
 import org.dive4elements.river.artifacts.access.FixRealizingAccess;
 import org.dive4elements.river.artifacts.access.IsOfficialAccess;
 import org.dive4elements.river.artifacts.access.RangeAccess;
+import org.dive4elements.river.artifacts.common.DefaultCalculationResults;
+import org.dive4elements.river.artifacts.common.ExportContextPDF;
+import org.dive4elements.river.artifacts.common.GeneralResultType;
+import org.dive4elements.river.artifacts.common.JasperDesigner;
+import org.dive4elements.river.artifacts.common.JasperReporter;
+import org.dive4elements.river.artifacts.common.MetaAndTableJRDataSource;
 import org.dive4elements.river.artifacts.model.CalculationResult;
 import org.dive4elements.river.artifacts.model.ConstantWQKms;
+import org.dive4elements.river.artifacts.model.DischargeTables;
 import org.dive4elements.river.artifacts.model.Segment;
 import org.dive4elements.river.artifacts.model.WKmsJRDataSource;
 import org.dive4elements.river.artifacts.model.WQCKms;
 import org.dive4elements.river.artifacts.model.WQKms;
 import org.dive4elements.river.artifacts.model.WQKmsResult;
 import org.dive4elements.river.artifacts.model.WstLine;
 import org.dive4elements.river.artifacts.resources.Resources;
+import org.dive4elements.river.artifacts.sinfo.util.CalculationUtils;
+import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
+import org.dive4elements.river.model.DischargeTable;
 import org.dive4elements.river.model.Gauge;
 import org.dive4elements.river.utils.Formatter;
 import org.dive4elements.river.utils.RiverUtils;
 import org.dive4elements.river.utils.RiverUtils.WQ_MODE;
 import au.com.bytecode.opencsv.CSVWriter;
 import gnu.trove.TDoubleArrayList;
 import net.sf.jasperreports.engine.JRException;
-import net.sf.jasperreports.engine.JasperExportManager;
-import net.sf.jasperreports.engine.JasperFillManager;
-import net.sf.jasperreports.engine.JasperPrint;
 /**
 * Generates different output formats (wst, csv, pdf) of data that resulted from
 * a waterlevel computation.
 *
 public static final String DEFAULT_CSV_LOCATION_HEADER = "Lage";
 public static final String DEFAULT_CSV_GAUGE_HEADER = "Bezugspegel";
 public static final String DEFAULT_CSV_NOT_IN_GAUGE_RANGE = "außerhalb des gewählten Bezugspegels";
 protected static final String PDF_HEADER_MODE = "export.waterlevel.pdf.mode";
-private static final String JASPER_FILE = "export.waterlevel.pdf.file";
 /** The storage that contains all WQKms objects that are calculated. */
 public List<WQKms[]> data;
 /** The storage that contains official fixings if available. */
 public List<WQKms> officalFixings;
+private final Map<String, Double> gaugeQ_W_Map = new HashMap<>();
 public WaterlevelExporter() {
 this.data = new ArrayList<>();
 }
 } else {
 return "Q=" + nf.format(rawValue);
 }
 }
-protected final boolean isQ() {
+private final boolean isQ() {
 final WQ_MODE mode = RiverUtils.getWQMode((D4EArtifact) this.master);
 return mode == WQ_MODE.QGAUGE || mode == WQ_MODE.QFREE;
 }
 @Override
 }
 return colDesc == null ? "" : colDesc;
 }
+private List<Segment> getSegments(final D4EArtifact flys) {
+if (flys instanceof AbstractFixBunduArtifact) {
+// Get W/Q input per gauge for this case.
+final FixRealizingAccess fixAccess = new FixRealizingAccess(flys);
+return fixAccess.getSegments();
+}
+return null;
+}
 /**
 * Write "rows" of csv data from wqkms with writer.
 */
 protected void wQKms2CSV(final CSVWriter writer, final WQKms wqkms, final boolean atGauge, final boolean isQ) {
 log.debug("WaterlevelExporter.wQKms2CSV");
 final String gaugeName = gauge.getName(); // RANDOMLY CHRASHES HERE, gauge is randomly null (especially on heavy load) - on production system as well
 // (it seems to)
 String desc = "";
 final String notinrange = msg(CSV_NOT_IN_GAUGE_RANGE, DEFAULT_CSV_NOT_IN_GAUGE_RANGE);
-List<Segment> segments = null;
-boolean isFixRealize = false;
 final double a = gauge.getRange().getA().doubleValue();
 final double b = gauge.getRange().getB().doubleValue();
 final long startTime = System.currentTimeMillis();
 desc = getDesc(wqkms, isQ);
-if (flys instanceof AbstractFixBunduArtifact) {
+boolean isFixRealize = false;
-// Get W/Q input per gauge for this case.
+final List<Segment> segments = getSegments(flys);
-final FixRealizingAccess fixAccess = new FixRealizingAccess(flys);
+if (segments != null && !segments.isEmpty()) {
-segments = fixAccess.getSegments();
+isFixRealize = true;
-if (segments != null && !segments.isEmpty()) {
+}
-isFixRealize = true;
-}
-}
 if (atGauge) { // "At gauge" needs more output.
 // Kms tend to be close together so caching the last sector
 // is a good time saving heuristic.
 Segment lastSegment = null;
 } else {
 writer.addColumn(wqkms.getName());
 }
 }
-public void doWritePdf(final WKmsJRDataSource source, final String jasperFile) {
-final String confPath = Config.getConfigDirectory().toString();
-final Map parameters = new HashMap();
-parameters.put("ReportTitle", "Exported Data");
-try {
-final JasperPrint print = JasperFillManager.fillReport(confPath + jasperFile, parameters, source);
-JasperExportManager.exportReportToPdfStream(print, this.out);
-}
-catch (final JRException je) {
-log.warn("Error generating PDF Report!", je);
-}
-}
 @Override
 protected void writePDF(final OutputStream out) {
 log.debug("write PDF");
-final WKmsJRDataSource source = createJRData();
-final String jasperFile = Resources.getMsg(this.context.getMeta(), JASPER_FILE, "/jasper/waterlevel_en.jasper");
+final boolean isQ = isQ();
-doWritePdf(source, jasperFile);
+final MetaAndTableJRDataSource source = new MetaAndTableJRDataSource();
+final String jasperFile = !isQ ? "/jasper/templates/waterlevel_new.jrxml" : "/jasper/templates/fix_waterlevel.jrxml";
-}
+((D4EArtifact) this.master).getData("calculation.mode");
-private WKmsJRDataSource createJRData() {
+if ((this.master instanceof WINFOArtifact)) {
-final WKmsJRDataSource source = new WKmsJRDataSource();
+addMetaData(source, "calc.surface.curve"); // Wasserspiegellage
+} else if
+(this.master instanceof AbstractFixBunduArtifact) {
+addMetaData(source, ((AbstractFixBunduArtifact) this.master).getCalculationModeString());// "calculation.vollmer"); // ausgelagerte
+// Wasserspiegellage
+// else
+// addMetaData(source, "calculation.analysis"); // Fixierungsanalyse
+}
+try {
+final List<String[]> sorted = getRows(); // Custom Result could be nice, too...
+for (final String[] list : sorted) {
+source.addData(list);
+}
+final JasperReporter reporter = new JasperReporter();
+final JasperDesigner d = reporter.addReport(jasperFile, source);
+d.removeColumn("delete"); // I don't want to mess with getRows(), so I prefer deleting the unwanted row directly in the report.
+reporter.exportPDF(this.out);
+}
+catch (final JRException je) {
+log.warn("Error generating PDF Report!", je);
+}
+}
+private List<String[]> getRows() {
+final List<String[]> list = new ArrayList<>();
 final WQ_MODE mode = RiverUtils.getWQMode((D4EArtifact) this.master);
 final boolean atGauge = mode == WQ_MODE.QGAUGE || mode == WQ_MODE.WGAUGE;
 final boolean isQ = mode == WQ_MODE.QGAUGE || mode == WQ_MODE.QFREE;
 Double first = Double.NaN;
 Double last = Double.NaN;
-addMetaData(source);
 for (final WQKms[] tmp : this.data) {
 for (final WQKms wqkms : tmp) {
-addWKmsData(wqkms, atGauge, isQ, source);
+list.addAll(getRows2(wqkms, atGauge, isQ));
 final double[] firstLast = wqkms.getFirstLastKM();
 if (first.isNaN()) {
 /* Initialize */
 first = firstLast[0];
 last = firstLast[1];
 }
 }
 /* Append the official fixing at the bottom */
 for (final WQKms wqkms : this.officalFixings) {
+list.addAll(getRows2(filterWQKms(wqkms, first, last), atGauge, isQ));
+}
+return list;
+}
+private List<String[]> getRows2(final WQKms wqkms, final boolean atGauge, final boolean isQ) {
+log.debug("WaterlevelExporter.addWKmsData"); // OLD CODE :-/
+final List<String[]> list = new ArrayList<>();
+// Skip constant data.
+if (wqkms instanceof ConstantWQKms) {
+return null;
+}
+final NumberFormat kmf = getKmFormatter();
+final NumberFormat wf = getWFormatter();
+final NumberFormat qf = getQFormatter();
+final int size = wqkms.size();
+double[] result = new double[3];
+final D4EArtifact flys = (D4EArtifact) this.master;
+final RangeAccess rangeAccess = new RangeAccess(flys);
+final Gauge gauge = rangeAccess.getRiver().determineRefGauge(rangeAccess.getKmRange(), rangeAccess.isRange());
+final String gaugeName = gauge.getName();
+String desc = "";
+final String notinrange = msg(CSV_NOT_IN_GAUGE_RANGE, DEFAULT_CSV_NOT_IN_GAUGE_RANGE);
+final double a = gauge.getRange().getA().doubleValue();
+final double b = gauge.getRange().getB().doubleValue();
+final WaterlevelDescriptionBuilder wldb = new WaterlevelDescriptionBuilder(flys, this.context);
+desc = wldb.getDesc(wqkms);// class getDesc(wqkms, isQ);
+Segment lastSegment = null;
+final Gauge lastGauge = null;
+final List<Segment> segments = getSegments(flys);
+final NumberFormat nf = Formatter.getFormatter(this.context.getMeta(), 0, 0);
+for (int i = 0; i < size; ++i) {
+result = wqkms.get(i, result);
+final double km = result[2];
+if (segments != null) {
+final Segment found = lastSegment != null && lastSegment.inside(km) ? lastSegment : findSegment(km, segments);
+if (found != null) {
+desc = nf.format(found.getValues()[0]);
+}
+lastSegment = found;
+}
+final double q = result[1];
+final String waterlevel = this.getWaterlevel(q, gauge); // THIS IS NEW (and makes common super method
+// difficult)
+if (atGauge) {
+list.add(new String[] { kmf.format(result[2]), wf.format(result[0]), waterlevel, qf.format(RiverUtils.roundQ(result[1])), desc,
+RiverUtils.getLocationDescription(flys, result[2]), result[2] >= a && result[2] <= b ? gaugeName : notinrange });
+} else {
+list.add(new String[] { kmf.format(result[2]), wf.format(result[0]), waterlevel, qf.format(RiverUtils.roundQ(result[1])), desc,
+RiverUtils.getLocationDescription(flys, result[2]), result[2] >= a && result[2] <= b ? gaugeName : notinrange });
+}
+}
+return list;
+}
+protected final void addMetaData(final MetaAndTableJRDataSource source, final String calculation) {
+final D4EArtifact flys = (D4EArtifact) this.master;
+final String user = CalculationUtils.findArtifactUser(this.context, flys);
+final RangeAccess ra = new RangeAccess(flys);
+final RiverInfo ri = new RiverInfo(ra.getRiver());
+final DefaultCalculationResults results = new DefaultCalculationResults(msg(calculation), user, ri, ra.getRange());
+final ExportContextPDF contextPdf = new ExportContextPDF(this.context, results);
+contextPdf.addJRMetaDataDefaults(source);
+contextPdf.addJRMetaDataForModules(source);
+/* column headings */
+contextPdf.addJRMetadata(source, "station_header", GeneralResultType.station);
+contextPdf.addJRMetadata(source, "fix_w", msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER, new Object[] { ri.getWstUnit() }));
+contextPdf.addJRMetadata(source, "w_at_gauge_header", msg("fix.export.csv.w_at_gauge"));
+contextPdf.addJRMetadata(source, "fix_q", msg(CSV_Q_HEADER));
+contextPdf.addJRMetadata(source, "waterlevel_name_header", msg("common.export.csv.header.mainvalue_label"));
+contextPdf.addJRMetadata(source, "location_header", msg("common.export.csv.header.location"));
+contextPdf.addJRMetadata(source, "w_at_gauge_header_2", msg("export.waterlevel.csv.header.w.desc"));
+contextPdf.addJRMetadata(source, "location_header", msg("common.export.csv.header.location"));
+contextPdf.addJRMetadata(source, "gauge_header", msg("common.export.csv.header.gauge"));
+}
+private WKmsJRDataSource createJRData() {
+final WKmsJRDataSource source = new WKmsJRDataSource();
+final WQ_MODE mode = RiverUtils.getWQMode((D4EArtifact) this.master);
+final boolean atGauge = mode == WQ_MODE.QGAUGE || mode == WQ_MODE.WGAUGE;
+final boolean isQ = mode == WQ_MODE.QGAUGE || mode == WQ_MODE.QFREE;
+Double first = Double.NaN;
+Double last = Double.NaN;
+addMetaData(source);
+for (final WQKms[] tmp : this.data) {
+for (final WQKms wqkms : tmp) {
+addWKmsData(wqkms, atGauge, isQ, source);
+final double[] firstLast = wqkms.getFirstLastKM();
+if (first.isNaN()) {
+/* Initialize */
+first = firstLast[0];
+last = firstLast[1];
+}
+if (firstLast[0] > firstLast[1]) {
+/*
+* Calculating upstream we assert that it is
+* impossible that the direction changes during this
+* loop
+*/
+first = Math.max(first, firstLast[0]);
+last = Math.min(last, firstLast[1]);
+} else if (firstLast[0] < firstLast[1]) {
+first = Math.min(first, firstLast[0]);
+last = Math.max(last, firstLast[1]);
+} else {
+first = last = firstLast[0];
+}
+}
+}
+/* Append the official fixing at the bottom */
+for (final WQKms wqkms : this.officalFixings) {
 addWKmsData(filterWQKms(wqkms, first, last), atGauge, isQ, source);
 }
 return source;
 }
 if (log.isDebugEnabled()) {
 log.debug("Writing PDF data took " + (stopTime - startTime) / 1000f + " secs.");
 }
 }
+protected final String getWaterlevel(final double discharge, final Gauge gauge) {
+final NumberFormat formatter = Formatter.getWaterlevelW(this.context);
+final Double waterlevel = this.getWforGaugeAndQ(gauge, discharge);
+if (waterlevel != null)
+return formatter.format(waterlevel);
+return "";
+}
+private Double getWforGaugeAndQ(final Gauge gauge, final double q) {
+final String key = gauge.getName() + String.valueOf(q);
+if (!this.gaugeQ_W_Map.containsKey(key)) {
+final DischargeTable dt = gauge.fetchMasterDischargeTable();
+final double[][] table = DischargeTables.loadDischargeTableValues(dt);
+final double[] qs = DischargeTables.getWsForQ(table, q);
+// final DischargeTables dct = new DischargeTables(gauge.getRiver().getName(), gauge.getName());
+// final double[] qs = DischargeTables.getWsForQ(dct.getFirstTable(), q); // TODO: KLÄREN, welche Abflusstabelle
+// genommen werden soll!
+if (qs != null && qs.length > 0) {
+this.gaugeQ_W_Map.put(key, qs[0]);
+}
+}
+return this.gaugeQ_W_Map.get(key);
+}
 }
 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

Mercurial > dive4elements > river

comparison artifacts/src/main/java/org/dive4elements/river/exports/WaterlevelExporter.java @ 9486:ce13a2f07290