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;
ingo@407: 
christian@3254: import au.com.bytecode.opencsv.CSVWriter;
christian@3254: 
teichmann@5831: import org.dive4elements.river.artifacts.model.CalculationResult;
teichmann@5831: import org.dive4elements.river.artifacts.model.WQCKms;
teichmann@5831: import org.dive4elements.river.artifacts.model.WQKms;
christian@3254: 
ingo@418: import java.text.NumberFormat;
ingo@418: 
ingo@407: import org.apache.log4j.Logger;
ingo@407: 
ingo@407: /**
ingo@407:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@407:  */
ingo@407: public class DischargeLongitudinalSectionExporter extends WaterlevelExporter {
ingo@407: 
ingo@407:     /** The logger used in this exporter.*/
ingo@407:     private static Logger logger =
ingo@407:         Logger.getLogger(DischargeLongitudinalSectionExporter.class);
ingo@407: 
ingo@407: 
ingo@416:     public static final String CSV_KM_HEADER =
ingo@416:         "export.discharge.longitudinal.section.csv.header.km";
ingo@416: 
ingo@416:     public static final String CSV_W_HEADER =
ingo@416:         "export.discharge.longitudinal.section.csv.header.w";
ingo@416: 
ingo@416:     public static final String CSV_CW_HEADER =
ingo@416:         "export.discharge.longitudinal.section.csv.header.cw";
ingo@416: 
ingo@416:     public static final String CSV_Q_HEADER =
ingo@416:         "export.discharge.longitudinal.section.csv.header.q";
ingo@416: 
ingo@416:     public static final String DEFAULT_CSV_KM_HEADER = "Fluss-Km";
ingo@416:     public static final String DEFAULT_CSV_W_HEADER  = "W [NN + m]";
ingo@416:     public static final String DEFAULT_CSV_CW_HEADER = "W korr.";
sascha@664:     public static final String DEFAULT_CSV_Q_HEADER  = "Q [m\u00b3/s]";
ingo@416: 
ingo@416: 
ingo@407:     @Override
sascha@701:     protected void addData(Object d) {
sascha@716:         if (d instanceof CalculationResult) {
sascha@716:             d = ((CalculationResult)d).getData();
sascha@716:             if (d instanceof WQKms []) {
sascha@716:                 data.add((WQKms [])d);
sascha@716:             }
sascha@742:         }
ingo@407:     }
ingo@407: 
ingo@407: 
ingo@407:     @Override
ingo@2068:     protected void writeCSVHeader(
ingo@2068:         CSVWriter writer,
ingo@2068:         boolean   atGauge,
ingo@2068:         boolean   isQ
ingo@2068:     ) {
ingo@416:         logger.info("WaterlevelExporter.writeCSVHeader");
ingo@416: 
ingo@416:         writer.writeNext(new String[] {
ingo@416:             msg(CSV_KM_HEADER, DEFAULT_CSV_KM_HEADER),
ingo@416:             msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER),
ingo@416:             msg(CSV_CW_HEADER, DEFAULT_CSV_CW_HEADER),
rrenkert@5141:             msg(CSV_Q_HEADER, DEFAULT_CSV_Q_HEADER),
rrenkert@5141:             msg(CSV_Q_DESC_HEADER, DEFAULT_CSV_Q_DESC_HEADER)
ingo@416:         });
ingo@416:     }
ingo@416: 
ingo@2068: 
ingo@2068:     protected void wQKms2CSV(
ingo@2068:         CSVWriter writer,
ingo@2068:         WQKms     wqkms,
ingo@2068:         boolean   atGauge,
rrenkert@5141:         boolean   isQ
ingo@2068:     ) {
ingo@407:         logger.debug("WaterlevelExporter.wQKms2CSV");
ingo@407: 
ingo@407:         int      size   = wqkms.size();
ingo@407:         double[] result = new double[4];
ingo@407: 
ingo@418:         NumberFormat kmf = getKmFormatter();
ingo@418:         NumberFormat wf  = getWFormatter();
ingo@418:         NumberFormat qf  = getQFormatter();
ingo@418: 
ingo@407:         for (int i = 0; i < size; i ++) {
ingo@407:             result = wqkms.get(i, result);
ingo@407: 
rrenkert@5141:             String name = wqkms.getName();
ingo@407:             String wc = "";
ingo@407:             if (wqkms instanceof WQCKms) {
ingo@418:                 wc = wf.format(result[3]);
ingo@407:             }
ingo@407: 
ingo@407:             writer.writeNext(new String[] {
ingo@418:                 kmf.format(result[2]),
ingo@418:                 wf.format(result[0]),
ingo@407:                 wc,
rrenkert@5141:                 qf.format(result[1]),
rrenkert@5141:                 name
ingo@407:             });
ingo@407:         }
ingo@407:     }
ingo@450: 
ingo@450: 
ingo@450:     @Override
ingo@450:     protected void addWSTColumn(WstWriter writer, WQKms wqkms) {
ingo@450:         String name = wqkms.getName();
ingo@450: 
ingo@450:         // is it a W or a Q mode?
ingo@450:         int wIdx = name.indexOf("W");
ingo@450:         int qIdx = name.indexOf("Q");
ingo@450: 
ingo@450:         String wq = null;
ingo@450:         if (wIdx >= 0) {
ingo@450:             wq = "W";
ingo@450:         }
ingo@450:         else if (qIdx >= 0) {
ingo@450:             wq = "Q";
ingo@450:         }
ingo@450: 
ingo@450:         // we just want to display the first W or Q value in the WST
ingo@450:         int start = name.indexOf("(");
ingo@450:         int end   = name.indexOf(")");
ingo@450: 
ingo@450:         String   tmp    = name.substring(start+1, end);
ingo@450:         String[] values = tmp.split(";");
ingo@450: 
ingo@450:         String column = wq + "=" + values[0];
ingo@450: 
ingo@450:         writer.addColumn(column);
ingo@450:     }
ingo@407: }
ingo@407: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :