Mercurial > dive4elements > river

annotate artifacts/src/main/java/org/dive4elements/river/exports/process/SedimentLoadProcessor.java @ 7153:fdf2d31b8955

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add SedimentLoadProcessor
author Andre Heinecke <aheinecke@intevation.de>
date Wed, 25 Sep 2013 17:40:43 +0200
parents
children e7690ed60b4c
rev   line source
7153
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
1 /* Copyright (C) 2013 by Bundesanstalt für Gewässerkunde
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
2 * Software engineering by Intevation GmbH
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
3 *
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
4 * This file is Free Software under the GNU AGPL (>=v3)
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
5 * and comes with ABSOLUTELY NO WARRANTY! Check out the
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
6 * documentation coming with Dive4Elements River for details.
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
7 */
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
8
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
9 package org.dive4elements.river.exports.process;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
10
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
11 import java.util.List;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
12 import java.util.Set;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
13
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
14 import org.apache.log4j.Logger;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
15 import org.jfree.data.xy.XYSeries;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
16 import org.dive4elements.river.artifacts.D4EArtifact;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
17
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
18 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
19 import org.dive4elements.artifacts.CallContext;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
20 import org.dive4elements.river.artifacts.access.SedimentLoadAccess;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
21 import org.dive4elements.river.artifacts.model.FacetTypes;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
22 import org.dive4elements.river.exports.DiagramGenerator;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
23 import org.dive4elements.river.exports.StyledSeriesBuilder;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
24 import org.dive4elements.river.jfree.StyledXYSeries;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
25 import org.dive4elements.river.themes.ThemeDocument;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
26
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
27 import org.dive4elements.river.artifacts.model.minfo.SedimentLoad;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
28 import org.dive4elements.river.artifacts.model.minfo.SedimentLoadFraction;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
29
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
30 public class SedimentLoadProcessor extends DefaultProcessor {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
31
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
32 private final static Logger logger =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
33 Logger.getLogger(SedimentLoadProcessor.class);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
34
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
35 public static final String I18N_YAXIS_LABEL_1 =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
36 "chart.sedimentload.ls.yaxis.label.tpera";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
37 public static final String I18N_YAXIS_LABEL_2 =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
38 "chart.sedimentload.ls.yaxis.label.m3pera";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
39 public static final String I18N_YAXIS_LABEL_DEFAULT_1 = "[t/a]";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
40 public static final String I18N_YAXIS_LABEL_DEFAULT_2 = "[m\u00b3/a]";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
41
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
42 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
43 public void doOut(
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
44 DiagramGenerator generator,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
45 ArtifactAndFacet bundle,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
46 ThemeDocument theme,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
47 boolean visible) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
48 CallContext context = generator.getCallContext();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
49 XYSeries series = new StyledXYSeries(bundle.getFacetDescription(),
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
50 theme);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
51 Object data = bundle.getData(context);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
52 String facetName = bundle.getFacetName();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
53 double [][] points;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
54
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
55 if (FacetTypes.IS.SEDIMENT_LOAD(facetName)) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
56 points = (double[][]) data;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
57 } else if (FacetTypes.IS.SEDIMENT_LOAD_UNKNOWN(facetName)) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
58 SedimentLoad load = (SedimentLoad) data;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
59 Set<Double> kms = load.getKms();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
60 points = new double[2][kms.size()];
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
61 int counter = 0;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
62 for (Double km: kms) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
63 SedimentLoadFraction fraction = load.getFraction(km);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
64 points[0][counter] = km;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
65 points[1][counter] = fraction.getUnknown();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
66 counter++;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
67 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
68 } else {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
69 logger.error("Unknown facet name: " + facetName);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
70 return;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
71 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
72 StyledSeriesBuilder.addPoints(series, points, true);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
73
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
74 generator.addAxisSeries(series, axisName, visible);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
75 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
76
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
77 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
78 public boolean canHandle(String facettype) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
79 return FacetTypes.IS.SEDIMENT_LOAD(facettype) ||
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
80 FacetTypes.IS.SEDIMENT_LOAD_UNKNOWN(facettype);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
81 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
82
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
83 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
84 public String getAxisLabel(DiagramGenerator generator) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
85 SedimentLoadAccess slaccess =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
86 new SedimentLoadAccess((D4EArtifact) generator.getMaster());
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
87 String unit = slaccess.getUnit();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
88 if (unit != null && unit.equals("m3_per_a")) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
89 return generator.msg(I18N_YAXIS_LABEL_2, I18N_YAXIS_LABEL_DEFAULT_2);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
90 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
91 else {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
92 return generator.msg(I18N_YAXIS_LABEL_1, I18N_YAXIS_LABEL_DEFAULT_1);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
93 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
94 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
95 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
96
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
97
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
98
http://dive4elements.wald.intevation.org