mschaefer@9035: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
mschaefer@9035:  * Software engineering by
mschaefer@9035:  *  Björnsen Beratende Ingenieure GmbH
mschaefer@9035:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
mschaefer@9035:  *
mschaefer@9035:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@9035:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@9035:  * documentation coming with Dive4Elements River for details.
mschaefer@9035:  */
mschaefer@9035: 
mschaefer@9035: package org.dive4elements.river.artifacts.sinfo.common;
mschaefer@9035: 
mschaefer@9035: import java.util.HashSet;
mschaefer@9035: import java.util.Map;
mschaefer@9035: import java.util.Set;
mschaefer@9035: 
mschaefer@9035: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
mschaefer@9035: import org.dive4elements.artifacts.Artifact;
mschaefer@9035: import org.dive4elements.artifacts.CallContext;
mschaefer@9035: import org.dive4elements.river.artifacts.sinfo.predefinedflowdepth.PredefinedFlowDepthQueryCalculationResult;
mschaefer@9035: import org.dive4elements.river.exports.DiagramGenerator;
mschaefer@9035: import org.dive4elements.river.exports.StyledSeriesBuilder;
mschaefer@9035: import org.dive4elements.river.jfree.StyledXYSeries;
mschaefer@9035: import org.dive4elements.river.themes.ThemeDocument;
mschaefer@9035: 
mschaefer@9035: /**
mschaefer@9035:  * Processor to generate a data series for flow depth data loaded from the database
mschaefer@9035:  *
mschaefer@9035:  * @author Matthias Schäfer
mschaefer@9035:  *
mschaefer@9035:  */
mschaefer@9035: public class PredefinedFlowDepthProcessor extends AbstractSInfoProcessor {
mschaefer@9035: 
mschaefer@9035:     public static final String FACET_PREDEFINED_FLOW_DEPTH = "sinfo_facet_predefined_flowdepth";
mschaefer@9035: 
mschaefer@9035:     private static final String I18N_AXIS_LABEL = "sinfo.chart.flow_depth.section.yaxis.label";
mschaefer@9035: 
mschaefer@9035:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
mschaefer@9035: 
mschaefer@9035:     static {
mschaefer@9035:         HANDLED_FACET_TYPES.add(FACET_PREDEFINED_FLOW_DEPTH);
mschaefer@9035:     }
mschaefer@9035: 
mschaefer@9035:     public PredefinedFlowDepthProcessor() {
mschaefer@9035:         super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES);
mschaefer@9035:     }
mschaefer@9035: 
mschaefer@9035:     @Override
mschaefer@9035:     protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
mschaefer@9035: 
gernotbelger@9123:         final CallContext context = generator.getContext();
mschaefer@9035:         final Map<String, String> metaData = bundle.getFacet().getMetaData();
mschaefer@9035: 
mschaefer@9035:         final Artifact artifact = bundle.getArtifact();
mschaefer@9035: 
mschaefer@9035:         final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
mschaefer@9035:         series.putMetaData(metaData, artifact, context);
mschaefer@9035: 
mschaefer@9035:         final String facetName = bundle.getFacetName();
mschaefer@9035:         final PredefinedFlowDepthQueryCalculationResult data = (PredefinedFlowDepthQueryCalculationResult) bundle.getData(context);
mschaefer@9035:         if (data == null) {
mschaefer@9035:             // Check has been here before so we keep it for security reasons
mschaefer@9035:             // this should never happen though.
mschaefer@9035:             throw new IllegalStateException("Data is null for facet: " + facetName);
mschaefer@9035:         }
mschaefer@9035: 
mschaefer@9035:         final double[][] points = data.getStationPoints(SInfoResultType.flowdepth);
mschaefer@9035: 
mschaefer@9035:         StyledSeriesBuilder.addPoints(series, points, true);
mschaefer@9035:         generator.addAxisSeries(series, getAxisName(), visible);
mschaefer@9035: 
mschaefer@9035:         return null;
mschaefer@9035:     }
mschaefer@9035: }