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