diff artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flowdepth/FlowDepthCalculationResult.java @ 9351:0c8f170945d7

no tkh bug fixed
author gernotbelger
date Tue, 31 Jul 2018 17:17:03 +0200
parents 7b2b086e45f0
children ba1e2e8f05d1
line wrap: on
line diff
--- a/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flowdepth/FlowDepthCalculationResult.java	Tue Jul 31 16:04:01 2018 +0200
+++ b/artifacts/src/main/java/org/dive4elements/river/artifacts/sinfo/flowdepth/FlowDepthCalculationResult.java	Tue Jul 31 17:17:03 2018 +0200
@@ -55,28 +55,28 @@
     }
 
     @Override
-    public void writeCSVHeader(final ExportContextCSV exportContextCSV, final RiverInfo river) {
+    public void writeCSVHeader(final ExportContextCSV exportContext, final RiverInfo river) {
         // getLog().info("FlowDepthExporter.writeCSVHeader");
 
         final Collection<String> header = new ArrayList<>(11);
 
-        header.add(exportContextCSV.formatCsvHeader(GeneralResultType.station));
-        header.add(exportContextCSV.msgUnitCSV(SInfoResultType.flowdepth));
+        header.add(exportContext.formatCsvHeader(GeneralResultType.station));
+        header.add(exportContext.msgUnitCSV(SInfoResultType.flowdepth));
 
-        if (this.hasTkh()) {
-            header.add(exportContextCSV.msgUnitCSV(SInfoResultType.flowdepthtkh));
-            header.add(exportContextCSV.msgUnitCSV(SInfoResultType.tkh));
+        if (((FlowDepthCalculationResults) exportContext.getResults()).hasTkh()) {
+            header.add(exportContext.msgUnitCSV(SInfoResultType.flowdepthtkh));
+            header.add(exportContext.msgUnitCSV(SInfoResultType.tkh));
         }
 
-        header.add(exportContextCSV.msgUnitCSV(SInfoResultType.waterlevel, river.getWstUnit()));
-        header.add(exportContextCSV.msgUnitCSV(SInfoResultType.discharge));
-        header.add(exportContextCSV.formatCsvHeader(GeneralResultType.waterlevelLabel));
-        header.add(exportContextCSV.formatCsvHeader(GeneralResultType.gaugeLabel));
-        header.add(exportContextCSV.formatCsvHeader(SInfoResultType.meanBedHeight));
-        header.add(exportContextCSV.formatCsvHeader(SInfoResultType.soundingLabel));
-        header.add(exportContextCSV.formatCsvHeader(GeneralResultType.location));
+        header.add(exportContext.msgUnitCSV(SInfoResultType.waterlevel, river.getWstUnit()));
+        header.add(exportContext.msgUnitCSV(SInfoResultType.discharge));
+        header.add(exportContext.formatCsvHeader(GeneralResultType.waterlevelLabel));
+        header.add(exportContext.formatCsvHeader(GeneralResultType.gaugeLabel));
+        header.add(exportContext.formatCsvHeader(SInfoResultType.meanBedHeight));
+        header.add(exportContext.formatCsvHeader(SInfoResultType.soundingLabel));
+        header.add(exportContext.formatCsvHeader(GeneralResultType.location));
 
-        exportContextCSV.writeCSVLine(header.toArray(new String[header.size()]));
+        exportContext.writeCSVLine(header.toArray(new String[header.size()]));
 
     }
 
@@ -134,7 +134,9 @@
         lines.add(exportContext.formatRowValue(row, SInfoResultType.flowdepth));
         final FlowDepthCalculationResults results = exportContext.getResults();
 
-        if (mode == ExportMode.pdf || results.isUseTkh()) {
+        // REMARK: for csv: only remove tkh columns, if no result has tkh
+        // for pdf: adding by default is necessary. If there is no tkh, columns will be removed later in the exporter.
+        if (mode == ExportMode.pdf || results.hasTkh()) {
             lines.add(exportContext.formatRowValue(row, SInfoResultType.flowdepthtkh));
             lines.add(exportContext.formatRowValue(row, SInfoResultType.tkh));
         }

http://dive4elements.wald.intevation.org