Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/exports/sq/SQRelationExporter.java @ 6601:5ecc6d4d73f2
Add official fixings to Waterlevel CSV Export (issue1384)
This searches the collection for staticwqkms artifacts that
contain official data and adds that data to the export.
The data is filtered by the calculation range and sorted by the
calculation direction.
author | Andre Heinecke <aheinecke@intevation.de> |
---|---|
date | Thu, 18 Jul 2013 13:16:33 +0200 |
parents | 8c976bb0362f |
children | 708416bb560c |
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.sq; import java.io.OutputStream; import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.HashMap; import java.util.Date; import java.util.Locale; import java.text.DateFormat; import org.w3c.dom.Document; import net.sf.jasperreports.engine.JasperExportManager; import net.sf.jasperreports.engine.JasperFillManager; import net.sf.jasperreports.engine.JasperPrint; import net.sf.jasperreports.engine.JRException; import au.com.bytecode.opencsv.CSVWriter; import org.dive4elements.artifacts.CallContext; import org.dive4elements.artifacts.CallMeta; import org.dive4elements.river.artifacts.model.CalculationResult; import org.dive4elements.river.artifacts.model.sq.SQFractionResult; import org.dive4elements.river.artifacts.model.sq.SQResult; import org.dive4elements.river.artifacts.model.sq.SQRelationJRDataSource; import org.dive4elements.river.artifacts.model.Parameters; import org.dive4elements.river.artifacts.model.DateRange; import org.dive4elements.river.artifacts.access.SQRelationAccess; import org.dive4elements.river.artifacts.resources.Resources; import org.dive4elements.river.artifacts.D4EArtifact; import org.dive4elements.river.exports.AbstractExporter; import org.dive4elements.river.utils.RiverUtils; import org.dive4elements.river.utils.Formatter; import org.dive4elements.artifacts.common.utils.Config; import org.apache.log4j.Logger; /** * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class SQRelationExporter extends AbstractExporter { /** Private logger. */ private static final Logger logger = Logger.getLogger(SQRelationExporter.class); public static final String CSV_PARAMETER = "export.sqrelation.csv.header.parameter"; public static final String CSV_STATION = "export.sqrelation.csv.header.station"; public static final String CSV_KM = "export.sqrelation.csv.header.km"; public static final String CSV_FUNCTION = "export.sqrelation.csv.header.function"; public static final String CSV_GAUGE = "export.sqrelation.csv.header.gauge"; public static final String CSV_COEFF_A = "export.sqrelation.csv.header.coeff.a"; public static final String CSV_COEFF_B = "export.sqrelation.csv.header.coeff.b"; public static final String CSV_COEFF_Q = "export.sqrelation.csv.header.coeff.q"; public static final String CSV_COEFF_R = "export.sqrelation.csv.header.coeff.r"; public static final String CSV_N_TOTAL = "export.sqrelation.csv.header.n.total"; public static final String CSV_N_OUTLIERS = "export.sqrelation.csv.header.n.outliers"; public static final String CSV_C_DUAN = "export.sqrelation.csv.header.c.duan"; public static final String CSV_C_FERGUSON = "export.sqrelation.csv.header.c.ferguson"; public static final String CSV_VARIANCE = "export.sqrelation.csv.header.variance"; public static final String PDF_TITLE= "export.sqrelation.pdf.title"; public static final String PDF_HEADER_MODE = "export.sqrelation.pdf.mode"; public static final String JASPER_FILE = "export.sqrelation.pdf.file"; protected List<SQResult []> data; public void init(Document request, OutputStream out, CallContext cc) { super.init(request, out, cc); data = new ArrayList<SQResult []>(); } @Override protected void addData(Object d) { if (d instanceof CalculationResult) { d = ((CalculationResult)d).getData(); if (d instanceof SQResult []) { data.add((SQResult [])d); } } } protected void writeCSVHeader(CSVWriter writer) { writer.writeNext(new String[] { msg(CSV_KM , CSV_KM), msg(CSV_PARAMETER, CSV_PARAMETER), msg(CSV_COEFF_A , CSV_COEFF_A), msg(CSV_COEFF_B , CSV_COEFF_B), msg(CSV_N_TOTAL , CSV_N_TOTAL), msg(CSV_N_OUTLIERS, CSV_N_OUTLIERS), msg(CSV_VARIANCE , CSV_VARIANCE) }); } @Override protected void writeCSVData(CSVWriter writer) { logger.debug("writeCSVData"); writeCSVHeader(writer); for (SQResult [] results: data) { for (SQResult result: results) { writer.writeAll(data2StringArrays(result)); } } } protected List<String[]> data2StringArrays(SQResult result) { String km = Formatter.getSQRelationKM(context ).format(result.getKm()); List<String[]> retval = new ArrayList<String[]>(); for (int i = 0; i < SQResult.NUMBER_FRACTIONS; ++i) { SQFractionResult fraction = result.getFraction(i); String name = result.getFractionName(i); Parameters parameters = fraction.getParameters(); if (parameters == null) { continue; } String a, b, sd, o, t; a = Formatter.getSQRelationA(context).format(parameters.getValue(0, "a")); b = Formatter.getSQRelationB(context).format(parameters.getValue(0, "b")); sd = Formatter.getVariance(context).format(Math.sqrt(parameters.getValue(0, "std_dev"))); o = String.valueOf(fraction.totalNumOutliers()); t = String.valueOf(fraction.numMeasurements() + o); retval.add(new String[] { km, name, a, b, t, o, sd }); } return retval; } protected SQRelationJRDataSource createJRData() { SQRelationJRDataSource source = new SQRelationJRDataSource(); addMetaData(source); for (SQResult [] results: data) { for (SQResult result: results) { for (String[] res: data2StringArrays(result)) { source.addData(res); } } } return source; } protected void addMetaData(SQRelationJRDataSource source) { CallMeta meta = context.getMeta(); D4EArtifact arti = (D4EArtifact) master; source.addMetaData ("river", RiverUtils.getRivername(arti)); Locale locale = Resources.getLocale(meta); DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, locale); source.addMetaData("date", df.format(new Date())); SQRelationAccess access = new SQRelationAccess(arti); source.addMetaData("location", "KM " + getKmFormatter().format(access.getLocation())); DateRange period = access.getPeriod(); source.addMetaData("periods", df.format(period.getFrom()) + " - " + df.format(period.getTo())); source.addMetaData("outliertest", Resources.getMsg(meta, access.getOutlierMethod(), access.getOutlierMethod())); source.addMetaData("outliers", access.getOutliers().toString()); source.addMetaData("calculation", Resources.getMsg( locale, PDF_HEADER_MODE, "SQRelation")); } @Override protected void writePDF(OutputStream out) { logger.debug("write PDF"); SQRelationJRDataSource source = createJRData(); String jasperFile = Resources.getMsg( context.getMeta(), JASPER_FILE, "/jasper/sqrelation_en.jasper"); String confPath = Config.getConfigDirectory().toString(); Map parameters = new HashMap(); parameters.put("ReportTitle", Resources.getMsg( context.getMeta(), PDF_TITLE, "Exported Data")); try { JasperPrint print = JasperFillManager.fillReport( confPath + jasperFile, parameters, source); JasperExportManager.exportReportToPdfStream(print, out); } catch(JRException je) { logger.warn("Error generating PDF Report!", je); } } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :