Changeset 202 for trunk/MESOSCALE/PLOT/PYTHON/mylib

Timestamp:

Jul 11, 2011, 12:02:13 AM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: python graphics interface. much better handling of picture names and a few improvements (new generic functions)

File:

• trunk/MESOSCALE/PLOT/PYTHON/mylib/myplot.py (modified) (7 diffs)

trunk/MESOSCALE/PLOT/PYTHON/mylib/myplot.py

@@ -62 +62 @@
     return lschar, zehour, zehourin
 
+def getprefix (nc):
+    prefix = 'LMD_MMM_'
+    prefix = prefix + 'd'+str(getattr(nc,'GRID_ID'))+'_'
+    prefix = prefix + str(int(getattr(nc,'DX')/1000.))+'km_'
+    return prefix
+
 def api_onelevel (  path_to_input   = None, \
                     input_name      = 'wrfout_d0?_????-??-??_??:00:00', \
@@ -177 +183 @@
         return improc
 
+def getwinds (nc,charu='Um',charv='Vm'):
+    import numpy as np
+    u = nc.variables[charu]
+    v = nc.variables[charv]
+    if charu == 'U': u = u[:, :, :, 0:len(u[0,0,0,:])-1]
+    if charv == 'V': v = v[:, :, 0:len(v[0,0,:,0])-1, :]
+                     ### ou alors prendre les coordonnees speciales
+    return u,v
+
 def vectorfield (u, v, x, y, stride=3, scale=15., factor=250., color='black', csmooth=1, key=True):
     ## scale regle la reference du vecteur
@@ -182 +197 @@
     import  matplotlib.pyplot               as plt
     import  numpy                           as np
-    #posx = np.max(x) + np.std(x) / 3.  ## pb pour les domaines globaux ...
-    #posy = np.mean(y)
-    #posx = np.min(x)
-    #posy = np.max(x)
-    #posx = np.max(x) - np.std(x) / 10.
-    #posy = np.max(y) + np.std(y) / 10.
     posx = np.min(x) - np.std(x) / 10.
     posy = np.min(y) - np.std(y) / 10.
@@ -199 +208 @@
                     angles='xy',color=color,\
                     scale=factor,width=widthvec )
-    if color=='white':     kcolor='black'
-    elif color=='yellow':  kcolor=color
-    else:                  kcolor=color
+    if color in ['white','yellow']:     kcolor='black'
+    else:                               kcolor=color
     if key: p = plt.quiverkey(q,posx,posy,scale,\
                    str(int(scale)),coordinates='data',color=kcolor,labelpos='S',labelsep = 0.03)
@@ -233 +241 @@
     else:                  blat = wlat[0]
     h = 2000.
-    radius = 3397200
+    radius = 3397200.
     if   char == "cyl":     m = Basemap(rsphere=radius,projection='cyl',\
                               llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
@@ -261 +269 @@
                 "def":          "%.1e",\
                 "PTOT":         "%.0f",\
+                "HGT":          "%.1e",\
                 "USTM":         "%.2f",\
                         }
@@ -266 +275 @@
                 whichvar = "def"
         return fmtvar[whichvar]
+
+def defcolorb (whichone="def"):
+        whichcolorb =    { \
+                "def":          "spectral",\
+                "HGT":          "spectral",\
+                "tk":           "gist_heat",\
+                "QH2O":         "PuBu",\
+                         }
+        if whichone not in whichcolorb:
+                whichone = "def"
+        return whichcolorb[whichone]
+
+def definecolorvec (whichone="def"):
+        whichcolor =    { \
+                "def":          "black",\
+                "vis":          "yellow",\
+                "vishires":     "yellow",\
+                "molabw":       "yellow",\
+                "mola":         "black",\
+                "gist_heat":    "white",\
+                "hot":          "tk",\
+                "gist_rainbow": "black",\
+                "spectral":     "black",\
+                "gray":         "red",\
+                "PuBu":         "black",\
+                        }
+        if whichone not in whichcolor:
+                whichone = "def"
+        return whichcolor[whichone]
 
 def marsmap (whichone="vishires"):