Changeset 240 for trunk/MESOSCALE_DEV/PLOT/PYTHON/scripts/winds.py

Timestamp:

Jul 22, 2011, 9:46:21 AM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: corrected a bug in the previous commit in makemeso, works now fine. added anomaly calculations in python interface + better treatment for output; default is now GUI where one can zoom etc... and further explore the plotted field

File:

• trunk/MESOSCALE_DEV/PLOT/PYTHON/scripts/winds.py (modified) (11 diffs)

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

@@ -26 +26 @@
            display=True,\
            itstep=None,\
-           hole=False):
+           hole=False,\
+           save="gui",\
+           anomaly=False):
 
     ####################################################################################################################
@@ -34 +36 @@
     ### Load librairies and functions
     from netCDF4 import Dataset
-    from myplot import getcoord2d,define_proj,makeplotpng,makeplotpngres,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy,\
+    from myplot import getcoord2d,define_proj,makeplotres,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy,\
                        fmtvar,definecolorvec,defcolorb,getprefix,putpoints,calculate_bounds,errormess,definesubplot,\
-                       zoomset,getcoorddef,getwinddef,whatkindfile,reducefield,bounds,getstralt,getfield
+                       zoomset,getcoorddef,getwinddef,whatkindfile,reducefield,bounds,getstralt,getfield,smooth
     from mymath import deg,max,min,mean
-    from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor
+    from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor, show
     from matplotlib.cm import get_cmap
     import numpy as np
     from numpy.core.defchararray import find
 
+    #rcParams['backend'] = 'PS'
+
     ######################
     ### Load NETCDF object
-    nc  = Dataset(namefile)
+    nc  = Dataset(namefile)  
     
     ##################################
@@ -60 +64 @@
     ### Define plot boundaries
     if proj == "npstere":             [wlon,wlat] = latinterv("North_Pole")
+    elif proj == "spstere":           [wlon,wlat] = latinterv("South_Pole")
     elif proj in ["lcc","laea"]:      [wlon,wlat] = wrfinterv(lon2d,lat2d)
     else:                             [wlon,wlat] = simplinterv(lon2d,lat2d)
@@ -119 +124 @@
            what_I_plot, error = reducefield( getfield(nc,var), d4=itime, d3=nvert )
            if not error: 
-               if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot)
+               fvar = var
+               ###
+               if anomaly:
+                   what_I_plot = 100. * ((what_I_plot / smooth(what_I_plot,10)) - 1.)
+                   fvar = 'anomaly'
+               ###
+               if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
                zevmin, zevmax = calculate_bounds(what_I_plot,vmin=vmin,vmax=vmax)
-               if colorb is True: colorb = defcolorb(var)
+               if colorb is True: colorb = defcolorb(fvar)
                palette = get_cmap(name=colorb)
                if not tile:
@@ -133 +144 @@
                elif colorb:              
                          ndiv = 10
-                         colorbar(fraction=0.05,pad=0.1,format=fmtvar(var),\
+                         colorbar(fraction=0.05,pad=0.1,format=fmtvar(fvar),\
                                            ticks=np.linspace(zevmin,zevmax,ndiv+1),\
-                                           extend='max',spacing='proportional')
-                                           # both min max neither
+                                           extend='neither',spacing='proportional')
+                                           # both min max neither 
                   
        ### Vector plot
@@ -144 +155 @@
            if not error:
                if typefile in ['mesoapi','meso']:    
-                   [vecx,vecy] = [dumpbdy(vecx,stag=uchar), dumpbdy(vecy,stag=vchar)]
+                   [vecx,vecy] = [dumpbdy(vecx,6,stag=uchar), dumpbdy(vecy,6,stag=vchar)]
                    key = True
                elif typefile in ['gcm']:            
@@ -178 +189 @@
     else:               zeplot = target + "/" + zeplot  
     ###
-    if found_lct:     makeplotpng(zeplot,pad_inches_value=0.35,disp=display)   
-    #if found_lct:     makeplotpngres(zeplot,200,disp=display)
+    if found_lct:     
+        pad_inches_value = 0.35
+        if save == 'png': 
+            makeplotres(zeplot,res=100.,pad_inches_value=pad_inches_value,erase=True)  ## a miniature
+            makeplotres(zeplot,res=200.,pad_inches_value=pad_inches_value,disp=False)
+        elif save in ['eps','svg','pdf']:
+            makeplotres(zeplot,         pad_inches_value=pad_inches_value,disp=False,ext=save)
+        elif save == 'gui':
+            show()
+        else: 
+            print "save mode not supported. using gui instead."
+            show()
     else:             print "Local time not found"
 
@@ -210 +231 @@
     from netCDF4 import Dataset
     from myplot import getlschar
+    from os import system
 
     #############################
@@ -233 +255 @@
     parser.add_option('-e', action='store',dest='itstep',       type="int",     default=None,  help='stride time (def=4)')
     parser.add_option('-H', action='store_true',dest='hole',                    default=False, help='holes above max and below min')
+    parser.add_option('-S', action='store',dest='save',         type="string",  default="gui", help='save mode (png,eps,svg,pdf or gui)(def=gui)')
+    parser.add_option('-a', action='store_true',dest='anomaly',                 default=False, help='compute and plot relative anomaly in %')
     #parser.add_option('-V', action='store', dest='comb',        type="float",   default=None,  help='a defined combination of variables')
     (opt,args) = parser.parse_args()
@@ -275 +299 @@
             zefile = api_onelevel (  path_to_input   = '', \
                                      input_name      = zefile, \
-                                     #path_to_output  = opt.target, \
                                      fields          = zefields, \
                                      interp_method   = opt.interp, \
@@ -303 +326 @@
                 addchar=lschar,interv=[zehour,zehourin],vmin=argvmin,vmax=argvmax,\
                 tile=opt.tile,zoom=opt.zoom,display=opt.display,\
-                itstep=opt.itstep,hole=opt.hole)
+                itstep=opt.itstep,hole=opt.hole,save=opt.save,\
+                anomaly=opt.anomaly)
             print 'Done: '+name
-    
+            system("rm -f to_be_erased")
+  
         #########################################################
         ### Generate a .sh file with the used command saved in it