Changeset 204

Timestamp:

Jul 11, 2011, 1:53:25 AM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: python: a small late commit to add zooming capabilities.

Location:

trunk/MESOSCALE/PLOT/PYTHON

Files:

• mylib/myplot.py (modified) (1 diff)
• scripts/winds.py (modified) (5 diffs)

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

@@ -264 +264 @@
 
 def fmtvar (whichvar="def"):
-        fmtvar    =     { \
-                "tk":           "%.0f",\
-                "tpot":         "%.0f",\
-                "def":          "%.1e",\
-                "PTOT":         "%.0f",\
-                "HGT":          "%.1e",\
-                "USTM":         "%.2f",\
-                        }
-        if whichvar not in fmtvar:
-                whichvar = "def"
-        return fmtvar[whichvar]
+    fmtvar    =     { \
+             "tk":           "%.0f",\
+             "tpot":         "%.0f",\
+             "def":          "%.1e",\
+             "PTOT":         "%.0f",\
+             "HGT":          "%.1e",\
+             "USTM":         "%.2f",\
+                    }
+    if whichvar not in fmtvar:
+        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]
+    whichcolorb =    { \
+             "def":          "spectral",\
+             "HGT":          "spectral",\
+             "tk":           "gist_heat",\
+             "QH2O":         "PuBu",\
+             "USTM":         "YlOrRd",\
+#"RdPu",\
+                     }
+    if whichone not in whichcolorb:
+        whichone = "def"
+    return whichcolorb[whichone]
 
 def definecolorvec (whichone="def"):

trunk/MESOSCALE/PLOT/PYTHON/scripts/winds.py

@@ -22 +22 @@
            vmin=None,\
            vmax=None,\
-           tile=False):
+           tile=False,\
+           zoom=None):
 
     ####################################################################################################################
@@ -84 +85 @@
     elif proj == "lcc":      [wlon,wlat] = wrfinterv(lon2d,lat2d)
     else:                    [wlon,wlat] = simplinterv(lon2d,lat2d)
+    if zoom:  
+        dlon = abs(wlon[1]-wlon[0])/2.
+        dlat = abs(wlat[1]-wlat[0])/2.
+        [wlon,wlat] = [ [wlon[0]+zoom*dlon/100.,wlon[1]-zoom*dlon/100.],\
+                        [wlat[0]+zoom*dlat/100.,wlat[1]-zoom*dlat/100.] ]
+        print "zoom %",zoom,wlon,wlat
 
     ##############################################################################
@@ -219 +226 @@
                if dimension == 2:                what_I_plot = field[:,:]
                elif dimension == 3:              what_I_plot = field[i,:,:]
+           palette = get_cmap(name=colorb)
+           #palette.set_over('b', 1.0)
            if not tile:
                zelevels = np.linspace(zevmin,zevmax)
-               contourf( x, y, what_I_plot, cmap = get_cmap(name=colorb), levels = zelevels )
+               contourf( x, y, what_I_plot, 10, cmap = palette, levels = zelevels )
            else:    
-               pcolor( x, y, what_I_plot, cmap = get_cmap(name=colorb), vmin=zevmin, vmax=zevmax )
+               pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
            if var in ['HGT']:        pass
-           elif colorb:              colorbar(fraction=0.05,pad=0.1,format=fmtvar(var))
-
+           elif colorb:              
+                                     ndiv = 10
+                                     colorbar(fraction=0.05,pad=0.1,format=fmtvar(var),\
+                                              ticks=np.linspace(zevmin,zevmax,ndiv+1),\
+                                              extend='max',spacing='proportional')
+                                                     # both min max neither
        ### Vector plot
        if winds:
@@ -296 +309 @@
     parser.add_option('-m', action='store', dest='vmin',        type="float",   default=None,  help='bounding minimum value for color plot')    
     parser.add_option('-M', action='store', dest='vmax',        type="float",   default=None,  help='bounding maximum value for color plot') 
-    parser.add_option('-T', action='store_true', dest='tile',                   default=False, help='draw a tiled plot')
+    parser.add_option('-T', action='store_true', dest='tile',                   default=False, help='draw a tiled plot (no blank zone)')
+    parser.add_option('-z', action='store', dest='zoom',        type="float",   default=None,  help='zoom factor in %')
     #parser.add_option('-V', action='store', dest='comb',        type="float",   default=None,  help='a defined combination of variables')
     (opt,args) = parser.parse_args()
@@ -332 +346 @@
     name = winds (zefile,int(zelevel),\
            proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=opt.var,numplot=opt.numplot,colorb=opt.colorb,winds=opt.winds,\
-           addchar=lschar,interv=[zehour,zehourin],vmin=opt.vmin,vmax=opt.vmax,tile=opt.tile)
+           addchar=lschar,interv=[zehour,zehourin],vmin=opt.vmin,vmax=opt.vmax,tile=opt.tile,zoom=opt.zoom)
     print 'Done: '+name