Mercurial > dive4elements > river
annotate artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flowdepth/FlowDepthProcessor.java @ 8916:5d5d0051723f
Working on outputmodes of tkh calculation
| author | gernotbelger |
|---|---|
| date | Wed, 28 Feb 2018 18:55:39 +0100 |
| parents | 90b7f45ff4ae |
| children |
| rev | line source |
|---|---|
| 8854 | 1 /* Copyright (C) 2017 by Bundesanstalt für Gewässerkunde |
| 8884 | 2 * Software engineering by |
| 3 * Björnsen Beratende Ingenieure GmbH | |
| 8854 | 4 * Dr. Schumacher Ingenieurbüro für Wasser und Umwelt |
| 5 * | |
| 6 * This file is Free Software under the GNU AGPL (>=v3) | |
| 7 * and comes with ABSOLUTELY NO WARRANTY! Check out the | |
| 8 * documentation coming with Dive4Elements River for details. | |
| 9 */ | |
| 10 | |
| 11 package org.dive4elements.river.artifacts.sinfo.flowdepth; | |
| 12 | |
| 8884 | 13 import java.util.HashSet; |
| 8886 | 14 import java.util.Map; |
| 8884 | 15 import java.util.Set; |
| 8854 | 16 |
| 8886 | 17 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet; |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
18 import org.dive4elements.artifacts.Artifact; |
| 8886 | 19 import org.dive4elements.artifacts.CallContext; |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
20 import org.dive4elements.river.artifacts.D4EArtifact; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
21 import org.dive4elements.river.artifacts.access.RiverAccess; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
22 import org.dive4elements.river.artifacts.context.RiverContext; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
23 import org.dive4elements.river.artifacts.math.MovingAverage; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
24 import org.dive4elements.river.artifacts.model.ZoomScale; |
| 8916 | 25 import org.dive4elements.river.artifacts.sinfo.common.AbstractSInfoProcessor; |
| 8886 | 26 import org.dive4elements.river.exports.DiagramGenerator; |
| 27 import org.dive4elements.river.exports.StyledSeriesBuilder; | |
| 28 import org.dive4elements.river.jfree.StyledXYSeries; | |
| 29 import org.dive4elements.river.themes.ThemeDocument; | |
| 30 | |
| 8884 | 31 public final class FlowDepthProcessor extends AbstractSInfoProcessor { |
| 8854 | 32 |
| 8884 | 33 /* Theme name, usually defined in 'FacetTypes', but that is soooo bad dependencies... */ |
| 34 // REMARK: these mustend with 'filtered' so extra handling happens in chart: point are always recalculated, because data | |
| 35 // changes depending on zoom state | |
| 8854 | 36 static String FACET_FLOW_DEPTH_FILTERED = "sinfo_flow_depth.filtered"; |
| 37 | |
| 8884 | 38 static String FACET_FLOW_DEPTH_TKH_FILTERED = "sinfo_flow_depth.tkh.filtered"; |
| 39 | |
| 40 private static final Set<String> HANDLED_FACET_TYPES = new HashSet<>(); | |
| 41 | |
| 42 static { | |
| 43 HANDLED_FACET_TYPES.add(FACET_FLOW_DEPTH_FILTERED); | |
| 44 HANDLED_FACET_TYPES.add(FACET_FLOW_DEPTH_TKH_FILTERED); | |
| 45 } | |
| 8854 | 46 |
| 47 private static final String I18N_AXIS_LABEL = "sinfo.chart.flow_depth.section.yaxis.label"; | |
| 48 | |
| 8884 | 49 public FlowDepthProcessor() { |
| 50 super(I18N_AXIS_LABEL, HANDLED_FACET_TYPES); | |
| 8854 | 51 } |
| 52 | |
| 8886 | 53 @Override |
| 54 protected final String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) { | |
| 55 | |
| 56 final CallContext context = generator.getCallContext(); | |
| 57 final Map<String, String> metaData = bundle.getFacet().getMetaData(); | |
| 58 | |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
59 final Artifact artifact = bundle.getArtifact(); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
60 |
| 8886 | 61 final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme); |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
62 series.putMetaData(metaData, artifact, context); |
| 8886 | 63 |
| 64 final String facetName = bundle.getFacetName(); | |
| 65 final FlowDepthCalculationResult data = (FlowDepthCalculationResult) bundle.getData(context); | |
| 66 if (data == null) { | |
| 67 // Check has been here before so we keep it for security reasons | |
| 68 // this should never happen though. | |
| 69 throw new IllegalStateException("Data is null for facet: " + facetName); | |
| 70 } | |
| 71 | |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
72 final Double radius = findRadius(context, artifact); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
73 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
74 final double[][] points = generatePoints(radius, data, facetName); |
| 8886 | 75 |
| 76 StyledSeriesBuilder.addPoints(series, points, true); | |
| 77 generator.addAxisSeries(series, getAxisName(), visible); | |
| 78 | |
| 79 return metaData.get("Y"); | |
| 80 } | |
| 81 | |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
82 private Double findRadius(final CallContext context, final Artifact artifact) { |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
83 final Double start = (Double) context.getContextValue("startkm"); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
84 final Double end = (Double) context.getContextValue("endkm"); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
85 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
86 if (start == null || end == null) |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
87 return null; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
88 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
89 final RiverContext fc = (RiverContext) context.globalContext(); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
90 final ZoomScale scales = (ZoomScale) fc.get("zoomscale"); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
91 final RiverAccess access = new RiverAccess((D4EArtifact) artifact); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
92 final String river = access.getRiverName(); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
93 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
94 return scales.getRadius(river, start, end); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
95 } |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
96 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
97 private double[][] generatePoints(final Double radius, final FlowDepthCalculationResult data, final String facetName) { |
| 8854 | 98 |
| 8884 | 99 if (FACET_FLOW_DEPTH_FILTERED.contentEquals(facetName)) |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
100 return movingAverage(radius, data.getFlowDepthPoints()); |
| 8854 | 101 |
| 8884 | 102 if (FACET_FLOW_DEPTH_TKH_FILTERED.contentEquals(facetName)) |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
103 return movingAverage(radius, data.getFlowDepthTkhPoints()); |
| 8854 | 104 |
| 8886 | 105 final String error = String.format("Unknown facet name: %s", facetName); |
| 106 throw new UnsupportedOperationException(error); | |
| 8854 | 107 } |
|
8890
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
108 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
109 private double[][] movingAverage(final Double radius, final double[][] points) { |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
110 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
111 if (radius == null) |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
112 return points; |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
113 |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
114 return MovingAverage.weighted(points, radius); |
|
90b7f45ff4ae
Calculating moving average of flow depth output
gernotbelger
parents:
8886
diff
changeset
|
115 } |
| 8854 | 116 } |