dive4elements/river: artifacts/src/main/java/org/dive4elements/river/exports/process/DischargeProcessor.java comparison

comparison artifacts/src/main/java/org/dive4elements/river/exports/process/DischargeProcessor.java @ 8182:0b4cd7a5f079

Changed discharge processor to plot only discharge curves and added processor for computed discharge curves.

author	Raimund Renkert <rrenkert@intevation.de>
date	Thu, 04 Sep 2014 11:14:24 +0200
parents	67e87c2d3748
children	59e50aa0c6c2

comparison

equal deleted inserted replaced

-:209f8a9e8a75
+:0b4cd7a5f079
 import java.util.ArrayList;
 import java.util.List;
 import org.apache.log4j.Logger;
+import org.jfree.chart.annotations.XYTextAnnotation;
 import org.jfree.data.xy.XYSeries;
 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
 import org.dive4elements.artifacts.CallContext;
+import org.dive4elements.river.artifacts.D4EArtifact;
+import org.dive4elements.river.artifacts.access.RiverAccess;
 import org.dive4elements.river.artifacts.model.FacetTypes;
 import org.dive4elements.river.artifacts.model.WQKms;
+import org.dive4elements.river.exports.DiagramGenerator;
 import org.dive4elements.river.exports.DischargeCurveGenerator;
+import org.dive4elements.river.exports.StyledSeriesBuilder;
 import org.dive4elements.river.exports.XYChartGenerator;
 import org.dive4elements.river.jfree.CollisionFreeXYTextAnnotation;
 import org.dive4elements.river.jfree.RiverAnnotation;
 import org.dive4elements.river.jfree.StickyAxisAnnotation;
 import org.dive4elements.river.jfree.StyledXYSeries;
+import org.dive4elements.river.model.Gauge;
+import org.dive4elements.river.model.River;
 import org.dive4elements.river.themes.ThemeDocument;
