Mercurial > dive4elements > river

comparison flys-artifacts/src/main/java/org/dive4elements/river/exports/WDifferencesExporter.java @ 5831:bd047b71ab37

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
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/WDifferencesExporter.java@b2ea89a665bc
children
comparison
equal deleted inserted replaced
5830:160f53ee0870 5831:bd047b71ab37
1 package org.dive4elements.river.exports;
2
3 import java.io.IOException;
4 import java.io.OutputStream;
5 import java.text.NumberFormat;
6 import java.util.ArrayList;
7 import java.util.List;
8 import java.util.Map;
9 import java.util.HashMap;
10 import java.util.Date;
11 import java.text.DateFormat;
12 import java.util.Locale;
13
14 import org.w3c.dom.Document;
15
16 import org.apache.log4j.Logger;
17
18 import au.com.bytecode.opencsv.CSVWriter;
19
20 import net.sf.jasperreports.engine.JasperExportManager;
21 import net.sf.jasperreports.engine.JasperFillManager;
22 import net.sf.jasperreports.engine.JasperPrint;
23 import net.sf.jasperreports.engine.JRException;
24
25 import org.dive4elements.artifacts.CallContext;
26 import org.dive4elements.artifacts.CallMeta;
27 import org.dive4elements.artifacts.common.utils.Config;
28
29 import org.dive4elements.river.artifacts.WINFOArtifact;
30
31 import org.dive4elements.river.artifacts.model.CalculationResult;
32 import org.dive4elements.river.artifacts.model.WKms;
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 * (CSV)Exporter for WDifferences.
41 */
42 public class WDifferencesExporter extends AbstractExporter {
43
44 /** The logger used in this exporter. */
45 private static Logger logger = Logger.getLogger(WDifferencesExporter.class);
46
47
48 public static final String WDIFF_CSV_KM_HEADER =
49 "export.w_differences.csv.header.km";
50
51 public static final String WDIFF_CSV_W_HEADER =
52 "export.w_differences.csv.header.w";
53
54 public static final String WDIFF_DEFAULT_CSV_KM_HEADER = "Fluss-Km";
55 public static final String WDIFF_DEFAULT_CSV_W_HEADER = "m";
56
57 public static final String PDF_HEADER_MODE = "export.wdifferences.pdf.mode";
58 public static final String JASPER_FILE = "export.wdifferences.pdf.file";
59
60 /** The storage that contains all WKms objects for the different facets. */
61 protected List<WKms[]> data;
62
63
64 public void init(Document request, OutputStream out, CallContext context) {
65 logger.debug("WDifferencesExporter.init");
66
67 super.init(request, out, context);
68
69 this.data = new ArrayList<WKms[]>();
70 }
71
72
73 /**
74 * Genereate data in csv format.
75 */
76 @Override
77 public void generate()
78 throws IOException
79 {
80 logger.debug("WDifferencesExporter.generate");
81
82 if (facet == null) {
83 throw new IOException("invalid (null) facet for exporter");
84 }
85 else if (facet.equals(AbstractExporter.FACET_CSV)) {
86 generateCSV();
87 }
88 else if (facet.equals(AbstractExporter.FACET_PDF)) {
89 generatePDF();
90 }
91 else {
92 throw new IOException("invalid facet (" + facet + ") for exporter");
93 }
94 }
95
96
97 /**
98 * Adds given data.
99 * @param d either a WKms or a CalculationResult to add to data.
100 */
101 @Override
102 protected void addData(Object d) {
103 if (d instanceof CalculationResult) {
104 d = ((CalculationResult)d).getData();
105 if (d instanceof WKms []) {
106 data.add((WKms [])d);
107 }
108 }
109 else if (d instanceof WKms) {
110 data.add(new WKms[] { (WKms) d });
111 }
112 }
113
114
115 /**
116 * Lets writer write all data (including header).
117 * @param writer Writer to write data with.
118 */
119 @Override
120 protected void writeCSVData(CSVWriter writer) {
121 logger.info("WDifferencesExporter.writeData");
122
123 writeCSVHeader(writer);
124
125 for (WKms[] tmp: data) {
126 for (WKms wkms: tmp) {
127 wKms2CSV(writer, wkms);
128 }
129 }
130 }
131
132
133 /**
134 * Lets csvwriter write the header (first line in file).
135 * @param writer Writer to write header with.
136 */
137 protected void writeCSVHeader(CSVWriter writer) {
138 logger.info("WDifferencesExporter.writeCSVHeader");
139
140 writer.writeNext(new String[] {
141 msg(WDIFF_CSV_KM_HEADER, WDIFF_DEFAULT_CSV_KM_HEADER),
142 msg(WDIFF_CSV_W_HEADER, WDIFF_DEFAULT_CSV_W_HEADER)
143 });
144 }
145
146
147 protected void wKms2CSV(CSVWriter writer, WKms wkms) {
148 logger.debug("WDifferencesExporter.wKms2CSV");
149
150 NumberFormat kmf = getKmFormatter();
151 NumberFormat wf = getWFormatter();
152 int size = wkms.size();
153
154 for (int i = 0; i < size; i ++) {
155
156 writer.writeNext(new String[] {
157 kmf.format(wkms.getKm(i)),
158 wf.format(wkms.getW(i))
159 });
160 }
161 }
162
163
164 /**
165 * Returns the number formatter for kilometer values.
166 *
167 * @return the number formatter for kilometer values.
168 */
169 protected NumberFormat getKmFormatter() {
170 return Formatter.getWaterlevelKM(context);
171 }
172
173
174 /**
175 * Returns the number formatter for W values.
176 *
177 * @return the number formatter for W values.
178 */
179 protected NumberFormat getWFormatter() {
180 return Formatter.getWaterlevelW(context);
181 }
182
183
184 @Override
185 protected void writePDF(OutputStream out) {
186 WKmsJRDataSource source = createJRData();
187
188 String jasperFile = Resources.getMsg(
189 context.getMeta(),
190 JASPER_FILE,
191 "/jasper/wdifferences_en.jasper");
192 String confPath = Config.getConfigDirectory().toString();
193
194 Map parameters = new HashMap();
195 parameters.put("ReportTitle", "Exported Data");
196 try {
197 JasperPrint print = JasperFillManager.fillReport(
198 confPath + jasperFile,
199 parameters,
200 source);
201 JasperExportManager.exportReportToPdfStream(print, out);
202 }
203 catch(JRException je) {
204 logger.warn("Error generating PDF Report!");
205 je.printStackTrace();
206 }
207 }
208
209 protected WKmsJRDataSource createJRData() {
210 WKmsJRDataSource source = new WKmsJRDataSource();
211
212 addMetaData(source);
213 for (WKms[] tmp: data) {
214 for (WKms wkms: tmp) {
215 addWKmsData(source, wkms);
216 }
217 }
218 return source;
219 }
220
221
222 protected void addMetaData(WKmsJRDataSource source) {
223 CallMeta meta = context.getMeta();
224
225 WINFOArtifact flys = (WINFOArtifact) master;
226
227 source.addMetaData ("river", FLYSUtils.getRivername(flys));
228
229 Locale locale = Resources.getLocale(meta);
230 DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, locale);
231
232 source.addMetaData("date", df.format(new Date()));
233
234 String differences = FLYSUtils.getWDifferences(flys, context);
235 source.addMetaData("differences", differences);
236
237 source.addMetaData("calculation", Resources.getMsg(
238 locale,
239 PDF_HEADER_MODE,
240 "W Differences"));
241 }
242
243 protected void addWKmsData(WKmsJRDataSource source, WKms wkms) {
244 NumberFormat kmf = getKmFormatter();
245 NumberFormat wf = getWFormatter();
246 int size = wkms.size();
247
248 for (int i = 0; i < size; i ++) {
249
250 source.addData(new String[] {
251 kmf.format(wkms.getKm(i)),
252 wf.format(wkms.getW(i))
253 });
254 }
255 }
256
257 }
258 // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :
http://dive4elements.wald.intevation.org