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

Timestamp:

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

Author:

aslmd

Message:

MESOSCALE: A GENERAL CLEAN-UP FOLLOWING UPDATING THE USER MANUAL. EVERYTHING ESSENTIAL IS IN MESOSCALE (much lighter than before). EVERYTHING FOR DEVELOPPERS OR EXPERTS IS IN MESOSCALE_DEV.

Location:

trunk/MESOSCALE_DEV

Files:

• . (added)
• PLOT (copied) (copied from trunk/MESOSCALE/PLOT)
• PLOT/PYTHON/scripts/winds.py (copied) (copied from trunk/MESOSCALE/PLOT/PYTHON/scripts/winds.py) (8 diffs)

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

@@ -71 +71 @@
         if not var:                                        back = "mola"    ## if no var:         draw mola
         elif typefile in ['mesoapi','meso'] \
-             and proj not in ['merc','lcc']:               back = "molabw"  ## if var but meso:   draw molabw
+             and proj not in ['merc','lcc','nsper','laea']:       back = "molabw"  ## if var but meso:   draw molabw
         else:                                              pass             ## else:              draw None
 
@@ -82 +82 @@
     elif typefile in ['gcm']:
         [lon2d,lat2d] = getcoord2d(nc,nlat="latitude",nlon="longitude",is1d=True)
-    if proj == "npstere":    [wlon,wlat] = latinterv("North_Pole")
-    elif proj == "lcc":      [wlon,wlat] = wrfinterv(lon2d,lat2d)
-    else:                    [wlon,wlat] = simplinterv(lon2d,lat2d)
+    if proj == "npstere":             [wlon,wlat] = latinterv("North_Pole")
+    elif proj in ["lcc","laea"]:      [wlon,wlat] = wrfinterv(lon2d,lat2d)
+    else:                             [wlon,wlat] = simplinterv(lon2d,lat2d)
     if zoom:  
         dlon = abs(wlon[1]-wlon[0])/2.
@@ -125 +125 @@
         ### some already defined colormaps
         if colorb is True:    colorb = defcolorb(var)
+    else:
+        dimension = 0
  
     ###########################
@@ -139 +141 @@
     if var:     basename = basename + var
     ###
-    if typefile is 'meso':                      stralt = "_lvl" + str(nvert)
-    elif typefile is 'mesoapi':  
-        zelevel = int(nc.variables['vert'][nvert])
-        if 'altitude'       in nc.dimensions:   stralt = "_"+str(zelevel)+"m-AMR"
-        elif 'altitude_abg' in nc.dimensions:   stralt = "_"+str(zelevel)+"m-ALS"
-        elif 'bottom_top'   in nc.dimensions:   stralt = "_"+str(zelevel)+"m"
-        elif 'pressure'     in nc.dimensions:   stralt = "_"+str(zelevel)+"Pa" 
-    else:                                       stralt = ""          
+    if dimension == 4:
+        if typefile is 'meso':                      stralt = "_lvl" + str(nvert)
+        elif typefile is 'mesoapi':  
+            zelevel = int(nc.variables['vert'][nvert])
+            if 'altitude'       in nc.dimensions:   stralt = "_"+str(zelevel)+"m-AMR"
+            elif 'altitude_abg' in nc.dimensions:   stralt = "_"+str(zelevel)+"m-ALS"
+            elif 'bottom_top'   in nc.dimensions:   stralt = "_"+str(zelevel)+"m"
+            elif 'pressure'     in nc.dimensions:   stralt = "_"+str(zelevel)+"Pa" 
+        else:                                       stralt = ""          
+    else:
+        stralt = ""
     ###
     basename = basename + stralt
@@ -175 +180 @@
             rcParams['font.size'] = int( rcParams['font.size'] * 1. / 2. )
         elif numplot == 9:
-            sub = 331 
+            sub = 331
             fig.subplots_adjust(wspace = 0.3, hspace = 0.3)
             rcParams['font.size'] = int( rcParams['font.size'] * 1. / 2. )
@@ -181 +186 @@
             sub = 99999
         else:
-            print "supported: 1,2,3,4,6,8"
+            print "supported: 1,2,3,4,6,8,9"
             exit()
         ### Prepare time loop
@@ -306 +311 @@
     parser.add_option('-i', action='store', dest='interp',      type="int",     default=None,  help='interpolation method (2: press, 3: z-amr, 4:z-als)')
     parser.add_option('-c', action='store', dest='colorb',      type="string",  default=True,  help='change colormap')
-    parser.add_option('-x', action='store_false', dest='winds',                 default=True,  help='flag: no wind vectors')
+    parser.add_option('-x', action='store_false', dest='winds',                 default=True,  help='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') 
     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('-N', action='store_true', dest='nocall',                 default=False, help='do not recreate api file')
     #parser.add_option('-V', action='store', dest='comb',        type="float",   default=None,  help='a defined combination of variables')
     (opt,args) = parser.parse_args()
@@ -339 +345 @@
                                  fields          = zefields, \
                                  interp_method   = opt.interp, \
-                                 onelevel        = zelevel )
+                                 onelevel        = zelevel, \
+                                 nocall          = opt.nocall )
+        print zefile
         zelevel = 0 ## so that zelevel could play again the role of nvert