Changeset 199 for trunk/MESOSCALE/PLOT/PYTHON

Timestamp:

Jul 5, 2011, 4:18:57 PM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: python graphics. autoscale + bug fixes.

Location:

trunk/MESOSCALE/PLOT/PYTHON

Files:

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

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

@@ -44 +44 @@
     from timestuff import sol2ls
     nc  = Dataset(namefile)
-    if 'Times' in nc.variables:
+    if 'Times' in nc.variables and 'vert' not in nc.variables:
         zetime = nc.variables['Times'][0]
         zetimestart = getattr(nc, 'START_DATE')

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

@@ -18 +18 @@
            colorb=None,\
            winds=True,\
-           addchar=None,
-           interv=[0,1]):
+           addchar=None,\
+           interv=[0,1],\
+           vmin=None,\
+           vmax=None):
 
     ####################################################################################################################
@@ -28 +30 @@
     from netCDF4 import Dataset
     from myplot import getcoord2d,define_proj,makeplotpng,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy
-    from mymath import deg
-    from matplotlib.pyplot import contourf, subplot, figure, rcParams, savefig, colorbar
+    from mymath import deg,max,min,mean
+    from matplotlib.pyplot import contourf, subplot, figure, rcParams, savefig, colorbar, pcolor
     from matplotlib.cm import get_cmap
     import numpy as np
@@ -109 +111 @@
             elif dimension == 3:   field = nc.variables[var][:,:,:]
             elif dimension == 4:   field = nc.variables[var][:,nvert,:,:]  
-
+        dev = np.std(field)*2.5
+        if vmin is None:  zevmin = mean(field) - dev
+        else:             zevmin = vmin
+        if vmax is None:  zevmax = mean(field) + dev
+        else:             zevmax = vmax
+        print "bounds ", zevmin, zevmax
+  
     ###########################
     ### Get length of time axis
@@ -208 +216 @@
                if dimension == 2:                what_I_plot = field[:,:]
                elif dimension == 3:              what_I_plot = field[i,:,:]
-           contourf( x, y, what_I_plot, 30, cmap = get_cmap(name=colorb) )
-           if colorb:     colorbar(fraction=0.05,pad=0.1)
+           #contourf( x, y, what_I_plot, 30, cmap = get_cmap(name=colorb), vmin=zevmin, vmax=zevmax )
+           pcolor( x, y, what_I_plot, cmap = get_cmap(name=colorb), vmin=zevmin, vmax=zevmax )
+           if colorb:     colorbar(fraction=0.05,pad=0.1,format='%.1e')
 
        ### Vector plot
@@ -281 +290 @@
     parser.add_option('-c', action='store', dest='colorb',      type="string",  default=None,  help='change colormap (and draw a colorbar)')
     parser.add_option('-x', action='store_false', dest='winds',                 default=True,  help='flag: no wind vectors')
+    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') 
     (opt,args) = parser.parse_args()
     if opt.namefile is None: 
@@ -315 +326 @@
     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])
+           addchar=lschar,interv=[zehour,zehourin],vmin=opt.vmin,vmax=opt.vmax)
 
     #########################################################