-import org.jfree.chart.annotations.XYTextAnnotation;
 /** Helper for data handling in discharge diagrams. */
 public class DischargeProcessor
 extends DefaultProcessor implements FacetTypes {
 /** Tolerance for comparison of kilometers. */
 public static final double KM_EPSILON = 0.001d;
 /** This processor needs to be constructed with a given km. */
-private DischargeProcessor() {
+public DischargeProcessor() {
 km = Double.NaN;
 }
 public DischargeProcessor(double km) {
 /** Process data, add it to plot. */
 @Override
 public void doOut(
-XYChartGenerator generator,
+DiagramGenerator generator,
 ArtifactAndFacet bundle,
 ThemeDocument theme,
-boolean visible,
+boolean visible
-int axisIndex
 ) {
 CallContext context = generator.getCallContext();
 Object data = bundle.getData(context);
 if (data instanceof WQKms) {
-doWQKmsPointOut(
+doDischargeOut(
-generator, (WQKms) data, bundle, theme, visible, axisIndex);
+generator,
-return;
+bundle,
-}
+bundle.getFacetName(),
-else if (data instanceof RiverAnnotation) {
+theme,
-doRiverAnnotationOut(
+visible);
-generator, (RiverAnnotation) data, bundle, theme, visible);
-return;
-}
-else if (data instanceof double[][]) {
-doMarksOut(
-generator, (double[][]) data, bundle, theme, visible);
-return;
 }
 else {
 logger.error("Can't process "
 + data.getClass().getName() + " objects of facet "
 + bundle.getFacetName());
 }
 }
 /** True if this processor knows how to deal with facetType. */
 @Override
 public boolean canHandle(String facetType) {
-return STATIC_WQKMS_W.equals(facetType)
+return DISCHARGE_CURVE.equals(facetType)
-|| COMPUTED_DISCHARGE_MAINVALUES_Q.equals(facetType)
+|| GAUGE_DISCHARGE_CURVE.equals(facetType);
-|| MAINVALUES_Q.equals(facetType)
-|| COMPUTED_DISCHARGE_MAINVALUES_W.equals(facetType)
-|| MAINVALUES_W.equals(facetType)
-|| STATIC_W_INTERPOL.equals(facetType);
 }
 /** The station of the current calculation/view. */
 protected double getKm() {
 return km;
 }
+/**
+* Add series with discharge curve to diagram.
+*/
+protected void doDischargeOut(
+DiagramGenerator generator,
+ArtifactAndFacet bundle,
+String        name,
+ThemeDocument theme,
+boolean       visible)
+{
+logger.debug("DischargeCurveGenerator.doDischargeOut");
+CallContext context = generator.getCallContext();
+WQKms data = (WQKms)bundle.getData(context);
+XYSeries series = new StyledXYSeries(name, theme);
+double[][] wqData = new double[2][data.size()];
+for (int i = 0; i < data.size(); i++) {
+wqData[0][i] = data.getQ(i);
+wqData[1][i] = data.getW(i);
+}
+StyledSeriesBuilder.addPoints(series, wqData, false);
-/** Handle WQKms data by finding w/q values at given km. */
+String gaugeName = data.getName();
-protected void doWQKmsPointOut(XYChartGenerator generator,
+RiverAccess access = new RiverAccess((D4EArtifact)bundle.getArtifact());
-WQKms wqkms,
+River river = access.getRiver();
-ArtifactAndFacet bundle,
-ThemeDocument theme,
-boolean visible,
-int axidx
-) {
-logger.debug("doWQKmsPointOut");
-String title = bundle.getFacetDescription();
-XYSeries series = new StyledXYSeries(
-title,
-theme);
-double[] kms = wqkms.getKms();
+if (river == null) {
+logger.debug("no river found");
-for (int i = 0 ; i< kms.length; i++) {
-if (Math.abs(kms[i] - getKm()) <= KM_EPSILON) {
-series.add(wqkms.getQ(i), wqkms.getW(i));
-generator.addAxisSeries(series, axidx, visible);
-if(visible && theme.parseShowPointLabel()) {
-List<XYTextAnnotation> textAnnos = new ArrayList<XYTextAnnotation>();
-XYTextAnnotation anno = new CollisionFreeXYTextAnnotation(
-title,
-wqkms.getQ(i),
-// TODO add a percentage to the extend of W axis
-wqkms.getW(i));
-textAnnos.add(anno);
-RiverAnnotation flysAnno = new RiverAnnotation(null, null, null, theme);
-flysAnno.setTextAnnotations(textAnnos);
-generator.addAnnotations(flysAnno);
-}
-return;
-}
-}
-logger.warn("No WQ found for km " + getKm());
-}
-protected void doRiverAnnotationOut(XYChartGenerator generator,
-RiverAnnotation annotations,
-ArtifactAndFacet bundle,
-ThemeDocument theme,
-boolean visible
-) {
-if (!(generator instanceof DischargeCurveGenerator)) {
-logger.error("DischargeProcessor can only be used in " +
-" in DischargeCurveGenerator-classes.");
-return;
-}
-logger.debug("doRiverAnnotationOut");
-DischargeCurveGenerator dGenerator =
-(DischargeCurveGenerator) generator;
-dGenerator.translateRiverAnnotation(annotations);
-dGenerator.doAnnotations(
-annotations,
-bundle, theme, visible);
-}
-/**
-* Put Sticky Axis Markers to Y-axis for each value.
-* @param data [[-,y1],[-,y2],...] ('x'-coordinates ignored)
-*/
-protected void doMarksOut(XYChartGenerator generator,
-double[][] data,
-ArtifactAndFacet bundle,
-ThemeDocument theme,
-boolean visible
-) {
-logger.debug("doMarksOut");
-if (!visible) {
 return;
 }
-// TODO subtract gauge null point if at gauge.
+Gauge gauge = river.determineGaugeByName(gaugeName);
-String title = bundle.getFacetDescription();
-List<StickyAxisAnnotation> yMarks = new ArrayList<StickyAxisAnnotation>();
-for (double yPos: data[1]) {
+if (gauge == null) {
-yMarks.add(new StickyAxisAnnotation(
+logger.debug("no gauge found");
-title,
+return;
-(float) yPos,
-StickyAxisAnnotation.SimpleAxis.Y_AXIS));
 }
-generator.doAnnotations(new RiverAnnotation(title, yMarks),
+generator.addAxisSeries(series, axisName, visible);
-bundle, theme, visible);
 }
 }
 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :

Mercurial > dive4elements > river

comparison artifacts/src/main/java/org/dive4elements/river/exports/process/DischargeProcessor.java @ 8182:0b4cd7a5f079