Mercurial > dive4elements > river
view flys-artifacts/src/main/java/de/intevation/flys/exports/ComputedDischargeCurveGenerator.java @ 1083:d0db31d1f64c
Enable plotting of some annotations that look like MainValues.
flys-artifacts/trunk@2580 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author | Felix Wolfsteller <felix.wolfsteller@intevation.de> |
---|---|
date | Thu, 25 Aug 2011 10:53:25 +0000 |
parents | d10efbe2e5c0 |
children | 07878836ee0d |
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.ui.TextAnchor; import de.intevation.artifacts.Artifact; import de.intevation.artifactdatabase.state.Facet; import de.intevation.flys.artifacts.FLYSArtifact; import de.intevation.flys.artifacts.model.FacetTypes; import de.intevation.flys.artifacts.model.WQKms; import de.intevation.flys.jfree.StickyAxisAnnotation; import de.intevation.flys.model.MainValue; /** * 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 MainValues (Annotations in plot). */ protected static List<MainValue> mainValues; /** Trivial Constructor. */ public ComputedDischargeCurveGenerator () { super(); mainValues = new ArrayList<MainValue>(); } @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); 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)) { doMainValueAnnotations(f.getData(artifact, context), attr); } else { logger.warn("Unknown facet type for computed discharge: " + name); return; } } /** * Add MainValues as annotations to plot. */ protected void doMainValueAnnotations(Object o, Document theme) { this.mainValues = (List<MainValue>) o; } /** Generate Chart with annotations. */ @Override public JFreeChart generateChart() { JFreeChart c = super.generateChart(); XYPlot p = (XYPlot) c.getPlot(); redoAnnotations(p, p.getDomainAxis()); return c; } /** * Recalculate some annotation positions and add them to plot. * Annotations represent MainValues. * @param plot Plot to add annotations to. * @param valueAxis ignored. */ protected void redoAnnotations(XYPlot plot, ValueAxis axis) { plot.clearAnnotations(); ValueAxis yAxis = plot.getRangeAxis(); float posY = 140.f; if (yAxis != null) { posY = (float) yAxis.getRange().getLowerBound(); // Add some (2%) space between Text and axis. // TODO have all the position logic in StickyAxisAnnotation // (then merge LongitudinalSectioNGenerator). posY += 0.02f * (yAxis.getRange().getUpperBound() - yAxis.getRange().getLowerBound()); } // Add all MainValues as annotations. // TODO Implement and handle second facet for annotations on // vertical axis. for (MainValue mv: mainValues) { if (mv.getMainValue().getType().getName() == "W") { continue; } float posX = mv.getValue().floatValue(); String text = mv.getMainValue().getName(); StickyAxisAnnotation ta = new StickyAxisAnnotation(text, posX, posY); plot.getRenderer().addAnnotation(ta); } } /** * 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 :