teichmann@5863: /* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde
teichmann@5863:  * Software engineering by Intevation GmbH
teichmann@5863:  *
teichmann@5863:  * This file is Free Software under the GNU AGPL (>=v3) 
teichmann@5863:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
teichmann@5863:  * documentation coming with Dive4Elements River for details. 
teichmann@5863:  */
teichmann@5863: 
teichmann@5831: package org.dive4elements.river.exports;
ingo@2714: 
ingo@2714: import java.io.OutputStream;
ingo@2714: import java.text.NumberFormat;
ingo@2714: import java.util.ArrayList;
ingo@2714: import java.util.List;
ingo@2714: 
ingo@2714: import org.w3c.dom.Document;
ingo@2714: 
ingo@2714: import org.apache.log4j.Logger;
ingo@2714: 
ingo@2714: import au.com.bytecode.opencsv.CSVWriter;
ingo@2714: 
teichmann@5831: import org.dive4elements.artifacts.CallContext;
ingo@2714: 
teichmann@5867: import org.dive4elements.river.artifacts.D4EArtifact;
teichmann@5831: import org.dive4elements.river.artifacts.model.CalculationResult;
teichmann@5831: import org.dive4elements.river.artifacts.model.MiddleBedHeightData;
teichmann@5865: import org.dive4elements.river.utils.RiverUtils;
teichmann@5831: import org.dive4elements.river.utils.Formatter;
ingo@2714: 
ingo@2714: 
ingo@2714: /**
ingo@2714:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@2714:  */
ingo@2714: public class MiddleBedHeightExporter extends AbstractExporter {
ingo@2714: 
felix@3042:     /** Private logger. */
ingo@2714:     private static final Logger logger =
ingo@2714:         Logger.getLogger(MiddleBedHeightExporter.class);
ingo@2714: 
ingo@2714:     public static final String CSV_KM =
ingo@2714:         "export.bedheight_middle.csv.header.km";
ingo@2714: 
ingo@2714:     public static final String CSV_SOUNDING =
ingo@2714:         "export.bedheight_middle.csv.header.sounding";
ingo@2714: 
ingo@2714:     public static final String CSV_HEIGHT =
ingo@2714:         "export.bedheight_middle.csv.header.height";
ingo@2714: 
ingo@2714:     public static final String CSV_UNCERTAINTY =
ingo@2714:         "export.bedheight_middle.csv.header.uncertainty";
ingo@2714: 
ingo@2714:     public static final String CSV_DATA_GAP =
ingo@2714:         "export.bedheight_middle.csv.header.datagap";
ingo@2714: 
ingo@2714:     public static final String CSV_SOUNDING_WIDTH =
ingo@2714:         "export.bedheight_middle.csv.header.soundingwidth";
ingo@2714: 
ingo@2714:     public static final String CSV_WIDTH =
ingo@2714:         "export.bedheight_middle.csv.header.width";
ingo@2714: 
ingo@2714:     public static final String CSV_LOCATIONS =
ingo@2714:         "export.bedheight_middle.csv.header.locations";
ingo@2714: 
ingo@2714: 
ingo@2714:     protected List<MiddleBedHeightData[]> data;
ingo@2714: 
ingo@2714: 
ingo@2714:     public void init(Document request, OutputStream out, CallContext cc) {
ingo@2714:         super.init(request, out, cc);
ingo@2714:         data = new ArrayList<MiddleBedHeightData[]>();
ingo@2714:     }
ingo@2714: 
ingo@2714: 
ingo@2714:     @Override
ingo@2714:     protected void addData(Object d) {
ingo@2714:         if (d instanceof CalculationResult) {
ingo@2714:             d = ((CalculationResult) d).getData();
ingo@2714: 
ingo@2714:             if (d instanceof MiddleBedHeightData[]) {
ingo@2714:                 logger.debug("Add new data of type MiddleBedHeightData");
ingo@2714:                 data.add((MiddleBedHeightData[]) d);
ingo@2714:             }
ingo@2714:         }
ingo@2714:     }
ingo@2714: 
ingo@2714: 
ingo@2714:     @Override
ingo@2714:     protected void writeCSVData(CSVWriter writer) {
ingo@2714:         logger.info("MiddleBedHeightExporter.writeCSVData");
ingo@2714:         logger.debug("CSV gets " + data.size() + " MiddleBedHeightData objects.");
ingo@2714: 
ingo@2714:         writeCSVHeader(writer);
ingo@2714: 
ingo@2714:         for (MiddleBedHeightData[] d: data) {
ingo@2714:             data2CSV(writer, d);
ingo@2714:         }
ingo@2714:     }
ingo@2714: 
ingo@2714: 
ingo@2714:     protected void writeCSVHeader(CSVWriter writer) {
ingo@2714:         writer.writeNext(new String[] {
ingo@2714:             msg(CSV_KM, CSV_KM),
ingo@2714:             msg(CSV_SOUNDING, CSV_SOUNDING),
ingo@2714:             msg(CSV_HEIGHT, CSV_HEIGHT),
ingo@2714:             msg(CSV_UNCERTAINTY, CSV_UNCERTAINTY),
ingo@2714:             msg(CSV_DATA_GAP, CSV_DATA_GAP),
ingo@2714:             msg(CSV_SOUNDING_WIDTH, CSV_SOUNDING_WIDTH),
ingo@2714:             msg(CSV_WIDTH, CSV_WIDTH),
ingo@2714:             msg(CSV_LOCATIONS, CSV_LOCATIONS)
ingo@2714:         });
ingo@2714:     }
ingo@2714: 
ingo@2714: 
ingo@2714:     protected void data2CSV(CSVWriter writer, MiddleBedHeightData[] mData) {
ingo@2714:         logger.debug("Add next MiddleBedHeightData to CSV");
ingo@2714: 
teichmann@5867:         D4EArtifact flys = (D4EArtifact) master;
ingo@2714: 
ingo@2714:         NumberFormat kmF     = Formatter.getMiddleBedHeightKM(context);
ingo@2714:         NumberFormat heightF = Formatter.getMiddleBedHeightHeight(context);
ingo@2714:         NumberFormat uncertF = Formatter.getMiddleBedHeightUncert(context);
ingo@2714:         NumberFormat gapF    = Formatter.getMiddleBedHeightDataGap(context);
ingo@2714:         NumberFormat soundF  = Formatter.getMiddleBedHeightSounding(context);
ingo@2714:         NumberFormat widthF  = Formatter.getMiddleBedHeightWidth(context);
ingo@2714: 
rrenkert@4929:         heightF.setMaximumFractionDigits(1);
rrenkert@4929:         soundF.setMaximumFractionDigits(1);
ingo@2714:         for (MiddleBedHeightData data: mData) {
ingo@2714:             for (int i = 0, n = data.size(); i < n; i++) {
ingo@2714:                 int start = data.getStartYear();
ingo@2714:                 int end   = data.getEndYear();
ingo@2714: 
ingo@2714:                 if (start == end) {
ingo@2714:                     writer.writeNext(new String[] {
ingo@2714:                         kmF.format(data.getKM(i)),
rrenkert@4929:                         data.getDescription(),
ingo@2714:                         heightF.format(data.getMiddleHeight(i)),
ingo@2714:                         uncertF.format(data.getUncertainty(i)),
ingo@2714:                         gapF.format(data.getDataGap(i)) + "%",
ingo@2714:                         soundF.format(data.getSoundingWidth(i)),
ingo@2714:                         widthF.format(data.getWidth(i)),
teichmann@5865:                         RiverUtils.getLocationDescription(flys, data.getKM(i)),
ingo@2714:                     });
ingo@2714:                 }
ingo@2714:                 else {
ingo@2714:                     writer.writeNext(new String[] {
ingo@2714:                         kmF.format(data.getKM(i)),
rrenkert@4929:                         data.getDescription(),
ingo@2714:                         heightF.format(data.getMiddleHeight(i)),
ingo@2714:                         "",
ingo@2714:                         "",
ingo@2714:                         "",
ingo@2714:                         "",
teichmann@5865:                         RiverUtils.getLocationDescription(flys, data.getKM(i)),
ingo@2714:                     });
ingo@2714:                 }
ingo@2714: 
ingo@2714:             }
ingo@2714:         }
ingo@2714:     }
ingo@2714: 
ingo@2714: 
ingo@2714:     @Override
ingo@2714:     protected void writePDF(OutputStream out) {
ingo@2714:         logger.error("TODO: Implement FlowVelocityExporter.writePDF");
ingo@2714:     }
ingo@2714: }
ingo@2714: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :