Changeset 351 for trunk/MESOSCALE/LMD_MM_MARS/SRC/PYTHON/planetoplot.py

Timestamp:

Nov 7, 2011, 12:53:40 AM (13 years ago)

Author:

aslmd

Message:

PYTHON GRAPHICS: for mesoscale files section with vertically interpolated fields with alitude or pressure (through api wrapper) is now supported. the same thing might be useful to be done with zrecast for GCM.

File:

• trunk/MESOSCALE/LMD_MM_MARS/SRC/PYTHON/planetoplot.py (modified) (6 diffs)

trunk/MESOSCALE/LMD_MM_MARS/SRC/PYTHON/planetoplot.py

@@ -7 +7 @@
 
 def planetoplot (namefiles,\
-           nvert=0,\
+           level=0,\
            vertmode=0,\
            proj=None,\
@@ -75 +75 @@
       ### Load NETCDF object
       nc  = Dataset(namefile)
-      
+
       ##################################
       ### Initial checks and definitions
@@ -104 +104 @@
       elif typefile in ['meso','mesoapi']:
           if mapmode == 0:
-              if var in ['PHTOT','W']:  vertdim='BOTTOM-TOP_PATCH_END_STAG'
-              else:                     vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
+              if var in ['PHTOT','W']:    vertdim='BOTTOM-TOP_PATCH_END_STAG'
+              else:                       vertdim='BOTTOM-TOP_PATCH_END_UNSTAG'
               if var in ['V']:  latdim='SOUTH-NORTH_PATCH_END_STAG'
               else:             latdim='SOUTH-NORTH_PATCH_END_UNSTAG'
@@ -112 +112 @@
               lon = np.arange(0,getattr(nc,londim),1)
               lat = np.arange(0,getattr(nc,latdim),1)
-              vert = np.arange(0,getattr(nc,vertdim),1)
+              if vertmode == 0:   vert = np.arange(0,getattr(nc,vertdim),1)
+              else:               vert = nc.variables["vert"][:]
               time = np.arange(0,len(nc.variables["Times"]),1)
        #if firstfile:
@@ -135 +136 @@
          ### Name for title and graphics save file
          basename = getname(var=var,winds=winds,anomaly=anomaly)
-         basename = basename + getstralt(nc,nvert)  ## can be moved elsewhere for a more generic routine
+         basename = basename + getstralt(nc,level)  ## can be moved elsewhere for a more generic routine
       
       print "var, var2: ", var, var2
@@ -196 +197 @@
                indexlon = None
                indexlat = None
-               indexvert = nvert  ## ou svert ??? 
+               indexvert = level  ## ou svert ??? 
                nlon = 1
                nlat = 1