gernotbelger@9150: /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@9150:  * Software engineering by
gernotbelger@9150:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@9150:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@9150:  *
gernotbelger@9150:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@9150:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@9150:  * documentation coming with Dive4Elements River for details.
gernotbelger@9150:  */
gernotbelger@9150: package org.dive4elements.river.artifacts.sinfo.collision;
gernotbelger@9150: 
gernotbelger@9150: import java.util.ArrayList;
gernotbelger@9150: import java.util.Collection;
mschaefer@9487: import java.util.List;
gernotbelger@9150: 
gernotbelger@9265: import org.apache.commons.collections.Predicate;
gernotbelger@9150: import org.dive4elements.river.artifacts.common.AbstractCalculationExportableResult;
gernotbelger@9150: import org.dive4elements.river.artifacts.common.ExportContextCSV;
gernotbelger@9195: import org.dive4elements.river.artifacts.common.ExportContextPDF;
gernotbelger@9150: import org.dive4elements.river.artifacts.common.GeneralResultType;
mschaefer@9204: import org.dive4elements.river.artifacts.common.IResultType;
gernotbelger@9150: import org.dive4elements.river.artifacts.common.MetaAndTableJRDataSource;
gernotbelger@9150: import org.dive4elements.river.artifacts.common.ResultRow;
mschaefer@9487: import org.dive4elements.river.artifacts.model.DateRange;
mschaefer@9157: import org.dive4elements.river.artifacts.sinfo.common.SInfoResultType;
gernotbelger@9150: import org.dive4elements.river.artifacts.sinfo.util.RiverInfo;
gernotbelger@9150: 
gernotbelger@9150: /**
mschaefer@9157:  * Contains the results of a {@link CollisionCalculation}.
gernotbelger@9150:  *
gernotbelger@9150:  * @author Gernot Belger
gernotbelger@9150:  */
gernotbelger@9195: final public class CollisionCalcOverviewResult extends AbstractCalculationExportableResult {
gernotbelger@9150: 
gernotbelger@9150:     private static final long serialVersionUID = 1L;
gernotbelger@9150: 
gernotbelger@9150:     private static final String JASPER_FILE = "/jasper/templates/sinfo.collision.overview.jrxml";
gernotbelger@9150: 
mschaefer@9487:     private final boolean isEpochs;
mschaefer@9204: 
mschaefer@9487:     private final List<DateRange> years;
mschaefer@9487: 
mschaefer@9487:     public CollisionCalcOverviewResult(final String label, final boolean isEpochs, final Collection<DateRange> years, final Collection<ResultRow> rows) {
gernotbelger@9150:         super(label, rows);
mschaefer@9487:         this.isEpochs = isEpochs;
mschaefer@9487:         this.years = new ArrayList<>(years);
mschaefer@9204:     }
mschaefer@9204: 
mschaefer@9487:     public boolean getIsEpochs() {
mschaefer@9487:         return this.isEpochs;
mschaefer@9204:     }
mschaefer@9204: 
mschaefer@9487:     public List<DateRange> getYears() {
mschaefer@9487:         return this.years;
mschaefer@9487:     }
mschaefer@9487: 
mschaefer@9487:     public double[][] getStationPointsByYear(final IResultType type, final String years) {
mschaefer@9204: 
gernotbelger@9265:         final Predicate filter = new Predicate() {
gernotbelger@9265:             @Override
gernotbelger@9265:             public boolean evaluate(final Object object) {
gernotbelger@9265:                 final ResultRow row = (ResultRow) object;
mschaefer@9487:                 return row.getValue(SInfoResultType.years).toString().equals(years);
gernotbelger@9265:             }
gernotbelger@9265:         };
mschaefer@9204: 
gernotbelger@9265:         return getPoints(GeneralResultType.station, type, filter);
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9195:     protected void writeCSVResultMetadata(final ExportContextCSV exportContextCSV) {
gernotbelger@9172:         // do nothing so far
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9195:     public void writeCSVHeader(final ExportContextCSV exportContextCSV, final RiverInfo river) {
gernotbelger@9171:         final int colSize = 3;
mschaefer@9432:         exportContextCSV.writeTitleForTabs("sinfo.export.csv.title.collision.overview", 3); // Voraussetzung für Tabs ist, dass der Titel vor den Headern
mschaefer@9204:         // geschrieben wird.
gernotbelger@9171:         // Das ist etwas doof.
gernotbelger@9150: 
gernotbelger@9171:         final Collection<String> header = new ArrayList<>(colSize);
gernotbelger@9150: 
gernotbelger@9150:         header.add(exportContextCSV.formatCsvHeader(GeneralResultType.station));
mschaefer@9157:         header.add(exportContextCSV.formatCsvHeader(SInfoResultType.years));
mschaefer@9157:         header.add(exportContextCSV.formatCsvHeader(SInfoResultType.collisionCount));
gernotbelger@9150: 
gernotbelger@9171:         exportContextCSV.writeCSVLine(header.toArray(new String[colSize]));
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9195:     protected String[] formatCSVRow(final ExportContextCSV exportContextCSV, final ResultRow row) {
gernotbelger@9195:         return formatRow(exportContextCSV, row);
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9195:     protected String[] formatPDFRow(final ExportContextPDF exportContextPDF, final ResultRow row) {
gernotbelger@9195: 
mschaefer@9157:         final Collection<String> lines = new ArrayList<>(3);
mschaefer@9157: 
gernotbelger@9195:         lines.add(exportContextPDF.formatRowValue(row, GeneralResultType.station));
gernotbelger@9195:         lines.add(exportContextPDF.formatRowValue(row, SInfoResultType.years));
gernotbelger@9195:         lines.add(exportContextPDF.formatRowValue(row, SInfoResultType.collisionCount));
gernotbelger@9195: 
mschaefer@9157:         return lines.toArray(new String[lines.size()]);
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9195:     private String[] formatRow(final ExportContextCSV exportContextCSV, final ResultRow row) {
gernotbelger@9150: 
mschaefer@9157:         final Collection<String> lines = new ArrayList<>(3);
gernotbelger@9150: 
gernotbelger@9150:         lines.add(exportContextCSV.formatRowValue(row, GeneralResultType.station));
mschaefer@9157:         lines.add(exportContextCSV.formatRowValue(row, SInfoResultType.years));
mschaefer@9157:         lines.add(exportContextCSV.formatRowValue(row, SInfoResultType.collisionCount));
gernotbelger@9150: 
gernotbelger@9150:         return lines.toArray(new String[lines.size()]);
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9150:     protected String getJasperFile() {
gernotbelger@9150:         return JASPER_FILE;
gernotbelger@9150:     }
gernotbelger@9150: 
gernotbelger@9150:     @Override
gernotbelger@9195:     protected void addJRTableHeader(final ExportContextPDF exportContext, final MetaAndTableJRDataSource source) {
gernotbelger@9150: 
gernotbelger@9150:         /* column headings */
gernotbelger@9195:         exportContext.addJRMetadata(source, "station_header", GeneralResultType.station);
gernotbelger@9195:         exportContext.addJRMetadata(source, "collision_years_header", SInfoResultType.years);
gernotbelger@9195:         exportContext.addJRMetadata(source, "collision_count_header", SInfoResultType.collisionCount);
gernotbelger@9150:     }
gernotbelger@9150: }