Changeset 187 for trunk/MESOSCALE/PLOT/PYTHON

Timestamp:

Jul 3, 2011, 4:53:57 AM (14 years ago)

Author:

aslmd

Message:

MESOSCALE: python program for plots: quiverkey and colorbar.

Location:

trunk/MESOSCALE/PLOT/PYTHON

Files:

• mylib/myplot.py (modified) (2 diffs)
• scripts/winds.py (modified) (6 diffs)

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

@@ -155 +155 @@
     import  matplotlib.pyplot               as plt
     import  numpy                           as np
-    posx = np.max(x) + np.std(x) / 3.  ## pb pour les domaines globaux ...
-    posy = np.mean(y)
+    #posx = np.max(x) + np.std(x) / 3.  ## pb pour les domaines globaux ...
+    #posy = np.mean(y)
     #posx = np.min(x)
     #posy = np.max(x)
+    #posx = np.max(x) - np.std(x) / 10.
+    #posy = np.max(y) + np.std(y) / 10.
+    posx = np.min(x) - np.std(x) / 10.
+    posy = np.min(y) - np.std(y) / 10.
     u = smooth(u,csmooth)
     v = smooth(v,csmooth)
@@ -172 +176 @@
     else:                  kcolor=color
     if key: p = plt.quiverkey(q,posx,posy,scale,\
-                   str(int(scale)),coordinates='data',color=kcolor)
+                   str(int(scale)),coordinates='data',color=kcolor,labelpos='S')
     return 
 

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

@@ -15 +15 @@
            stride=3,\
            numplot=4,\
-           var=None): 
+           var=None,\
+           colorb=None): 
 
     #################################
@@ -21 +22 @@
     from netCDF4 import Dataset
     from myplot import getcoord2d,define_proj,makeplotpng,simplinterv,vectorfield,ptitle,latinterv,getproj,wrfinterv,dumpbdy
-    from matplotlib.pyplot import contourf, subplot, figure, rcParams, savefig
+    from matplotlib.pyplot import contourf, subplot, figure, rcParams, savefig, colorbar
     import numpy as np
 
@@ -174 +175 @@
                elif dimension == 3:              what_I_plot = field[i,:,:]
            contourf(x, y, what_I_plot, 30)
+           if colorb:     colorbar(fraction=0.05,pad=0.1)
 
        ### Vector plot
@@ -231 +233 @@
     parser.add_option('-s', action='store', dest='stride',      type="int",     default=3,     help='stride vectors (def=3)')
     parser.add_option('-v', action='store', dest='var',         type="string",  default=None,  help='variable contoured')
-    parser.add_option('-n', action='store', dest='numplot',     type="int",     default=4,     help='number of plots (def=1)(if <0: 1 plot of LT -*numplot*)')
+    parser.add_option('-n', action='store', dest='numplot',     type="int",     default=4,     help='number of plots (def=1)(<0: 1 plot of LT -*numplot*)')
     parser.add_option('-i', action='store', dest='interp',      type="int",     default=None,  help='interpolation method (done at level *nvert* km)')
+    parser.add_option('-c', action='store', dest='colorb',      type="string",  default=None,  help='colorbar')
     (opt,args) = parser.parse_args()
     if opt.namefile is None: 
@@ -241 +244 @@
     zefile = opt.namefile    
     zelevel = opt.nvert   
-    if opt.nvert is 0 and opt.interp:   zelevel = 0.020 
     if opt.interp is not None:
+        if opt.nvert is 0 and opt.interp is 4:  zelevel = 0.010
         if   opt.var is None    :  zefields = 'uvmet'
         else                    :  zefields = 'uvmet,'+opt.var
@@ -253 +256 @@
         zelevel = 0
 
-    winds (zefile,int(zelevel),proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=opt.var,numplot=opt.numplot)
+    winds (zefile,int(zelevel),proj=opt.proj,back=opt.back,target=opt.target,stride=opt.stride,var=opt.var,numplot=opt.numplot,colorb=opt.colorb)