Mercurial > dive4elements > river
view flys-artifacts/src/main/java/de/intevation/flys/exports/ComputedDischargeCurveGenerator.java @ 1089:e298c4d28927
Improved mainvalues rendering.
flys-artifacts/trunk@2592 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author | Felix Wolfsteller <felix.wolfsteller@intevation.de> |
---|---|
date | Fri, 26 Aug 2011 11:15:24 +0000 |
parents | 07878836ee0d |
children | 0eb585cd3882 |
line wrap: on
line source
package de.intevation.flys.exports; import org.apache.log4j.Logger; import java.util.ArrayList; import java.util.List; import org.w3c.dom.Document; import org.jfree.chart.JFreeChart; import org.jfree.chart.title.TextTitle; import org.jfree.data.xy.XYSeries; import org.jfree.chart.axis.ValueAxis; import org.jfree.chart.plot.XYPlot; import org.jfree.chart.annotations.XYAnnotation; import de.intevation.artifacts.Artifact; import de.intevation.artifactdatabase.state.Facet; import de.intevation.flys.artifacts.FLYSArtifact; import de.intevation.flys.artifacts.model.NamedDouble; import de.intevation.flys.artifacts.model.FacetTypes; import de.intevation.flys.artifacts.model.WQKms; import de.intevation.flys.jfree.StickyAxisAnnotation; /** * An OutGenerator that generates discharge curves. * * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class ComputedDischargeCurveGenerator extends DischargeCurveGenerator implements FacetTypes { /** The logger used in this generator. */ private static Logger logger = Logger.getLogger(ComputedDischargeCurveGenerator.class); public static final String I18N_CHART_TITLE = "chart.computed.discharge.curve.title"; public static final String I18N_CHART_SUBTITLE = "chart.computed.discharge.curve.subtitle"; public static final String I18N_YAXIS_LABEL = "chart.computed.discharge.curve.yaxis.label"; public static final String I18N_CHART_TITLE_DEFAULT = "Abflusskurve"; public static final String I18N_YAXIS_LABEL_DEFAULT = "W [NN + m]"; /** List of Annotations (specifically, Main Values). */ protected List<XYAnnotation> annotations; // TODO Add pseudodataseries for having mainvalue-text in legend. // TODO Let theme pass through to annotations-facets. /** Trivial Constructor. */ public ComputedDischargeCurveGenerator () { super(); annotations= new ArrayList<XYAnnotation>(); } @Override protected String getChartTitle() { return msg(I18N_CHART_TITLE, I18N_CHART_TITLE_DEFAULT); } @Override protected void addSubtitles(JFreeChart chart) { double[] dist = getRange(); Object[] args = new Object[] { getRiverName(), dist[0] }; String subtitle = msg(I18N_CHART_SUBTITLE, "", args); chart.addSubtitle(new TextTitle(subtitle)); } @Override protected String getYAxisLabel() { return msg(I18N_YAXIS_LABEL, I18N_YAXIS_LABEL_DEFAULT); } @Override public void doOut(Artifact artifact, Facet facet, Document attr) { String name = (facet != null) ? facet.getName() : null; logger.debug("ComputedDischargeCurveGenerator.doOut: " + name); if (name == null) { logger.warn("Broken facet in computed discharge out generation."); return; } FLYSArtifact flys = (FLYSArtifact) artifact; Facet f = flys.getNativeFacet(facet); if (name.equals(COMPUTED_DISCHARGE_Q)) { doQOut((WQKms) f.getData(artifact, context), attr); } else if (name.equals(COMPUTED_DISCHARGE_MAINVALUES_Q)) { doMainValueQAnnotations(f.getData(artifact, context), attr); } else if (name.equals(COMPUTED_DISCHARGE_MAINVALUES_W)) { doMainValueWAnnotations(f.getData(artifact, context), attr); } else { logger.warn("Unknown facet type for computed discharge: " + name); return; } } /** * Store W MainValues as annotations for later plotting. */ protected void doMainValueWAnnotations(Object o, Document theme) { logger.debug("ComputedDischargeCurveGenerator set W MainValues."); List<NamedDouble> mainValuesW = (List<NamedDouble>) o; for (NamedDouble mv: mainValuesW) { float pos = (float) mv.getValue(); String text = mv.getName(); StickyAxisAnnotation ta = new StickyAxisAnnotation(text, pos, StickyAxisAnnotation.SimpleAxis.Y_AXIS); logger.debug("Adding W: " + text + " : " + pos); this.annotations.add(ta); } } /** * Store Q MainValues as annotations for later plotting. */ protected void doMainValueQAnnotations(Object o, Document theme) { logger.debug("ComputedDischargeCurveGenerator set Q MainValues."); List<NamedDouble> mainValuesQ = (List<NamedDouble>) o; for (NamedDouble mv: mainValuesQ) { float pos = (float) mv.getValue(); String text = mv.getName(); StickyAxisAnnotation ta = new StickyAxisAnnotation(text, pos); this.annotations.add(ta); } } /** Generate Chart with annotations. */ @Override public JFreeChart generateChart() { JFreeChart c = super.generateChart(); XYPlot p = (XYPlot) c.getPlot(); redoAnnotations(p); return c; } /** * Recalculate some annotation positions and add them to plot. * Annotations represent MainValues. * @param plot Plot to add annotations to. */ protected void redoAnnotations(XYPlot plot) { plot.clearAnnotations(); for (XYAnnotation a: annotations) { plot.addAnnotation(a, false); } } /** * Add Q-Series to plot. * @param wqkms actual data * @param theme theme to use. */ protected void doQOut(WQKms wqkms, Document theme) { int size = wqkms.size(); double[] res = new double[3]; XYSeries series = new StyledXYSeries(getSeriesName(wqkms), theme); for (int i = 0; i < size; i++) { res = wqkms.get(i, res); series.add(res[1], res[0]); } addFirstAxisSeries(series); } protected String getSeriesName(WQKms wqkms) { Object[] args = new Object[] { getRiverName(), wqkms.getName() }; return msg( "chart.computed.discharge.curve.curve.label", "", args); } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :