Changeset 639 in lmdz_wrf for trunk

Timestamp:

Sep 19, 2015, 11:50:23 AM (9 years ago)

Author:

lfita

Message:

Adding axes also cutting if a range on the dimensions is provided in 'plot_2D_shadow_time'

File:

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

trunk/tools/drawing.py

@@ -321 +321 @@
 
     if len(objdimx.shape) <= 2:
-        odimxv = objdimx[:]
-        odimyv = objdimy[:]
+        odimxv0 = objdimx[:]
+        odimyv0 = objdimy[:]
 
     elif len(objdimx.shape) == 3:
-        odimxv = objdimx[0,:]
-        odimyv = objdimy[0,:]
+        odimxv0 = objdimx[0,:]
+        odimyv0 = objdimy[0,:]
     else:
         print errormsg
@@ -339 +339 @@
     timelabel = timevals.split('|')[4].replace('!',' ')
 
+# Dimensional values
+    dimscut = {}
+    for dv in dimvals:
+        print dv
+        dvn = dv.split('|')[0]
+        dvv = dv.split('|')[1]
+        if dvv.find('@') != -1:
+            dimscut[dvn] = slice(int(dvv.split('@')[0]),int(dvv.split('@')[1]))
+        else:
+            if dvv == -1:
+                dnL = len(objsf.dimensions[dvn]) 
+                dimscut[dvn] = slice(0,dnL)
+            else:
+                dimscut[dvn] = slice(dvv,dvv)
+
+    if not dr.searchInlist(dimscut.keys(), vdimxn):
+        dnL = len(objsf.dimensions[vdimxn]) 
+        dimscut[vdimxn] = slice(0,dnL)
+    if not dr.searchInlist(dimscut.keys(), vdimyn):
+        dnL = len(objsf.dimensions[vdimyn]) 
+        dimscut[vdimyn] = slice(0,dnL)
+
     if vdimxn == timename:
-        odimxv = objsf.variables[vdimxn][:]
+#        odimxv = objsf.variables[vdimxn][:]
+        odimxv = objsf.variables[vdimxn][tuple(dimscut[vdimxn])]
         odimxu = timelabel
         timeaxis = 'x'
-        odimyv = objsf.variables[vdimyn]
+        odimyv = objsf.variables[vdimyn][tuple(dimscut[vdimyn])]
         odimyu = odimyv.getncattr('units')
         timepos, timelabels = drw.CFtimes_plot(odimxv, timeunit, timekind, timefmt)
     elif vdimyn == timename:
-        odimyv = objsf.variables[vdimyn][:]
+        odimyv = objsf.variables[vdimyn][tuple(dimscut[vdimyn])]
         odimyu = timelabel
         timeaxis = 'y'
-        odimxv = objsf.variables[vdimxn]
+        odimxv = objsf.variables[vdimxn][tuple(dimscut[vdimxn])]
         odimxu = odimxv.getncattr('units')
         timepos, timelabels = drw.CFtimes_plot(odimyv, timeunit, timekind, timefmt)