Changeset 428

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.

Location:

trunk/UTIL/PYTHON

Files:

• myplot.py (modified) (5 diffs)
• planetoplot.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:

trunk/UTIL/PYTHON/planetoplot.py

@@ -92 +92 @@
     numplot = zelen*nslices
     print "********** FILES, SLICES, VARS, TOTAL PLOTS: ", len(namefiles), nslices, len(var), numplot
-    print "********** MAPMODE: ", mapmode
     if ope is not None:
         if fileref is not None:       zelen = zelen + 2
@@ -116 +115 @@
       ### ... TYPEFILE
       typefile = whatkindfile(nc)                                  
+      if typefile in ['les']:
+          mapmode=0
+          winds=False
+      if firstfile: print "********** MAPMODE: ", mapmode
       if firstfile:                 typefile0 = typefile
       elif typefile != typefile0:   errormess("Not the same kind of files !", [typefile0, typefile])
@@ -130 +133 @@
       [lon2d,lat2d] = getcoorddef(nc)
       ### ... PROJECTION
-      if proj == None:   proj = getproj(nc)                 
+      if ((proj == None) and (typefile not in ['les'])):   proj = getproj(nc)                 
 
 ##########################################################
@@ -139 +142 @@
           else:                           errormess("no time axis found.")
           vert = nc.variables["altitude"][:]
-      elif typefile in ['meso','mesoapi','geo']:
+      elif typefile in ['meso','mesoapi','geo','les']:
           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]
@@ -329 +332 @@
                      #zelevels = np.linspace(zevmin*(1. + 1.e-7),zevmax*(1. - 1.e-7)) #,num=20)
                      zelevels = np.linspace(zevmin,zevmax,num=ticks)
+                     print np.array(x).shape
+                     print np.array(y).shape
+                     print np.array(what_I_plot).shape
+
                      if mapmode == 1:       m.contourf( x, y, what_I_plot, zelevels, cmap = palette)
                      elif mapmode == 0:     contourf( x, y, what_I_plot, zelevels, cmap = palette)