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

Timestamp:

Nov 28, 2011, 10:59:33 AM (13 years ago)

Author:

aslmd

Message:

LMDZ.MARS: corrected wrong datafile default link. MESOSCALE: added 4 to 5 nests files. GRAPHICS: added geo files handling. LMDZ.GENERIC: replaced makegcm by a symboli link to makegcm_whateverversion.

File:

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

trunk/UTIL/PYTHON/planetoplot.py

@@ -139 +139 @@
           else:                           errormess("no time axis found.")
           vert = nc.variables["altitude"][:]
-      elif typefile in ['meso','mesoapi']:
+      elif typefile in ['meso','mesoapi','geo']:
           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]
@@ -150 +150 @@
           else:                 londim='WEST-EAST_PATCH_END_UNSTAG'
           lon = np.arange(0,getattr(nc,londim),1) ; lat = np.arange(0,getattr(nc,latdim),1)
-          if vertmode is None:  vertmode=0
-          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 typefile in ['geo']:   vert = [0.] ; stime = readslices(str(0))
+          else:
+              if vertmode is None:  vertmode=0
+              if vertmode == 0:     vert = np.arange(0,getattr(nc,vertdim),1)
+              else:                 vert = nc.variables["vert"][:]
        #if firstfile:
        #   lat0 = lat
@@ -230 +232 @@
     print "********************************************"
     while error is False:
-       print "********** nplot", nplot, "error",error
+       print "********** NPLOT", nplot
       
        ### General plot settings
@@ -247 +249 @@
        indexlat  = getsindex(slat,((nplot-1)//nlon)%nlat,lat)
        indexvert = getsindex(svert,((nplot-1)//(nlon*nlat))%nvert,vert) 
-
        if ope is not None:
            if fileref is not None:      index_f = ((nplot-1)//(nlon*nlat*nvert*ntime))%(len(namefiles)+2)  ## OK only 1 var,  see test in the beginning
@@ -261 +262 @@
        if mapmode == 1 and stime is None:   indextime = 1
        else:                                indextime = getsindex(stime,((nplot-1)//(nlon*nlat*nvert))%ntime,time)
+       ltst = None 
        if typefile in ['mesoapi','meso'] and indextime is not None:  ltst = localtime ( interv[0]+indextime*interv[1], 0.5*(wlon[0]+wlon[1]) )  
        print "********** index lon, lat, vert, time ",indexlon,indexlat,indexvert,indextime
@@ -329 +331 @@
                      if mapmode == 1:       m.pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
                      elif mapmode == 0:     pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
-                 if colorb != 'nobar' and varname != 'HGT' :        
-                     if (fileref is not None) and (index_f is numplot-1):
-                        colorbar(fraction=0.05,pad=0.03,format="%.3f",\
-                                           ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),\
-                                           extend='neither',spacing='proportional')
-                     else:
-                        colorbar(fraction=0.05,pad=0.03,format=fmtvar(fvar.upper()),\
-                                           ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),\
-                                           extend='neither',spacing='proportional')
-                                           # both min max neither
+                 if colorb != 'nobar':        
+                     if (fileref is not None) and (index_f is numplot-1):   daformat = "%.3f"
+                     else:                                                  daformat = fmtvar(fvar.upper())
+                     colorbar( fraction=0.05,pad=0.03,format=daformat,\
+                               ticks=np.linspace(zevmin,zevmax,num=min([ticks/2+1,20])),extend='neither',spacing='proportional' ) 
                ##### 1D field
                elif len(what_I_plot.shape) is 1:
@@ -346 +343 @@
                #### Other cases: (maybe plot 3-D field one day or movie ??)
                else:
-                 print "WARNING!!! ",len(what_I_plot.shape),"-D PLOT NOT SUPPORTED !!!"
-                 print "field dimensions: ", what_I_plot.shape
-                 exit()
-               
+                 print "WARNING!!! ",len(what_I_plot.shape),"-D PLOT NOT SUPPORTED !!! dimensions: ",what_I_plot.shape
+                 errormess("Are you sure you did not forget to prescribe a dimension ?")
            else:
                errormess("There is an error in reducing field !")
@@ -382 +377 @@
             if addchar:  
                 [addchar,gogol,gogol2] = getlschar ( all_namefile[index_f] )
-                plottitle = plottitle + addchar + "_LT"+str(ltst)
-            else:        plottitle = plottitle + "_LT"+str(ltst)
+                plottitle = plottitle + addchar + "_LT"
+            else:        plottitle = plottitle + "_LT"
+            if ltst is not None: plottitle = plottitle + str(ltst)
        else:
             if fileref is not None: