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@9200: import java.util.Map;
gernotbelger@8884: import java.util.Set;
gernotbelger@8884: 
gernotbelger@8884: import org.apache.log4j.Logger;
gernotbelger@8884: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
gernotbelger@9203: import org.dive4elements.artifactdatabase.state.Facet;
gernotbelger@9200: import org.dive4elements.artifacts.Artifact;
gernotbelger@9200: import org.dive4elements.artifacts.CallContext;
gernotbelger@9200: import org.dive4elements.river.artifacts.D4EArtifact;
gernotbelger@9200: import org.dive4elements.river.artifacts.access.RiverAccess;
gernotbelger@9200: import org.dive4elements.river.artifacts.common.AbstractCalculationResult;
gernotbelger@9200: import org.dive4elements.river.artifacts.common.IResultType;
gernotbelger@9200: import org.dive4elements.river.artifacts.context.RiverContext;
gernotbelger@9200: import org.dive4elements.river.artifacts.math.MovingAverage;
gernotbelger@9200: import org.dive4elements.river.artifacts.model.ZoomScale;
gernotbelger@9203: import org.dive4elements.river.artifacts.resources.Resources;
gernotbelger@9203: import org.dive4elements.river.artifacts.states.DefaultState.ComputeType;
gernotbelger@8884: import org.dive4elements.river.exports.DiagramGenerator;
gernotbelger@9200: import org.dive4elements.river.exports.StyledSeriesBuilder;
gernotbelger@8884: import org.dive4elements.river.exports.process.DefaultProcessor;
gernotbelger@9200: import org.dive4elements.river.jfree.StyledAreaSeriesCollection;
gernotbelger@9200: import org.dive4elements.river.jfree.StyledXYSeries;
gernotbelger@8884: import org.dive4elements.river.themes.ThemeDocument;
gernotbelger@8884: 
gernotbelger@8884: /**
gernotbelger@8884:  * Abstraction for some processor implementation within S-INFO. Probably this abstraction could also be used for other
gernotbelger@8884:  * cases as well.
gernotbelger@8884:  *
gernotbelger@8884:  * @author Gernot Belger
gernotbelger@8884:  *
gernotbelger@8884:  */
gernotbelger@8916: public abstract class AbstractSInfoProcessor extends DefaultProcessor {
gernotbelger@8884: 
gernotbelger@9200:     protected static final double GAP_DISTANCE = 0.101;
gernotbelger@9200: 
gernotbelger@8884:     private final static Logger log = Logger.getLogger(AbstractSInfoProcessor.class);
gernotbelger@8884: 
gernotbelger@8884:     private String yAxisLabel;
gernotbelger@8884: 
gernotbelger@8884:     private final Set<String> handled_facet_types;
gernotbelger@8884: 
gernotbelger@8884:     private final String i18n_axis_label;
gernotbelger@8884: 
gernotbelger@8884:     public AbstractSInfoProcessor(final String i18n_axis_label, final Set<String> handled_facet_types) {
gernotbelger@8884:         this.i18n_axis_label = i18n_axis_label;
gernotbelger@8884:         this.handled_facet_types = handled_facet_types;
gernotbelger@8884:     }
gernotbelger@8884: 
gernotbelger@8884:     @Override
gernotbelger@8884:     public final void doOut(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
gernotbelger@8886:         try {
gernotbelger@8886:             this.yAxisLabel = generateSeries(generator, bundle, theme, visible);
gernotbelger@8884:         }
gernotbelger@8886:         catch (final Exception e) {
gernotbelger@8886:             log.error(e.getMessage(), e);
gernotbelger@8886:         }
gernotbelger@8884:     }
gernotbelger@8884: 
gernotbelger@8884:     /**
gernotbelger@8886:      * @return The axis label
gernotbelger@8884:      */
gernotbelger@8886:     protected abstract String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible);
gernotbelger@8884: 
gernotbelger@9200:     protected final String buildSeriesForType(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible,
gernotbelger@9200:             final IResultType resultType, final Double gapDistance) {
gernotbelger@9200:         final CallContext context = generator.getContext();
gernotbelger@9200:         final Map<String, String> metaData = bundle.getFacet().getMetaData();
gernotbelger@9200: 
gernotbelger@9200:         final Artifact artifact = bundle.getArtifact();
gernotbelger@9200: 
gernotbelger@9200:         final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
gernotbelger@9200:         series.putMetaData(metaData, artifact, context);
gernotbelger@9200: 
gernotbelger@9200:         final String facetName = bundle.getFacetName();
gernotbelger@9200: 
gernotbelger@9200:         final AbstractCalculationResult data = (AbstractCalculationResult) bundle.getData(context);
gernotbelger@9200:         if (data == null) {
gernotbelger@9200:             // Check has been here before so we keep it for security reasons
gernotbelger@9200:             // this should never happen though.
gernotbelger@9200:             throw new IllegalStateException("Data is null for facet: " + facetName);
gernotbelger@9200:         }
gernotbelger@9200: 
gernotbelger@9200:         final double[][] points = generatePoints(context, artifact, data, facetName, resultType);
gernotbelger@9200: 
gernotbelger@9200:         if (gapDistance == null)
gernotbelger@9200:             StyledSeriesBuilder.addPoints(series, points, true);
gernotbelger@9200:         else
gernotbelger@9200:             StyledSeriesBuilder.addPoints(series, points, true, gapDistance);
gernotbelger@9200: 
gernotbelger@9200:         generator.addAxisSeries(series, getAxisName(), visible);
gernotbelger@9200: 
gernotbelger@9200:         return metaData.get("Y");
gernotbelger@9200:     }
gernotbelger@9200: 
mschaefer@9204:     protected final String buildSeriesForType1(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme,
mschaefer@9204:             final boolean visible,
mschaefer@9204:             final IResultType resultType, final Double gapDistance) {
mschaefer@9204:         final CallContext context = generator.getContext();
mschaefer@9204:         final Map<String, String> metaData = bundle.getFacet().getMetaData();
mschaefer@9204: 
mschaefer@9204:         final Artifact artifact = bundle.getArtifact();
mschaefer@9204: 
mschaefer@9204:         final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
mschaefer@9204:         series.putMetaData(metaData, artifact, context);
mschaefer@9204: 
mschaefer@9204:         final String facetName = bundle.getFacetName();
mschaefer@9204: 
mschaefer@9204:         final AbstractCalculationResult data = (AbstractCalculationResult) bundle.getData(context);
mschaefer@9204:         if (data == null) {
mschaefer@9204:             // Check has been here before so we keep it for security reasons
mschaefer@9204:             // this should never happen though.
mschaefer@9204:             throw new IllegalStateException("Data is null for facet: " + facetName);
mschaefer@9204:         }
mschaefer@9204: 
mschaefer@9204:         final double[][] points = generatePoints(context, artifact, data, facetName, resultType, bundle.getFacet().getIndex());
mschaefer@9204: 
mschaefer@9204:         if (gapDistance == null)
mschaefer@9204:             StyledSeriesBuilder.addPoints(series, points, true);
mschaefer@9204:         else
mschaefer@9204:             StyledSeriesBuilder.addPoints(series, points, true, gapDistance);
mschaefer@9204: 
mschaefer@9204:         generator.addAxisSeries(series, getAxisName(), visible);
mschaefer@9204: 
mschaefer@9204:         return metaData.get("Y");
mschaefer@9204:     }
mschaefer@9204: 
gernotbelger@9200:     protected final String buildSeriesForTkh(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme,
gernotbelger@9200:             final boolean visible) {
gernotbelger@9200:         final CallContext context = generator.getContext();
gernotbelger@9200: 
gernotbelger@9200:         final String facetName = bundle.getFacetName();
gernotbelger@9200:         final AbstractTkhCalculationResult data = (AbstractTkhCalculationResult) bundle.getData(context);
gernotbelger@9200:         if (data == null) {
gernotbelger@9200:             // Check has been here before so we keep it for security reasons
gernotbelger@9200:             // this should never happen though.
gernotbelger@9200:             throw new IllegalStateException("Data is null for facet: " + facetName);
gernotbelger@9200:         }
gernotbelger@9200: 
gernotbelger@9200:         final StyledXYSeries seriesUp = new StyledXYSeries(bundle.getFacetDescription(), theme);
gernotbelger@9200:         final double[][] pointsUp = data.getTkhUpPoints();
gernotbelger@9200:         StyledSeriesBuilder.addPoints(seriesUp, pointsUp, true);
gernotbelger@9200: 
gernotbelger@9200:         // REMARK: we add " " because the description is misused as id, which must be unique.
gernotbelger@9200:         final StyledXYSeries seriesDown = new StyledXYSeries(bundle.getFacetDescription() + " ", theme);
gernotbelger@9200:         final double[][] pointsDown = data.getTkhDownPoints();
gernotbelger@9200:         StyledSeriesBuilder.addPoints(seriesDown, pointsDown, true);
gernotbelger@9200: 
gernotbelger@9200:         final StyledAreaSeriesCollection area = new StyledAreaSeriesCollection(theme);
gernotbelger@9200:         area.setMode(StyledAreaSeriesCollection.FILL_MODE.BETWEEN);
gernotbelger@9200:         area.addSeries(seriesUp);
gernotbelger@9200:         area.addSeries(seriesDown);
gernotbelger@9200: 
gernotbelger@9200:         generator.addAreaSeries(area, getAxisName(), visible);
gernotbelger@9200: 
gernotbelger@9200:         return null;
gernotbelger@9200:     }
gernotbelger@9200: 
gernotbelger@9200:     private Double findRadius(final CallContext context, final Artifact artifact) {
gernotbelger@9200:         final Double start = (Double) context.getContextValue("startkm");
gernotbelger@9200:         final Double end = (Double) context.getContextValue("endkm");
gernotbelger@9200: 
gernotbelger@9200:         if (start == null || end == null)
gernotbelger@9200:             return null;
gernotbelger@9200: 
gernotbelger@9200:         final RiverContext fc = (RiverContext) context.globalContext();
gernotbelger@9200:         final ZoomScale scales = (ZoomScale) fc.get("zoomscale");
gernotbelger@9200:         final RiverAccess access = new RiverAccess((D4EArtifact) artifact);
gernotbelger@9200:         final String river = access.getRiverName();
gernotbelger@9200: 
gernotbelger@9200:         return scales.getRadius(river, start, end);
gernotbelger@9200:     }
gernotbelger@9200: 
gernotbelger@9200:     private double[][] generatePoints(final CallContext context, final Artifact artifact, final AbstractCalculationResult data, final String facetName,
gernotbelger@9200:             final IResultType resultType) {
gernotbelger@9200: 
gernotbelger@9200:         final double[][] points = data.getStationPoints(resultType);
gernotbelger@9200:         if (facetName.endsWith(".filtered")) {
gernotbelger@9200:             final Double radius = findRadius(context, artifact);
gernotbelger@9200:             return movingAverage(radius, points);
gernotbelger@9200:         }
gernotbelger@9200: 
gernotbelger@9200:         return points;
gernotbelger@9200:     }
gernotbelger@9200: 
mschaefer@9204:     private double[][] generatePoints(final CallContext context, final Artifact artifact, final AbstractCalculationResult data, final String facetName,
mschaefer@9204:             final IResultType resultType, final int index) {
mschaefer@9204: 
mschaefer@9204:         final double[][] points = data.getStationPoints(resultType, index);
mschaefer@9204:         if (facetName.endsWith(".filtered")) {
mschaefer@9204:             final Double radius = findRadius(context, artifact);
mschaefer@9204:             return movingAverage(radius, points);
mschaefer@9204:         }
mschaefer@9204: 
mschaefer@9204:         return points;
mschaefer@9204:     }
mschaefer@9204: 
gernotbelger@9200:     private double[][] movingAverage(final Double radius, final double[][] points) {
gernotbelger@9200: 
gernotbelger@9200:         if (radius == null)
gernotbelger@9200:             return points;
gernotbelger@9200: 
gernotbelger@9200:         return MovingAverage.weighted(points, radius);
gernotbelger@9200:     }
gernotbelger@9200: 
gernotbelger@8884:     @Override
gernotbelger@8884:     public final boolean canHandle(final String facettype) {
gernotbelger@8884:         return this.handled_facet_types.contains(facettype);
gernotbelger@8884:     }
gernotbelger@8884: 
gernotbelger@8884:     @Override
gernotbelger@8884:     public final String getAxisLabel(final DiagramGenerator generator) {
gernotbelger@8884:         if (this.yAxisLabel != null && !this.yAxisLabel.isEmpty()) {
gernotbelger@8884:             // REMARK/UNINTENDED: yAxisLabel may also be a resolved message (side-effect of StyledXYSeries#putMetadata),
gernotbelger@8884:             // and cannot be resolved, so we need to give the resolved value as default
gernotbelger@9040:             // TODO: In other implementations (i.e. FlowVelocityProcessor), an explicit (German) default label is given here,
gernotbelger@8884:             // probably the English version will also show German (CHECK)
gernotbelger@8884:             return generator.msg(this.yAxisLabel, this.yAxisLabel);
gernotbelger@8884:         }
gernotbelger@8884:         return generator.msg(this.i18n_axis_label, "MISSING");
gernotbelger@8884:     }
gernotbelger@9203: 
gernotbelger@9203:     // Moved from SInfoLineProcessor:
gernotbelger@9203:     protected static final Facet createFacet(final CallContext context, final String hash, final String id, final AbstractCalculationResult result,
gernotbelger@9203:             final int index, final String axisLabel, final String facetId, final String description) {
gernotbelger@9203:         final String facetFlowDepthFilteredDescription = Resources.getMsg(context.getMeta(), description, description, result.getLabel());
gernotbelger@9203:         return new SInfoResultFacet(index, facetId, facetFlowDepthFilteredDescription, axisLabel, ComputeType.ADVANCE, id, hash);
gernotbelger@9203:     }
gernotbelger@9203: 
gernotbelger@8884: }