gernotbelger@9313: /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9313:  * Software engineering by
gernotbelger@9313:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@9313:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@9313:  *
gernotbelger@9313:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9313:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9313:  * documentation coming with Dive4Elements River for details.
gernotbelger@9313:  */
gernotbelger@9313: package org.dive4elements.river.artifacts.bundu.bezugswst;
gernotbelger@9313: 
gernotbelger@9313: import java.util.ArrayList;
gernotbelger@9313: import java.util.Collection;
gernotbelger@9313: 
gernotbelger@9318: import org.dive4elements.river.artifacts.bundu.BunduResultType;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.AbstractCalculationExportableResult;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.ExportContextCSV;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.ExportContextPDF;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.GeneralResultType;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.IExportContext;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.MetaAndTableJRDataSource;
gernotbelger@9313: import org.dive4elements.river.artifacts.common.ResultRow;
gernotbelger@9313: import org.dive4elements.river.artifacts.sinfo.util.BedHeightInfo;
gernotbelger@9313: import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
gernotbelger@9313: import org.dive4elements.river.artifacts.sinfo.util.WstInfo;
gernotbelger@9313: 
gernotbelger@9313: /**
gernotbelger@9313:  * Contains the results of a //TODO LINK{@link BezugswstCalculation}.
gernotbelger@9313:  *
gernotbelger@9313:  * @author Gernot Belger
gernotbelger@9313:  */
gernotbelger@9313: final class BezugswstCalculationResult extends AbstractCalculationExportableResult {
gernotbelger@9313: 
gernotbelger@9313:     private static final long serialVersionUID = 1L;
gernotbelger@9313: 
gernotbelger@9313:     private static final String JASPER_FILE = "/jasper/templates/bundu.bezugswst.jrxml";
gernotbelger@9313: 
gernotbelger@9323:     private final WstInfo wst;
gernotbelger@9323:     private final BedHeightInfo sounding;
gernotbelger@9323:     private final String function;
gernotbelger@9323:     private final boolean preprocessing;
gernotbelger@9323:     private final int startYear;
gernotbelger@9323:     private final int endYear;
gernotbelger@9323:     private final Double missingVolumeFrom;
gernotbelger@9323:     private final Double missingVolumeTo;
gernotbelger@9323:     private final Integer ud;
gernotbelger@9323:     private final double[] qOfGauges;
gernotbelger@9313: 
gernotbelger@9323:     public BezugswstCalculationResult(final String label, final Collection<ResultRow> rows, final BedHeightInfo sounding, final WstInfo wst,
gernotbelger@9323:             final String function, final boolean preprocessing, final int startYear, final int endYear, final Integer ud, final double[] qOfGauges,
gernotbelger@9323:             final Double missingVolumeFrom, final Double missingVolumeTo) {
gernotbelger@9313:         super(label, rows);
gernotbelger@9323:         this.wst = wst;
gernotbelger@9323:         this.sounding = sounding;
gernotbelger@9323:         this.function = function;
gernotbelger@9323:         this.preprocessing = preprocessing;
gernotbelger@9323:         this.startYear = startYear;
gernotbelger@9323:         this.endYear = endYear;
gernotbelger@9323:         this.missingVolumeTo = missingVolumeTo;
gernotbelger@9323:         this.missingVolumeFrom = missingVolumeFrom;
gernotbelger@9323:         this.ud = ud;
gernotbelger@9323:         this.qOfGauges = qOfGauges;
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     protected void writeCSVResultMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9313: 
gernotbelger@9318:         // TODO: wenn wst und peilung vorhanden sind, sollte dieser Teil wieder einkommentiert werden
gernotbelger@9313:         // exportContextCSV.writeCSVWaterlevelMetadata(wst);
gernotbelger@9313:         // exportContextCSV.writeBlankLine();
gernotbelger@9323:         // exportContextCSV.writeCSVSoundingMetadata(this.sounding, " ");
gernotbelger@9323: 
gernotbelger@9323:         // Reihenfolge in der LV anders; ich finde sie aber so sinnvoller
gernotbelger@9323:         exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.function", this.function);
gernotbelger@9323:         exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.preprocessing", String.valueOf(this.preprocessing));
gernotbelger@9323:         exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.q_period", String.valueOf(this.startYear), String.valueOf(this.endYear));
gernotbelger@9323:         if (this.ud != null)
gernotbelger@9323:             exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.ud", this.ud);
gernotbelger@9323:         else if (this.qOfGauges != null)
gernotbelger@9323:             exportContextCSV.writeCSVMetaEntry("common.export.waterlevel.csv.meta.q", this.qOfGauges); // was auch immer dieser String beinhalten mag
gernotbelger@9323: 
gernotbelger@9323:         exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.function", this.function);
gernotbelger@9323:         exportContextCSV.writeCSVMetaEntry("bundu.export.bezugswst.csv.meta.missing_volume_range", this.missingVolumeFrom, this.missingVolumeTo);
gernotbelger@9323: 
gernotbelger@9323:         exportContextCSV.writeBlankLine();
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     protected String getJasperFile() {
gernotbelger@9313:         return JASPER_FILE;
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     protected String[] formatRow(final IExportContext exportContext, final ResultRow row) {
gernotbelger@9313: 
gernotbelger@9313:         final Collection<String> lines = new ArrayList<>(10);
gernotbelger@9313: 
gernotbelger@9313:         lines.add(exportContext.formatRowValue(row, GeneralResultType.station));
gernotbelger@9318:         lines.add(exportContext.formatRowValue(row, BunduResultType.bezugswst));
gernotbelger@9313:         lines.add(exportContext.formatRowValue(row, GeneralResultType.dischargeQwithUnit));
gernotbelger@9313: 
gernotbelger@9318:         lines.add(exportContext.formatRowValue(row, GeneralResultType.waterlevelLabel));
gernotbelger@9318:         lines.add(exportContext.formatRowValue(row, GeneralResultType.gaugeLabel));
gernotbelger@9313:         lines.add(exportContext.formatRowValue(row, GeneralResultType.location));
gernotbelger@9313: 
gernotbelger@9313:         return lines.toArray(new String[lines.size()]);
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     public void writeCSVHeader(final ExportContextCSV exportContextCSV, final RiverInfo river) {
gernotbelger@9313: 
gernotbelger@9313:         final Collection<String> header = new ArrayList<>(11);
gernotbelger@9313: 
gernotbelger@9313:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.station));
gernotbelger@9318:         header.add(exportContextCSV.formatCsvHeader(BunduResultType.bezugswst));
gernotbelger@9313:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.dischargeQwithUnit));
gernotbelger@9318:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.waterlevelLabel));
gernotbelger@9318:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.gaugeLabel));
gernotbelger@9313:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.location));
gernotbelger@9313: 
gernotbelger@9313:         exportContextCSV.writeCSVLine(header.toArray(new String[header.size()]));
gernotbelger@9313: 
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     protected String[] formatCSVRow(final ExportContextCSV exportContextCSV, final ResultRow row) {
gernotbelger@9313: 
gernotbelger@9313:         return this.formatRow(exportContextCSV, row);
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     protected String[] formatPDFRow(final ExportContextPDF exportContextPDF, final ResultRow row) {
gernotbelger@9313: 
gernotbelger@9313:         return this.formatRow(exportContextPDF, row);
gernotbelger@9313:     }
gernotbelger@9313: 
gernotbelger@9313:     @Override
gernotbelger@9313:     protected void addJRTableHeader(final ExportContextPDF exportContextPDF, final MetaAndTableJRDataSource source) {
gernotbelger@9313:         /* column headings */
gernotbelger@9313:         exportContextPDF.addJRMetadata(source, "station_header", GeneralResultType.station);
gernotbelger@9318:         exportContextPDF.addJRMetadata(source, "bezugswst_header", BunduResultType.bezugswst);
gernotbelger@9318:         exportContextPDF.addJRMetadata(source, "discharge_header", GeneralResultType.dischargeQwithUnit);
gernotbelger@9318:         exportContextPDF.addJRMetadata(source, "waterlevel_name_header", GeneralResultType.waterlevelLabel);
gernotbelger@9318:         exportContextPDF.addJRMetadata(source, "gauge_header", GeneralResultType.gaugeLabel);
gernotbelger@9313:         exportContextPDF.addJRMetadata(source, "location_header", GeneralResultType.location);
gernotbelger@9313: 
gernotbelger@9313:     }
gernotbelger@9313: }