ingo@403: package de.intevation.flys.exports;
ingo@403: 
ingo@420: import java.awt.Color;
ingo@420: 
ingo@403: import org.apache.log4j.Logger;
ingo@403: 
ingo@420: import org.jfree.chart.JFreeChart;
ingo@420: import org.jfree.chart.plot.XYPlot;
ingo@420: import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
ingo@403: import org.jfree.data.xy.XYSeries;
ingo@420: import org.jfree.data.xy.XYSeriesCollection;
ingo@403: 
ingo@403: import org.w3c.dom.Document;
ingo@403: 
ingo@403: import de.intevation.artifacts.Artifact;
ingo@403: 
ingo@403: import de.intevation.flys.artifacts.WINFOArtifact;
ingo@403: import de.intevation.flys.artifacts.model.WQCKms;
ingo@403: import de.intevation.flys.artifacts.model.WQKms;
ingo@403: 
ingo@403: 
ingo@403: /**
ingo@403:  * An OutGenerator that generates discharge longitudinal section curves.
ingo@403:  *
ingo@403:  * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a>
ingo@403:  */
ingo@403: public class DischargeLongitudinalSectionGenerator
ingo@403: extends      LongitudinalSectionGenerator
ingo@403: {
ingo@403:     private static Logger logger =
ingo@403:         Logger.getLogger(DischargeLongitudinalSectionGenerator.class);
ingo@403: 
ingo@403: 
ingo@403:     /** The facet for Ws.*/
ingo@403:     public static final String FACET_W = "discharge_longitudinal_section.w";
ingo@403: 
ingo@403:     /** The facet for Qs.*/
ingo@403:     public static final String FACET_Q = "discharge_longitudinal_section.q";
ingo@403: 
ingo@403: 
ingo@420:     /** The storage for the corrected W series to be drawn in this chart.*/
ingo@420:     protected XYSeriesCollection cw;
ingo@420: 
ingo@420: 
ingo@420:     public DischargeLongitudinalSectionGenerator() {
ingo@420:         super();
ingo@420: 
ingo@420:         this.cw = new XYSeriesCollection();
ingo@420:     }
ingo@420: 
ingo@420: 
ingo@420:     public void addDatasets(JFreeChart chart) {
ingo@420:         super.addDatasets(chart);
ingo@420: 
ingo@420:         XYPlot plot = (XYPlot) chart.getPlot();
ingo@420: 
ingo@420:         plot.setDataset(2, cw);
ingo@420:     }
ingo@420: 
ingo@420: 
ingo@420:     protected void adjustPlot(XYPlot plot) {
ingo@420:         super.adjustPlot(plot);
ingo@420: 
ingo@420:         // TODO REMOVE THIS CODE, IF WE HAVE INTRODUCED THEMES!
ingo@420:         XYLineAndShapeRenderer rw = (XYLineAndShapeRenderer)
ingo@420:             plot.getRendererForDataset(w);
ingo@420: 
ingo@420:         XYLineAndShapeRenderer rcw = null;
ingo@420:         try {
ingo@420:             rcw = (XYLineAndShapeRenderer) rw.clone();
ingo@420:         }
ingo@420:         catch (Exception e) {
ingo@420:             logger.error(e, e);
ingo@420:         }
ingo@420: 
ingo@420:         int cwNum = cw.getSeriesCount();
ingo@420: 
ingo@420:         for (int i = 0; i < cwNum; i++) {
ingo@420:             rcw.setSeriesPaint(i, Color.RED);
ingo@420:         }
ingo@420: 
ingo@420:         plot.setRenderer(2, rcw);
ingo@420:     }
ingo@420: 
ingo@420: 
ingo@672:     protected void adjustAxes(XYPlot plot) {
ingo@672:         super.adjustAxes(plot);
ingo@672: 
ingo@672:         plot.mapDatasetToRangeAxis(2, 1);
ingo@672:     }
ingo@672: 
ingo@672: 
ingo@403: 
ingo@403:     @Override
ingo@403:     public void doOut(Artifact artifact, String facet, Document attr) {
ingo@403:         logger.debug("DischargeLongitudinalSectionGenerator.doOut");
ingo@403: 
ingo@403:         if (facet != null && facet.equals(FACET_W)) {
ingo@403:             doWOut(getDischargeLongitudinalSectionData(artifact));
ingo@403:         }
ingo@403:         else if (facet != null && facet.equals(FACET_Q)) {
ingo@403:             doQOut(getDischargeLongitudinalSectionData(artifact));
ingo@403:         }
ingo@403:         else {
ingo@403:             logger.warn("Unknown facet name: " + facet);
ingo@403:             return;
ingo@403:         }
ingo@403:     }
ingo@403: 
ingo@403: 
ingo@403:     protected WQKms[] getDischargeLongitudinalSectionData(Artifact artifact) {
ingo@403:         WINFOArtifact winfoArtifact = (WINFOArtifact) artifact;
ingo@403:         return winfoArtifact.getDischargeLongitudinalSectionData();
ingo@403:     }
ingo@403: 
ingo@403: 
ingo@403:     /**
ingo@403:      * This method adds new Series for the wqkms objects. If there are instances
ingo@403:      * of WQCKms in this array, there is a further curve that displays the
ingo@403:      * corrected W values.
ingo@403:      *
ingo@403:      * @param wqkms An array of WQKms objects to be displayed.
ingo@403:      */
ingo@403:     protected void doWOut(WQKms[] wqkms) {
ingo@403:         logger.debug("LongitudinalSectionGenerator.doWOut");
ingo@403: 
ingo@403:         int idx = 0;
ingo@403:         for (WQKms tmp: wqkms) {
ingo@403:             if (tmp instanceof WQCKms) {
ingo@403:                 doCorrectedWOut((WQCKms) tmp);
ingo@403:             }
ingo@403: 
ingo@403:             doWOut(tmp);
ingo@403:         }
ingo@403:     }
ingo@403: 
ingo@403: 
ingo@403:     /**
ingo@403:      * Adds a new series for the W curve.
ingo@403:      *
ingo@403:      * @param wqkms The object that contains the W values.
ingo@403:      */
ingo@403:     protected void doWOut(WQKms wqkms) {
ingo@403:         logger.debug("DischargeLongitudinalSectionGenerator.doWOut");
ingo@403: 
ingo@450:         XYSeries series = new XYSeries(getSeriesName(wqkms, "W"));
ingo@403: 
ingo@403:         double[] target = new double[4];
ingo@403:         int      size   = wqkms.size();
ingo@403: 
ingo@403:         for (int i = 0; i < size; i++) {
ingo@403:             target = wqkms.get(i, target);
ingo@403: 
ingo@403:             series.add(target[2], target[0]);
ingo@403:         }
ingo@403: 
ingo@403:         w.addSeries(series);
ingo@403:     }
ingo@403: 
ingo@403: 
ingo@403:     /**
ingo@403:      * Adds a new series for the corrected W curve.
ingo@403:      *
ingo@403:      * @param wqckms The object that contains the corrected W values.
ingo@403:      */
ingo@403:     protected void doCorrectedWOut(WQCKms wqckms) {
ingo@403:         logger.debug("DischargeLongitudinalSectionGenerator.doCorrectedWOut");
ingo@403: 
ingo@450:         XYSeries series = new XYSeries(getSeriesNameForCorrected(wqckms, "W"));
ingo@403: 
ingo@403:         double[] target = new double[4];
ingo@403:         int      size   = wqckms.size();
ingo@403: 
ingo@403:         for (int i = 0; i < size; i++) {
ingo@403:             target = wqckms.get(i, target);
ingo@403: 
ingo@403:             series.add(target[2], target[3]);
ingo@403:         }
ingo@403: 
ingo@403:         if (series.getItemCount() > 0) {
ingo@420:             cw.addSeries(series);
ingo@403:         }
ingo@403:     }
ingo@450: 
ingo@450: 
ingo@450:     protected String getSeriesNameForCorrected(WQKms wqkms, String mode) {
ingo@450:         String name = wqkms.getName();
ingo@450: 
ingo@450:         name = name.replace(
ingo@450:             "Benutzerdefiniert",
ingo@450:             "Benutzerdefiniert [korrigiert]");
ingo@450: 
ingo@450:         String prefix = name != null && name.indexOf(mode) >= 0 ? null : mode;
ingo@450: 
ingo@450:         return prefix != null && prefix.length() > 0
ingo@450:             ? prefix + "(" + name +")"
ingo@450:             : name;
ingo@450:     }
ingo@403: }
ingo@403: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :