Mercurial > dive4elements > river

comparison artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flowdepth/FlowDepthProcessor.java @ 8890:90b7f45ff4ae

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Calculating moving average of flow depth output
author gernotbelger
date Wed, 14 Feb 2018 18:45:42 +0100
parents cc86b0f9b3c3
children 5d5d0051723f
comparison
equal deleted inserted replaced
8889:f87f435df856 8890:90b7f45ff4ae
13 import java.util.HashSet; 13 import java.util.HashSet;
14 import java.util.Map; 14 import java.util.Map;
15 import java.util.Set; 15 import java.util.Set;
16 16
17 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet; 17 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
18 import org.dive4elements.artifacts.Artifact;
18 import org.dive4elements.artifacts.CallContext; 19 import org.dive4elements.artifacts.CallContext;
20 import org.dive4elements.river.artifacts.D4EArtifact;
21 import org.dive4elements.river.artifacts.access.RiverAccess;
22 import org.dive4elements.river.artifacts.context.RiverContext;
23 import org.dive4elements.river.artifacts.math.MovingAverage;
24 import org.dive4elements.river.artifacts.model.ZoomScale;
19 import org.dive4elements.river.exports.DiagramGenerator; 25 import org.dive4elements.river.exports.DiagramGenerator;
20 import org.dive4elements.river.exports.StyledSeriesBuilder; 26 import org.dive4elements.river.exports.StyledSeriesBuilder;
21 import org.dive4elements.river.jfree.StyledXYSeries; 27 import org.dive4elements.river.jfree.StyledXYSeries;
22 import org.dive4elements.river.themes.ThemeDocument; 28 import org.dive4elements.river.themes.ThemeDocument;
23 29
47 protected final String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) { 53 protected final String generateSeries(final DiagramGenerator generator, final ArtifactAndFacet bundle, final ThemeDocument theme, final boolean visible) {
48 54
49 final CallContext context = generator.getCallContext(); 55 final CallContext context = generator.getCallContext();
50 final Map<String, String> metaData = bundle.getFacet().getMetaData(); 56 final Map<String, String> metaData = bundle.getFacet().getMetaData();
51 57
58 final Artifact artifact = bundle.getArtifact();
59
52 final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme); 60 final StyledXYSeries series = new StyledXYSeries(bundle.getFacetDescription(), theme);
53 series.putMetaData(metaData, bundle.getArtifact(), context); 61 series.putMetaData(metaData, artifact, context);
54 62
55 final String facetName = bundle.getFacetName(); 63 final String facetName = bundle.getFacetName();
56 final FlowDepthCalculationResult data = (FlowDepthCalculationResult) bundle.getData(context); 64 final FlowDepthCalculationResult data = (FlowDepthCalculationResult) bundle.getData(context);
57 if (data == null) { 65 if (data == null) {
58 // Check has been here before so we keep it for security reasons 66 // Check has been here before so we keep it for security reasons
59 // this should never happen though. 67 // this should never happen though.
60 throw new IllegalStateException("Data is null for facet: " + facetName); 68 throw new IllegalStateException("Data is null for facet: " + facetName);
61 } 69 }
62 70
63 final double[][] points = generatePoints(data, facetName); 71 final Double radius = findRadius(context, artifact);
72
73 final double[][] points = generatePoints(radius, data, facetName);
64 74
65 StyledSeriesBuilder.addPoints(series, points, true); 75 StyledSeriesBuilder.addPoints(series, points, true);
66 generator.addAxisSeries(series, getAxisName(), visible); 76 generator.addAxisSeries(series, getAxisName(), visible);
67 77
68 return metaData.get("Y"); 78 return metaData.get("Y");
69 } 79 }
70 80
71 // FIXME: do filtering 81 private Double findRadius(final CallContext context, final Artifact artifact) {
72 private double[][] generatePoints(final FlowDepthCalculationResult data, final String facetName) { 82 final Double start = (Double) context.getContextValue("startkm");
83 final Double end = (Double) context.getContextValue("endkm");
84
85 if (start == null || end == null)
86 return null;
87
88 final RiverContext fc = (RiverContext) context.globalContext();
89 final ZoomScale scales = (ZoomScale) fc.get("zoomscale");
90 final RiverAccess access = new RiverAccess((D4EArtifact) artifact);
91 final String river = access.getRiverName();
92
93 return scales.getRadius(river, start, end);
94 }
95
96 private double[][] generatePoints(final Double radius, final FlowDepthCalculationResult data, final String facetName) {
73 97
74 if (FACET_FLOW_DEPTH_FILTERED.contentEquals(facetName)) 98 if (FACET_FLOW_DEPTH_FILTERED.contentEquals(facetName))
75 return data.getFlowDepthPoints(); 99 return movingAverage(radius, data.getFlowDepthPoints());
76 100
77 if (FACET_FLOW_DEPTH_TKH_FILTERED.contentEquals(facetName)) 101 if (FACET_FLOW_DEPTH_TKH_FILTERED.contentEquals(facetName))
78 return data.getFlowDepthTkhPoints(); 102 return movingAverage(radius, data.getFlowDepthTkhPoints());
79 103
80 final String error = String.format("Unknown facet name: %s", facetName); 104 final String error = String.format("Unknown facet name: %s", facetName);
81 throw new UnsupportedOperationException(error); 105 throw new UnsupportedOperationException(error);
82 } 106 }
107
108 private double[][] movingAverage(final Double radius, final double[][] points) {
109
110 if (radius == null)
111 return points;
112
113 return MovingAverage.weighted(points, radius);
114 }
83 } 115 }
http://dive4elements.wald.intevation.org