Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/exports/MiddleBedHeightExporter.java @ 6759:6eb2b8087cfa
issue1416: exporter for sedimentload and i18n.
author | Felix Wolfsteller <felix.wolfsteller@intevation.de> |
---|---|
date | Tue, 06 Aug 2013 16:56:45 +0200 |
parents | 2c7a16d59f67 |
children | 0a337f0005c2 |
line wrap: on
line source
/* Copyright (C) 2011, 2012, 2013 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; import java.io.OutputStream; import java.text.NumberFormat; import java.util.ArrayList; import java.util.List; import org.w3c.dom.Document; import org.apache.log4j.Logger; import au.com.bytecode.opencsv.CSVWriter; import org.dive4elements.artifacts.CallContext; import org.dive4elements.river.artifacts.D4EArtifact; import org.dive4elements.river.artifacts.model.CalculationResult; import org.dive4elements.river.artifacts.model.MiddleBedHeightData; import org.dive4elements.river.utils.RiverUtils; import org.dive4elements.river.utils.Formatter; /** * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class MiddleBedHeightExporter extends AbstractExporter { /** Private logger. */ private static final Logger logger = Logger.getLogger(MiddleBedHeightExporter.class); public static final String CSV_KM = "export.bedheight_middle.csv.header.km"; public static final String CSV_SOUNDING = "export.bedheight_middle.csv.header.sounding"; public static final String CSV_HEIGHT = "export.bedheight_middle.csv.header.height"; public static final String CSV_UNCERTAINTY = "export.bedheight_middle.csv.header.uncertainty"; public static final String CSV_DATA_GAP = "export.bedheight_middle.csv.header.datagap"; public static final String CSV_SOUNDING_WIDTH = "export.bedheight_middle.csv.header.soundingwidth"; public static final String CSV_WIDTH = "export.bedheight_middle.csv.header.width"; public static final String CSV_LOCATIONS = "export.bedheight_middle.csv.header.locations"; protected List<MiddleBedHeightData[]> data; public void init(Document request, OutputStream out, CallContext cc) { super.init(request, out, cc); data = new ArrayList<MiddleBedHeightData[]>(); } @Override protected void addData(Object d) { if (d instanceof CalculationResult) { d = ((CalculationResult) d).getData(); if (d instanceof MiddleBedHeightData[]) { logger.debug("Add new data of type MiddleBedHeightData"); data.add((MiddleBedHeightData[]) d); } } } @Override protected void writeCSVData(CSVWriter writer) { logger.info("MiddleBedHeightExporter.writeCSVData"); logger.debug("CSV gets " + data.size() + " MiddleBedHeightData objects."); writeCSVHeader(writer); for (MiddleBedHeightData[] d: data) { data2CSV(writer, d); } } protected void writeCSVHeader(CSVWriter writer) { writer.writeNext(new String[] { msg(CSV_KM, CSV_KM), msg(CSV_SOUNDING, CSV_SOUNDING), msg(CSV_HEIGHT, CSV_HEIGHT), msg(CSV_UNCERTAINTY, CSV_UNCERTAINTY), msg(CSV_DATA_GAP, CSV_DATA_GAP), msg(CSV_SOUNDING_WIDTH, CSV_SOUNDING_WIDTH), msg(CSV_WIDTH, CSV_WIDTH), msg(CSV_LOCATIONS, CSV_LOCATIONS) }); } protected void data2CSV(CSVWriter writer, MiddleBedHeightData[] mData) { logger.debug("Add next MiddleBedHeightData to CSV"); D4EArtifact flys = (D4EArtifact) master; NumberFormat kmF = Formatter.getMiddleBedHeightKM(context); NumberFormat heightF = Formatter.getMiddleBedHeightHeight(context); NumberFormat uncertF = Formatter.getMiddleBedHeightUncert(context); NumberFormat gapF = Formatter.getMiddleBedHeightDataGap(context); NumberFormat soundF = Formatter.getMiddleBedHeightSounding(context); NumberFormat widthF = Formatter.getMiddleBedHeightWidth(context); heightF.setMaximumFractionDigits(1); soundF.setMaximumFractionDigits(1); for (MiddleBedHeightData data: mData) { for (int i = 0, n = data.size(); i < n; i++) { int start = data.getStartYear(); int end = data.getEndYear(); if (start == end) { String uncert = !Double.isNaN(data.getUncertainty(i)) ? uncertF.format(data.getUncertainty(i)) : ""; String gap = !Double.isNaN(data.getDataGap(i)) ? gapF.format(data.getDataGap(i)) + "%" : ""; String sound = !Double.isNaN(data.getSoundingWidth(i)) ? soundF.format(data.getSoundingWidth(i)) : ""; writer.writeNext(new String[] { kmF.format(data.getKM(i)), data.getDescription(), heightF.format(data.getMiddleHeight(i)), uncert, gap, sound, widthF.format(data.getWidth(i)), RiverUtils.getLocationDescription(flys, data.getKM(i)), }); } else { writer.writeNext(new String[] { kmF.format(data.getKM(i)), data.getDescription(), heightF.format(data.getMiddleHeight(i)), "", "", "", "", RiverUtils.getLocationDescription(flys, data.getKM(i)), }); } } } } @Override protected void writePDF(OutputStream out) { logger.error("TODO: Implement FlowVelocityExporter.writePDF"); } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :