Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedQualityExporter.java @ 8587:07c9ac22f611
(issue1755) Generalise BedQuality result handling
The bedquality calculation now produces a result for each time period
which has BedQualityResultValues for each specific result type.
Formally this was split up in density, porosity and diameter classes
with some bedload diameter classes mixed in for extra fun.
The intent of this commit is to allow more shared code and generic
access patterns to the BedQuality results.
author | Andre Heinecke <andre.heinecke@intevation.de> |
---|---|
date | Wed, 18 Mar 2015 18:42:08 +0100 |
parents | 073ea4bcea58 |
children | 717b58b158c9 |
line wrap: on
line source
/* Copyright (C) 2011, 2012, 2013, 2015 by Bundesanstalt für Gewässerkunde * Software engineering by Intevation GmbH * * 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.exports.minfo; import gnu.trove.TDoubleArrayList; import java.io.IOException; import java.io.OutputStream; import java.text.DateFormat; import java.text.NumberFormat; import java.util.Arrays; import java.util.ArrayList; import java.util.List; import org.apache.log4j.Logger; import au.com.bytecode.opencsv.CSVWriter; import org.dive4elements.river.artifacts.model.CalculationResult; import org.dive4elements.river.artifacts.model.minfo.BedQualityResult; import org.dive4elements.river.artifacts.model.minfo.BedQualityResultValue; import org.dive4elements.river.exports.AbstractExporter; import org.dive4elements.river.utils.Formatter; import org.dive4elements.river.artifacts.access.RangeAccess; import org.dive4elements.river.artifacts.D4EArtifact; public class BedQualityExporter extends AbstractExporter { /** Private log. */ private static Logger log = Logger.getLogger(BedQualityExporter.class); private static final String CSV_HEADER_KM = "export.minfo.bedquality.km"; private static final String CSV_HEADER_BASE = "export.minfo.bedquality"; private BedQualityResult[] results; public BedQualityExporter() { results = new BedQualityResult[0]; } /** Create double[] containing the data for rows in csv. */ private List<double[]> createDataRows() { double[] kms = new RangeAccess((D4EArtifact) master).getKmSteps(); int cols = 1; for (BedQualityResult result: results) { cols += result.getValues().size(); } List<double[]> rows = new ArrayList<double[]>(kms.length); for (double km: kms) { double[] row = new double[cols]; row[0] = km; for (BedQualityResult result: results) { int i = 1; for (BedQualityResultValue value: result.getValues()) { row[i++] = value.getDataInterpolated(km); } } rows.add(row); } return rows; } @Override protected void writeCSVData(CSVWriter writer) throws IOException { writeCSVHeader(writer); NumberFormat nf = Formatter.getFormatter(context, 1, 3); for (double[] d : createDataRows()) { log.debug("row + " + Arrays.toString(d)); List<String> cells = new ArrayList<String>(d.length); for (int i = 0; i < d.length; i++) { if (!Double.isNaN(d[i])) { cells.add(nf.format(d[i])); } else { cells.add(""); } } writer.writeNext(cells.toArray(new String[cells.size()])); } } @Override protected void writePDF(OutputStream out) { // TODO Auto-generated method stub } @Override protected void addData(Object data) { log.debug("addData()"); if (!(data instanceof CalculationResult)) { log.warn("Invalid data type."); return; } Object[] d = (Object[])((CalculationResult)data).getData(); if (!(d instanceof BedQualityResult[])) { log.warn("Invalid result object."); return; } results = (BedQualityResult[])d; } protected void writeCSVHeader(CSVWriter writer) { log.debug("writeCSVHeader()"); List<String> header = new ArrayList<String>(); if (results == null) { writer.writeNext(header.toArray(new String[header.size()])); return; } header.add(msg(CSV_HEADER_KM, "km")); DateFormat df = Formatter.getDateFormatter(context.getMeta(), "dd.MM.yyyy"); for (BedQualityResult result: results) { String d1 = df.format(result.getDateRange().getFrom()); String d2 = df.format(result.getDateRange().getTo()); for (BedQualityResultValue value: result.getValues()) { String i18n; if (value.isDiameterResult()) { i18n = CSV_HEADER_BASE + ".diameter." + value.getType(); header.add(msg(i18n, i18n) + " - " + value.getName() + " - " + d1 + "-" + d2); } else { i18n = CSV_HEADER_BASE + "." + value.getName() + "." + value.getType(); header.add(msg(i18n, i18n) + " - " + d1 + "-" + d2); } } } writer.writeNext(header.toArray(new String[header.size()])); } }