Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/collision/CollisionCalcOverviewResult.java @ 9215:0fc9c82e744e
work on collison, flood_duration
author | gernotbelger |
---|---|
date | Tue, 03 Jul 2018 17:00:48 +0200 |
parents | 3dae6b78e1da |
children | e5367900dd6d |
line wrap: on
line source
/* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde * Software engineering by * Björnsen Beratende Ingenieure GmbH * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt * * This file is Free Software under the GNU AGPL (>=v3) * and comes with ABSOLUTELY NO WARRANTY! Check out the * documentation coming with Dive4Elements River for details. */ package org.dive4elements.river.artifacts.sinfo.collision; import java.util.ArrayList; import java.util.Collection; import org.dive4elements.river.artifacts.common.AbstractCalculationExportableResult; import org.dive4elements.river.artifacts.common.ExportContextCSV; import org.dive4elements.river.artifacts.common.ExportContextPDF; import org.dive4elements.river.artifacts.common.GeneralResultType; import org.dive4elements.river.artifacts.common.IResultType; import org.dive4elements.river.artifacts.common.MetaAndTableJRDataSource; import org.dive4elements.river.artifacts.common.ResultRow; import org.dive4elements.river.artifacts.sinfo.common.SInfoResultType; import org.dive4elements.river.artifacts.sinfo.util.RiverInfo; import gnu.trove.TDoubleArrayList; /** * Contains the results of a {@link CollisionCalculation}. * * @author Gernot Belger */ final public class CollisionCalcOverviewResult extends AbstractCalculationExportableResult { private static final long serialVersionUID = 1L; private static final String JASPER_FILE = "/jasper/templates/sinfo.collision.overview.jrxml"; private final int[] singleYears; public CollisionCalcOverviewResult(final String label, final int[] singleYears, final Collection<ResultRow> rows) { super(label, rows); this.singleYears = singleYears; } public int[] getSingleYears() { return this.singleYears; } public double[][] getStationPointsByYear(final IResultType type, final int year) { final TDoubleArrayList xPoints = new TDoubleArrayList(this.rows.size()); final TDoubleArrayList yPoints = new TDoubleArrayList(this.rows.size()); for (final ResultRow row : this.rows) { if ((this.singleYears == null) || (Integer.valueOf(row.getValue(SInfoResultType.years).toString()) == year)) { final double station = row.getDoubleValue(GeneralResultType.station); final double value = row.getDoubleValue(type); xPoints.add(station); yPoints.add(value); } } return new double[][] { xPoints.toNativeArray(), yPoints.toNativeArray() }; } @Override protected void writeCSVResultMetadata(final ExportContextCSV exportContextCSV) { // do nothing so far } @Override public void writeCSVHeader(final ExportContextCSV exportContextCSV, final RiverInfo river) { final int colSize = 3; exportContextCSV.writeTitleForTabs("sinfo.export.csv.title.collison.overview", 3); // Voraussetzung für Tabs ist, dass der Titel vor den Headern // geschrieben wird. // Das ist etwas doof. final Collection<String> header = new ArrayList<>(colSize); header.add(exportContextCSV.formatCsvHeader(GeneralResultType.station)); header.add(exportContextCSV.formatCsvHeader(SInfoResultType.years)); header.add(exportContextCSV.formatCsvHeader(SInfoResultType.collisionCount)); exportContextCSV.writeCSVLine(header.toArray(new String[colSize])); } @Override protected String[] formatCSVRow(final ExportContextCSV exportContextCSV, final ResultRow row) { return formatRow(exportContextCSV, row); } @Override protected String[] formatPDFRow(final ExportContextPDF exportContextPDF, final ResultRow row) { final Collection<String> lines = new ArrayList<>(3); lines.add(exportContextPDF.formatRowValue(row, GeneralResultType.station)); lines.add(exportContextPDF.formatRowValue(row, SInfoResultType.years)); lines.add(exportContextPDF.formatRowValue(row, SInfoResultType.collisionCount)); return lines.toArray(new String[lines.size()]); } private String[] formatRow(final ExportContextCSV exportContextCSV, final ResultRow row) { final Collection<String> lines = new ArrayList<>(3); lines.add(exportContextCSV.formatRowValue(row, GeneralResultType.station)); lines.add(exportContextCSV.formatRowValue(row, SInfoResultType.years)); lines.add(exportContextCSV.formatRowValue(row, SInfoResultType.collisionCount)); return lines.toArray(new String[lines.size()]); } @Override protected String getJasperFile() { return JASPER_FILE; } @Override protected void addJRTableHeader(final ExportContextPDF exportContext, final MetaAndTableJRDataSource source) { /* column headings */ exportContext.addJRMetadata(source, "station_header", GeneralResultType.station); exportContext.addJRMetadata(source, "collision_years_header", SInfoResultType.years); exportContext.addJRMetadata(source, "collision_count_header", SInfoResultType.collisionCount); } }