FIX: GateMapper will now map source field to destination radar even i…

…f field not in destination radar (#1418) * FIX: GateMapper will now map source field to destination radar even if field does not exist in destination. * FIX: Whitespace --------- Co-authored-by: Robert Jackson <[email protected]>
ARM-DOE · Apr 13, 2023 · 1c8d989 · 1c8d989
1 parent 6c8bc9c
commit 1c8d989
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/pyart/map/gate_mapper.py b/pyart/map/gate_mapper.py
@@ -222,9 +222,15 @@ def mapped_radar(self, field_list):
 
         src_fields = {}
         for field in field_list:
-            mapped_radar.fields[field]["data"] = np.ma.masked_where(
-                True, mapped_radar.fields[field]["data"]
-            )
+            if field in list(mapped_radar.fields.keys()):
+                mapped_radar.fields[field]["data"] = np.ma.masked_where(
+                    True, mapped_radar.fields[field]["data"]
+                )
+            else:
+                mapped_radar.fields[field] = deepcopy(self.src_radar.fields[field])
+                mapped_radar.fields[field]["data"] = np.ma.masked_where(
+                    True, np.ma.zeros((mapped_radar.nrays, mapped_radar.ngates))
+                )
             src_fields[field] = np.ma.masked_where(
                 self.gatefilter_src.gate_excluded, self.src_radar.fields[field]["data"]
             )

diff --git a/tests/map/test_gatemapper.py b/tests/map/test_gatemapper.py
@@ -11,15 +11,23 @@ def test_gatemapper():
     new_radar = deepcopy(old_radar)
     new_radar.latitude["data"] = old_radar.latitude["data"] + 0.001
     new_radar.longitude["data"] = old_radar.longitude["data"] + 0.001
-    gate_mapper = pyart.map.GateMapper(new_radar, old_radar)
+    old_radar.fields["reflectivity_copy"] = old_radar.fields["reflectivity"]
+    gate_mapper = pyart.map.GateMapper(old_radar, new_radar)
     mapped_radar = gate_mapper.mapped_radar(["reflectivity"])
 
     # Test point outside of 1 min tolerance
     assert gate_mapper[20, 20] == (None, None)
-    assert gate_mapper[4, 4] == (26, 11)
+    assert gate_mapper[40, 40] == (40, 33)
+    assert (
+        mapped_radar.fields["reflectivity"]["data"][40, 33]
+        == old_radar.fields["reflectivity"]["data"][40, 40]
+    )
+
+    # Check case where source radar has field destination doesn't
+    mapped_radar = gate_mapper.mapped_radar(["reflectivity_copy"])
     assert (
-        mapped_radar.fields["reflectivity"]["data"][26, 11]
-        == old_radar.fields["reflectivity"]["data"][4, 4]
+        mapped_radar.fields["reflectivity_copy"]["data"][40, 33]
+        == old_radar.fields["reflectivity_copy"]["data"][40, 40]
     )