Changeset 232 for trunk/MESOSCALE_DEV/PLOT

Timestamp:

Jul 16, 2011, 1:06:45 AM (13 years ago)

Author:

aslmd

Message:

MESOSCALE: corrections petits bugs graphiques.

Location:

trunk/MESOSCALE_DEV/PLOT/PYTHON

Files:

• mylib/api_ifort.sh (added)
• mylib/myplot.py (modified) (5 diffs)
• scripts/winds.py (modified) (5 diffs)

trunk/MESOSCALE_DEV/PLOT/PYTHON/mylib/myplot.py

@@ -35 +35 @@
         return wlon,wlat
 
-#def landers (map) 
-#    map.plot(blue_calf_lon,blue_calf_lat, 'gs')
-#    return
-
 def getlschar ( namefile ):
     #### strangely enough this does not work for api or ncrcat results!
@@ -256 +252 @@
     return m
 
+#### test temporaire
+def putpoints (map,plot):
+    #### from http://www.scipy.org/Cookbook/Matplotlib/Maps
+    # lat/lon coordinates of five cities.
+    lats = [18.4]
+    lons = [-134.0]
+    points=['Olympus Mons']
+    # compute the native map projection coordinates for cities.
+    x,y = map(lons,lats)
+    # plot filled circles at the locations of the cities.
+    map.plot(x,y,'bo')
+    # plot the names of those five cities.
+    wherept = 0 #1000 #50000
+    for name,xpt,ypt in zip(points,x,y):
+       plot.text(xpt+wherept,ypt+wherept,name)
+    ## le nom ne s'affiche pas...
+    return
+
+
 def fmtvar (whichvar="def"):
     fmtvar    =     { \
@@ -265 +280 @@
              "USTM":         "%.2f",\
              "HFX":          "%.0f",\
+             "ICETOT":       "%.1e",\
                     }
     if whichvar not in fmtvar:
@@ -278 +294 @@
              "USTM":         "YlOrRd",\
              "HFX":          "RdYlBu",\
+             "ICETOT":       "YlGnBu",\
                      }
     if whichone not in whichcolorb:
@@ -305 +322 @@
                 "vis":          "http://maps.jpl.nasa.gov/pix/mar0kuu2.jpg",\
                 "vishires":     "http://dl.dropbox.com/u/11078310/MarsMap_2500x1250.jpg",\
+                "geolocal":     "http://dl.dropbox.com/u/11078310/geolocal.jpg",\
                 "mola":         "http://www.lns.cornell.edu/~seb/celestia/mars-mola-2k.jpg",\
                 "molabw":       "http://dl.dropbox.com/u/11078310/MarsElevation_2500x1250.jpg",\

trunk/MESOSCALE_DEV/PLOT/PYTHON/scripts/winds.py

@@ -32 +32 @@
     from netCDF4 import Dataset
     from myplot import getcoord2d,define_proj,makeplotpng,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy,\
-                       fmtvar,definecolorvec,getwinds,defcolorb,getprefix
+                       fmtvar,definecolorvec,getwinds,defcolorb,getprefix,putpoints
     from mymath import deg,max,min,mean
     from matplotlib.pyplot import contour,contourf, subplot, figure, rcParams, savefig, colorbar, pcolor
@@ -124 +124 @@
             elif dimension == 3:   field = nc.variables[var][:,:,:]
             elif dimension == 4:   field = nc.variables[var][:,nvert,:,:]  
-        dev = np.std(field)*3.0
-        if vmin is None:  zevmin = mean(field) - dev
+        fieldcalc = field[ field < 9e+35 ]
+        dev = np.std(fieldcalc)*2.0
+        if vmin is None:  zevmin = mean(fieldcalc) - dev
         else:             zevmin = vmin
-        if vmax is None:  zevmax = mean(field) + dev
+        if vmax is None:  zevmax = mean(fieldcalc) + dev
         else:             zevmax = vmax
         if vmin == vmax:   
-                          zevmin = mean(field) - dev  ### for continuity
-                          zevmax = mean(field) + dev  ### for continuity            
-        print "field ", min(field), max(field)
+                          zevmin = mean(fieldcalc) - dev  ### for continuity
+                          zevmax = mean(fieldcalc) + dev  ### for continuity            
+        print "field ", min(fieldcalc), max(fieldcalc)
         print "bounds ", zevmin, zevmax
         ### some already defined colormaps
@@ -157 +158 @@
         elif typefile is 'mesoapi':  
             zelevel = int(nc.variables['vert'][nvert])
-            if 'altitude'       in nc.dimensions:   stralt = "_"+str(zelevel)+"m-AMR"
-            elif 'altitude_abg' in nc.dimensions:   stralt = "_"+str(zelevel)+"m-ALS"
-            elif 'bottom_top'   in nc.dimensions:   stralt = "_"+str(zelevel)+"m"
+            if abs(zelevel) < 10000.:   strheight=str(zelevel)+"m"
+            else:                       strheight=str(int(zelevel/1000.))+"km"
+            if 'altitude'       in nc.dimensions:   stralt = "_"+strheight+"-AMR"
+            elif 'altitude_abg' in nc.dimensions:   stralt = "_"+strheight+"-ALS"
+            elif 'bottom_top'   in nc.dimensions:   stralt = "_"+strheight
             elif 'pressure'     in nc.dimensions:   stralt = "_"+str(zelevel)+"Pa" 
         else:                                       stralt = ""          
@@ -179 +182 @@
         elif numplot == 2: 
             sub = 121
-            fig.subplots_adjust(wspace = 0.3)
+            fig.subplots_adjust(wspace = 0.35)
             rcParams['font.size'] = int( rcParams['font.size'] * 3. / 4. )
         elif numplot == 3: 
@@ -253 +256 @@
            if not tile:
                zelevels = np.linspace(zevmin,zevmax)
-               what_I_plot[ what_I_plot < zevmin ] = zevmin*(1. + 1.e-7)
-               what_I_plot[ what_I_plot > zevmax ] = zevmax*(1. - 1.e-7)
+               hole = True
+               if not hole:
+                   what_I_plot[ what_I_plot < zevmin ] = zevmin*(1. + 1.e-7)
+                   what_I_plot[ what_I_plot > 9e+35  ] = -9e+35
+                   what_I_plot[ what_I_plot > zevmax ] = zevmax*(1. - 1.e-7)
                contourf( x, y, what_I_plot, 10, cmap = palette, levels = zelevels )
            else:    
                pcolor( x, y, what_I_plot, cmap = palette, vmin=zevmin, vmax=zevmax )
+           #putpoints(m,fig)
            if var in ['HGT']: pass
            elif colorb: