Mercurial > dive4elements > river

annotate artifacts/src/main/java/org/dive4elements/river/exports/process/SedimentLoadProcessor.java @ 7159:1508ee33f85f

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Removed obsolete imports.
author Sascha L. Teichmann <teichmann@intevation.de>
date Thu, 26 Sep 2013 11:41:02 +0200
parents e7690ed60b4c
children 4c8ce0c41dc2
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.Set;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
12
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
13 import org.apache.log4j.Logger;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
14 import org.jfree.data.xy.XYSeries;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
15 import org.dive4elements.river.artifacts.D4EArtifact;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
16
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
17 import org.dive4elements.artifactdatabase.state.ArtifactAndFacet;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
18 import org.dive4elements.artifacts.CallContext;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
19 import org.dive4elements.river.artifacts.access.SedimentLoadAccess;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
20 import org.dive4elements.river.artifacts.model.FacetTypes;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
21 import org.dive4elements.river.exports.DiagramGenerator;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
22 import org.dive4elements.river.exports.StyledSeriesBuilder;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
23 import org.dive4elements.river.jfree.StyledXYSeries;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
24 import org.dive4elements.river.themes.ThemeDocument;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
25
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
26 import org.dive4elements.river.artifacts.model.minfo.SedimentLoad;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
27 import org.dive4elements.river.artifacts.model.minfo.SedimentLoadFraction;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
28
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
29 public class SedimentLoadProcessor extends DefaultProcessor {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
30
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
31 private final static Logger logger =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
32 Logger.getLogger(SedimentLoadProcessor.class);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
33
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
34 public static final String I18N_YAXIS_LABEL_1 =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
35 "chart.sedimentload.ls.yaxis.label.tpera";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
36 public static final String I18N_YAXIS_LABEL_2 =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
37 "chart.sedimentload.ls.yaxis.label.m3pera";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
38 public static final String I18N_YAXIS_LABEL_DEFAULT_1 = "[t/a]";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
39 public static final String I18N_YAXIS_LABEL_DEFAULT_2 = "[m\u00b3/a]";
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
40
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
41 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
42 public void doOut(
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
43 DiagramGenerator generator,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
44 ArtifactAndFacet bundle,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
45 ThemeDocument theme,
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
46 boolean visible) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
47 CallContext context = generator.getCallContext();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
48 XYSeries series = new StyledXYSeries(bundle.getFacetDescription(),
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
49 theme);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
50 Object data = bundle.getData(context);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
51 String facetName = bundle.getFacetName();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
52 double [][] points;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
53
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
54 if (FacetTypes.IS.SEDIMENT_LOAD(facetName)) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
55 points = (double[][]) data;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
56 } else if (FacetTypes.IS.SEDIMENT_LOAD_UNKNOWN(facetName)) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
57 SedimentLoad load = (SedimentLoad) data;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
58 Set<Double> kms = load.getKms();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
59 points = new double[2][kms.size()];
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
60 int counter = 0;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
61 for (Double km: kms) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
62 SedimentLoadFraction fraction = load.getFraction(km);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
63 points[0][counter] = km;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
64 points[1][counter] = fraction.getUnknown();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
65 counter++;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
66 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
67 } else {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
68 logger.error("Unknown facet name: " + facetName);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
69 return;
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
70 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
71 StyledSeriesBuilder.addPoints(series, points, true);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
72
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
73 generator.addAxisSeries(series, axisName, visible);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
74 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
75
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
76 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
77 public boolean canHandle(String facettype) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
78 return FacetTypes.IS.SEDIMENT_LOAD(facettype) ||
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
79 FacetTypes.IS.SEDIMENT_LOAD_UNKNOWN(facettype);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
80 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
81
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
82 @Override
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
83 public String getAxisLabel(DiagramGenerator generator) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
84 SedimentLoadAccess slaccess =
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
85 new SedimentLoadAccess((D4EArtifact) generator.getMaster());
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
86 String unit = slaccess.getUnit();
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
87 if (unit != null && unit.equals("m3_per_a")) {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
88 return generator.msg(I18N_YAXIS_LABEL_2, I18N_YAXIS_LABEL_DEFAULT_2);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
89 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
90 else {
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
91 return generator.msg(I18N_YAXIS_LABEL_1, I18N_YAXIS_LABEL_DEFAULT_1);
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
92 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
93 }
fdf2d31b8955 Add SedimentLoadProcessor
Andre Heinecke <aheinecke@intevation.de>
parents:
diff changeset
94 }
http://dive4elements.wald.intevation.org