ENH: Added share_tickers parameter to axes._AxesBase.twinx/y

madphysicist · madphysicist · commit a7618739e68f · 2016-11-29T10:18:12.000-05:00
Added copy constructor to axis.Ticker
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -470,6 +470,13 @@ def __init__(self, fig, rect,
                              to share the x-axis with
           *sharey*           an class:`~matplotlib.axes.Axes` instance
                              to share the y-axis with
+          *share_tickers*    [ *True* | *False* ] whether the major and
+                             minor `Formatter` and `Locator` instances
+                             are always shared (if `True`) or can be set
+                             independently (if `False`) between this set
+                             of axes and `sharex` and `sharey`. This
+                             argument has no meaning if neither `sharex`
+                             nor `sharey` are set. Defaults to `True`.
           *title*            the title string
           *visible*          [ *True* | *False* ] whether the axes is
                              visible
@@ -499,6 +506,10 @@ def __init__(self, fig, rect,
         self.set_anchor('C')
         self._sharex = sharex
         self._sharey = sharey
+        # share_tickers is only used as a modifier for sharex/y. It
+        # should not remain in kwargs by the time kwargs updates the
+        # instance dictionary.
+        self._share_tickers = kwargs.pop('share_tickers', True)
         if sharex is not None:
             self._shared_x_axes.join(self, sharex)
             if sharex._adjustable == 'box':
@@ -974,50 +985,52 @@ def cla(self):
         self.callbacks = cbook.CallbackRegistry()
 
         if self._sharex is not None:
-            # major and minor are class instances with
-            # locator and formatter attributes
-            self.xaxis.major = self._sharex.xaxis.major
-            self.xaxis.minor = self._sharex.xaxis.minor
+            # The tickers need to exist but can be empty until after the
+            # call to Axis._set_scale since they will be overwritten
+            # anyway
+            self.xaxis.major = maxis.Ticker()
+            self.xaxis.minor = maxis.Ticker()
+
+            # Copy the axis limits
             x0, x1 = self._sharex.get_xlim()
             self.set_xlim(x0, x1, emit=False, auto=None)
 
-            # Save the current formatter/locator so we don't lose it
-            majf = self._sharex.xaxis.get_major_formatter()
-            minf = self._sharex.xaxis.get_minor_formatter()
-            majl = self._sharex.xaxis.get_major_locator()
-            minl = self._sharex.xaxis.get_minor_locator()
-
             # This overwrites the current formatter/locator
             self.xaxis._set_scale(self._sharex.xaxis.get_scale())
 
-            # Reset the formatter/locator
-            self.xaxis.set_major_formatter(majf)
-            self.xaxis.set_minor_formatter(minf)
-            self.xaxis.set_major_locator(majl)
-            self.xaxis.set_minor_locator(minl)
+            # Reset the formatter/locator. Axis handle gets marked as
+            # stale in previous line, no need to repeat.
+            if self._share_tickers:
+                self.xaxis.major = self._sharex.xaxis.major
+                self.xaxis.minor = self._sharex.xaxis.minor
+            else:
+                self.xaxis.major = maxis.Ticker(self._sharex.xaxis.major)
+                self.xaxis.minor = maxis.Ticker(self._sharex.xaxis.minor)
         else:
             self.xaxis._set_scale('linear')
 
         if self._sharey is not None:
-            self.yaxis.major = self._sharey.yaxis.major
-            self.yaxis.minor = self._sharey.yaxis.minor
+            # The tickers need to exist but can be empty until after the
+            # call to Axis._set_scale since they will be overwritten
+            # anyway
+            self.yaxis.major = maxis.Ticker()
+            self.yaxis.minor = maxis.Ticker()
+
+            # Copy the axis limits
             y0, y1 = self._sharey.get_ylim()
             self.set_ylim(y0, y1, emit=False, auto=None)
 
-            # Save the current formatter/locator so we don't lose it
-            majf = self._sharey.yaxis.get_major_formatter()
-            minf = self._sharey.yaxis.get_minor_formatter()
-            majl = self._sharey.yaxis.get_major_locator()
-            minl = self._sharey.yaxis.get_minor_locator()
-
             # This overwrites the current formatter/locator
             self.yaxis._set_scale(self._sharey.yaxis.get_scale())
 
-            # Reset the formatter/locator
-            self.yaxis.set_major_formatter(majf)
-            self.yaxis.set_minor_formatter(minf)
-            self.yaxis.set_major_locator(majl)
-            self.yaxis.set_minor_locator(minl)
+            # Reset the formatter/locator. Axis handle gets marked as
+            # stale in previous line, no need to repeat.
+            if self._share_tickers:
+                self.yaxis.major = self._sharey.yaxis.major
+                self.yaxis.minor = self._sharey.yaxis.minor
+            else:
+                self.yaxis.major = maxis.Ticker(self._sharey.yaxis.major)
+                self.yaxis.minor = maxis.Ticker(self._sharey.yaxis.minor)
         else:
             self.yaxis._set_scale('linear')
 
@@ -3894,21 +3907,27 @@ def _make_twin_axes(self, *kl, **kwargs):
         ax2 = self.figure.add_axes(self.get_position(True), *kl, **kwargs)
         return ax2
 
-    def twinx(self):
+    def twinx(self, share_tickers=True):
         """
         Create a twin Axes sharing the xaxis
 
-        create a twin of Axes for generating a plot with a sharex
+        Create a twin of Axes for generating a plot with a sharex
         x-axis but independent y axis.  The y-axis of self will have
         ticks on left and the returned axes will have ticks on the
         right. To ensure tick marks of both axis align, see
         :class:`~matplotlib.ticker.LinearLocator`
 
+        `share_tickers` determines if the shared axis will always have
+        the same major and minor `Formatter` and `Locator` objects as
+        this one. This is usually desirable since the axes overlap.
+        However, if one of the axes is shifted so that they are both
+        visible, it may be useful to set this parameter to ``False``.
+
         .. note::
             For those who are 'picking' artists while using twinx, pick
             events are only called for the artists in the top-most axes.
         """
-        ax2 = self._make_twin_axes(sharex=self)
+        ax2 = self._make_twin_axes(sharex=self, share_tickers=share_tickers)
         ax2.yaxis.tick_right()
         ax2.yaxis.set_label_position('right')
         ax2.yaxis.set_offset_position('right')
@@ -3917,21 +3936,27 @@ def twinx(self):
         ax2.patch.set_visible(False)
         return ax2
 
-    def twiny(self):
+    def twiny(self, share_tickers=True):
         """
         Create a twin Axes sharing the yaxis
 
-        create a twin of Axes for generating a plot with a shared
+        Create a twin of Axes for generating a plot with a shared
         y-axis but independent x axis.  The x-axis of self will have
         ticks on bottom and the returned axes will have ticks on the
         top.
 
+        `share_tickers` determines if the shared axis will always have
+        the same major and minor `Formatter` and `Locator` objects as
+        this one. This is usually desirable since the axes overlap.
+        However, if one of the axes is shifted so that they are both
+        visible, it may be useful to set this parameter to ``False``.
+
         .. note::
             For those who are 'picking' artists while using twiny, pick
             events are only called for the artists in the top-most axes.
         """
 
-        ax2 = self._make_twin_axes(sharey=self)
+        ax2 = self._make_twin_axes(sharey=self, share_tickers=share_tickers)
         ax2.xaxis.tick_top()
         ax2.xaxis.set_label_position('top')
         self.xaxis.tick_bottom()
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -604,6 +604,14 @@ class Ticker(object):
     locator = None
     formatter = None
 
+    def __init__(self, ticker=None):
+        """
+        Constructor that optionally allows a copy to be made.
+        """
+        if ticker:
+            self.locator = ticker.locator
+            self.formatter = ticker.formatter
+
 
 class Axis(artist.Artist):
     """
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -26,6 +26,7 @@
 import matplotlib.markers as mmarkers
 import matplotlib.patches as mpatches
 import matplotlib.colors as mcolors
+import matplotlib.ticker as mticker
 from numpy.testing import assert_allclose, assert_array_equal
 from matplotlib.cbook import IgnoredKeywordWarning
 import matplotlib.colors as mcolors
@@ -144,6 +145,58 @@ def test_twinx_cla():
     assert ax.patch.get_visible()
     assert ax.yaxis.get_visible()
 
+@cleanup
+def test_twin_xy_sharing():
+    fig, ax = plt.subplots()
+
+    # Make some twinned axes to play with (with share_tickers=True)
+    ax2 = ax.twinx()
+    ax3 = ax2.twiny()
+    plt.draw()
+
+    assert ax.xaxis.major is ax2.xaxis.major
+    assert ax.xaxis.minor is ax2.xaxis.minor
+    assert ax2.yaxis.major is ax3.yaxis.major
+    assert ax2.yaxis.minor is ax3.yaxis.minor
+
+    # Verify that for share_tickers=True, the formatters and locators
+    # are identical no matter what
+    ax2.xaxis.set_major_formatter(mticker.PercentFormatter())
+    ax3.yaxis.set_major_locator(mticker.MaxNLocator())
+
+    assert ax.xaxis.get_major_formatter() is ax2.xaxis.get_major_formatter()
+    assert ax2.yaxis.get_major_locator() is ax3.yaxis.get_major_locator()
+
+    # Make some more twinned axes to play with (with share_tickers=False)
+    ax4 = ax.twinx(share_tickers=False)
+    ax5 = ax2.twiny(share_tickers=False)
+    plt.draw()
+
+    assert ax4 is not ax2
+    assert ax5 is not ax3
+
+    assert ax.xaxis.major is not ax4.xaxis.major
+    assert ax.xaxis.minor is not ax4.xaxis.minor
+    assert ax.xaxis.get_major_formatter() is ax4.xaxis.get_major_formatter()
+    assert ax.xaxis.get_minor_formatter() is ax4.xaxis.get_minor_formatter()
+    assert ax.xaxis.get_major_locator() is ax4.xaxis.get_major_locator()
+    assert ax.xaxis.get_minor_locator() is ax4.xaxis.get_minor_locator()
+
+    assert ax2.yaxis.major is not ax5.yaxis.major
+    assert ax2.yaxis.minor is not ax5.yaxis.minor
+    assert ax2.yaxis.get_major_formatter() is ax5.yaxis.get_major_formatter()
+    assert ax2.yaxis.get_minor_formatter() is ax5.yaxis.get_minor_formatter()
+    assert ax2.yaxis.get_major_locator() is ax5.yaxis.get_major_locator()
+    assert ax2.yaxis.get_minor_locator() is ax5.yaxis.get_minor_locator()
+
+    # Verify that for share_tickers=False, the formatters and locators
+    # can be changed independently
+    ax4.xaxis.set_minor_formatter(mticker.PercentFormatter())
+    ax5.yaxis.set_minor_locator(mticker.MaxNLocator())
+
+    assert ax.xaxis.get_minor_formatter() is not ax4.xaxis.get_minor_formatter()
+    assert ax2.yaxis.get_minor_locator() is not ax4.yaxis.get_minor_locator()
+
 
 @image_comparison(baseline_images=["minorticks_on_rcParams_both"], extensions=['png'])
 def test_minorticks_on_rcParams_both():
