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

Timestamp:

Mar 31, 2012, 7:33:41 PM (13 years ago)

Author:

aslmd

Message:

UTIL PYTHON : for mesoscale plots 1. improved the case of pressure interpolation -i 1 or 2 ; 2. fixed the mean capabilities e.g. --lat 15.,35. now working fine the problem was the wrong dimension being considered in slon, my bad.

File:

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

trunk/UTIL/PYTHON/planetoplot.py

@@ -173 +173 @@
 ##########################################################
       if typefile in ["gcm","earthgcm","ecmwf"]:
-          if slon is not None: sslon = slon  ## a fix of the sslon not initialized. but what is the use of sslon?
-          if slat is not None: sslat = slat  ## a fix of the sslat not initialized. but what is the use of sslat?
+          if slon is not None: sslon = slon  
+          if slat is not None: sslat = slat  
           ### LAT
           if "latitude" in nc.variables:  lat = nc.variables["latitude"][:]
@@ -208 +208 @@
               else:                                                                           iwantawhereplot = None   #do not show anything, just select indices
               numlon = 1 ; numlat = 1 
-              if slon is not None:   numlon = slon.shape[0]    
-              if slat is not None:   numlat = slat.shape[0]
+              if slon is not None:   numlon = slon.shape[1]    
+              if slat is not None:   numlat = slat.shape[1]
               indices = np.ones([numlon,numlat,2]) ; vlon = None ; vlat = None
               for iii in range(numlon):  
                for jjj in range(numlat):
-                 if slon is not None:  vlon = slon[iii][0]  ### note: slon[:][0] does not work
-                 if slat is not None:  vlat = slat[jjj][0]  ### note: slon[:][0] does not work
+                 if slon is not None:  vlon = slon[0][iii]  ### note: slon[:][0] does not work
+                 if slat is not None:  vlat = slat[0][jjj]  ### note: slon[:][0] does not work
                  indices[iii,jjj,:] = bidimfind(lon2d,lat2d,vlon,vlat,file=iwantawhereplot)  
                  lonp,latp = ( lon2d[indices[iii,jjj,0],indices[iii,jjj,1]] , lat2d[indices[iii,jjj,0],indices[iii,jjj,1]] )
@@ -220 +220 @@
               for iii in range(numlon):
                for jjj in range(numlat):
-                 if slon is not None: sslon[iii][0] = indices[iii,0,1] ; sslon[iii][1] = indices[iii,0,1]  #...this is idx
-                 if slat is not None: sslat[jjj][0] = indices[0,jjj,0] ; sslat[jjj][1] = indices[0,jjj,0]  #...this is idy
+                 if slon is not None: sslon[0][iii] = indices[iii,0,1] #...this is idx
+                 if slat is not None: sslat[0][jjj] = indices[0,jjj,0] #...this is idy
               lonp,latp = ( lon2d[indices[0,0,0],indices[0,0,1]] , lat2d[indices[0,0,0],indices[0,0,1]] )
           ######
@@ -262 +262 @@
               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"][:]
+              elif vertmode == 1 or vertmode == 2:  vert = nc.variables["vert"][:]        ## pressure in Pa
+              else:                                 vert = nc.variables["vert"][:]/1000.  ## altitude in km
        #if firstfile:
        #   lat0 = lat