mschaefer@9176: /** Copyright (C) 2017 by Bundesanstalt für Gewässerkunde
mschaefer@9176:  * Software engineering by
mschaefer@9176:  *  Björnsen Beratende Ingenieure GmbH
mschaefer@9176:  *  Dr. Schumacher Ingenieurbüro für Wasser und Umwelt
mschaefer@9176:  *
mschaefer@9176:  * This file is Free Software under the GNU AGPL (>=v3)
mschaefer@9176:  * and comes with ABSOLUTELY NO WARRANTY! Check out the
mschaefer@9176:  * documentation coming with Dive4Elements River for details.
mschaefer@9176:  */
mschaefer@9176: 
mschaefer@9176: package org.dive4elements.river.artifacts.sinfo.common;
mschaefer@9176: 
gernotbelger@9205: import java.util.Collection;
mschaefer@9176: import java.util.HashSet;
gernotbelger@9205: import java.util.List;
mschaefer@9176: import java.util.Set;
gernotbelger@9205: import java.util.function.Function;
mschaefer@9176: 
gernotbelger@9203: import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
mschaefer@9176: import org.dive4elements.artifactdatabase.state.Facet;
mschaefer@9176: import org.dive4elements.artifacts.CallContext;
mschaefer@9176: import org.dive4elements.river.artifacts.common.AbstractCalculationResult;
gernotbelger@9205: import org.dive4elements.river.artifacts.common.GeneralResultType;
gernotbelger@9205: import org.dive4elements.river.artifacts.common.ResultRow;
mschaefer@9202: import org.dive4elements.river.artifacts.resources.Resources;
gernotbelger@9205: import org.dive4elements.river.artifacts.sinfo.flood_duration.DurationWaterlevel;
mschaefer@9202: import org.dive4elements.river.artifacts.sinfo.flood_duration.FloodDurationCalculationResult;
mschaefer@9202: import org.dive4elements.river.artifacts.states.DefaultState.ComputeType;
gernotbelger@9203: import org.dive4elements.river.exports.DiagramGenerator;
gernotbelger@9203: import org.dive4elements.river.themes.ThemeDocument;
mschaefer@9176: 
gernotbelger@9205: import gnu.trove.TDoubleArrayList;
gernotbelger@9205: 
mschaefer@9176: /**
mschaefer@9176:  * Processor to generate the facet and data series of infrastructure flood durations
mschaefer@9176:  *
mschaefer@9176:  * @author Matthias Schäfer
mschaefer@9176:  *
mschaefer@9176:  */
gernotbelger@9203: public final class FloodDurationProcessor extends AbstractSInfoProcessor {
mschaefer@9176: 
gernotbelger@9205:     private static final String FACET_FLOOD_DURATION = "sinfo_facet_flood_duration";
mschaefer@9176: 
gernotbelger@9205:     private static final String FACET_FLOOD_DURATION_DESCRIPTION = "sinfo_facet_flood_duration.description";
mschaefer@9176: 
gernotbelger@9205:     private static final String FACET_MAIN_VALUE_DURATION = "mainvalue.duration";
mschaefer@9202: 
gernotbelger@9205:     private static final String FACET_MAIN_VALUE_DURATION_DESCRIPTION = "mainvalue.duration.description";
mschaefer@9176: 
mschaefer@9176:     private static final String I18N_AXIS_LABEL = "sinfo.chart.flood_duration.section.yaxis.label";
mschaefer@9176: 
mschaefer@9176:     private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>();
mschaefer@9176: 
mschaefer@9176:     static {
mschaefer@9176:         HANDLED_FACET_TYPES.add(FACET_FLOOD_DURATION);
gernotbelger@9205:         HANDLED_FACET_TYPES.add(FACET_MAIN_VALUE_DURATION);
gernotbelger@9205: 
mschaefer@9176:     }
mschaefer@9176: 
mschaefer@9176:     public FloodDurationProcessor() {
mschaefer@9176:         super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES);
mschaefer@9176:     }
mschaefer@9176: 
gernotbelger@9203:     public static Facet createFloodDurationFacet(final CallContext context, final String hash, final String id, final AbstractCalculationResult result,
gernotbelger@9205:             final int resultIndex) {
mschaefer@9176: 
gernotbelger@9205:         return AbstractSInfoProcessor.createFacet(context, hash, id, result, resultIndex, -1, I18N_AXIS_LABEL, FACET_FLOOD_DURATION,
gernotbelger@9205:                 FACET_FLOOD_DURATION_DESCRIPTION);
mschaefer@9176:     }
mschaefer@9176: 
gernotbelger@9203:     public static Facet createMainValueDurationFacet(final CallContext context, final String hash, final String id, final FloodDurationCalculationResult result,
gernotbelger@9205:             final int facetIndex, final int resultIndex, final int dataIndex) {
mschaefer@9202: 
mschaefer@9202:         final String description = Resources.getMsg(context.getMeta(), FACET_MAIN_VALUE_DURATION_DESCRIPTION, FACET_MAIN_VALUE_DURATION_DESCRIPTION,
gernotbelger@9205:                 result.getMainValueLabel(resultIndex));
gernotbelger@9205: 
gernotbelger@9205:         return new SInfoResultFacet(facetIndex, resultIndex, dataIndex, FACET_MAIN_VALUE_DURATION, description, I18N_AXIS_LABEL, ComputeType.ADVANCE, id, hash);
mschaefer@9176:     }
gernotbelger@9203: 
gernotbelger@9203:     @Override
gernotbelger@9203:     protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
gernotbelger@9205:         final String facetName = bundle.getFacetName();
gernotbelger@9203: 
gernotbelger@9205:         if (FACET_FLOOD_DURATION.contentEquals(facetName))
gernotbelger@9205:             return buildSeriesForType(generator, bundle, theme, visible, SInfoResultType.floodDuration, null);
gernotbelger@9205: 
gernotbelger@9205:         if (FACET_MAIN_VALUE_DURATION.contentEquals(facetName)) {
gernotbelger@9205: 
gernotbelger@9205:             final Function<DurationWaterlevel, Double> valueGetter = new Function<DurationWaterlevel, Double>() {
gernotbelger@9205:                 @Override
gernotbelger@9205:                 public Double apply(final DurationWaterlevel waterlevel) {
gernotbelger@9205:                     return (double) waterlevel.getFloodDurDaysPerYear();
gernotbelger@9205:                 }
gernotbelger@9205:             };
gernotbelger@9205: 
gernotbelger@9205:             // FIXME: instead (see CollisionXXProcessor) delegate to result
gernotbelger@9205:             final double[][] points = getMainValueDurationPoints(generator, bundle, valueGetter);
gernotbelger@9205: 
gernotbelger@9205:             return buildSeriesForType(points, generator, bundle, theme, visible, null);
gernotbelger@9205:         }
gernotbelger@9205: 
gernotbelger@9205:         final String error = String.format("Unknown facet name: %s", facetName);
gernotbelger@9205:         throw new UnsupportedOperationException(error);
gernotbelger@9205:     }
gernotbelger@9205: 
gernotbelger@9205:     public static final double[][] getMainValueDurationPoints(final DiagramGenerator generator, final ArtifactAndFacet bundle,
gernotbelger@9205:             final Function<DurationWaterlevel, Double> valueGetter) {
gernotbelger@9205: 
gernotbelger@9205:         final AbstractCalculationResult data = getResult(generator, bundle);
gernotbelger@9205: 
gernotbelger@9205:         final int index = getDataIndex(bundle);
gernotbelger@9205: 
gernotbelger@9205:         ((SInfoResultFacet) bundle.getFacet()).getDataIndex();
gernotbelger@9205: 
gernotbelger@9205:         final Collection<ResultRow> rows = data.getRows();
gernotbelger@9205: 
gernotbelger@9205:         final TDoubleArrayList xPoints = new TDoubleArrayList(rows.size());
gernotbelger@9205:         final TDoubleArrayList yPoints = new TDoubleArrayList(rows.size());
gernotbelger@9205: 
gernotbelger@9205:         for (final ResultRow row : rows) {
gernotbelger@9205: 
gernotbelger@9205:             final double station = row.getDoubleValue(GeneralResultType.station);
gernotbelger@9205: 
gernotbelger@9205:             final List<DurationWaterlevel> waterlevels = (List<DurationWaterlevel>) row.getValue(SInfoResultType.customMultiRowColWaterlevel);
gernotbelger@9205:             final DurationWaterlevel waterlevel = waterlevels.get(index);
gernotbelger@9205: 
gernotbelger@9205:             final Double value = valueGetter.apply(waterlevel);
gernotbelger@9205: 
gernotbelger@9205:             xPoints.add(station);
gernotbelger@9205:             yPoints.add(value);
gernotbelger@9205:         }
gernotbelger@9205: 
gernotbelger@9205:         return new double[][] { xPoints.toNativeArray(), yPoints.toNativeArray() };
gernotbelger@9203:     }
mschaefer@9176: }