Changeset 562 for trunk/UTIL

Timestamp:

Mar 6, 2012, 1:49:07 AM (13 years ago)

Author:

aslmd

Message:

UTIL PYTHON added -i -1 i.e. an option to use model-level geopotential height without the need to call api [useful for big files such as LES results where api cannot handle it]. introduced automatic correction in define_axis. other changes are purely cosmetic.

Location:

trunk/UTIL/PYTHON

Files:

• myplot.py (modified) (3 diffs)
• myscript.py (modified) (1 diff)
• planetoplot.py (modified) (5 diffs)
• pp.py (modified) (2 diffs)

trunk/UTIL/PYTHON/myplot.py

@@ -684 +684 @@
     dev = np.std(fieldcalc)*3.0
     ###
-    if vmin is None:
-        zevmin = mean(fieldcalc) - dev
+    if vmin is None:  zevmin = mean(fieldcalc) - dev
     else:             zevmin = vmin
     ###
@@ -778 +777 @@
              "W":            "%.1f",\
              "WMAX_TH":      "%.1f",\
+             "WSTAR":        "%.1f",\
              "QSURFICE":     "%.0f",\
              "UM":           "%.0f",\
@@ -1080 +1080 @@
    print "CHECK SHAPE: what_I_plot, x, y", what_I_plot.shape, x.shape, y.shape
    if len(shape) == 1:
-      if shape[0] != len(x):
-         print "WARNING HERE !!!"
-         x = y
+       if shape[0] != len(x):           print "WARNING: shape[0] != len(x). Correcting." ; what_I_plot = what_I_plot[0:len(x)]
    elif len(shape) == 2:
-      if shape[1] == len(y) and shape[0] == len(x) and shape[0] != shape[1]:
-         what_I_plot = swapaxes(what_I_plot,0,1)
-         print "INFO: swapaxes", what_I_plot.shape, shape
+       if shape[1] == len(y) and shape[0] == len(x) and shape[0] != shape[1]:
+           print "INFO: swapaxes: ",what_I_plot.shape,shape ; what_I_plot = swapaxes(what_I_plot,0,1)
+       else:
+           if shape[0] != len(y):       print "WARNING: shape[0] != len(y). Correcting." ; what_I_plot = what_I_plot[0:len(y),:]
+           elif shape[1] != len(x):     print "WARNING: shape[1] != len(x). Correcting." ; what_I_plot = what_I_plot[:,0:len(x)]
+   elif len(shape) == 3:
+       if vertmode < 0: print "not supported. must check array dimensions at some point. not difficult to implement though."
    return what_I_plot,x,y
 

trunk/UTIL/PYTHON/myscript.py

@@ -24 +24 @@
     ### VERTICAL INTERPOLATION
     parser.add_option('-l', '--level',  action='store',dest='lvl',       type="string",  default="0",   help='level / start,stop,step (-i 2: p,Pa)(-i 3,4: z,km) [0]')
-    parser.add_option('-i', '--interp', action='store',dest='itp',       type="int",     default=None,  help='interpolation (2: p, 3: z-amr, 4:z-als)')
+    parser.add_option('-i', '--interp', action='store',dest='itp',       type="int",     default=None,  help='interpolation (2: p, 3: z-amr, 4:z-als, -1)')
     parser.add_option('--intas',        action='store',dest='intas',     type="string",  default=None,  help='specify "mcs" or "tes" for gcm P interpolation grid')
     parser.add_option('-N', '--no-api', action='store_true',dest='nocall',               default=False, help='do not recreate api file')

trunk/UTIL/PYTHON/planetoplot.py

@@ -196 +196 @@
           ###### principle: calculate correct indices then repopulate slon and slat
           if slon is not None or slat is not None:
-              if firstfile and save == 'png' and typefile == 'meso':   iwantawhereplot = nc     #show a topo map with a cross on the chosen point
-              else:                                                    iwantawhereplot = None   #do not show anything, just select indices
+              if firstfile and save == 'png' and typefile == 'meso' and "HGT" in varinfile:   iwantawhereplot = nc     #show a topo map with a cross on the chosen point
+              else:                                                                           iwantawhereplot = None   #do not show anything, just select indices
               numlon = 1 ; numlat = 1 
               if slon is not None:   numlon = slon.shape[0]    
@@ -245 +245 @@
               if vertmode is None:  vertmode=0
               if vertmode == 0:     vert = np.arange(0,getattr(nc,vertdim),1)
+              elif vertmode == -1:  vert = nc.variables["PHTOT"][0,:,0,0]/3.72 ; vert = np.array(vert[0:len(vert)-1]) #; print vert
               else:                 vert = nc.variables["vert"][:]
        #if firstfile:
@@ -398 +399 @@
                if mapmode == 0:    ### could this be moved inside imov loop ?
                    itime=indextime
-                   if len(what_I_plot.shape) is 3:itime=[0]
+                   if len(what_I_plot.shape) is 3: itime=[0]
                    m = None ; x = None ; y = None
                    what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
@@ -431 +432 @@
                  elif len(what_I_plot.shape) == 1:
                     which = "unidim"
+                    if what_I_plot.shape[-1] == 1:      print "VALUE VALUE VALUE VALUE ::: ", what_I_plot[0] ; save = 'donothing'
                  ##### IMOV LOOP #### IMOV LOOP
                  while imov <= iend:
@@ -508 +510 @@
                                                               #200.         ## or csmooth=stride
                     elif which == "contour":
-                        zevminc, zevmaxc = calculate_bounds(what_I_plot_frame)
+                        zevminc, zevmaxc = calculate_bounds(what_I_plot_frame, vmin=min(what_I_plot_frame), vmax=max(what_I_plot_frame))
                         zelevels = np.linspace(zevminc,zevmaxc,ticks/2) #20)
                         if var2 == 'HGT': zelevels = np.arange(-10000.,30000.,2000.)

trunk/UTIL/PYTHON/pp.py

@@ -53 +53 @@
 
       typefile = whatkindfile(Dataset(zefiles[0])) ; stralt = None
-      if typefile in ["meso","mesoapi","mesoideal"]:          
+      if typefile in ["meso"]:          
           [lschar,zehour,zehourin] = getlschar ( zefiles[0] )
           if opt.var is None:  opt.var = ["HGT"] ; opt.clb = "nobar"
@@ -90 +90 @@
         ######################################################### 
         if opt.itp is not None:
+         if opt.itp > 0:
           #####
           ##### MESOSCALE : written by AS
           #####
-          if typefile in ["meso","mesoideal"]:
+          if typefile in ["meso"]:
             if zelevel == 0. and opt.itp == 4:  zelevel = 0.010
             ### winds or no winds