Changeset 21426

Timestamp:

11/29/16 02:48:25 (8 years ago)

Author:

aplach

Message:

plot_unit.py: fixed issue with continous colorbar when setting colorlevels; now both plot and colorbar are descrete if colorlevels are set in plotmodel()

Location:

issm/trunk-jpl/src/m/plot

Files:

• plot_unit.py (modified) (1 diff)
• plotmodel.py (modified) (5 diffs)

issm/trunk-jpl/src/m/plot/plot_unit.py

@@ -29 +29 @@
         # {{{ define wich colormap to use 
         try:
-                defaultmap=plt.cm.get_cmap('viridis')
+                defaultmap=plt.cm.get_cmap('viridis',colorlevels)
         except AttributeError:
                 print("Viridis can't be found (probably too old Matplotlib) reverting to gnuplot colormap")

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

@@ -6 +6 @@
         import pylab as p
         import matplotlib.pyplot as plt
-        from mpl_toolkits.axes_grid1 import ImageGrid, AxesGrid
+        from mpl_toolkits.axes_grid1 import ImageGrid, AxesGrid
 except ImportError:
         print "could not import pylab, matplotlib has not been installed, no plotting capabilities enabled"
@@ -17 +17 @@
         '''
 
-        #First process options 
+        #First process options
         options=plotoptions(*args)
 
@@ -36 +36 @@
                 nrows=np.ceil(numberofplots/subplotwidth)
                 nr=False
-        
+
         if options.list[0].exist('ncols'):
                 ncols=options.list[0].getfieldvalue('ncols')
@@ -45 +45 @@
         ncols=int(ncols)
         nrows=int(nrows)
-        
+
         #check that nrows and ncols were given at the same time!
         if not nr==nc:
                 raise StandardError('error: nrows and ncols need to be specified together, or not at all')
-        
+
         #Go through plots
         if numberofplots:
@@ -82 +82 @@
                 cbar_size=options.list[0].getfieldvalue('colorbarsize','5%')
                 cbar_pad=options.list[0].getfieldvalue('colorbarpad','2.5%') # None or %
-                
+
                 axgrid=ImageGrid(fig, 111,
                                 nrows_ncols=(nrows,ncols),