gernotbelger@8854: /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@8884:  * Software engineering by
gernotbelger@8884:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@8854:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@8854:  *
gernotbelger@8854:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@8854:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@8854:  * documentation coming with Dive4Elements River for details.
gernotbelger@8854:  */
gernotbelger@8854: 
gernotbelger@8854: package org.dive4elements.river.artifacts.sinfo.flowdepth;
gernotbelger@8854: 
gernotbelger@8884: import java.util.HashSet;
gernotbelger@8886: import java.util.Map;
gernotbelger@8884: import java.util.Set;
gernotbelger@8854: 
gernotbelger@8886: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
gernotbelger@8886: import org.dive4elements.artifacts.CallContext;
gernotbelger@8886: import org.dive4elements.river.exports.DiagramGenerator;
gernotbelger@8886: import org.dive4elements.river.exports.StyledSeriesBuilder;
gernotbelger@8886: import org.dive4elements.river.jfree.StyledXYSeries;
gernotbelger@8886: import org.dive4elements.river.themes.ThemeDocument;
gernotbelger@8886: 
gernotbelger@8884: public final class FlowDepthProcessor extends AbstractSInfoProcessor {
gernotbelger@8854: 
gernotbelger@8884:     /* Theme name, usually defined in 'FacetTypes', but that is soooo bad dependencies... */
gernotbelger@8884:     // REMARK: these mustend with 'filtered' so extra handling happens in chart: point are always recalculated, because data
gernotbelger@8884:     // changes depending on zoom state
gernotbelger@8854:     static String FACET_FLOW_DEPTH_FILTERED = "sinfo_flow_depth.filtered";
gernotbelger@8854: 
gernotbelger@8884:     static String FACET_FLOW_DEPTH_TKH_FILTERED = "sinfo_flow_depth.tkh.filtered";
gernotbelger@8884: 
gernotbelger@8884:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
gernotbelger@8884: 
gernotbelger@8884:     static {
gernotbelger@8884:         HANDLED_FACET_TYPES.add(FACET_FLOW_DEPTH_FILTERED);
gernotbelger@8884:         HANDLED_FACET_TYPES.add(FACET_FLOW_DEPTH_TKH_FILTERED);
gernotbelger@8884:     }
gernotbelger@8854: 
gernotbelger@8854:     private static final String I18N_AXIS_LABEL = "sinfo.chart.flow_depth.section.yaxis.label";
gernotbelger@8854: 
gernotbelger@8884:     public FlowDepthProcessor() {
gernotbelger@8884:         super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES);
gernotbelger@8854:     }
gernotbelger@8854: 
gernotbelger@8886:     @Override
gernotbelger@8886:     protected final String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
gernotbelger@8886: 
gernotbelger@8886:         final CallContext context = generator.getCallContext();
gernotbelger@8886:         final Map<String, String> metaData = bundle.getFacet().getMetaData();
gernotbelger@8886: 
gernotbelger@8886:         final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
gernotbelger@8886:         series.putMetaData(metaData, bundle.getArtifact(), context);
gernotbelger@8886: 
gernotbelger@8886:         final String facetName = bundle.getFacetName();
gernotbelger@8886:         final FlowDepthCalculationResult data = (FlowDepthCalculationResult) bundle.getData(context);
gernotbelger@8886:         if (data == null) {
gernotbelger@8886:             // Check has been here before so we keep it for security reasons
gernotbelger@8886:             // this should never happen though.
gernotbelger@8886:             throw new IllegalStateException("Data is null for facet: " + facetName);
gernotbelger@8886:         }
gernotbelger@8886: 
gernotbelger@8886:         final double[][] points = generatePoints(data, facetName);
gernotbelger@8886: 
gernotbelger@8886:         StyledSeriesBuilder.addPoints(series, points, true);
gernotbelger@8886:         generator.addAxisSeries(series, getAxisName(), visible);
gernotbelger@8886: 
gernotbelger@8886:         return metaData.get("Y");
gernotbelger@8886:     }
gernotbelger@8886: 
gernotbelger@8884:     // FIXME: do filtering
gernotbelger@8886:     private double[][] generatePoints(final FlowDepthCalculationResult data, final String facetName) {
gernotbelger@8854: 
gernotbelger@8884:         if (FACET_FLOW_DEPTH_FILTERED.contentEquals(facetName))
gernotbelger@8884:             return data.getFlowDepthPoints();
gernotbelger@8854: 
gernotbelger@8884:         if (FACET_FLOW_DEPTH_TKH_FILTERED.contentEquals(facetName))
gernotbelger@8884:             return data.getFlowDepthTkhPoints();
gernotbelger@8854: 
gernotbelger@8886:         final String error = String.format("Unknown facet name: %s", facetName);
gernotbelger@8886:         throw new UnsupportedOperationException(error);
gernotbelger@8854:     }
gernotbelger@8854: }