Mercurial > dive4elements > river
view flys-artifacts/src/main/java/de/intevation/flys/exports/LongitudinalSectionGenerator.java @ 637:f0c1250d1e7b
Make "Berechnungsart 4" work independent of flow direction. Commented out back jump correction temporarily.
flys-artifacts/trunk@2013 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author | Sascha L. Teichmann <sascha.teichmann@intevation.de> |
---|---|
date | Thu, 26 May 2011 16:54:16 +0000 |
parents | 88d9e1d75d64 |
children | e006da5679d2 |
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.axis.NumberAxis; import org.jfree.chart.axis.ValueAxis; import org.jfree.chart.plot.XYPlot; import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer; import org.jfree.chart.title.TextTitle; 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.model.River; import de.intevation.flys.artifacts.FLYSArtifact; import de.intevation.flys.artifacts.WINFOArtifact; import de.intevation.flys.artifacts.model.WQKms; /** * An OutGenerator that generates discharge curves. * * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class LongitudinalSectionGenerator extends XYChartGenerator { /** The logger that is used in this generator.*/ private static Logger logger = Logger.getLogger(LongitudinalSectionGenerator.class); public static final String LONGITUDINAL_SECTION_W = "longitudinal_section.w"; public static final String LONGITUDINAL_SECTION_Q = "longitudinal_section.q"; public static final String I18N_CHART_TITLE = "chart.longitudinal.section.title"; public static final String I18N_CHART_SUBTITLE = "chart.longitudinal.section.subtitle"; public static final String I18N_XAXIS_LABEL = "chart.longitudinal.section.xaxis.label"; public static final String I18N_YAXIS_LABEL = "chart.longitudinal.section.yaxis.label"; public static final String I18N_2YAXIS_LABEL = "chart.longitudinal.section.yaxis.second.label"; public static final String I18N_CHART_TITLE_DEFAULT = "W-Längsschnitt"; public static final String I18N_XAXIS_LABEL_DEFAULT = "km"; public static final String I18N_YAXIS_LABEL_DEFAULT = "W [NN + m]"; public static final String I18N_2YAXIS_LABEL_DEFAULT = "Q [m³/s]"; /** The storage for the W series to be drawn in this chart.*/ protected XYSeriesCollection w; /** The storage for the Q series to be drawn in this chart.*/ protected XYSeriesCollection q; public LongitudinalSectionGenerator() { super(); this.w = new XYSeriesCollection(); this.q = new XYSeriesCollection(); } 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], dist[1] }; String subtitle = msg(I18N_CHART_SUBTITLE, "", args); chart.addSubtitle(new TextTitle(subtitle)); } protected String getXAxisLabel() { return msg(I18N_XAXIS_LABEL, I18N_XAXIS_LABEL_DEFAULT); } protected String getYAxisLabel() { return msg(I18N_YAXIS_LABEL, I18N_YAXIS_LABEL_DEFAULT); } protected void addDatasets(JFreeChart chart) { XYPlot plot = (XYPlot) chart.getPlot(); plot.setDataset(0, w); plot.setDataset(1, q); } protected void adjustPlot(XYPlot plot) { super.adjustPlot(plot); // TODO REMOVE THIS CODE, IF WE HAVE INTRODUCED THEMES! XYLineAndShapeRenderer rw = (XYLineAndShapeRenderer) plot.getRendererForDataset(w); XYLineAndShapeRenderer rq = null; try { rq = (XYLineAndShapeRenderer) rw.clone(); } catch (Exception e) { logger.error(e, e); } int wNum = w.getSeriesCount(); int qNum = q.getSeriesCount(); for (int i = 0; i < wNum; i++) { rw.setSeriesPaint(i, Color.BLUE); } for (int i = 0; i < qNum; i++) { rq.setSeriesPaint(i, Color.GREEN); } plot.setRenderer(0, rw); plot.setRenderer(1, rq); } protected void adjustAxes(XYPlot plot) { super.adjustAxes(plot); NumberAxis qAxis = new NumberAxis( msg(I18N_2YAXIS_LABEL, I18N_2YAXIS_LABEL_DEFAULT)); plot.setRangeAxis(2, qAxis); plot.mapDatasetToRangeAxis(1, 2); invertXAxis(plot.getDomainAxis()); } /** * This method inverts the x-axis based on the kilometer information of the * selected river. If the head of the river is at kilometer 0, the axis is * not inverted, otherwise it is. * * @param xaxis The domain axis. */ protected void invertXAxis(ValueAxis xaxis) { FLYSArtifact artifact = (FLYSArtifact) master; River river = artifact.getRiver(); boolean up = river.getKmUp(); if (up) { logger.info("Invert X-Axis."); xaxis.setInverted(true); } } public void doOut(Artifact artifact, String facet, Document attr) { logger.debug("LongitudinalSectionGenerator.doOut: " + facet); if (facet == null) { logger.error("No facet name for doOut(). No output generated!"); return; } if (facet.equals(LONGITUDINAL_SECTION_W)) { doWOut(getWaterlevelData(artifact)); } else if (facet.equals(LONGITUDINAL_SECTION_Q)) { doQOut(getWaterlevelData(artifact)); } else { logger.warn("Unknown facet name: " + facet); return; } } /** * Returns the waterlevel data computed by the WINFOArtifact. * * @param artifact The WINFOArtifact. * * @return the computed waterlevel data. */ protected WQKms[] getWaterlevelData(Artifact artifact) { WINFOArtifact winfoArtifact = (WINFOArtifact) artifact; WQKms[] wqkms = winfoArtifact.getWaterlevelData(); logger.debug("Got " + wqkms.length + " WQKms objects."); return wqkms; } /** * Process the output for W facets in a longitudinal section curve. * * @param wqkms An array of WQKms values. */ protected void doWOut(WQKms[] wqkms) { logger.debug("LongitudinalSectionGenerator.doWOut"); int idx = 0; for (WQKms tmp: wqkms) { XYSeries series = new XYSeries(getSeriesName(tmp, "W")); double[] target = new double[3]; int size = tmp.size(); if (logger.isDebugEnabled()) { if (tmp.size() > 0) { logger.debug("Generate series: " + series.getKey()); logger.debug("Start km: " + tmp.getKms(0)); logger.debug("End km: " + tmp.getKms(size-1)); logger.debug("Values : " + size); } } for (int i = 0; i < size; i++) { target = tmp.get(i, target); series.add(target[2], target[0]); } w.addSeries(series); } } /** * Process the output for Q facets in a longitudinal section curve. * * @param wqkms An array of WQKms values. */ protected void doQOut(WQKms[] wqkms) { logger.debug("LongitudinalSectionGenerator.doQOut"); int idx = 0; for (WQKms tmp: wqkms) { XYSeries series = new XYSeries(getSeriesName(tmp, "Q")); double[] target = new double[3]; int size = tmp.size(); if (logger.isDebugEnabled()) { if (tmp.size() > 0) { logger.debug("Generate series: " + series.getKey()); logger.debug("Start km: " + tmp.getKms(0)); logger.debug("End km: " + tmp.getKms(size-1)); logger.debug("Values : " + size); } } for (int i = 0; i < size; i++) { target = tmp.get(i, target); //logger.debug("++ Q Tuple: " + target[2] + " -> " + target[1]); series.add(target[2], target[1]); } q.addSeries(series); } } protected String getSeriesName(WQKms wqkms, String mode) { String name = wqkms.getName(); 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 :