Deprecated the set_colorbar method on a scalar mappable.

pelson · pelson · commit b50237da013f · 2013-05-23T15:56:35.000+01:00
diff --git a/lib/matplotlib/cm.py b/lib/matplotlib/cm.py
@@ -193,12 +193,14 @@ def __init__(self, norm=None, cmap=None):
         self.norm = norm
         #: The Colormap instance of this ScalarMappable.
         self.cmap = get_cmap(cmap)
+        #: The last colorbar associated with this ScalarMappable. May be None.
         self.colorbar = None
         self.update_dict = {'array': False}
 
+    @cbook.deprecated('1.3', alternative='the colorbar attribute')
     def set_colorbar(self, im, ax):
-        'set the colorbar image and axes associated with mappable'
-        self.colorbar = im, ax
+        """set the colorbar and axes instances associated with mappable"""
+        self.colorbar = im
 
     def to_rgba(self, x, alpha=None, bytes=False):
         """
diff --git a/lib/matplotlib/colorbar.py b/lib/matplotlib/colorbar.py
@@ -1269,6 +1269,6 @@ def colorbar_factory(cax, mappable, **kwargs):
         cb = Colorbar(cax, mappable, **kwargs)
 
     mappable.callbacksSM.connect('changed', cb.on_mappable_changed)
-    mappable.set_colorbar(cb, cax)
+    mappable.colorbar = cb
 
     return cb
diff --git a/lib/mpl_toolkits/axes_grid/colorbar.py b/lib/mpl_toolkits/axes_grid/colorbar.py
@@ -824,6 +824,6 @@ def on_changed(m):
         cb.update_bruteforce(m)
 
     cbid = mappable.callbacksSM.connect('changed', on_changed)
-    mappable.set_colorbar(cb, cax)
+    mappable.colorbar = cb
     ax.figure.sca(ax)
     return cb
diff --git a/lib/mpl_toolkits/axes_grid1/axes_grid.py b/lib/mpl_toolkits/axes_grid1/axes_grid.py
@@ -92,7 +92,7 @@ def on_changed(m):
             cb.update_bruteforce(m)
 
         self.cbid = mappable.callbacksSM.connect('changed', on_changed)
-        mappable.set_colorbar(cb, self)
+        mappable.colorbar = cb
 
         self.locator = cb.cbar_axis.get_major_locator()
 
diff --git a/lib/mpl_toolkits/axes_grid1/colorbar.py b/lib/mpl_toolkits/axes_grid1/colorbar.py
@@ -825,6 +825,6 @@ def on_changed(m):
         cb.update_bruteforce(m)
 
     cbid = mappable.callbacksSM.connect('changed', on_changed)
-    mappable.set_colorbar(cb, cax)
+    mappable.colorbar = cb
     ax.figure.sca(ax)
     return cb
