added ContourMappable attribute 'level_upper' (needed for colorbar)

Jeff Whitaker · Jeff Whitaker · commit f29f8dcf20a0 · 2005-08-26T14:45:17.000Z
svn path=/trunk/matplotlib/; revision=1675
diff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py
@@ -86,6 +86,7 @@ def changed(self):
 
         ScalarMappable.changed(self)
 
+
 class ContourLabeler:
     def __init__(self, ax):
         self.ax = ax
@@ -691,6 +692,8 @@ def contour(self, *args, **kwargs):
 
         tcolors, mappable, collections = self._process_colors(colors,
                                                             alpha, lev, cmap)
+        if mappable is not None:
+            mappable.level_upper = None
 
         if linewidths == None:
             tlinewidths = [rcParams['lines.linewidth']] *Nlev
@@ -803,6 +806,7 @@ def contourf(self, *args, **kwargs):
         tcolors, mappable, collections = self._process_colors(colors,
                                                                alpha,
                                                                lev[:-1], cmap)
+        mappable.level_upper = lev[-1]
 
         C = _contour.Cntr(x, y, z.filled(), z.mask())
         for level, level_upper, color in zip(lev[:-1], lev[1:], tcolors):
