aheinecke@7101: /* Copyright (C) 2013 by Bundesanstalt für Gewässerkunde
aheinecke@7101:  * Software engineering by Intevation GmbH
aheinecke@7101:  *
aheinecke@7101:  * This file is Free Software under the GNU AGPL (>=v3)
aheinecke@7101:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
aheinecke@7101:  * documentation coming with Dive4Elements River for details.
aheinecke@7101:  */
aheinecke@7101: 
aheinecke@7101: 
aheinecke@7101: package org.dive4elements.river.exports.process;
aheinecke@7101: 
aheinecke@7101: import org.apache.log4j.Logger;
aheinecke@7101: 
aheinecke@7101: 
aheinecke@7101: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
aheinecke@7101: import org.dive4elements.artifacts.CallContext;
aheinecke@7101: 
aheinecke@7101: import org.dive4elements.river.artifacts.geom.Lines;
aheinecke@7101: import org.dive4elements.river.artifacts.model.WKms;
aheinecke@7101: import org.dive4elements.river.artifacts.model.WQKms;
aheinecke@7714: import org.dive4elements.river.artifacts.model.WQCKms;
aheinecke@7101: import org.dive4elements.river.artifacts.model.AreaFacet;
aheinecke@7101: import org.dive4elements.river.artifacts.model.FacetTypes;
aheinecke@7101: import org.dive4elements.river.exports.DiagramGenerator;
aheinecke@7101: import org.dive4elements.river.exports.StyledSeriesBuilder;
aheinecke@7101: import org.dive4elements.river.jfree.StyledAreaSeriesCollection;
aheinecke@7101: import org.dive4elements.river.jfree.StyledXYSeries;
aheinecke@7101: import org.dive4elements.river.themes.ThemeDocument;
aheinecke@7101: 
aheinecke@7101: import org.jfree.data.xy.XYSeries;
aheinecke@7101: 
aheinecke@7101: public class AreaProcessor extends DefaultProcessor {
aheinecke@7101: 
teichmann@8202:     private static final Logger log = Logger.getLogger(AreaProcessor.class);
aheinecke@7101: 
tom@8856:     protected XYSeries getSeries(
gernotbelger@9556:         String facetName,
tom@8856:         String seriesName,
tom@8856:         Object data,
tom@8856:         String facetType,
tom@8856:         ThemeDocument theme
tom@8856:     ) {
aheinecke@7714:         if (data == null || facetType == null) {
aheinecke@7714:             return null;
aheinecke@7714:         }
aheinecke@7714: 
gernotbelger@9556:         XYSeries series = new StyledXYSeries(facetName,seriesName, false, theme);
aheinecke@7714: 
aheinecke@7714:         if (facetType.equals(FacetTypes.DISCHARGE_LONGITUDINAL_C)) {
aheinecke@7714:             WQCKms wqckms = (WQCKms) data;
aheinecke@7714:             int size = wqckms.size();
aheinecke@7714:             for (int i = 0; i < size; i++) {
aheinecke@7714:                 series.add(wqckms.getKm(i), wqckms.getC(i), false);
aheinecke@7714:             }
aheinecke@7714:         } else if (data instanceof WQKms) {
aheinecke@7714:             if (FacetTypes.IS.Q(facetType)) {
aheinecke@7714:                 StyledSeriesBuilder.addPointsKmQ(series, (WQKms) data);
aheinecke@7714:             }
aheinecke@7714:             else {
aheinecke@7714:                 StyledSeriesBuilder.addPoints(series, (WKms) data);
aheinecke@7714:             }
aheinecke@7714:         }
aheinecke@7714:         else if (data instanceof double[][]) {
aheinecke@7714:             StyledSeriesBuilder.addPoints(series, (double [][]) data, false);
aheinecke@7714:         }
aheinecke@7714:         else if (data instanceof WKms) {
aheinecke@7714:             StyledSeriesBuilder.addPoints(series, (WKms) data);
aheinecke@7714:         }
aheinecke@7714:         else if (data instanceof Lines.LineData) {
tom@8856:             StyledSeriesBuilder.addPoints(
tom@8856:                 series, ((Lines.LineData) data).points, false);
aheinecke@7714:         }
aheinecke@7714:         else {
teichmann@8202:             log.error("Do not know how to deal with (up) area info from: "
aheinecke@7714:                     + data);
aheinecke@7714:         }
aheinecke@7714:         return series;
aheinecke@7714:     }
aheinecke@7714: 
aheinecke@7714: 
aheinecke@7101:     @Override
aheinecke@7101:     public void doOut(
aheinecke@7101:             DiagramGenerator generator,
aheinecke@7101:             ArtifactAndFacet bundle,
aheinecke@7101:             ThemeDocument    theme,
aheinecke@7101:             boolean          visible) {
gernotbelger@9123:         CallContext context = generator.getContext();
aheinecke@7101:         String seriesName = bundle.getFacetDescription();
gernotbelger@9556:         StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(bundle.getFacetName(),theme);
aheinecke@7101: 
teichmann@8202:         log.debug("Area Processor processing: " + seriesName);
aheinecke@7101: 
aheinecke@7101:         AreaFacet.Data data = (AreaFacet.Data) bundle.getData(context);
aheinecke@7101: 
aheinecke@7714:         String lowerFacetName = data.getLowerFacetName();
aheinecke@7714:         String upperFacetName = data.getUpperFacetName();
aheinecke@7101: 
gernotbelger@9556:         XYSeries up = getSeries(bundle.getFacetName(),
tom@8856:             seriesName, data.getUpperData(), upperFacetName, theme);
gernotbelger@9556:         XYSeries down = getSeries(bundle.getFacetName(),
tom@8856:             seriesName + " ", data.getLowerData(), lowerFacetName, theme);
aheinecke@7714: 
aheinecke@7101: 
tom@8856:         // TODO Depending on style, the area (e.g. 20m^2)
tom@8856:         // should be added as annotation.
aheinecke@7101: 
aheinecke@7101:         if (up == null && down != null) {
aheinecke@7101:             area.setMode(StyledAreaSeriesCollection.FILL_MODE.ABOVE);
aheinecke@7101:             down.setKey(seriesName);
aheinecke@7101:             area.addSeries(down);
aheinecke@7101:             area.addSeries(StyledSeriesBuilder.createGroundAtInfinity(down));
aheinecke@7101:         }
aheinecke@7101:         else if (up != null && down == null) {
aheinecke@7101:             area.setMode(StyledAreaSeriesCollection.FILL_MODE.UNDER);
aheinecke@7101:             area.addSeries(up);
aheinecke@7101:             area.addSeries(StyledSeriesBuilder.createGroundAtInfinity(up));
aheinecke@7101:         }
aheinecke@7101:         else if (up != null && down != null) {
aheinecke@7101:             if (data.doPaintBetween()) {
aheinecke@7101:                 area.setMode(StyledAreaSeriesCollection.FILL_MODE.BETWEEN);
aheinecke@7101:             }
aheinecke@7101:             else {
aheinecke@7101:                 area.setMode(StyledAreaSeriesCollection.FILL_MODE.ABOVE);
aheinecke@7101:             }
aheinecke@7101:             area.addSeries(up);
aheinecke@7101:             area.addSeries(down);
aheinecke@7101:         }
aheinecke@7101: 
aheinecke@7714:         String facetNameForAxis = lowerFacetName == null
aheinecke@7714:             ? upperFacetName
aheinecke@7714:             : lowerFacetName;
aheinecke@7101:         /* Decide axis name based on facet name */
gernotbelger@9496:         generator.addAxisDataset(area,
aheinecke@7714:                 axisNameForFacet(facetNameForAxis), visible);
aheinecke@7101:     }
aheinecke@7101: 
aheinecke@7101:     /** Look up the axis identifier for a given facet type. */
aheinecke@7101:     private String axisNameForFacet(String facetName) {
aheinecke@7713:         if (FacetTypes.W_DIFFERENCES.equals(facetName)) {
aheinecke@7713:             return "diffW";
aheinecke@7713:         }
aheinecke@7714:         if (FacetTypes.IS.W(facetName) ||
aheinecke@7714:             FacetTypes.DISCHARGE_LONGITUDINAL_C.equals(facetName)) {
aheinecke@7101:             return "W";
aheinecke@7101:         }
aheinecke@7713:         if (FacetTypes.IS.Q(facetName)) {
aheinecke@7101:             return "Q";
aheinecke@7101:         }
teichmann@8202:         log.warn("Could not find axis for facet " + facetName);
aheinecke@7713:         return "W";
aheinecke@7101:     }
aheinecke@7101: 
aheinecke@7101:     @Override
aheinecke@7101:     public boolean canHandle(String facetType) {
aheinecke@7101:         if (facetType == null) {
aheinecke@7101:             return false;
aheinecke@7101:         }
aheinecke@7101:         return FacetTypes.IS.AREA(facetType);
aheinecke@7101:     }
aheinecke@7101: }