Mercurial > dive4elements > river
view artifacts/src/main/java/org/dive4elements/river/exports/process/DischargeProcessor.java @ 6914:695a5f6599b4
Minor refactoring, renamed variable, extracted method.
author | Felix Wolfsteller <felix.wolfsteller@intevation.de> |
---|---|
date | Mon, 26 Aug 2013 14:28:19 +0200 |
parents | 1b35b2ddfc28 |
children | 6fb355b87f50 |
line wrap: on
line source
/* Copyright (C) 2011, 2012, 2013 by Bundesanstalt für Gewässerkunde * Software engineering by Intevation GmbH * * This file is Free Software under the GNU AGPL (>=v3) * and comes with ABSOLUTELY NO WARRANTY! Check out the * documentation coming with Dive4Elements River for details. */ package org.dive4elements.river.exports.process; import java.util.ArrayList; import java.util.List; import org.apache.log4j.Logger; import org.jfree.data.xy.XYSeries; import org.dive4elements.artifactdatabase.state.ArtifactAndFacet; import org.dive4elements.artifacts.CallContext; import org.dive4elements.river.artifacts.model.FacetTypes; import org.dive4elements.river.artifacts.model.WQKms; import org.dive4elements.river.exports.DischargeCurveGenerator; import org.dive4elements.river.exports.XYChartGenerator; import org.dive4elements.river.jfree.CollisionFreeXYTextAnnotation; import org.dive4elements.river.jfree.RiverAnnotation; import org.dive4elements.river.jfree.StyledXYSeries; import org.dive4elements.river.themes.ThemeDocument; import org.jfree.chart.annotations.XYTextAnnotation; /** Helper for data handling in discharge diagrams. */ public class DischargeProcessor implements Processor, FacetTypes { private final static Logger logger = Logger.getLogger(DischargeProcessor.class); /** Station for which the diagram is shown. */ private double km; /** 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() { km = Double.NaN; } public DischargeProcessor(double km) { this.km = km; } /** Process data, add it to plot. */ @Override public void doOut( XYChartGenerator generator, ArtifactAndFacet aandf, ThemeDocument theme, boolean visible, int axisIndex ) { CallContext context = generator.getCallContext(); Object data = aandf.getData(context); if (false && data instanceof WQKms) { doWQKmsPointOut( generator, (WQKms) data, aandf, theme, visible, axisIndex); return; } else if (data instanceof RiverAnnotation) { doRiverAnnotationOut( generator, (RiverAnnotation) data, aandf, theme, visible); return; } logger.error("Can't process " + data.getClass().getName() + " objects"); } /** True if this processor knows how to deal with facetType. */ @Override public boolean canHandle(String facetType) { return /*STATIC_WQKMS_W.equals(facetType) ||*/ COMPUTED_DISCHARGE_MAINVALUES_Q.equals(facetType) || MAINVALUES_Q.equals(facetType) || COMPUTED_DISCHARGE_MAINVALUES_W.equals(facetType) || MAINVALUES_W.equals(facetType); } /** The station of the current calculation/view. */ protected double getKm() { return km; } /** Handle WQKms data by finding w/q values at given km. */ protected void doWQKmsPointOut(XYChartGenerator generator, WQKms wqkms, ArtifactAndFacet aandf, ThemeDocument theme, boolean visible, int axidx ) { logger.debug("doWQKmsPointOut"); String title = aandf.getFacetDescription(); XYSeries series = new StyledXYSeries( title, theme); double[] kms = wqkms.getKms(); 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()); } } // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :