Changeset 2658 in lmdz_wrf

Timestamp:

Jul 3, 2019, 3:55:27 AM (5 years ago)

Author:

lfita

Message:

Managing better rotation of missing values

File:

• trunk/tools/geometry_tools.py (modified) (1 diff)

trunk/tools/geometry_tools.py

@@ -180 +180 @@
     rotvecs = np.zeros(vectors.shape, dtype=np.float)
 
+    mavec = False
+    if type(vectors) == type(gen.mamat):
+        mavec = True
+        vectors = ma.filled(vectors,gen.fillValueF)
+
     Nvecs = vectors.shape[0]
     for iv in range(Nvecs):
         rotvecs[iv,:] = rotate_2D(vectors[iv,:], angle)
+
+    if mavec: 
+        rotvecs = ma.masked_equal(rotvecs, gen.fillValueF)
 
     return rotvecs