Mercurial > dive4elements > river
diff artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/common/FloodDurationProcessor.java @ 9205:3dae6b78e1da
inundationDuration/floodDuration multiple columns+chartLines refactoring
author | gernotbelger |
---|---|
date | Mon, 02 Jul 2018 19:01:09 +0200 |
parents | 9831f6c60473 |
children | 53cc5b496692 |
line wrap: on
line diff
--- a/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/common/FloodDurationProcessor.java Mon Jul 02 17:40:39 2018 +0200 +++ b/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/common/FloodDurationProcessor.java Mon Jul 02 19:01:09 2018 +0200 @@ -10,19 +10,27 @@ package org.dive4elements.river.artifacts.sinfo.common; +import java.util.Collection; import java.util.HashSet; +import java.util.List; import java.util.Set; +import java.util.function.Function; import org.dive4elements.artifactdatabase.state.ArtifactAndFacet; import org.dive4elements.artifactdatabase.state.Facet; import org.dive4elements.artifacts.CallContext; import org.dive4elements.river.artifacts.common.AbstractCalculationResult; +import org.dive4elements.river.artifacts.common.GeneralResultType; +import org.dive4elements.river.artifacts.common.ResultRow; import org.dive4elements.river.artifacts.resources.Resources; +import org.dive4elements.river.artifacts.sinfo.flood_duration.DurationWaterlevel; import org.dive4elements.river.artifacts.sinfo.flood_duration.FloodDurationCalculationResult; import org.dive4elements.river.artifacts.states.DefaultState.ComputeType; import org.dive4elements.river.exports.DiagramGenerator; import org.dive4elements.river.themes.ThemeDocument; +import gnu.trove.TDoubleArrayList; + /** * Processor to generate the facet and data series of infrastructure flood durations * @@ -31,17 +39,13 @@ */ public final class FloodDurationProcessor extends AbstractSInfoProcessor { - public static final String FACET_FLOOD_DURATION = "sinfo_facet_flood_duration"; - - public static final String FACET_FLOOD_DURATION_DESCRIPTION = "sinfo_facet_flood_duration.description"; + private static final String FACET_FLOOD_DURATION = "sinfo_facet_flood_duration"; - public static final String FACET_MAIN_VALUE_1_DURATION = "mainvalue.1.duration"; + private static final String FACET_FLOOD_DURATION_DESCRIPTION = "sinfo_facet_flood_duration.description"; - public static final String FACET_MAIN_VALUE_2_DURATION = "mainvalue.2.duration"; + private static final String FACET_MAIN_VALUE_DURATION = "mainvalue.duration"; - public static final String FACET_MAIN_VALUE_3_DURATION = "mainvalue.3.duration"; - - public static final String FACET_MAIN_VALUE_DURATION_DESCRIPTION = "mainvalue.duration.description"; + private static final String FACET_MAIN_VALUE_DURATION_DESCRIPTION = "mainvalue.duration.description"; private static final String I18N_AXIS_LABEL = "sinfo.chart.flood_duration.section.yaxis.label"; @@ -49,72 +53,83 @@ static { HANDLED_FACET_TYPES.add(FACET_FLOOD_DURATION); - HANDLED_FACET_TYPES.add(FACET_MAIN_VALUE_1_DURATION); - HANDLED_FACET_TYPES.add(FACET_MAIN_VALUE_2_DURATION); - HANDLED_FACET_TYPES.add(FACET_MAIN_VALUE_3_DURATION); + HANDLED_FACET_TYPES.add(FACET_MAIN_VALUE_DURATION); + } public FloodDurationProcessor() { super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES); } - // @Override TODO: REMOVE (has been replaced by doGetType - // protected double[][] doGetPoints(final AbstractCalculationResult data, final String facetName) { - // - // if (FACET_FLOOD_DURATION.contentEquals(facetName)) - // return ((FloodDurationCalculationResult) data).fetchInfrastructurePoints(SInfoResultType.floodDuration); - // - // if (FACET_MAIN_VALUE_1_DURATION.contentEquals(facetName)) - // return ((FloodDurationCalculationResult) data).fetchMainValuePoints(SInfoResultType.mainValue1Duration); - // - // if (FACET_MAIN_VALUE_2_DURATION.contentEquals(facetName)) - // return ((FloodDurationCalculationResult) data).fetchMainValuePoints(SInfoResultType.mainValue2Duration); - // - // if (FACET_MAIN_VALUE_3_DURATION.contentEquals(facetName)) - // return ((FloodDurationCalculationResult) data).fetchMainValuePoints(SInfoResultType.mainValue3Duration); - // - // final String error = String.format("Unknown facet name: %s", facetName); - // throw new UnsupportedOperationException(error); - // } - public static Facet createFloodDurationFacet(final CallContext context, final String hash, final String id, final AbstractCalculationResult result, - final int index) { - - return AbstractSInfoProcessor.createFacet(context, hash, id, result, index, I18N_AXIS_LABEL, FACET_FLOOD_DURATION, FACET_FLOOD_DURATION_DESCRIPTION); - } - - private SInfoResultType doGetType(final String facetName) { - - if (FACET_FLOOD_DURATION.contentEquals(facetName)) - return SInfoResultType.floodDuration; + final int resultIndex) { - if (FACET_MAIN_VALUE_1_DURATION.contentEquals(facetName)) - return SInfoResultType.mainValue1Duration; - - if (FACET_MAIN_VALUE_2_DURATION.contentEquals(facetName)) - return SInfoResultType.mainValue2Duration; - - if (FACET_MAIN_VALUE_3_DURATION.contentEquals(facetName)) - return SInfoResultType.mainValue3Duration; - - final String error = String.format("Unknown facet name: %s", facetName); - throw new UnsupportedOperationException(error); - + return AbstractSInfoProcessor.createFacet(context, hash, id, result, resultIndex, -1, I18N_AXIS_LABEL, FACET_FLOOD_DURATION, + FACET_FLOOD_DURATION_DESCRIPTION); } public static Facet createMainValueDurationFacet(final CallContext context, final String hash, final String id, final FloodDurationCalculationResult result, - final int index) { + final int facetIndex, final int resultIndex, final int dataIndex) { final String description = Resources.getMsg(context.getMeta(), FACET_MAIN_VALUE_DURATION_DESCRIPTION, FACET_MAIN_VALUE_DURATION_DESCRIPTION, - result.getMainValueLabel(index)); - assert ((index >= 0) && (index <= 2)); - final String facetName = new String[] { FACET_MAIN_VALUE_1_DURATION, FACET_MAIN_VALUE_2_DURATION, FACET_MAIN_VALUE_3_DURATION }[index]; - return new SInfoResultFacet(0, facetName, description, I18N_AXIS_LABEL, ComputeType.ADVANCE, id, hash); + result.getMainValueLabel(resultIndex)); + + return new SInfoResultFacet(facetIndex, resultIndex, dataIndex, FACET_MAIN_VALUE_DURATION, description, I18N_AXIS_LABEL, ComputeType.ADVANCE, id, hash); } @Override protected String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) { + final String facetName = bundle.getFacetName(); - return buildSeriesForType(generator, bundle, theme, visible, doGetType(bundle.getFacetName()), GAP_DISTANCE); + if (FACET_FLOOD_DURATION.contentEquals(facetName)) + return buildSeriesForType(generator, bundle, theme, visible, SInfoResultType.floodDuration, null); + + if (FACET_MAIN_VALUE_DURATION.contentEquals(facetName)) { + + final Function<DurationWaterlevel, Double> valueGetter = new Function<DurationWaterlevel, Double>() { + @Override + public Double apply(final DurationWaterlevel waterlevel) { + return (double) waterlevel.getFloodDurDaysPerYear(); + } + }; + + // FIXME: instead (see CollisionXXProcessor) delegate to result + final double[][] points = getMainValueDurationPoints(generator, bundle, valueGetter); + + return buildSeriesForType(points, generator, bundle, theme, visible, null); + } + + final String error = String.format("Unknown facet name: %s", facetName); + throw new UnsupportedOperationException(error); + } + + public static final double[][] getMainValueDurationPoints(final DiagramGenerator generator, final ArtifactAndFacet bundle, + final Function<DurationWaterlevel, Double> valueGetter) { + + final AbstractCalculationResult data = getResult(generator, bundle); + + final int index = getDataIndex(bundle); + + ((SInfoResultFacet) bundle.getFacet()).getDataIndex(); + + final Collection<ResultRow> rows = data.getRows(); + + final TDoubleArrayList xPoints = new TDoubleArrayList(rows.size()); + final TDoubleArrayList yPoints = new TDoubleArrayList(rows.size()); + + for (final ResultRow row : rows) { + + final double station = row.getDoubleValue(GeneralResultType.station); + + final List<DurationWaterlevel> waterlevels = (List<DurationWaterlevel>) row.getValue(SInfoResultType.customMultiRowColWaterlevel); + final DurationWaterlevel waterlevel = waterlevels.get(index); + + final Double value = valueGetter.apply(waterlevel); + + xPoints.add(station); + yPoints.add(value); + } + + return new double[][] { xPoints.toNativeArray(), yPoints.toNativeArray() }; } } \ No newline at end of file