Changeset 399 for trunk/UTIL/PYTHON/planetoplot.py

Timestamp:

Nov 18, 2011, 8:43:59 PM (13 years ago)

Author:

aslmd

Message:

GRAPHICS: improved the selection of lon,lat with pp.py for mesoscale files

File:

• trunk/UTIL/PYTHON/planetoplot.py (modified) (11 diffs)

trunk/UTIL/PYTHON/planetoplot.py

@@ -18 +18 @@
            var=None,\
            colorb="def",\
-           winds=True,\
+           winds=False,\
            addchar=None,\
            interv=[0,1],\
@@ -64 +64 @@
                        fmtvar,definecolorvec,defcolorb,getprefix,putpoints,calculate_bounds,errormess,definesubplot,\
                        zoomset,getcoorddef,getwinddef,whatkindfile,reducefield,bounds,getstralt,getfield,smooth,nolow,\
-                       getname,localtime,polarinterv,getsindex,define_axis,determineplot,readslices
+                       getname,localtime,polarinterv,getsindex,define_axis,determineplot,readslices,bidimfind,getlschar
     from mymath import deg,max,min,mean,get_tsat
     import matplotlib as mpl
@@ -86 +86 @@
     ################################
     nlon, nlat, nvert, ntime, mapmode, nslices = determineplot(slon, slat, svert, stime)
+    vlon = None ; vlat = None
+    if slon is not None: vlon = slon[0][0]
+    if slat is not None: vlat = slat[0][0]
     if mapmode == 0:       winds=False
     elif mapmode == 1:     
@@ -93 +96 @@
     print "********** FILES, SLICES, VARS, TOTAL PLOTS: ", len(namefiles), nslices, len(var), numplot
     print "********** MAPMODE: ", mapmode
-    if fileref is not None:       all_var  = [[]]*(zelen+2) ;  all_varname = [[]]*(zelen+2)
-    else:                         all_var  = [[]]*zelen     ;  all_var2  = [[]]*zelen ;  all_title = [[]]*zelen ;  all_varname = [[]]*zelen
+    if fileref is not None:       all_var  = [[]]*(zelen+2) ;  all_varname = [[]]*(zelen+2)  ;  all_namefile = [[]]*(zelen+2) 
+    else:                         all_var  = [[]]*zelen     ;  all_var2  = [[]]*zelen ;  all_title = [[]]*zelen ;  all_varname = [[]]*zelen ; all_namefile = [[]]*zelen
  
     #################################################################################################
     ### Loop over the files + vars initially separated by commas to be plotted on the same figure ###
     #################################################################################################
-    k = 0
-    firstfile = True
+    k = 0 ; firstfile = True
     for nnn in range(len(namefiles)):
      for vvv in range(len(var)): 
@@ -127 +129 @@
       if not varname and not winds: errormess("please set at least winds or var",printvar=nc.variables)
       ### ... COORDINATES, could be moved below
-      [lon2d,lat2d] = getcoorddef(nc)                       
+      [lon2d,lat2d] = getcoorddef(nc)
       ### ... PROJECTION
       if proj == None:   proj = getproj(nc)                 
@@ -144 +146 @@
           vert = nc.variables["altitude"][:]
       elif typefile in ['meso','mesoapi']:
+          if vlon is not None or vlat is not None:   indices = bidimfind(lon2d,lat2d,vlon,vlat) ; print '********** INDICES: ', indices
+          if slon is not None: slon[0][0] = indices[0] ; slon[0][1] = indices[0]
+          if slat is not None: slat[0][0] = indices[1] ; slat[0][1] = indices[1]
           if varname in ['PHTOT','W']:    vertdim='BOTTOM-TOP_PATCH_END_STAG'
           else:                           vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
@@ -173 +178 @@
          else:                             [wlon,wlat] = simplinterv(lon2d,lat2d)
          if zoom:                          [wlon,wlat] = zoomset(wlon,wlat,zoom) 
-         ### ... NAME FOR TITLE and FILE (have to do something for multiple vars...)
-         basename = getname(var=varname,winds=winds,anomaly=anomaly)
-         basename = basename + getstralt(nc,level)  ## can be moved elsewhere for a more generic routine
 
       ##### SPECIFIC
@@ -183 +185 @@
           else:                                 tinput=tt
           all_var[k]=get_tsat(vert,tinput,zlon=lon,zlat=lat,zalt=vert,ztime=time)
-          all_varname[k] = varname
       else:
       ##### GENERAL STUFF HERE
           all_var[k] = getfield(nc,varname)
-          all_varname[k] = varname
+      all_varname[k] = varname
+      all_namefile[k] = namefile
       if var2: all_var2[k] = getfield(nc,var2)
       
@@ -208 +210 @@
                 numplot = numplot+2
              elif ope in ["cat"]:
-                tab = all_var[0];k = 0
+                tab = all_var[0];k = 1
                 while k != len(namefiles)-1: 
                     tab = np.append(tab,all_var[k],axis=0)
@@ -276 +278 @@
        #print slon, slat, svert, stime       
        #print index_f
-       #print "index lon,lat,vert,time", max(indexlon), max(indexlat), max(indexvert), max(indextime)
+       #print "index lon,lat,vert,time", indexlon, indexlat, indexvert, indextime
        #if varname != all_varname[index_f]:  indextime = first
 ####################################################################
@@ -401 +403 @@
        ### Next subplot
        basename = getname(var=varname,winds=winds,anomaly=anomaly)
+       basename = basename + getstralt(nc,level)  
        if typefile in ['mesoapi','meso']:
+            if slon is not None: basename = basename + "_lon_" + str(int(lon2d[indices[1],indices[0]]))
+            if slat is not None: basename = basename + "_lat_" + str(int(lat2d[indices[1],indices[0]]))
             plottitle = basename
-            if addchar:  plottitle = plottitle + addchar + "_LT"+str(ltst)
+            if addchar:  
+                [addchar,gogol,gogol2] = getlschar ( all_namefile[index_f] )
+                plottitle = plottitle + addchar + "_LT"+str(ltst)
             else:        plottitle = plottitle + "_LT"+str(ltst)
        else: