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: 
gernotbelger@9312: import java.text.NumberFormat;
christian@3254: 
gernotbelger@9312: import org.apache.log4j.Logger;
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: 
gernotbelger@9312: import au.com.bytecode.opencsv.CSVWriter;
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: 
gernotbelger@9312:     /** The log used in this exporter. */
gernotbelger@9312:     private static Logger log = Logger.getLogger(DischargeLongitudinalSectionExporter.class);
ingo@416: 
gernotbelger@9312:     public static final String CSV_KM_HEADER = "export.discharge.longitudinal.section.csv.header.km";
ingo@416: 
gernotbelger@9312:     public static final String CSV_W_HEADER = "export.discharge.longitudinal.section.csv.header.w";
ingo@416: 
gernotbelger@9312:     public static final String CSV_CW_HEADER = "export.discharge.longitudinal.section.csv.header.cw";
gernotbelger@9312: 
gernotbelger@9312:     public static final String CSV_Q_HEADER = "common.export.csv.header.q";
ingo@416: 
ingo@416:     public static final String DEFAULT_CSV_KM_HEADER = "Fluss-Km";
gernotbelger@9312:     public static final String DEFAULT_CSV_W_HEADER = "W [NN + m]";
ingo@416:     public static final String DEFAULT_CSV_CW_HEADER = "W korr.";
gernotbelger@9312:     public static final String DEFAULT_CSV_Q_HEADER = "Q [m\u00b3/s]";
ingo@416: 
ingo@407:     @Override
sascha@701:     protected void addData(Object d) {
sascha@716:         if (d instanceof CalculationResult) {
gernotbelger@9312:             d = ((CalculationResult) d).getData();
gernotbelger@9312:             if (d instanceof WQKms[]) {
gernotbelger@9312:                 this.data.add((WQKms[]) d);
sascha@716:             }
sascha@742:         }
ingo@407:     }
ingo@407: 
ingo@407:     @Override
gernotbelger@9312:     protected void writeCSVHeader(final CSVWriter writer, final boolean atGauge, final boolean isQ) {
teichmann@8202:         log.info("WaterlevelExporter.writeCSVHeader");
ingo@416: 
gernotbelger@9312:         writer.writeNext(new String[] { msg(CSV_KM_HEADER, DEFAULT_CSV_KM_HEADER), msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER),
gernotbelger@9312:                 msg(CSV_CW_HEADER, DEFAULT_CSV_CW_HEADER), msg(CSV_Q_HEADER, DEFAULT_CSV_Q_HEADER), msg(CSV_Q_DESC_HEADER, DEFAULT_CSV_Q_DESC_HEADER) });
ingo@416:     }
ingo@416: 
gernotbelger@9312:     @Override
gernotbelger@9312:     protected void wQKms2CSV(final CSVWriter writer, final WQKms wqkms, final boolean atGauge, final boolean isQ) {
teichmann@8202:         log.debug("WaterlevelExporter.wQKms2CSV");
ingo@407: 
gernotbelger@9312:         final int size = wqkms.size();
ingo@407:         double[] result = new double[4];
ingo@407: 
gernotbelger@9312:         final NumberFormat kmf = getKmFormatter();
gernotbelger@9312:         final NumberFormat wf = getWFormatter();
gernotbelger@9312:         final NumberFormat qf = getQFormatter();
ingo@418: 
gernotbelger@9312:         for (int i = 0; i < size; i++) {
ingo@407:             result = wqkms.get(i, result);
ingo@407: 
gernotbelger@9312:             final 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: 
gernotbelger@9312:             writer.writeNext(new String[] { kmf.format(result[2]), wf.format(result[0]), wc, qf.format(result[1]), name });
ingo@407:         }
ingo@407:     }
ingo@450: 
ingo@450:     @Override
gernotbelger@9312:     protected void addWSTColumn(final WstWriter writer, final WQKms wqkms) {
gernotbelger@9312:         final String name = wqkms.getName();
ingo@450: 
ingo@450:         // is it a W or a Q mode?
gernotbelger@9312:         final int wIdx = name.indexOf("W");
gernotbelger@9312:         final int qIdx = name.indexOf("Q");
ingo@450: 
ingo@450:         String wq = null;
ingo@450:         if (wIdx >= 0) {
ingo@450:             wq = "W";
gernotbelger@9312:         } 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
gernotbelger@9312:         final int start = name.indexOf("(");
gernotbelger@9312:         final int end = name.indexOf(")");
ingo@450: 
gernotbelger@9312:         final String tmp = name.substring(start + 1, end);
gernotbelger@9312:         final String[] values = tmp.split(";");
ingo@450: 
gernotbelger@9312:         final 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 :