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

Timestamp:

Dec 1, 2011, 11:10:25 AM (13 years ago)

Author:

acolaitis

Message:

PYTHON. Contour plots now working in movies. I had to modify the call_contour (get out the reducefield part, introduce a new variable what_I_plot_contour dedicated to var2) in order to conserve the speed of movie rendering. (otherwise there were calls to reducefield for each frame in call_contour)

File:

• trunk/UTIL/PYTHON/myplot.py (modified) (1 diff)

trunk/UTIL/PYTHON/myplot.py

@@ -1034 +1034 @@
 
     return nlon, nlat, nvert, ntime, mapmode, nslices
+
+# Author: AC
+# Reduce complexity of main script by moving the contour part here. Also allow to call it from elsewhere
+# which can be usefull
+
+def call_contour(what_I_plot,error,lon,lat,vert,time,vertmode,ze_var2,indextime,indexlon,indexlat,indexvert,yintegral,mapmode,typefile,var2,ticks):
+      from matplotlib.pyplot import contour, plot
+      import numpy as np
+      #what_I_plot = what_I_plot*mult
+      if not error:
+         if mapmode == 1:
+             if typefile in ['mesoapi','meso']:    what_I_plot = dumpbdy(what_I_plot,6)
+         zevmin, zevmax = calculate_bounds(what_I_plot)
+         zelevels = np.linspace(zevmin,zevmax,ticks) #20)
+         if var2 == 'HGT':  zelevels = np.arange(-10000.,30000.,2000.)
+         if mapmode == 0:
+             #if typefile in ['mesoideal']:    what_I_plot = dumpbdy(what_I_plot,0,stag='W')
+             itime=indextime
+             if len(what_I_plot.shape) is 3: itime=[0]
+             what_I_plot, x, y = define_axis(lon,lat,vert,time,indexlon,indexlat,indexvert,\
+                   itime,what_I_plot, len(ze_var2.shape),vertmode)
+         ### If we plot a 2-D field
+         if len(what_I_plot.shape) is 2:
+             #zelevels=[0.2,0.4,0.6,0.8,1.]
+             cs = contour(x,y,what_I_plot, zelevels, colors='k', linewidths = 1 ) #0.33 colors='w' )# , alpha=0.5)
+             #clabel(cs,zelevels,inline=3,fmt='%1.1f',fontsize=7)
+         ### If we plot a 1-D field
+         elif len(what_I_plot.shape) is 1:
+             plot(what_I_plot,x)
+      else:
+         errormess("There is an error in reducing field !")
+      return error 
+