gernotbelger@8884: /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
gernotbelger@8884:  * Software engineering by
gernotbelger@8884:  *  Björnsen Beratende Ingenieure GmbH
gernotbelger@8884:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
gernotbelger@8884:  *
gernotbelger@8884:  * This file is Free Software under the GNU AGPL (>=v3)
gernotbelger@8884:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
gernotbelger@8884:  * documentation coming with Dive4Elements River for details.
gernotbelger@8884:  */
gernotbelger@8884: 
gernotbelger@8916: package org.dive4elements.river.artifacts.sinfo.common;
gernotbelger@8884: 
gernotbelger@8884: import java.util.HashSet;
gernotbelger@8884: import java.util.Set;
gernotbelger@8884: 
gernotbelger@8886: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
gernotbelger@8916: import org.dive4elements.artifactdatabase.state.Facet;
gernotbelger@8886: import org.dive4elements.artifacts.CallContext;
gernotbelger@8916: import org.dive4elements.river.artifacts.resources.Resources;
gernotbelger@8916: import org.dive4elements.river.artifacts.states.DefaultState.ComputeType;
gernotbelger@8886: import org.dive4elements.river.exports.DiagramGenerator;
gernotbelger@8886: import org.dive4elements.river.exports.StyledSeriesBuilder;
gernotbelger@8886: import org.dive4elements.river.jfree.StyledAreaSeriesCollection;
gernotbelger@8886: import org.dive4elements.river.jfree.StyledXYSeries;
gernotbelger@8886: import org.dive4elements.river.themes.ThemeDocument;
gernotbelger@8886: 
gernotbelger@8884: public final class TkhProcessor extends AbstractSInfoProcessor {
gernotbelger@8884: 
gernotbelger@8940:     private static String FACET_TKH = "sinfo_flow_depth.tkh";
gernotbelger@8916: 
gernotbelger@8940:     private static final String I18N_AXIS_LABEL = "sinfo.chart.tkh.section.yaxis.label";
gernotbelger@8916: 
gernotbelger@8940:     private static final String I18N_FACET_TKH_DESCRIPTION = "sinfo.facet.tkh.description";
gernotbelger@8940: 
gernotbelger@8940:     private static final String SINFO_CHART_TKX_YAXIS_LABEL = "sinfo.chart.tkh.yaxis.label";
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_TKH);
gernotbelger@8884:     }
gernotbelger@8884: 
gernotbelger@8884:     public TkhProcessor() {
gernotbelger@8884:         super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES);
gernotbelger@8884:     }
gernotbelger@8884: 
gernotbelger@8884:     @Override
gernotbelger@8886:     protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
gernotbelger@8886:         final CallContext context = generator.getCallContext();
gernotbelger@8884: 
gernotbelger@8886:         final String facetName = bundle.getFacetName();
gernotbelger@8916:         final AbstractSInfoCalculationResult<?> data = (AbstractSInfoCalculationResult<?>) 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 StyledXYSeries seriesUp = new StyledXYSeries(bundle.getFacetDescription(), theme);
gernotbelger@8886:         final double[][] pointsUp = data.getTkhUpPoints();
gernotbelger@8886:         StyledSeriesBuilder.addPoints(seriesUp, pointsUp, true);
gernotbelger@8886: 
gernotbelger@8886:         // REMARK: we add " " because the description is misused as id, which must be unique.
gernotbelger@8886:         final StyledXYSeries seriesDown = new StyledXYSeries(bundle.getFacetDescription() + " ", theme);
gernotbelger@8886:         final double[][] pointsDown = data.getTkhDownPoints();
gernotbelger@8886:         StyledSeriesBuilder.addPoints(seriesDown, pointsDown, true);
gernotbelger@8886: 
gernotbelger@8886:         final StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(theme);
gernotbelger@8886:         area.setMode(StyledAreaSeriesCollection.FILL_MODE.BETWEEN);
gernotbelger@8886:         area.addSeries(seriesUp);
gernotbelger@8886:         area.addSeries(seriesDown);
gernotbelger@8886: 
gernotbelger@8886:         generator.addAreaSeries(area, getAxisName(), visible);
gernotbelger@8886: 
gernotbelger@8886:         return null;
gernotbelger@8884:     }
gernotbelger@8916: 
gernotbelger@8916:     public static Facet createTkhFacet(final CallContext context, final String hash, final String id, final AbstractSInfoCalculationResult<?> result,
gernotbelger@8916:             final int index) {
gernotbelger@8916: 
gernotbelger@8916:         final String facetTkhDescription = Resources.getMsg(context.getMeta(), I18N_FACET_TKH_DESCRIPTION, I18N_FACET_TKH_DESCRIPTION,
gernotbelger@8916:                 result.getLabel());
gernotbelger@8916:         // FIXME: doe not work yet...
gernotbelger@8916:         return new SInfoResultFacet(index, TkhProcessor.FACET_TKH, facetTkhDescription, TkhProcessor.SINFO_CHART_TKX_YAXIS_LABEL, ComputeType.ADVANCE, id,
gernotbelger@8916:                 hash);
gernotbelger@8916:     }
gernotbelger@8884: }