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 <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
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<SQResult []> 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<SQResult []>();
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<String[]> data2StringArrays(SQResult result) {
aheinecke@6322:         String km = Formatter.getSQRelationKM(context
aheinecke@6322:                     ).format(result.getKm());
aheinecke@6172:         List<String[]> retval = new ArrayList<String[]>();
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@6830:         source.addMetaData("outliers", Formatter.getRawFormatter(context).format(
aheinecke@6830:                     access.getOutliers()));
aheinecke@6172: 
aheinecke@6172:         source.addMetaData("calculation", Resources.getMsg(
aheinecke@6172:                                             locale,
aheinecke@6172:                                             PDF_HEADER_MODE,
aheinecke@6172:                                             "SQRelation"));
aheinecke@6859: 
aheinecke@6859:         String measurementStationName = access.getMeasurementStationName();
aheinecke@6859: 
aheinecke@6859:         if (measurementStationName != null) {
aheinecke@6859:             source.addMetaData("msName", measurementStationName);
aheinecke@6859:         }
aheinecke@6859: 
aheinecke@6859:         String measurementStationGaugeName = access.getMeasurementStationGaugeName();
aheinecke@6859: 
aheinecke@6859:         if (measurementStationGaugeName != null) {
aheinecke@6859:             source.addMetaData("msGauge", measurementStationGaugeName);
aheinecke@6859:         }
aheinecke@6859: 
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 :