diff artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedDifferenceExporter.java @ 6242:34f393434913

Implement pdf export for bedheight differences
author Andre Heinecke <aheinecke@intevation.de>
date Mon, 10 Jun 2013 13:06:03 +0200
parents e213b538f78c
children 5733d7f27196
line wrap: on
line diff
--- a/artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedDifferenceExporter.java	Mon Jun 10 13:04:48 2013 +0200
+++ b/artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedDifferenceExporter.java	Mon Jun 10 13:06:03 2013 +0200
@@ -7,13 +7,33 @@
 import java.text.NumberFormat;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
+import java.util.HashMap;
+import java.util.Locale;
+import java.util.Date;
+import java.text.DateFormat;
 
 import org.apache.log4j.Logger;
+import org.apache.commons.lang.StringUtils;
+
+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 org.dive4elements.artifacts.CallContext;
+import org.dive4elements.artifacts.CallMeta;
+import org.dive4elements.river.artifacts.D4EArtifact;
+import org.dive4elements.river.artifacts.access.BedDifferencesAccess;
+import org.dive4elements.artifacts.common.utils.Config;
+import org.dive4elements.river.artifacts.resources.Resources;
 import org.dive4elements.river.artifacts.model.CalculationResult;
 import org.dive4elements.river.artifacts.model.minfo.BedDifferencesResult;
+import org.dive4elements.river.artifacts.model.minfo.BedDifferenceJRDataSource;
 import org.dive4elements.river.exports.AbstractExporter;
 import org.dive4elements.river.utils.Formatter;
+import org.dive4elements.river.utils.RiverUtils;
+
 import org.w3c.dom.Document;
 
 import au.com.bytecode.opencsv.CSVWriter;
@@ -27,18 +47,30 @@
     private static Logger logger =
         Logger.getLogger(BedDifferenceExporter.class);
 
-    private static final String CSV_HEADER_KM = "export.minfo.beddifference.km";
+    private static final String CSV_HEADER_KM =
+        "export.minfo.beddifference.km";
+
     private static final String CSV_HEADER_DIFF =
         "export.minfo.beddifference.diff";
 
+    public static final String JASPER_FILE =
+        "export.minfo.beddifference.pdf.file";
+
+    public static final String PDF_TITLE=
+        "export.minfo.beddifference.pdf.title";
+
+    public static final String PDF_HEADER_MODE=
+        "export.minfo.beddifference.pdf.mode";
+
     private BedDifferencesResult[] results;
 
     @Override
     public void init(Document request, OutputStream out, CallContext context) {
-        logger.debug("BedQualityExporter.init");
+        logger.debug("init");
         super.init(request, out, context);
         results = new BedDifferencesResult[0];
     }
+
     @Override
     protected void writeCSVData(CSVWriter writer) throws IOException {
         writeCSVHeader(writer);
@@ -56,12 +88,6 @@
     }
 
     @Override
-    protected void writePDF(OutputStream out) {
-        // TODO implement me!
-
-    }
-
-    @Override
     protected void addData(Object data) {
         if (!(data instanceof CalculationResult)) {
             logger.warn("Invalid data type.");
@@ -76,6 +102,35 @@
         results = (BedDifferencesResult[])d;
     }
 
+    protected void addMetaData(BedDifferenceJRDataSource 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()));
+
+        source.addMetaData("calculation", Resources.getMsg(
+                                            locale,
+                                            PDF_HEADER_MODE,
+                                            "Bedheight difference"));
+
+        // Now the dynamic parts
+
+        BedDifferencesAccess access = new BedDifferencesAccess(arti);
+        source.addMetaData("ye_mode", Resources.getMsg(locale,
+                    "state.minfo." + access.getYearEpoch(),
+                    access.getYearEpoch()));
+
+        source.addMetaData("differences", StringUtils.join(
+                access.getDifferenceArtifactNamePairs(), "\n"));
+    }
+
+
     protected void writeCSVHeader(CSVWriter writer) {
         logger.debug("writeCSVHeader()");
 
@@ -86,4 +141,48 @@
         }
         writer.writeNext(header.toArray(new String[header.size()]));
     }
+
+    protected BedDifferenceJRDataSource createJRData() {
+        BedDifferenceJRDataSource source = new BedDifferenceJRDataSource();
+
+        addMetaData(source);
+        NumberFormat kmf = Formatter.getCalculationKm(context.getMeta());
+        NumberFormat mf = Formatter.getMeterFormat(context);
+        for (BedDifferencesResult result: results) {
+            double[][] kms = result.getDifferencesData();
+            for (int j = 0; j < kms[0].length; j++) {
+                source.addData(new String[] {
+                    kmf.format(kms[0][j]), mf.format(kms[1][j])});
+            }
+        }
+        return source;
+    }
+
+    @Override
+    protected void writePDF(OutputStream out) {
+        logger.debug("write PDF");
+        BedDifferenceJRDataSource source = createJRData();
+
+        String jasperFile = Resources.getMsg(
+                                context.getMeta(),
+                                JASPER_FILE,
+                                "/jasper/beddifference_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);
+        }
+    }
+
 }

http://dive4elements.wald.intevation.org