Changeset 871 for trunk/UTIL

Timestamp:

Feb 1, 2013, 6:10:25 AM (12 years ago)

Author:

aslmd

Message:

UTIL PYTHON. Corrected a bug in getdimfromvar so that it works quite in all the cases now.

Location:

trunk/UTIL/PYTHON

Files:

• myplot.py (modified) (5 diffs)
• planetoplot.py (modified) (1 diff)

trunk/UTIL/PYTHON/myplot.py

@@ -732 +732 @@
 def getdimfromvar (nc):
     varinfile = nc.variables.keys()
-    dim = nc.variables[varinfile[-1]].shape ## usually the last variable is 4D or 3D
-    return dim
+    sav = [0.]
+    for var in varinfile:
+        dim = nc.variables[var].shape ## usually the last variable is 4D or 3D
+        if len(dim) > len(sav): sav=dim
+    return sav
 
 ## FROM COOKBOOK http://www.scipy.org/Cookbook/SignalSmooth
@@ -814 +817 @@
     elif 'u' in varinfile:  [uchar,vchar] = ['u','v']   #; print "this is GCM file"
     elif 'vitu' in varinfile:  [uchar,vchar] = ['vitu','vitv']   #; print "this is GCM v5 file"
+    elif 'ucomp' in varinfile:  [uchar,vchar] = ['ucomp','vcomp']
     ### you can add choices here !
     else:                   [uchar,vchar] = ['not found','not found']
@@ -1118 +1122 @@
              "TSK":          "RdBu_r",\
              "QH2O":         "PuBu",\
+             "PSFC":         "RdYlBu_r",\
              "USTM":         "YlOrRd",\
              "WIND":         "YlOrRd",\
@@ -1235 +1240 @@
         "Southern_Hemisphere":   [[-90., 60.],[-180., 180.]],\
         "Northern_Hemisphere":   [[-60., 90.],[-180., 180.]],\
+        "Tharsis_alt":           [[-30., 60.],[ 190., 350.]],\
         "Tharsis":               [[-30., 60.],[-170.,- 10.]],\
         "Whole_No_High":         [[-60., 60.],[-180., 180.]],\
@@ -1249 +1255 @@
         "Xanadu":                [[-40., 20.],[  40., 120.]],\
         "Hyperboreae":           [[ 80., 87.],[- 70.,- 10.]],\
+        "VM_alt":                [[-15.,  0.],[ 270., 300.]],\
               }
     if area not in list:   area = "Whole"

trunk/UTIL/PYTHON/planetoplot.py

@@ -228 +228 @@
           ### --> add a line here if your reference is not present
           else: 
-              print "No altitude found. Try to build a simple axis." ; dadim = getdimfromvar(nc) 
+              dadim = getdimfromvar(nc) ; print "No altitude found. Try to build a simple axis.",dadim
               if   len(dadim) == 4:  print "-- 4D field. Assume z is dim 2." ; vert = np.arange(dadim[-3])
               elif len(dadim) == 3:  print "-- 3D field. Assume z is dim 1." ; vert = [0.]