Account for no data in calculator

HydrologicEngineeringCenter · Jan 26, 2023 · db48178 · db48178
1 parent f2839af
commit db48178
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/vortex-api/src/main/java/mil/army/usace/hec/vortex/math/Calculator.java b/vortex-api/src/main/java/mil/army/usace/hec/vortex/math/Calculator.java
@@ -2,6 +2,8 @@
 
 import mil.army.usace.hec.vortex.VortexGrid;
 
+import java.util.Arrays;
+
 public class Calculator {
 
     private final VortexGrid inputGrid;
@@ -75,27 +77,31 @@ public VortexGrid calculate(){
         int size = data.length;
         float[] resultantData = new float[size];
 
+        float noDataValue = (float) inputGrid.noDataValue();
+
+        Arrays.fill(resultantData, noDataValue);
+
         if (!Double.isNaN(multiplyValue)) {
             for (int i = 0; i < size; i++) {
-                resultantData[i] = data[i] * multiplyValue;
+                resultantData[i] = data[i] != noDataValue ? data[i] * multiplyValue : noDataValue;
             }
         }
 
         if (!Double.isNaN(divideValue)) {
             for (int i = 0; i < size; i++) {
-                resultantData[i] = data[i] / divideValue;
+                resultantData[i] = data[i] != noDataValue ? data[i] / divideValue : noDataValue;
             }
         }
 
         if (!Double.isNaN(addValue)) {
             for (int i = 0; i < size; i++) {
-                resultantData[i] = data[i] + addValue;
+                resultantData[i] = data[i] != noDataValue ? data[i] + addValue : noDataValue;
             }
         }
 
         if (!Double.isNaN(subtractValue)) {
             for (int i = 0; i < size; i++) {
-                resultantData[i] = data[i] - subtractValue;
+                resultantData[i] = data[i] != noDataValue ? data[i] - subtractValue : noDataValue;
             }
         }