teichmann@5863: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
teichmann@5863: * Software engineering by Intevation GmbH
teichmann@5863: *
teichmann@5994: * This file is Free Software under the GNU AGPL (>=v3)
teichmann@5863: * and comes with ABSOLUTELY NO WARRANTY! Check out the
teichmann@5994: * documentation coming with Dive4Elements River for details.
teichmann@5863: */
teichmann@5863:
teichmann@5831: package org.dive4elements.river.exports.sq;
ingo@3077:
ingo@3077: import java.io.OutputStream;
ingo@3077: import java.util.ArrayList;
ingo@3077: import java.util.List;
aheinecke@6172: import java.util.Map;
aheinecke@6172: import java.util.HashMap;
aheinecke@6172: import java.util.Date;
aheinecke@6172: import java.util.Locale;
aheinecke@6172: import java.text.DateFormat;
aheinecke@6828: import java.text.NumberFormat;
ingo@3077:
ingo@3077: import org.w3c.dom.Document;
ingo@3077:
aheinecke@6172: import net.sf.jasperreports.engine.JasperExportManager;
aheinecke@6172: import net.sf.jasperreports.engine.JasperFillManager;
aheinecke@6172: import net.sf.jasperreports.engine.JasperPrint;
aheinecke@6172: import net.sf.jasperreports.engine.JRException;
aheinecke@6172:
ingo@3077: import au.com.bytecode.opencsv.CSVWriter;
ingo@3077:
teichmann@5831: import org.dive4elements.artifacts.CallContext;
aheinecke@6172: import org.dive4elements.artifacts.CallMeta;
sascha@3392:
teichmann@5831: import org.dive4elements.river.artifacts.model.CalculationResult;
teichmann@5831: import org.dive4elements.river.artifacts.model.sq.SQFractionResult;
teichmann@5831: import org.dive4elements.river.artifacts.model.sq.SQResult;
aheinecke@6172: import org.dive4elements.river.artifacts.model.sq.SQRelationJRDataSource;
teichmann@5831: import org.dive4elements.river.artifacts.model.Parameters;
aheinecke@6172: import org.dive4elements.river.artifacts.model.DateRange;
aheinecke@6172: import org.dive4elements.river.artifacts.access.SQRelationAccess;
aheinecke@6172:
aheinecke@6172: import org.dive4elements.river.artifacts.resources.Resources;
aheinecke@6172:
aheinecke@6172: import org.dive4elements.river.artifacts.D4EArtifact;
teichmann@5831:
teichmann@5831: import org.dive4elements.river.exports.AbstractExporter;
ingo@3077:
aheinecke@6172: import org.dive4elements.river.utils.RiverUtils;
aheinecke@6172: import org.dive4elements.river.utils.Formatter;
aheinecke@6172:
aheinecke@6172: import org.dive4elements.artifacts.common.utils.Config;
aheinecke@6172:
sascha@3392: import org.apache.log4j.Logger;
ingo@3077:
ingo@3077: /**
ingo@3077: * @author Ingo Weinzierl
ingo@3077: */
ingo@3077: public class SQRelationExporter extends AbstractExporter {
ingo@3077:
ingo@3077: /** Private logger. */
ingo@3077: private static final Logger logger =
ingo@3077: Logger.getLogger(SQRelationExporter.class);
ingo@3077:
ingo@3077:
ingo@3077: public static final String CSV_PARAMETER =
ingo@3077: "export.sqrelation.csv.header.parameter";
ingo@3077:
ingo@3077: public static final String CSV_STATION =
ingo@3077: "export.sqrelation.csv.header.station";
ingo@3077:
ingo@3077: public static final String CSV_KM =
ingo@3077: "export.sqrelation.csv.header.km";
ingo@3077:
ingo@3077: public static final String CSV_FUNCTION =
ingo@3077: "export.sqrelation.csv.header.function";
ingo@3077:
ingo@3077: public static final String CSV_GAUGE =
ingo@3077: "export.sqrelation.csv.header.gauge";
ingo@3077:
ingo@3077: public static final String CSV_COEFF_A =
ingo@3077: "export.sqrelation.csv.header.coeff.a";
ingo@3077:
ingo@3077: public static final String CSV_COEFF_B =
ingo@3077: "export.sqrelation.csv.header.coeff.b";
ingo@3077:
ingo@3077: public static final String CSV_COEFF_Q =
ingo@3077: "export.sqrelation.csv.header.coeff.q";
ingo@3077:
ingo@3077: public static final String CSV_COEFF_R =
ingo@3077: "export.sqrelation.csv.header.coeff.r";
ingo@3077:
ingo@3077: public static final String CSV_N_TOTAL =
ingo@3077: "export.sqrelation.csv.header.n.total";
ingo@3077:
ingo@3077: public static final String CSV_N_OUTLIERS =
ingo@3077: "export.sqrelation.csv.header.n.outliers";
ingo@3077:
ingo@3077: public static final String CSV_C_DUAN =
ingo@3077: "export.sqrelation.csv.header.c.duan";
ingo@3077:
ingo@3077: public static final String CSV_C_FERGUSON =
ingo@3077: "export.sqrelation.csv.header.c.ferguson";
ingo@3077:
aheinecke@6828: public static final String CSV_QMAX =
aheinecke@6828: "export.sqrelation.csv.header.qmax";
aheinecke@6828:
aheinecke@6828: public static final String CSV_SD =
aheinecke@6828: "export.sqrelation.csv.header.sd";
ingo@3077:
aheinecke@6172: public static final String PDF_TITLE=
aheinecke@6172: "export.sqrelation.pdf.title";
aheinecke@6172:
aheinecke@6172: public static final String PDF_HEADER_MODE =
aheinecke@6172: "export.sqrelation.pdf.mode";
aheinecke@6172:
aheinecke@6172: public static final String JASPER_FILE =
aheinecke@6172: "export.sqrelation.pdf.file";
ingo@3077:
sascha@3392: protected List data;
ingo@3077:
ingo@3077:
ingo@3077: public void init(Document request, OutputStream out, CallContext cc) {
ingo@3077: super.init(request, out, cc);
sascha@3392: data = new ArrayList();
ingo@3077: }
ingo@3077:
ingo@3077:
ingo@3077: @Override
ingo@3077: protected void addData(Object d) {
ingo@3077: if (d instanceof CalculationResult) {
sascha@3392: d = ((CalculationResult)d).getData();
sascha@3392: if (d instanceof SQResult []) {
sascha@3392: data.add((SQResult [])d);
ingo@3077: }
ingo@3077: }
ingo@3077: }
ingo@3077:
sascha@3392: protected void writeCSVHeader(CSVWriter writer) {
sascha@3392: writer.writeNext(new String[] {
sascha@3392: msg(CSV_KM , CSV_KM),
sascha@3392: msg(CSV_PARAMETER, CSV_PARAMETER),
sascha@3392: msg(CSV_COEFF_A , CSV_COEFF_A),
sascha@3392: msg(CSV_COEFF_B , CSV_COEFF_B),
aheinecke@6828: msg(CSV_SD, CSV_SD),
aheinecke@6828: msg(CSV_QMAX, CSV_QMAX),
aheinecke@6828: msg(CSV_COEFF_R, CSV_COEFF_R),
sascha@3392: msg(CSV_N_TOTAL , CSV_N_TOTAL),
sascha@3392: msg(CSV_N_OUTLIERS, CSV_N_OUTLIERS),
aheinecke@6828: msg(CSV_C_DUAN, CSV_C_DUAN),
aheinecke@6828: msg(CSV_C_FERGUSON, CSV_C_FERGUSON)
sascha@3392: });
sascha@3392: }
ingo@3077:
ingo@3077: @Override
ingo@3077: protected void writeCSVData(CSVWriter writer) {
ingo@3077: logger.debug("writeCSVData");
ingo@3077:
ingo@3077: writeCSVHeader(writer);
ingo@3077:
sascha@3392: for (SQResult [] results: data) {
sascha@3392: for (SQResult result: results) {
aheinecke@6172: writer.writeAll(data2StringArrays(result));
sascha@3392: }
ingo@3077: }
ingo@3077: }
ingo@3077:
aheinecke@6172: protected List data2StringArrays(SQResult result) {
aheinecke@6322: String km = Formatter.getSQRelationKM(context
aheinecke@6322: ).format(result.getKm());
aheinecke@6172: List retval = new ArrayList();
ingo@3077:
aheinecke@6828: NumberFormat sqAFormatter = Formatter.getSQRelationA(context);
aheinecke@6828: NumberFormat sqBFormatter = Formatter.getSQRelationB(context);
aheinecke@6828: NumberFormat fThreeFormatter = Formatter.getFormatter(context, 3, 3);
aheinecke@6828: NumberFormat fTwoFormatter = Formatter.getFormatter(context, 2, 2);
aheinecke@6828: NumberFormat fZeroFormatter = Formatter.getFormatter(context, 0, 0);
aheinecke@6828:
sascha@3392: for (int i = 0; i < SQResult.NUMBER_FRACTIONS; ++i) {
sascha@3392: SQFractionResult fraction = result.getFraction(i);
sascha@3552:
sascha@3552: String name = result.getFractionName(i);
sascha@3552:
sascha@3552: Parameters parameters = fraction.getParameters();
sascha@3552:
sascha@3552: if (parameters == null) {
sascha@3392: continue;
sascha@3392: }
sascha@3552:
aheinecke@6828: String a, b, sd, o, t, max_q, c_ferguson, c_duan, r2;
aheinecke@6828: a = sqAFormatter.format(parameters.getValue(0, "a"));
aheinecke@6828: b = sqBFormatter.format(parameters.getValue(0, "b"));
aheinecke@6828:
aheinecke@6828: /* The std_dev parameter contains the standard error actually */
aheinecke@6828: sd = fThreeFormatter.format(parameters.getValue(0, "std_dev"));
aheinecke@6828: max_q = fZeroFormatter.format(parameters.getValue(0, "max_q"));
aheinecke@6828: c_ferguson = fTwoFormatter.format(parameters.getValue(0, "c_ferguson"));
aheinecke@6828: c_duan = fTwoFormatter.format(parameters.getValue(0, "c_duan"));
aheinecke@6828: r2 = fTwoFormatter.format(parameters.getValue(0, "r2"));
aheinecke@6828:
sascha@3552:
aheinecke@6172: o = String.valueOf(fraction.totalNumOutliers());
aheinecke@6172: t = String.valueOf(fraction.numMeasurements() + o);
aheinecke@6172:
aheinecke@6172: retval.add(new String[] {
sascha@3392: km,
sascha@3392: name,
aheinecke@6172: a,
aheinecke@6172: b,
aheinecke@6828: sd, // 4
aheinecke@6828: max_q, // 5
aheinecke@6828: r2, // 6
aheinecke@6828: t, // 7
aheinecke@6828: o, // 8
aheinecke@6828: c_duan, // 9
aheinecke@6828: c_ferguson // 10
sascha@3392: });
sascha@3392: }
aheinecke@6172: return retval;
ingo@3077: }
ingo@3077:
ingo@3077:
aheinecke@6172: protected SQRelationJRDataSource createJRData() {
aheinecke@6172: SQRelationJRDataSource source = new SQRelationJRDataSource();
aheinecke@6172:
aheinecke@6172: addMetaData(source);
aheinecke@6172: for (SQResult [] results: data) {
aheinecke@6172: for (SQResult result: results) {
aheinecke@6172: for (String[] res: data2StringArrays(result)) {
aheinecke@6172: source.addData(res);
aheinecke@6172: }
aheinecke@6172: }
aheinecke@6172: }
aheinecke@6172: return source;
aheinecke@6172: }
aheinecke@6172:
aheinecke@6172: protected void addMetaData(SQRelationJRDataSource source) {
aheinecke@6172: CallMeta meta = context.getMeta();
aheinecke@6172:
aheinecke@6172: D4EArtifact arti = (D4EArtifact) master;
aheinecke@6172:
aheinecke@6172: source.addMetaData ("river", RiverUtils.getRivername(arti));
aheinecke@6172:
aheinecke@6172: Locale locale = Resources.getLocale(meta);
aheinecke@6172: DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, locale);
aheinecke@6172:
aheinecke@6172: source.addMetaData("date", df.format(new Date()));
aheinecke@6172:
aheinecke@6172: SQRelationAccess access = new SQRelationAccess(arti);
aheinecke@6175: source.addMetaData("location", "KM " + getKmFormatter().format(access.getLocation()));
aheinecke@6172:
aheinecke@6172: DateRange period = access.getPeriod();
aheinecke@6172: source.addMetaData("periods", df.format(period.getFrom()) + " - " +
aheinecke@6172: df.format(period.getTo()));
aheinecke@6172:
aheinecke@6175: source.addMetaData("outliertest", Resources.getMsg(meta,
aheinecke@6175: access.getOutlierMethod(),
aheinecke@6175: access.getOutlierMethod()));
aheinecke@6172: source.addMetaData("outliers", access.getOutliers().toString());
aheinecke@6172:
aheinecke@6172: source.addMetaData("calculation", Resources.getMsg(
aheinecke@6172: locale,
aheinecke@6172: PDF_HEADER_MODE,
aheinecke@6172: "SQRelation"));
aheinecke@6172: }
aheinecke@6172:
ingo@3077: @Override
ingo@3077: protected void writePDF(OutputStream out) {
aheinecke@6172: logger.debug("write PDF");
aheinecke@6172: SQRelationJRDataSource source = createJRData();
aheinecke@6172:
aheinecke@6172: String jasperFile = Resources.getMsg(
aheinecke@6172: context.getMeta(),
aheinecke@6172: JASPER_FILE,
aheinecke@6172: "/jasper/sqrelation_en.jasper");
aheinecke@6172: String confPath = Config.getConfigDirectory().toString();
aheinecke@6172:
aheinecke@6172:
aheinecke@6172: Map parameters = new HashMap();
aheinecke@6172: parameters.put("ReportTitle", Resources.getMsg(
aheinecke@6172: context.getMeta(), PDF_TITLE, "Exported Data"));
aheinecke@6172: try {
aheinecke@6172: JasperPrint print = JasperFillManager.fillReport(
aheinecke@6172: confPath + jasperFile,
aheinecke@6172: parameters,
aheinecke@6172: source);
aheinecke@6172: JasperExportManager.exportReportToPdfStream(print, out);
aheinecke@6172: }
aheinecke@6172: catch(JRException je) {
aheinecke@6172: logger.warn("Error generating PDF Report!", je);
aheinecke@6172: }
ingo@3077: }
ingo@3077: }
ingo@3077: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :