teichmann@5831: package org.dive4elements.river.exports;
ingo@2702: 
ingo@2702: import java.io.OutputStream;
ingo@2702: import java.text.NumberFormat;
ingo@2702: import java.util.ArrayList;
ingo@2702: import java.util.List;
ingo@2702: 
ingo@2702: import org.w3c.dom.Document;
ingo@2702: 
ingo@2702: import org.apache.log4j.Logger;
ingo@2702: 
ingo@2702: import au.com.bytecode.opencsv.CSVWriter;
ingo@2702: 
teichmann@5831: import org.dive4elements.artifacts.CallContext;
ingo@2702: 
teichmann@5831: import org.dive4elements.river.artifacts.FLYSArtifact;
teichmann@5831: import org.dive4elements.river.artifacts.model.CalculationResult;
teichmann@5831: import org.dive4elements.river.artifacts.model.FlowVelocityData;
teichmann@5831: import org.dive4elements.river.utils.FLYSUtils;
teichmann@5831: import org.dive4elements.river.utils.Formatter;
ingo@2702: 
ingo@2702: 
ingo@2702: /**
ingo@2702:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@2702:  */
ingo@2702: public class FlowVelocityExporter extends AbstractExporter {
ingo@2702: 
ingo@2702:     private static final Logger logger =
ingo@2702:         Logger.getLogger(FlowVelocityExporter.class);
ingo@2702: 
ingo@2702: 
ingo@2702:     public static final String CSV_KM =
ingo@2702:         "export.flow_velocity.csv.header.km";
ingo@2702: 
ingo@2702:     public static final String CSV_V_TOTAL =
ingo@2702:         "export.flow_velocity.csv.header.v_total";
ingo@2702: 
ingo@2702:     public static final String CSV_V_MAIN =
ingo@2702:         "export.flow_velocity.csv.header.v_main";
ingo@2702: 
ingo@2702:     public static final String CSV_TAU_MAIN =
ingo@2702:         "export.flow_velocity.csv.header.tau_main";
ingo@2702: 
ingo@2702:     public static final String CSV_Q =
ingo@2702:         "export.flow_velocity.csv.header.q";
ingo@2702: 
ingo@2702:     public static final String CSV_LOCATIONS =
ingo@2702:         "export.flow_velocity.csv.header.locations";
ingo@2702: 
ingo@2702: 
ingo@2702:     protected List<FlowVelocityData[]> data;
ingo@2702: 
ingo@2702: 
ingo@2702:     public void init(Document request, OutputStream out, CallContext cc) {
ingo@2702:         super.init(request, out, cc);
ingo@2702:         data = new ArrayList<FlowVelocityData[]>();
ingo@2702:     }
ingo@2702: 
ingo@2702: 
ingo@2702:     @Override
ingo@2702:     protected void addData(Object d) {
ingo@2702:         if (d instanceof CalculationResult) {
ingo@2702:             d = ((CalculationResult) d).getData();
ingo@2702: 
ingo@2702:             if (d instanceof FlowVelocityData[]) {
ingo@2702:                 logger.debug("Add new data of type FlowVelocityData");
ingo@2702:                 data.add((FlowVelocityData[]) d);
ingo@2702:             }
ingo@2702:         }
ingo@2702:     }
ingo@2702: 
ingo@2702: 
ingo@2702:     @Override
ingo@2702:     protected void writeCSVData(CSVWriter writer) {
ingo@2702:         logger.info("FlowVelocityExporter.writeCSVData");
ingo@2702:         logger.debug("CSV gets " + data.size() + " FlowVelocityData objects.");
ingo@2702: 
ingo@2702:         writeCSVHeader(writer);
ingo@2702: 
ingo@2702:         for (FlowVelocityData[] d: data) {
ingo@2702:             data2CSV(writer, d);
ingo@2702:         }
ingo@2702:     }
ingo@2702: 
ingo@2702: 
ingo@2702:     protected void writeCSVHeader(CSVWriter writer) {
ingo@2702:         writer.writeNext(new String[] {
ingo@2702:             msg(CSV_KM, CSV_KM),
ingo@2702:             msg(CSV_V_TOTAL, CSV_V_TOTAL),
ingo@2702:             msg(CSV_V_MAIN, CSV_V_MAIN),
ingo@2702:             msg(CSV_TAU_MAIN, CSV_TAU_MAIN),
ingo@2702:             msg(CSV_Q, CSV_Q),
ingo@2702:             msg(CSV_LOCATIONS, CSV_LOCATIONS)
ingo@2702:         });
ingo@2702:     }
ingo@2702: 
ingo@2702: 
ingo@2702:     protected void data2CSV(CSVWriter writer, FlowVelocityData[] fData) {
ingo@2702:         logger.debug("Add next FlowVelocityData to CSV");
ingo@2702: 
ingo@2703:         FLYSArtifact flys = (FLYSArtifact) master;
ingo@2703: 
ingo@2702:         for (FlowVelocityData data: fData) {
ingo@2702:             for (int i = 0, n = data.size(); i < n; i++) {
ingo@2702:                 NumberFormat kmF  = Formatter.getFlowVelocityKM(context);
ingo@2702:                 NumberFormat valF = Formatter.getFlowVelocityValues(context);
ingo@2702:                 NumberFormat qF   = Formatter.getFlowVelocityQ(context);
ingo@2702: 
ingo@2702:                 writer.writeNext(new String[] {
ingo@2702:                     kmF.format(data.getKM(i)),
ingo@2702:                     valF.format(data.getVMain(i)),
ingo@2702:                     valF.format(data.getVTotal(i)),
ingo@2702:                     valF.format(data.getTauMain(i)),
ingo@2702:                     qF.format(data.getQ(i)) + "=" + data.getZone(),
ingo@2703:                     FLYSUtils.getLocationDescription(flys, data.getKM(i)),
ingo@2702:                 });
ingo@2702:             }
ingo@2702:         }
ingo@2702:     }
ingo@2702: 
ingo@2702: 
ingo@2702:     @Override
ingo@2702:     protected void writePDF(OutputStream out) {
ingo@2702:         logger.error("TODO: Implement FlowVelocityExporter.writePDF");
ingo@2702:     }
ingo@2702: }
ingo@2702: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf-8 :