added getVariables

Author: TomBlock

core/src/main/java/com/bc/fiduceo/reader/smos/SmosL1CDailyGriddedReader.java

@@ -33,14 +33,27 @@ class SmosL1CDailyGriddedReader extends NetCDFReader {
 
     private final ReaderContext readerContext;
     private final Rectangle2D.Float boundary;
+    private final List<String> variablesToSkip;
+    private final List<String> variables2D;
 
     private File productDir;
     private PixelLocator pixelLocator;
     private TimeLocator timeLocator;
 
+
     SmosL1CDailyGriddedReader(ReaderContext readerContext) {
         this.readerContext = readerContext;
         this.boundary = new Rectangle2D.Float(-180.f, -86.72f, 360.f, 173.44f);
+
+        variablesToSkip = new ArrayList<>();
+        variablesToSkip.add("lat");
+        variablesToSkip.add("lon");
+        variablesToSkip.add("inc");
+        variablesToSkip.add("dinc");
+
+        variables2D = new ArrayList<>();
+        variables2D.add("X_Swath");
+        variables2D.add("Grid_Point_Mask");
     }
 
     @Override
@@ -172,7 +185,26 @@ public ArrayInt.D2 readAcquisitionTime(int x, int y, Interval interval) throws I
 
     @Override
     public List<Variable> getVariables() throws InvalidRangeException, IOException {
-        throw new IllegalStateException("not implemented");
+        final List<Variable> variablesInFile = netcdfFile.getVariables();
+        final ArrayList<Variable> exportVariables = new ArrayList<>();
+
+        for (Variable variable : variablesInFile) {
+            final String variableName = variable.getShortName();
+            if (variablesToSkip.contains(variableName)) {
+                continue;
+            }
+            if (variables2D.contains(variableName)) {
+                exportVariables.add(variable);
+                continue;
+            }
+
+            // the remainder is 3d with 15 angle layers tb 2022-10-12
+            final ArrayList<Variable> bandVariables = new ArrayList<>();
+            addLayered3DVariables(bandVariables, variable, 15, 0);
+            exportVariables.addAll(bandVariables);
+        }
+
+        return exportVariables;
     }
 
     @Override

core/src/test/java/com/bc/fiduceo/NCTestUtils.java

@@ -152,6 +152,8 @@ public static void assertAttribute(Variable variable, String attributeName, Stri
             assertEquals(expected, attribute.getStringValue());
         } else if (dataType == DataType.FLOAT) {
             assertEquals(Float.parseFloat(expected), attribute.getNumericValue().floatValue(), 1e-8);
+        } else if (dataType == DataType.INT) {
+            assertEquals(Integer.parseInt(expected), attribute.getNumericValue().intValue(), 1e-8);
         } else if (dataType == DataType.SHORT) {
             if (attribute.isArray()) {
                 final String[] segments = expected.split(",");

core/src/test/java/com/bc/fiduceo/reader/smos/SmosL1CDailyGriddedReader_IO_Test.java

@@ -1,6 +1,7 @@
 package com.bc.fiduceo.reader.smos;
 
 import com.bc.fiduceo.IOTestRunner;
+import com.bc.fiduceo.NCTestUtils;
 import com.bc.fiduceo.TestUtil;
 import com.bc.fiduceo.core.Dimension;
 import com.bc.fiduceo.core.NodeType;
@@ -14,11 +15,14 @@
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import ucar.ma2.InvalidRangeException;
+import ucar.nc2.Variable;
 
 import java.awt.geom.Point2D;
 import java.io.File;
 import java.io.IOException;
 import java.util.Date;
+import java.util.List;
 
 import static org.junit.Assert.*;
 
@@ -163,6 +167,57 @@ public void testGetPixelLocator_CDF3TA() throws IOException {
         }
     }
 
+    @Test
+    public void testGetVariables_CDF3TD() throws IOException, InvalidRangeException {
+        // X_Swath, Grid_Point_Mask
+        // BT_H(15), BT_V(15), BT_3(15), BT_4(15),
+        // Pixel_Radiometric_Accuracy_H(15), Pixel_Radiometric_Accuracy_V(15), Pixel_Radiometric_Accuracy_3(15), Pixel_Radiometric_Accuracy_4(15)
+        // Pixel_BT_Standard_Deviation_H(15), Pixel_BT_Standard_Deviation_V(15), Pixel_BT_Standard_Deviation_3(15), Pixel_BT_Standard_Deviation_4(15)
+        // Incidence_Angle(15), Azimuth_Angle(15), Footprint_Axis1(15), Footprint_Axis2(15),
+        // Xi(15), Eta(15), Nviews(15), Nb_RFI_Flags(15), Nb_SUN_Flags(15)
+        // Days(15), UTC_Seconds(15), UTC_Microseconds(15)
+        // 362 variables total
+
+        final File file = getCDF3TAFile();
+
+        try {
+            reader.open(file);
+
+            final List<Variable> variables = reader.getVariables();
+            assertEquals(362, variables.size());
+
+            Variable variable = variables.get(0);
+            assertEquals("X_Swath", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "9.96921E36");
+
+            variable = variables.get(3);
+            assertEquals("BT_H_ch02", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-32768");
+
+            variable = variables.get(38);
+            assertEquals("BT_3_ch07", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-32768");
+
+            variable = variables.get(175);
+            assertEquals("Pixel_BT_Standard_Deviation_4_ch09", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-32768");
+
+            variable = variables.get(219);
+            assertEquals("Footprint_Axis1_ch08", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-32768");
+
+            variable = variables.get(305);
+            assertEquals("Nb_SUN_Flags_ch04", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-32768");
+
+            variable = variables.get(361);
+            assertEquals("UTC_Microseconds_ch15", variable.getShortName());
+            NCTestUtils.assertAttribute(variable, "_FillValue", "-2147483647");
+        } finally {
+            reader.close();
+        }
+    }
+
     @Test
     public void testGetSubScenePixelLocator_CDF3TD() throws IOException {
         final File file = getCDF3TDFile();