Switch to a private, simpler AxesStack.

anntzer · anntzer · commit acd4bb237b63 · 2016-11-01T22:29:33.000-07:00
The current implementation of AxesStack subclasses cbook.Stack, which
requires hashable keys, which leads to additional complexity on the
caller's side (`_make_key`).

Instead, switch to using two lists (keys and axes) and relying on
`list.index`, which makes the implementation much simpler.

Also make the new class private and deprecate the previous one.
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
@@ -16,8 +16,9 @@
 
 import six
 
-import warnings
 from operator import itemgetter
+import warnings
+from weakref import WeakKeyDictionary
 
 import numpy as np
 
@@ -73,6 +74,7 @@ class AxesStack(Stack):
 
     """
     def __init__(self):
+        cbook.warn_deprecated("2.0")
         Stack.__init__(self)
         self._ind = 0
 
@@ -157,6 +159,63 @@ def __contains__(self, a):
         return a in self.as_list()
 
 
+class _AxesStack(object):
+    """Lightweight stack that tracks Axes in a Figure.
+    """
+
+    # We do not subclass the Stack class from cbook to avoid hashability
+    # issues.
+
+    def __init__(self):
+        self._keys = []
+        self._axes = []
+        self._axes_order = WeakKeyDictionary()
+        self._index = -1
+
+    axes = property(
+        lambda self: sorted(self._axes, key=self._axes_order.__getitem__),
+        doc="Copy of the list of axes, in the order of insertion.")
+
+    def get(self, key):
+        """Find the axes corresponding to a key; defaults to `None`.
+        """
+        try:
+            return self._axes[self._keys.index(key)]
+        except ValueError:
+            return None
+
+    def current_key_axes(self):
+        """Return the topmost key, axes pair, or `None, None` if empty.
+        """
+        return (self._keys[-1], self._axes[-1]) if self._keys else (None, None)
+
+    def add(self, key, ax):
+        """Append a key, axes pair.
+        """
+        self._keys.append(key)
+        self._axes.append(ax)
+        self._axes_order[ax] = self._index = self._index + 1
+
+    def bubble(self, ax):
+        """Move an axes and its corresponding key to the top.
+        """
+        idx = self._axes.index(ax)
+        self._keys.append(self._keys[idx])
+        self._axes.append(self._axes[idx])
+        del self._keys[idx], self._axes[idx]
+
+    def remove(self, ax):
+        """Remove an axes and its corresponding key.
+        """
+        idx = self._axes.index(ax)
+        del self._keys[idx], self._axes[idx]
+
+    def clear(self):
+        """Clear the stack.
+        """
+        del self._keys[:], self._axes[:]  # Py2 doesn't have list.clear.
+
+
 class SubplotParams(object):
     """
     A class to hold the parameters for a subplot
@@ -350,7 +409,7 @@ def __init__(self,
         self.subplotpars = subplotpars
         self.set_tight_layout(tight_layout)
 
-        self._axstack = AxesStack()  # track all figure axes and current axes
+        self._axstack = _AxesStack()  # track all figure axes and current axes
         self.clf()
         self._cachedRenderer = None
 
@@ -399,7 +458,7 @@ def show(self, warn=True):
                 "so cannot show the figure")
 
     def _get_axes(self):
-        return self._axstack.as_list()
+        return self._axstack.axes
 
     axes = property(fget=_get_axes, doc="Read-only: list of axes in Figure")
 
@@ -812,36 +871,6 @@ def delaxes(self, a):
             func(self)
         self.stale = True
 
-    def _make_key(self, *args, **kwargs):
-        'make a hashable key out of args and kwargs'
-
-        def fixitems(items):
-            #items may have arrays and lists in them, so convert them
-            # to tuples for the key
-            ret = []
-            for k, v in items:
-                # some objects can define __getitem__ without being
-                # iterable and in those cases the conversion to tuples
-                # will fail. So instead of using the iterable(v) function
-                # we simply try and convert to a tuple, and proceed if not.
-                try:
-                    v = tuple(v)
-                except Exception:
-                    pass
-                ret.append((k, v))
-            return tuple(ret)
-
-        def fixlist(args):
-            ret = []
-            for a in args:
-                if iterable(a):
-                    a = tuple(a)
-                ret.append(a)
-            return tuple(ret)
-
-        key = fixlist(args), fixitems(six.iteritems(kwargs))
-        return key
-
     @docstring.dedent_interpd
     def add_axes(self, *args, **kwargs):
         """
@@ -895,9 +924,9 @@ def add_axes(self, *args, **kwargs):
 
         # shortcut the projection "key" modifications later on, if an axes
         # with the exact args/kwargs exists, return it immediately.
-        key = self._make_key(*args, **kwargs)
+        key = (args, kwargs)
         ax = self._axstack.get(key)
-        if ax is not None:
+        if ax:
             self.sca(ax)
             return ax
 
@@ -914,7 +943,7 @@ def add_axes(self, *args, **kwargs):
             # check that an axes of this type doesn't already exist, if it
             # does, set it as active and return it
             ax = self._axstack.get(key)
-            if ax is not None and isinstance(ax, projection_class):
+            if isinstance(ax, projection_class):
                 self.sca(ax)
                 return ax
 
@@ -988,15 +1017,14 @@ def add_subplot(self, *args, **kwargs):
                 raise ValueError(msg)
             # make a key for the subplot (which includes the axes object id
             # in the hash)
-            key = self._make_key(*args, **kwargs)
+            key = (args, kwargs)
         else:
             projection_class, kwargs, key = process_projection_requirements(
                 self, *args, **kwargs)
 
             # try to find the axes with this key in the stack
             ax = self._axstack.get(key)
-
-            if ax is not None:
+            if ax:
                 if isinstance(ax, projection_class):
                     # the axes already existed, so set it as active & return
                     self.sca(ax)
@@ -1496,7 +1524,7 @@ def _gci(self):
         do not use elsewhere.
         """
         # Look first for an image in the current Axes:
-        cax = self._axstack.current_key_axes()[1]
+        ckey, cax = self._axstack.current_key_axes()
         if cax is None:
             return None
         im = cax._gci()
diff --git a/lib/matplotlib/projections/__init__.py b/lib/matplotlib/projections/__init__.py
@@ -103,11 +103,7 @@ def process_projection_requirements(figure, *args, **kwargs):
         raise TypeError('projection must be a string, None or implement a '
                             '_as_mpl_axes method. Got %r' % projection)
 
-    # Make the key without projection kwargs, this is used as a unique
-    # lookup for axes instances
-    key = figure._make_key(*args, **kwargs)
-
-    return projection_class, kwargs, key
+    return projection_class, kwargs, (args, kwargs)
 
 
 def get_projection_names():
