Changeset 182 for trunk/MESOSCALE/PLOT/PYTHON/scripts

Timestamp:

Jun 30, 2011, 12:47:42 PM (14 years ago)

Author:

aslmd

Message:

MESOSCALE: PYTHON: a much improved winds.py script. try it out with full options: winds.py -f /home/aslmd/MESOWATERCYCLE/THARSIS_noBUG.171021/wrfout_d01_2024-05-02_01\:00\:00 -v QH2O -l 10 -t ~ -p ortho -s 7 -b vishires

File:

• trunk/MESOSCALE/PLOT/PYTHON/scripts/winds.py (modified) (3 diffs)

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

@@ -1 +1 @@
 #!/usr/bin/env python
 
-### A. Spiga LMD 29/05/2011
+### A. Spiga and A. Colaitis -- LMD -- 30/05/2011
 
-def usage():
-    print 'USAGE:  winds.py nvert file (target)'
-    print 'file  : name of input netcdf file.'
-    print 'nvert : vertical level.'
-    print '(target) : a directory with write rights.'
-    print 'Example: winds.py 0 /d5/aslmd/LMD_MM_MARS_SIMUS/OM/OM6_TI85/wrfout_d01_2024-06-43_06:00:00_zabg ~/'
-
-def winds (namefile,nvert,proj="cyl",back=None,target=None): 
+def winds (namefile,\
+           nvert,\
+           proj="cyl",\
+           back=None,\
+           target=None,
+           stride=5,\
+           var='HGT'): 
     from netCDF4 import Dataset
     from myplot import getcoord2d,define_proj,makeplotpng,simplinterv,vectorfield,ptitle
     from matplotlib.pyplot import contourf, subplot, figure
+    import numpy as np
     nc  = Dataset(namefile)
     [lon2d,lat2d] = getcoord2d(nc)
@@ -19 +19 @@
     [u,v] = getwinds(nc)
     nt = len(u[:,0,0,0])
+    if var not in nc.variables: 
+        print "not found in file:",var
+        exit()
+    else:   
+        dimension = np.array(nc.variables[var]).ndim
+        if dimension == 3:     what_I_plot = nc.variables[var][:,:,:]
+        elif dimension == 4:   what_I_plot = nc.variables[var][:,nvert,:,:]  
     fig = figure()
     fig.subplots_adjust(wspace = 0.0, hspace = 0.3)
@@ -27 +34 @@
        m = define_proj(proj,wlon,wlat,back=back)
        x, y = m(lon2d, lat2d)
-       contourf(x, y, nc.variables['HGT'][0,:,:] / 1000., 30)
+       contourf(x, y, what_I_plot[i,:,:], 30)
        vectorfield(u[i,nvert,:,:], v[i,nvert,:,:],\
-                      x, y, stride=5, csmooth=3,\
+                      x, y, stride=stride, csmooth=2,\
                       scale=20., factor=300., color='k')
        sub += 1
-    if not target:   zeplot = namefile+".winds"+str(nvert)
-    else:            zeplot = target+"winds"+str(nvert)
+    if not target:   zeplot = namefile+".winds"+var+str(nvert)
+    else:            zeplot = target+"/winds"+var+str(nvert)
     makeplotpng(zeplot,pad_inches_value=0.35)   
 
-def getwinds (nc,charu='Um',charv='Vm'):
+def getwinds (nc,charu='U',charv='V'):
+    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, :]
+    #print np.array(u).shape, np.array(v).shape
     return u,v
 
 if __name__ == "__main__":
     import sys
-    if (len(sys.argv)) == 3:     winds( str(sys.argv[2]) , int(sys.argv[1]) )
-    elif (len(sys.argv)) == 4:   winds( str(sys.argv[2]) , int(sys.argv[1]) , target=str(sys.argv[3]))
+    ### to be replaced by argparse
+    from optparse import OptionParser
+    parser = OptionParser()
+    parser.add_option('-f', action='store', dest='namefile',    type="string",  default=None,  help='name of WRF file [NEEDED]')
+    parser.add_option('-l', action='store', dest='nvert',       type="int",     default=0,     help='subscript for vertical level')
+    parser.add_option('-p', action='store', dest='proj',        type="string",  default='cyl', help='projection')
+    parser.add_option('-b', action='store', dest='back',        type="string",  default=None,  help='background')
+    parser.add_option('-t', action='store', dest='target',      type="string",  default=None,  help='destination folder')
+    parser.add_option('-s', action='store', dest='stride',      type="int",     default=5,     help='stride vectors')
+    parser.add_option('-v', action='store', dest='var',         type="string",  default='HGT', help='variable contoured')
+    (opt,args) = parser.parse_args()
+    if opt.namefile is None: 
+        print "I want to eat one file at least ! Use winds.py -f name_of_my_file. Or type winds.py -h"
+        exit()
+    print "Options:", opt
+    winds (opt.namefile,opt.nvert,proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=opt.var)