Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedDifferenceExporter.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 | 5733d7f27196 |
children | 84349d440987 |
line wrap: on
line source
package org.dive4elements.river.exports.minfo; import java.io.IOException; import java.io.OutputStream; 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; public class BedDifferenceExporter extends AbstractExporter { /** Private logger. */ 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_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("init"); super.init(request, out, context); results = new BedDifferencesResult[0]; } @Override protected void writeCSVData(CSVWriter writer) throws IOException { writeCSVHeader(writer); NumberFormat kmf = Formatter.getCalculationKm(context.getMeta()); NumberFormat mf = Formatter.getMeterFormat(context); for (int i = 0; i < results.length; i++) { BedDifferencesResult result = results[i]; double[][] kms = result.getDifferencesData(); for (int j = 0; j < kms[0].length; j++) { writer.writeNext(new String[] { kmf.format(kms[0][j]), mf.format(kms[1][j])}); } } } @Override protected void addData(Object data) { if (!(data instanceof CalculationResult)) { logger.warn("Invalid data type."); return; } Object[] d = (Object[])((CalculationResult)data).getData(); if (!(d instanceof BedDifferencesResult[])) { logger.warn("Invalid result object."); return; } 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()"); List<String> header = new LinkedList<String>(); if (results != null) { header.add(msg(CSV_HEADER_KM, "km")); header.add(msg(CSV_HEADER_DIFF, "m")); } 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); } } }