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@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: 
felix@7540:     private static final Logger logger = Logger.getLogger(AreaProcessor.class);
aheinecke@7101: 
aheinecke@7101:     @Override
aheinecke@7101:     public void doOut(
aheinecke@7101:             DiagramGenerator generator,
aheinecke@7101:             ArtifactAndFacet bundle,
aheinecke@7101:             ThemeDocument    theme,
aheinecke@7101:             boolean          visible) {
aheinecke@7101:         CallContext context = generator.getCallContext();
aheinecke@7101:         String seriesName = bundle.getFacetDescription();
aheinecke@7101:         StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(theme);
aheinecke@7101: 
aheinecke@7101:         logger.debug("Area Processor processing: " + seriesName);
aheinecke@7101: 
aheinecke@7101:         AreaFacet.Data data = (AreaFacet.Data) bundle.getData(context);
aheinecke@7101: 
aheinecke@7101:         XYSeries up   = null;
aheinecke@7101:         XYSeries down = null;
aheinecke@7101: 
aheinecke@7101:         if (data.getUpperData() != null) {
aheinecke@7101:             up = new StyledXYSeries(seriesName, false, theme);
aheinecke@7101:             if (data.getUpperData() instanceof WQKms) {
aheinecke@7101:                 if (FacetTypes.IS.Q(data.getRootFacetName())) {
aheinecke@7101:                     StyledSeriesBuilder.addPointsKmQ(up, (WQKms) data.getUpperData());
aheinecke@7101:                 }
aheinecke@7101:                 else {
aheinecke@7101:                     StyledSeriesBuilder.addPoints(up, (WKms) data.getUpperData());
aheinecke@7101:                 }
aheinecke@7101:             }
aheinecke@7101:             else if (data.getUpperData() instanceof double[][]) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(up, (double [][]) data.getUpperData(), false);
aheinecke@7101:             }
aheinecke@7101:             else if (data.getUpperData() instanceof WKms) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(up, (WKms) data.getUpperData());
aheinecke@7101:             }
aheinecke@7101:             else if (data.getUpperData() instanceof Lines.LineData) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(up, ((Lines.LineData) data.getUpperData()).points, false);
aheinecke@7101:             }
aheinecke@7101:             else {
aheinecke@7101:                 logger.error("Do not know how to deal with (up) area info from: "
aheinecke@7101:                         + data.getUpperData());
aheinecke@7101:             }
aheinecke@7101:         }
aheinecke@7101: 
aheinecke@7101:         // TODO Depending on style, the area (e.g. 20m^2) should be added as annotation.
aheinecke@7101: 
aheinecke@7101:         if (data.getLowerData() != null) {
aheinecke@7101:             // TODO: Sort this out: when the two series have the same name,
aheinecke@7101:             // the renderer (or anything in between) will not work correctly.
aheinecke@7101:             down = new StyledXYSeries(seriesName + " ", false, theme);
aheinecke@7101:             if (data.getLowerData() instanceof WQKms) {
aheinecke@7101:                 if (FacetTypes.IS.Q(data.getRootFacetName())) {
aheinecke@7101:                     StyledSeriesBuilder.addPointsKmQ(down, (WQKms) data.getLowerData());
aheinecke@7101:                 }
aheinecke@7101:                 else {
aheinecke@7101:                     StyledSeriesBuilder.addPoints(down, (WQKms) data.getLowerData());
aheinecke@7101:                 }
aheinecke@7101:             }
aheinecke@7101:             else if (data.getLowerData() instanceof double[][]) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(down, (double[][]) data.getLowerData(), false);
aheinecke@7101:             }
aheinecke@7101:             else if (data.getLowerData() instanceof WKms) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(down, (WKms) data.getLowerData());
aheinecke@7101:             }
aheinecke@7101:             else if (data.getLowerData() instanceof Lines.LineData) {
aheinecke@7101:                 StyledSeriesBuilder.addPoints(down, ((Lines.LineData) data.getLowerData()).points, false);
aheinecke@7101:             }
aheinecke@7101:             else {
aheinecke@7101:                 logger.error("Do not know how to deal with (down) area info from: "
aheinecke@7101:                         + data.getLowerData());
aheinecke@7101:             }
aheinecke@7101:         }
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@7101:         /* Decide axis name based on facet name */
aheinecke@7101:         generator.addAreaSeries(area,
aheinecke@7101:                 axisNameForFacet(data.getRootFacetName()), 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@7101:         if (FacetTypes.IS.W(facetName)) {
aheinecke@7101:             return "W";
aheinecke@7101:         }
aheinecke@7101:         else if (FacetTypes.IS.Q(facetName)) {
aheinecke@7101:             return "Q";
aheinecke@7101:         }
aheinecke@7101:         else {
aheinecke@7101:             logger.warn("Could not find axis for facet " + facetName);
aheinecke@7101:             return "W";
aheinecke@7101:         }
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: }