Mercurial > dive4elements > river
view flys-artifacts/src/main/java/de/intevation/flys/exports/DischargeCurveGenerator.java @ 403:1ed48e2ddc1b
Added an OutGenerator to create discharge longitudinal section charts.
flys-artifacts/trunk@1844 c6561f87-3c4e-4783-a992-168aeb5c3f6f
author | Ingo Weinzierl <ingo.weinzierl@intevation.de> |
---|---|
date | Fri, 06 May 2011 14:20:26 +0000 |
parents | 52b6de9d0ab4 |
children | fc3ac59c3c8b |
line wrap: on
line source
package de.intevation.flys.exports; import java.util.ArrayList; import java.util.List; import java.util.Map; import org.apache.log4j.Logger; import org.w3c.dom.Document; import org.jfree.chart.JFreeChart; import org.jfree.chart.plot.XYPlot; import org.jfree.data.xy.DefaultXYDataset; import de.intevation.artifacts.Artifact; import de.intevation.flys.model.Gauge; import de.intevation.flys.model.River; import de.intevation.flys.artifacts.FLYSArtifact; import de.intevation.flys.artifacts.model.DischargeTables; import de.intevation.flys.artifacts.model.GaugesFactory; /** * An OutGenerator that generates discharge curves. * * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> */ public class DischargeCurveGenerator extends XYChartGenerator { /** The logger used in this generator.*/ private static Logger logger = Logger.getLogger(DischargeCurveGenerator.class); /** The Dataset that contains the data for the chart.*/ protected DefaultXYDataset dataset; public DischargeCurveGenerator() { super(); this.dataset = new DefaultXYDataset(); } protected String getChartTitle() { return "Abflusskurven"; } protected String getXAxisLabel() { return "Q[m³/s]"; } protected String getYAxisLabel() { return "W[cm]"; } protected void addDatasets(JFreeChart chart) { XYPlot plot = (XYPlot) chart.getPlot(); plot.setDataset(0, dataset); } public void doOut(Artifact artifact, String facet, Document attr) { logger.debug("DischargeCurveGenerator.doOut: " + facet); if (!(artifact instanceof FLYSArtifact)) { logger.error("Artifact is no instance of FLYSArtifact."); return; } FLYSArtifact flysArtifact = (FLYSArtifact) artifact; River river = flysArtifact.getRiver(); double[] distance = flysArtifact.getDistance(); List<double[]> ranges = new ArrayList<double[]>(); ranges.add(distance); List<Gauge> gauges = GaugesFactory.getGauges(river); List<Gauge> filtered = GaugesFactory.filterRanges(gauges, ranges); if (logger.isDebugEnabled()) { int numGauges = gauges != null ? gauges.size() : 0; int numFiltered = filtered != null ? filtered.size() : 0; logger.debug("++++++++++++++++++++"); logger.debug("Search gauges for river: " + river.getName()); logger.debug("-> ... from range: " + distance[0]); logger.debug("-> ... to range: " + distance[1]); logger.debug("-> Found " + numGauges + " gauges in total."); logger.debug("-> Found " + numFiltered + " gauges after filtering."); logger.debug("++++++++++++++++++++"); } String[] gaugeNames = new String[filtered.size()]; int idx = 0; for (Gauge gauge: filtered) { gaugeNames[idx++] = gauge.getName(); } DischargeTables dt = new DischargeTables(river.getName(), gaugeNames); Map<String, double[][]> gaugeValues = dt.getValues(100d); for (String gauge: gaugeNames) { double[][] values = gaugeValues.get(gauge); dataset.addSeries(gauge, values); } } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :