Mercurial > dive4elements > river
comparison flys-artifacts/src/main/java/org/dive4elements/river/exports/DurationCurveExporter.java @ 5831:bd047b71ab37
Repaired internal references
author | Sascha L. Teichmann <teichmann@intevation.de> |
---|---|
date | Thu, 25 Apr 2013 12:06:39 +0200 |
parents | flys-artifacts/src/main/java/de/intevation/flys/exports/DurationCurveExporter.java@56fbdcdb7ff2 |
children |
comparison
equal
deleted
inserted
replaced
5830:160f53ee0870 | 5831:bd047b71ab37 |
---|---|
1 package org.dive4elements.river.exports; | |
2 | |
3 import java.io.OutputStream; | |
4 import java.text.NumberFormat; | |
5 import java.util.ArrayList; | |
6 import java.util.List; | |
7 import java.util.Map; | |
8 import java.util.HashMap; | |
9 import java.util.Date; | |
10 import java.text.DateFormat; | |
11 import java.util.Locale; | |
12 | |
13 import org.w3c.dom.Document; | |
14 | |
15 import org.apache.log4j.Logger; | |
16 | |
17 import au.com.bytecode.opencsv.CSVWriter; | |
18 | |
19 import net.sf.jasperreports.engine.JasperExportManager; | |
20 import net.sf.jasperreports.engine.JasperFillManager; | |
21 import net.sf.jasperreports.engine.JasperPrint; | |
22 import net.sf.jasperreports.engine.JRException; | |
23 | |
24 import org.dive4elements.artifacts.CallContext; | |
25 import org.dive4elements.artifacts.CallMeta; | |
26 import org.dive4elements.artifacts.common.utils.Config; | |
27 | |
28 import org.dive4elements.river.artifacts.FLYSArtifact; | |
29 | |
30 import org.dive4elements.river.artifacts.access.RangeAccess; | |
31 import org.dive4elements.river.artifacts.model.WQDay; | |
32 import org.dive4elements.river.artifacts.model.CalculationResult; | |
33 import org.dive4elements.river.artifacts.model.WKmsJRDataSource; | |
34 import org.dive4elements.river.artifacts.resources.Resources; | |
35 | |
36 import org.dive4elements.river.utils.FLYSUtils; | |
37 import org.dive4elements.river.utils.Formatter; | |
38 | |
39 | |
40 /** | |
41 * @author <a href="mailto:ingo.weinzierl@intevation.de">Ingo Weinzierl</a> | |
42 */ | |
43 public class DurationCurveExporter extends AbstractExporter { | |
44 | |
45 /** The logger used in this exporter. */ | |
46 private static Logger logger = Logger.getLogger(DurationCurveExporter.class); | |
47 | |
48 | |
49 public static final String CSV_DURATION_HEADER = | |
50 "export.duration.curve.csv.header.duration"; | |
51 | |
52 public static final String CSV_W_HEADER = | |
53 "export.duration.curve.csv.header.w"; | |
54 | |
55 public static final String CSV_Q_HEADER = | |
56 "export.duration.curve.csv.header.q"; | |
57 | |
58 public static final String DEFAULT_CSV_DURATION_HEADER = "D [Tagen]"; | |
59 public static final String DEFAULT_CSV_W_HEADER = "W [NN + m]"; | |
60 public static final String DEFAULT_CSV_Q_HEADER = "Q [m\u00b3/s]"; | |
61 | |
62 public static final String PDF_HEADER_MODE = "export.duration.pdf.mode"; | |
63 public static final String JASPER_FILE = "export.duration.pdf.file"; | |
64 | |
65 /** The storage that contains all WQKms objects for the different facets. */ | |
66 protected List<WQDay> data; | |
67 | |
68 | |
69 public void init(Document request, OutputStream out, CallContext context) { | |
70 logger.debug("DurationCurveExporter.init"); | |
71 | |
72 super.init(request, out, context); | |
73 | |
74 this.data = new ArrayList<WQDay>(); | |
75 } | |
76 | |
77 | |
78 @Override | |
79 protected void addData(Object d) { | |
80 if (d instanceof CalculationResult) { | |
81 d = ((CalculationResult)d).getData(); | |
82 if (d instanceof WQDay) { | |
83 data.add((WQDay)d); | |
84 } | |
85 } | |
86 } | |
87 | |
88 | |
89 protected void writeCSVData(CSVWriter writer) { | |
90 logger.info("DurationCurveExporter.writeData"); | |
91 | |
92 writeCSVHeader(writer); | |
93 | |
94 for (WQDay wqday: data) { | |
95 wQDay2CSV(writer, wqday); | |
96 } | |
97 } | |
98 | |
99 | |
100 protected void writeCSVHeader(CSVWriter writer) { | |
101 logger.info("DurationCurveExporter.writeCSVHeader"); | |
102 | |
103 writer.writeNext(new String[] { | |
104 msg(CSV_W_HEADER, DEFAULT_CSV_W_HEADER), | |
105 msg(CSV_Q_HEADER, DEFAULT_CSV_Q_HEADER), | |
106 msg(CSV_DURATION_HEADER, DEFAULT_CSV_DURATION_HEADER) | |
107 }); | |
108 } | |
109 | |
110 | |
111 protected void wQDay2CSV(CSVWriter writer, WQDay wqday) { | |
112 logger.debug("DurationCurveExporter.wQDay2CSV"); | |
113 | |
114 int size = wqday.size(); | |
115 | |
116 NumberFormat wf = getWFormatter(); | |
117 NumberFormat qf = getQFormatter(); | |
118 NumberFormat df = getDFormatter(); | |
119 | |
120 if (wqday.isIncreasing()) { | |
121 for (int i = size-1; i >= 0; i --) { | |
122 writer.writeNext(new String[] { | |
123 wf.format(wqday.getW(i)), | |
124 qf.format(wqday.getQ(i)), | |
125 df.format(wqday.getDay(i)) | |
126 }); | |
127 } | |
128 } | |
129 else { | |
130 for (int i = 0; i < size; i ++) { | |
131 writer.writeNext(new String[] { | |
132 wf.format(wqday.getW(i)), | |
133 qf.format(wqday.getQ(i)), | |
134 df.format(wqday.getDay(i)) | |
135 }); | |
136 } | |
137 } | |
138 } | |
139 | |
140 | |
141 /** | |
142 * Returns the number formatter for W values. | |
143 * | |
144 * @return the number formatter for W values. | |
145 */ | |
146 @Override | |
147 protected NumberFormat getWFormatter() { | |
148 return Formatter.getDurationW(context); | |
149 } | |
150 | |
151 | |
152 /** | |
153 * Returns the number formatter for Q values. | |
154 * | |
155 * @return the number formatter for Q values. | |
156 */ | |
157 @Override | |
158 protected NumberFormat getQFormatter() { | |
159 return Formatter.getDurationQ(context); | |
160 } | |
161 | |
162 | |
163 /** | |
164 * Returns the number formatter for duration values. | |
165 * | |
166 * @return the number formatter for duration values. | |
167 */ | |
168 protected NumberFormat getDFormatter() { | |
169 return Formatter.getDurationD(context); | |
170 } | |
171 | |
172 | |
173 @Override | |
174 protected void writePDF(OutputStream out) { | |
175 WKmsJRDataSource source = createJRData(); | |
176 | |
177 String jasperFile = Resources.getMsg( | |
178 context.getMeta(), | |
179 JASPER_FILE, | |
180 "/jasper/duration_en.jasper"); | |
181 String confPath = Config.getConfigDirectory().toString(); | |
182 | |
183 Map parameters = new HashMap(); | |
184 parameters.put("ReportTitle", "Exported Data"); | |
185 try { | |
186 JasperPrint print = JasperFillManager.fillReport( | |
187 confPath + jasperFile, | |
188 parameters, | |
189 source); | |
190 JasperExportManager.exportReportToPdfStream(print, out); | |
191 } | |
192 catch(JRException je) { | |
193 logger.warn("Error generating PDF Report!"); | |
194 je.printStackTrace(); | |
195 } | |
196 } | |
197 | |
198 protected WKmsJRDataSource createJRData() { | |
199 WKmsJRDataSource source = new WKmsJRDataSource(); | |
200 | |
201 addMetaData(source); | |
202 for (WQDay wqday: data) { | |
203 addWQDayData(source, wqday); | |
204 } | |
205 | |
206 return source; | |
207 } | |
208 | |
209 | |
210 protected void addMetaData(WKmsJRDataSource source) { | |
211 CallMeta meta = context.getMeta(); | |
212 | |
213 FLYSArtifact flys = (FLYSArtifact) master; | |
214 | |
215 source.addMetaData ("river", FLYSUtils.getRivername(flys)); | |
216 | |
217 Locale locale = Resources.getLocale(meta); | |
218 DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, locale); | |
219 | |
220 source.addMetaData("date", df.format(new Date())); | |
221 | |
222 RangeAccess rangeAccess = new RangeAccess(flys, null); | |
223 double[] kms = rangeAccess.getKmRange(); | |
224 source.addMetaData("range", String.valueOf(kms[0])); | |
225 | |
226 source.addMetaData("calculation", Resources.getMsg( | |
227 locale, | |
228 PDF_HEADER_MODE, | |
229 "Duration")); | |
230 } | |
231 | |
232 protected void addWQDayData(WKmsJRDataSource source, WQDay wqday) { | |
233 int size = wqday.size(); | |
234 | |
235 NumberFormat wf = getWFormatter(); | |
236 NumberFormat qf = getQFormatter(); | |
237 NumberFormat df = getDFormatter(); | |
238 | |
239 if (wqday.isIncreasing()) { | |
240 for (int i = size-1; i >= 0; i --) { | |
241 source.addData(new String[] { | |
242 "", | |
243 wf.format(wqday.getW(i)), | |
244 qf.format(wqday.getQ(i)), | |
245 "", "", "", | |
246 df.format(wqday.getDay(i)) | |
247 }); | |
248 } | |
249 } | |
250 else { | |
251 for (int i = 0; i < size; i ++) { | |
252 source.addData(new String[] { | |
253 "", | |
254 wf.format(wqday.getW(i)), | |
255 qf.format(wqday.getQ(i)), | |
256 "", "", "", | |
257 df.format(wqday.getDay(i)) | |
258 }); | |
259 } | |
260 } | |
261 } | |
262 } | |
263 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 : |