Changeset 17756

Timestamp:

04/17/14 10:46:47 (11 years ago)

Author:

cborstad

Message:

working on python plotmodel, added plot_contour

Location:

issm/trunk-jpl/src/m/plot

Files:

• applyoptions.py (modified) (7 diffs)
• plot_contour.py (added)
• plot_manager.py (modified) (2 diffs)
• plot_overlay.py (modified) (3 diffs)
• plotmodel.py (modified) (1 diff)

issm/trunk-jpl/src/m/plot/applyoptions.py ¶

@@ -1 +1 @@
 import numpy as npy
 from cmaptools import truncate_colormap
+from plot_contour import plot_contour
 
 try:
@@ -23 +24 @@
         #if not isnan(md.mesh.hemisphere):
         #       options.addfielddefault('hemisphere',md.mesh.hemisphere)
+
+        # get handle to current figure and axes instance
+        fig = p.gcf()
+        ax=p.gca()
 
         #font {{{
@@ -113 +118 @@
 
         #ticklabel notation {{{
-        p.gca().ticklabel_format(style='sci',scilimits=(0,0))
+        #ax.ticklabel_format(style='sci',scilimits=(0,0))
         #}}}
 
         #ticklabelfontsize {{{
         if options.exist('ticklabelfontsize'):
-                ax=p.gca()
                 for label in ax.get_xticklabels() + ax.get_yticklabels():
                         label.set_fontsize(options.getfieldvalue('ticklabelfontsize'))
@@ -131 +135 @@
         if options.exist('axis'):
                 if options.getfieldvalue('axis',True)=='off':
-                        p.gca().ticklabel_format(style='plain')
-                        p.setp(p.gca().get_xticklabels(), visible=False)
-                        p.setp(p.gca().get_yticklabels(), visible=False)
+                        ax.ticklabel_format(style='plain')
+                        p.setp(ax.get_xticklabels(), visible=False)
+                        p.setp(ax.get_yticklabels(), visible=False)
         # }}}
 
@@ -177 +181 @@
         defaultmap=truncate_colormap(mpl.cm.gnuplot2,0.1,0.9,128)
         cmap=options.getfieldvalue('colormap',defaultmap)
+        norm = mpl.colors.Normalize(vmin=lims[0], vmax=lims[1])
+        options.addfield('colornorm',norm)
+        cbar_extend=0
+        if options.exist('cmap_set_over'):
+                over=options.getfieldvalue('cmap_set_over','0.5')
+                cmap.set_over(over)
+                cbar_extend+=1
+        if options.exist('cmap_set_under'):
+                under=options.getfieldvalue('cmap_set_under','0.5')
+                cmap.set_under(under)
+                cbar_extend+=2
+        options.addfield('colormap',cmap)
         #}}}
 
@@ -183 +199 @@
         #colorbar {{{
         if options.getfieldvalue('colorbar',1)==1:
-                fig = p.gcf()
-                ax = p.gca()
                 divider = make_axes_locatable(ax)
                 cax = divider.new_horizontal("5%", pad=0.05, axes_class=mpl.axes.Axes)
                 fig.add_axes(cax) 
-                norm = mpl.colors.Normalize(vmin=lims[0], vmax=lims[1])
-                cbar_extend=0
-                if options.exist('cmap_set_over'):
-                        over=options.getfieldvalue('cmap_set_over','0.5')
-                        cmap.set_over(over)
-                        cbar_extend+=1
-                if options.exist('cmap_set_under'):
-                        under=options.getfieldvalue('cmap_set_under','0.5')
-                        cmap.set_under(under)
-                        cbar_extend+=2
                 if cbar_extend==0:
                         extend='neither'
@@ -226 +230 @@
         #streamlines
 
-        #contours
+        #contours {{{
+        if options.exist('contourlevels'):
+                plot_contour(md,data,options)
+        #}}}
 
         #axis positions

issm/trunk-jpl/src/m/plot/plot_manager.py ¶

@@ -30 +30 @@
         #initialize plot handle variable
         handle=None
+
+        # initialize subplot
+        p.subplot(nlines,ncols,i,aspect='equal')
 
         ##basemap plot
@@ -70 +73 @@
 
         #standard plot
-        if not handle:
-                p.subplot(nlines,ncols,i,aspect='equal')
+        #if not handle:
+        #       p.subplot(nlines,ncols,i,aspect='equal')
 
         #plot unit

issm/trunk-jpl/src/m/plot/plot_overlay.py ¶

@@ -52 +52 @@
         gtif=gdal.Open(inputname)
         arr=gtif.ReadAsArray()
-        os.system('rm -rf ./temp.tif')
+        #os.system('rm -rf ./temp.tif')
         
         if gtif.RasterCount>=3:  # RGB array
@@ -62 +62 @@
         # normalize array
         arr=arr/npy.float(npy.max(arr.ravel()))
+
+        if options.getfieldvalue('overlayhist',0)==1:
+                ax=plt.gca()
+                num=2
+                while True:
+                        if not plt.fignum_exists(num):
+                                break
+                        else:
+                                num+=1
+                plt.figure(num)
+                plt.hist(arr.flatten(),bins=256,range=(0.,1.))
+                plt.title('histogram of overlay image, use for setting overlaylims')
+                plt.sca(ax) # return to original axes/figure
+                
 
         # get parameters from cropped geotiff
@@ -89 +103 @@
 
         norm=mpl.colors.Normalize(vmin=overlaylims[0],vmax=overlaylims[1])
+
         handle.pcolormesh(xgmap, ygmap, npy.flipud(arr), cmap=mpl.cm.Greys, norm=norm)
-        plt.axes().set_aspect('equal','box')
+        handle.set_aspect('equal','box')
         return handle

issm/trunk-jpl/src/m/plot/plotmodel.py ¶

@@ -56 +56 @@
 
                 if not hold: # TODO need to also check whether figurenumber is a new plot so that old plots are not mistakenly cleared
-                        p.clf()
+                        p.cla()
 
+                #TODO fig, axarray = plt.subplots(nrows,ncols), then pass fix and axarr to plot_manager
                 #if figsize specified
                 if options.list[0].exist('figsize'):