Mercurial > dive4elements > river
comparison artifacts/src/main/java/org/dive4elements/river/exports/minfo/BedDifferenceExporter.java @ 6217:e213b538f78c
Added exporter for bed height differences.
| author | Raimund Renkert <rrenkert@intevation.de> |
|---|---|
| date | Thu, 06 Jun 2013 16:41:27 +0200 |
| parents | |
| children | 34f393434913 |
comparison
equal
deleted
inserted
replaced
| 6216:9fdb2b59b992 | 6217:e213b538f78c |
|---|---|
| 1 package org.dive4elements.river.exports.minfo; | |
| 2 | |
| 3 import gnu.trove.TDoubleArrayList; | |
| 4 | |
| 5 import java.io.IOException; | |
| 6 import java.io.OutputStream; | |
| 7 import java.text.NumberFormat; | |
| 8 import java.util.LinkedList; | |
| 9 import java.util.List; | |
| 10 | |
| 11 import org.apache.log4j.Logger; | |
| 12 import org.dive4elements.artifacts.CallContext; | |
| 13 import org.dive4elements.river.artifacts.model.CalculationResult; | |
| 14 import org.dive4elements.river.artifacts.model.minfo.BedDifferencesResult; | |
| 15 import org.dive4elements.river.exports.AbstractExporter; | |
| 16 import org.dive4elements.river.utils.Formatter; | |
| 17 import org.w3c.dom.Document; | |
| 18 | |
| 19 import au.com.bytecode.opencsv.CSVWriter; | |
| 20 | |
| 21 | |
| 22 public class BedDifferenceExporter | |
| 23 extends AbstractExporter | |
| 24 { | |
| 25 | |
| 26 /** Private logger. */ | |
| 27 private static Logger logger = | |
| 28 Logger.getLogger(BedDifferenceExporter.class); | |
| 29 | |
| 30 private static final String CSV_HEADER_KM = "export.minfo.beddifference.km"; | |
| 31 private static final String CSV_HEADER_DIFF = | |
| 32 "export.minfo.beddifference.diff"; | |
| 33 | |
| 34 private BedDifferencesResult[] results; | |
| 35 | |
| 36 @Override | |
| 37 public void init(Document request, OutputStream out, CallContext context) { | |
| 38 logger.debug("BedQualityExporter.init"); | |
| 39 super.init(request, out, context); | |
| 40 results = new BedDifferencesResult[0]; | |
| 41 } | |
| 42 @Override | |
| 43 protected void writeCSVData(CSVWriter writer) throws IOException { | |
| 44 writeCSVHeader(writer); | |
| 45 | |
| 46 NumberFormat kmf = Formatter.getCalculationKm(context.getMeta()); | |
| 47 NumberFormat mf = Formatter.getMeterFormat(context); | |
| 48 for (int i = 0; i < results.length; i++) { | |
| 49 BedDifferencesResult result = results[i]; | |
| 50 double[][] kms = result.getDifferencesData(); | |
| 51 for (int j = 0; j < kms[0].length; j++) { | |
| 52 writer.writeNext(new String[] { | |
| 53 kmf.format(kms[0][j]), mf.format(kms[1][j])}); | |
| 54 } | |
| 55 } | |
| 56 } | |
| 57 | |
| 58 @Override | |
| 59 protected void writePDF(OutputStream out) { | |
| 60 // TODO implement me! | |
| 61 | |
| 62 } | |
| 63 | |
| 64 @Override | |
| 65 protected void addData(Object data) { | |
| 66 if (!(data instanceof CalculationResult)) { | |
| 67 logger.warn("Invalid data type."); | |
| 68 return; | |
| 69 } | |
| 70 Object[] d = (Object[])((CalculationResult)data).getData(); | |
| 71 | |
| 72 if (!(d instanceof BedDifferencesResult[])) { | |
| 73 logger.warn("Invalid result object."); | |
| 74 return; | |
| 75 } | |
| 76 results = (BedDifferencesResult[])d; | |
| 77 } | |
| 78 | |
| 79 protected void writeCSVHeader(CSVWriter writer) { | |
| 80 logger.debug("writeCSVHeader()"); | |
| 81 | |
| 82 List<String> header = new LinkedList<String>(); | |
| 83 if (results != null) { | |
| 84 header.add(msg(CSV_HEADER_KM, "km")); | |
| 85 header.add(msg(CSV_HEADER_DIFF, "m")); | |
| 86 } | |
| 87 writer.writeNext(header.toArray(new String[header.size()])); | |
| 88 } | |
| 89 } |