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

Timestamp:

Apr 8, 2015, 4:01:25 PM (10 years ago)

Author:

lfita

Message:

Fixing 'slice_variable' for that dimensions which are not given a value to slice

File:

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

trunk/tools/drawing.py

@@ -519 +519 @@
     odimyu = objdimy.getncattr('units')
 
-# Getting only tat dimensions with coincident names
+# Getting only that dimensions with coincident names
     dimnvx = objdimx.dimensions
     cutslice = []
@@ -580 +580 @@
 
     if mapvalue == 'None': mapvalue = None
-
-    print '  ' + fname + ': Lluis shapes valshad:',valshad.shape,'valcont:',valcont.shape
 
     drw.plot_2D_shadow_contour(valshad, valcont, vnamesfig, odimxv, odimyv, odimxu,  \
@@ -764 +762 @@
     if vdimxn == timename:
         timevals = objsf.variables[vdimxn][:]
+        timedims = objsf.variables[vdimxn].dimensions
         dimt = 'x'
         ovalaxis = objsf.variables[vdimyn]
@@ -769 +768 @@
     elif vdimyn == timename:
         timevals = objsf.variables[vdimyn][:]
+        timedims = objsf.variables[vdimyn].dimensions
         dimt = 'y'
         ovalaxis = objsf.variables[vdimxn]
@@ -779 +779 @@
     timepos, timelabels = drw.CFtimes_plot(timevals, timeunit, timekind, timefmt)
 
-    if len(ovalaxis.shape) <= 2:
-        ovalaxisv = ovalaxis[:]
-
-    elif len(ovalaxis.shape) == 3:
-        ovalaxisv = ovalaxis[0,:]
-    else:
-        print errormsg
-        print '  ' + fname + ': shape of dimension variable:', ovalaxis.shape,       \
-          ' not ready!!'
-        quit(-1)
+# Getting only that dimensions with coincident names
+    dimnvx = ovalaxis.dimensions
+
+    cutslice = []
+    for idimn in dimsshad:
+        found = False
+        for dimsn in dimnvx:
+            if idimn == dimsn:
+                cutslice.append(slice(0,len(objsf.dimensions[idimn])))
+                found = True
+        if not found: cutslice.append(0)
+
+    ovalaxisv = ovalaxis[tuple(cutslice)]
+
+##    if len(ovalaxis.shape) <= 2:
+##        ovalaxisv = ovalaxis[:]
+
+##    elif len(ovalaxis.shape) == 3:
+##        ovalaxisv = ovalaxis[0,:]
+##    else:
+##        print errormsg
+##        print '  ' + fname + ': shape of dimension variable:', ovalaxis.shape,       \
+##          ' not ready!!'
+##        quit(-1)
 
     if countlabelfmt == 'None': 
@@ -1922 +1936 @@
     dimxv = dimtvalues[0:dimtsqx]
     dimyv = dimtvalues[0:dimt:dimtsqx]
-
-    print 'Lluis lens: dimxv, dimyv:',len(dimxv), len(dimyv)
 
     dimn = ['time','time']