Changeset 428 for trunk/UTIL/PYTHON/myplot.py

Timestamp:

Nov 30, 2011, 11:22:27 AM (13 years ago)

Author:

acolaitis

Message:

PYTHON. Very simple modifications to allow for slices with LES files. Performance is not optimized and turbulence computations are not implemented. More to come in time.

File:

• trunk/UTIL/PYTHON/myplot.py (modified) (5 diffs)

trunk/UTIL/PYTHON/myplot.py

@@ -35 +35 @@
     return ltst
 
-## Author: AS
+## Author: AS, AC
 def whatkindfile (nc):
-    if 'controle' in nc.variables:   typefile = 'gcm'
-    elif 'phisinit' in nc.variables: typefile = 'gcm' 
-    elif 'vert' in nc.variables:     typefile = 'mesoapi'
-    elif 'U' in nc.variables:        typefile = 'meso'
-    elif 'HGT_M' in nc.variables:    typefile = 'geo'
+    if 'controle' in nc.variables:     typefile = 'gcm'
+    elif 'phisinit' in nc.variables:   typefile = 'gcm'
+    elif '9999' in getattr(nc,'START_DATE') : typefile = 'les'
+    elif 'vert' in nc.variables:       typefile = 'mesoapi'
+    elif 'U' in nc.variables:          typefile = 'meso'
+    elif 'HGT_M' in nc.variables:      typefile = 'geo'
     #else:                            errormess("whatkindfile: typefile not supported.")
     else:                            typefile = 'gcm' # for lslin-ed files from gcm
@@ -394 +395 @@
     elif typefile in ['geo']:
         [lon2d,lat2d] = getcoord2d(nc,nlat='XLAT_M',nlon='XLONG_M')
+    elif typefile in ['les']:
+        nx=getattr(nc,'WEST-EAST_GRID_DIMENSION')
+        ny=getattr(nc,'SOUTH-NORTH_GRID_DIMENSION')
+        [lon2d,lat2d] = np.meshgrid(np.arange(nx),np.arange(ny))
     return lon2d,lat2d    
 
@@ -487 +492 @@
     if typefile is 'mesoapi':    [uchar,vchar] = ['Um','Vm']
     elif typefile is 'gcm':      [uchar,vchar] = ['u','v']
-    elif typefile is 'meso':     [uchar,vchar] = ['U','V']
+    elif typefile in ['meso','les']:     [uchar,vchar] = ['U','V']
     else:                        [uchar,vchar] = ['not found','not found']
     ###
@@ -1001 +1006 @@
    return what_I_plot,x,y
 
-# Author: TN + AS
+# Author: TN + AS 
 def determineplot(slon, slat, svert, stime):
     nlon = 1 # number of longitudinal slices -- 1 is None
@@ -1022 +1027 @@
     #else:
     #    nslices = 2  
-
     mapmode = 0
     if slon is None and slat is None: