Changeset 382 in lmdz_wrf for trunk/tools/drawing.py

Timestamp:

Mar 23, 2015, 11:37:10 AM (10 years ago)

Author:

lfita

Message:

Second attempt to fix dimensions issue on 'draw_2D_shad_cont'

File:

• trunk/tools/drawing.py (modified) (2 diffs)

trunk/tools/drawing.py

@@ -523 +523 @@
     cutslice = []
     for idimn in objdimx.dimensions:
-        if idimn == dimxvn:
-            cutslice.append(slice(0,len(objsf.dimensions(idimn))))
-        else:
-            cutslice.append(0)
+        found = False
+        for dimsn in dimsshad:
+            if idimn == dimsn:
+                cutslice.append(slice(0,len(objsf.dimensions(idimn))))
+                found = True
+        if not found: cutslice.append(0)
 
     odimxv = objdimx[tuple(cutslice)]
@@ -533 +535 @@
     cutslice = []
     for idimn in objdimy.dimensions:
-        if idimn == dimyvn:
-            cutslice.append(slice(0,len(objsf.dimensions(idimn))))
-        else:
-            cutslice.append(0)
+        found = False
+        for dimsn in dimsshad:
+            if idimn == dimsn:
+                cutslice.append(slice(0,len(objsf.dimensions(idimn))))
+                found = True
+        if not found: cutslice.append(0)
 
     odimyv = objdimy[tuple(cutslice)]