gernotbelger@9243: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9243:  * Software engineering by
gernotbelger@9243:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@9243:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@9243:  *
gernotbelger@9243:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9243:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9243:  * documentation coming with Dive4Elements River for details.
gernotbelger@9243:  */
gernotbelger@9243: package org.dive4elements.river.artifacts.uinfo.salix;
gernotbelger@9243: 
gernotbelger@9243: import java.text.NumberFormat;
gernotbelger@9243: import java.util.Collection;
gernotbelger@9243: import java.util.List;
gernotbelger@9243: 
gernotbelger@9243: import org.dive4elements.river.artifacts.common.ExportContextCSV;
mschaefer@9316: import org.dive4elements.river.artifacts.common.GeneralResultType;
gernotbelger@9243: import org.dive4elements.river.artifacts.common.IExportContext;
gernotbelger@9243: import org.dive4elements.river.artifacts.common.ResultRow;
gernotbelger@9243: import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
gernotbelger@9243: import org.dive4elements.river.artifacts.uinfo.commons.UInfoResultType;
gernotbelger@9243: 
mschaefer@9316: import gnu.trove.TDoubleArrayList;
mschaefer@9316: 
gernotbelger@9243: /**
gernotbelger@9243:  * @author Domenico Nardi Tironi
gernotbelger@9243:  */
gernotbelger@9243: final class SalixLineCalculationRegionalResult extends SalixLineCalculationNoScenarioResult {
gernotbelger@9243: 
gernotbelger@9243:     private static final long serialVersionUID = 1L;
gernotbelger@9243: 
mschaefer@9316:     private final String[] scenarioLabels;
mschaefer@9316: 
mschaefer@9316:     public SalixLineCalculationRegionalResult(final String label, final String[] scenarioLabels, final Collection<ResultRow> rows) {
mschaefer@9316:         super(label, rows);
mschaefer@9316:         this.scenarioLabels = scenarioLabels;
mschaefer@9316:     }
mschaefer@9316: 
mschaefer@9316:     public final int getScenarioCount() {
mschaefer@9316:         return (this.scenarioLabels != null) ? this.scenarioLabels.length : 0;
mschaefer@9316:     }
mschaefer@9316: 
mschaefer@9316:     public String getScenarioLabel(final int i) {
mschaefer@9316:         if ((this.scenarioLabels != null) && (i < this.scenarioLabels.length))
mschaefer@9316:             return this.scenarioLabels[i];
mschaefer@9316:         else
mschaefer@9316:             return "";
gernotbelger@9243:     }
gernotbelger@9243: 
gernotbelger@9243:     @Override
gernotbelger@9243:     public void writeCSVHeader(final ExportContextCSV exportContextCSV, final RiverInfo river) {
gernotbelger@9243: 
gernotbelger@9243:         final Collection<String> header = super.writeNoScenarioHeader(exportContextCSV);
gernotbelger@9243: 
mschaefer@9316:         for (int i = 1; i <= getScenarioCount(); i++) {
mschaefer@9316:             header.add(exportContextCSV.msg(SalixScenario.getScenarioValueHeader()));
mschaefer@9316:             header.add(exportContextCSV.msg(SalixScenario.getDwsplHeader()));
mschaefer@9316:         }
gernotbelger@9243:         exportContextCSV.writeCSVLine(header.toArray(new String[header.size()]));
gernotbelger@9243:     }
gernotbelger@9243: 
gernotbelger@9243:     @Override
gernotbelger@9243:     protected void writeCSVResultMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9243:         writeRegionalEffectsCSVMetadata(exportContextCSV);
gernotbelger@9243:         exportContextCSV.writeBlankLine();
gernotbelger@9243: 
gernotbelger@9243:     }
gernotbelger@9243: 
gernotbelger@9243:     @Override
mschaefer@9316:     protected String[] formatRow(final IExportContext context, final ResultRow row) {
gernotbelger@9243: 
gernotbelger@9243:         final Collection<String> lines = super.getNoScenarioFormat(row, context);
gernotbelger@9243: 
gernotbelger@9243:         if (context instanceof ExportContextCSV) { // enum gespart - pdf-export hat ja nur die drei spalten
gernotbelger@9243:             final NumberFormat formatter = ((ExportContextCSV) context).getSalixScenFormatter();
gernotbelger@9243:             final List<SalixScenario> scenList = (List<SalixScenario>) row.getValue(UInfoResultType.customMultiRowColSalixRegionalValue_Dwspl);
gernotbelger@9243: 
gernotbelger@9243:             for (final SalixScenario scenario : scenList) {
gernotbelger@9243:                 // scenario unused; index wurde ja verworfen, stattdessen 2 spalten
gernotbelger@9243:                 lines.add(scenario.getSalixValueFormatted(formatter));
gernotbelger@9243:                 lines.add(scenario.getDwsplFormatted());
gernotbelger@9243:             }
gernotbelger@9243:         }
gernotbelger@9243:         return lines.toArray(new String[lines.size()]);
gernotbelger@9243:     }
gernotbelger@9243: 
gernotbelger@9243:     private void writeRegionalEffectsCSVMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9243:         final String main = "uinfo.export.csv.meta.header.salix.regional";
gernotbelger@9243:         // "##Regional wirkende Eingriffe"
gernotbelger@9243:         exportContextCSV.writeCSVMetaEntry(main);
gernotbelger@9243:         writeRegionalCommonCSVMeta(exportContextCSV);
gernotbelger@9243:     }
gernotbelger@9243: 
mschaefer@9316:     /**
mschaefer@9316:      * Gets a longitudinal section of W, Q, or flood duration of one of the waterlevels
mschaefer@9316:      */
mschaefer@9316:     public final double[][] getScenarioPoints(final int dataIndex) {
mschaefer@9316: 
mschaefer@9316:         final Collection<ResultRow> rows = getRows();
mschaefer@9316: 
mschaefer@9316:         final TDoubleArrayList xPoints = new TDoubleArrayList(rows.size());
mschaefer@9316:         final TDoubleArrayList yPoints = new TDoubleArrayList(rows.size());
mschaefer@9316: 
mschaefer@9316:         for (final ResultRow row : rows) {
mschaefer@9316: 
mschaefer@9316:             final double station = row.getDoubleValue(GeneralResultType.station);
mschaefer@9316: 
mschaefer@9316:             final List<SalixScenario> scenarios = (List<SalixScenario>) row.getValue(UInfoResultType.customMultiRowColSalixRegionalValue_Dwspl);
mschaefer@9316:             if (dataIndex <= scenarios.size() - 1) {
mschaefer@9316:                 final double value = scenarios.get(dataIndex).getSalixValue();
mschaefer@9316:                 xPoints.add(station);
mschaefer@9316:                 yPoints.add(value);
mschaefer@9316:             }
mschaefer@9316:         }
mschaefer@9316: 
mschaefer@9316:         return new double[][] { xPoints.toNativeArray(), yPoints.toNativeArray() };
mschaefer@9316:     }
mschaefer@9316: 
gernotbelger@9243:     // TODO: move to SupraRegionalEffectsResult-Class; remove here
gernotbelger@9243:     // private void writeExtendedRegionalEffectsCSVMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9243:     // final String main = "uinfo.export.csv.meta.header.salix.regionalextended";
gernotbelger@9243:     // // "##Überregional wirkende Eingriffe"
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(main);
gernotbelger@9243:     //
gernotbelger@9243:     // writeRegionalCommonCSVMeta(exportContextCSV);
gernotbelger@9243:     // }
gernotbelger@9243: 
gernotbelger@9243:     // TODO: move to historical class and delete here
gernotbelger@9243:     // private void writeHistoricalViewCSVMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9243:     // final String main = "uinfo.export.csv.meta.header.salix.historical";
gernotbelger@9243:     // // "##Historische Betrachtung"
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(main);
gernotbelger@9243:     //
gernotbelger@9243:     // final String mainSub = "uinfo.export.csv.meta.header.salix";
gernotbelger@9243:     // // "# Szenariotyp: "
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(mainSub + ".szenariotyp");
gernotbelger@9243:     // // "# Teilabschnitt: "
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(mainSub + ".teilabschnitt");
gernotbelger@9243:     //
gernotbelger@9243:     // // "# Art des Zeitraums: "
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(main + ".zeitart");
gernotbelger@9243:     // // "# Historischer Zeitpunkt: "
gernotbelger@9243:     // exportContextCSV.writeCSVMetaEntry(main + ".zeitpunkt");
gernotbelger@9243:     // }
gernotbelger@9243: }