Merge pull request matplotlib#2417 from fariza/remlim-with-invisible-option

Adding possibility to remove invisible lines and patches from relim.

Author: pelson

lib/matplotlib/axes/_base.py

@@ -1590,9 +1590,11 @@ def add_container(self, container):
         container.set_remove_method(lambda h: self.containers.remove(h))
         return container
 
-    def relim(self):
+    def relim(self, visible_only=False):
         """
-        Recompute the data limits based on current artists.
+        Recompute the data limits based on current artists. If you want to exclude
+        invisible artists from the calculation, set
+        `visible_only=True`
 
         At present, :class:`~matplotlib.collections.Collection`
         instances are not supported.
@@ -1604,10 +1606,12 @@ def relim(self):
         self.ignore_existing_data_limits = True
 
         for line in self.lines:
-            self._update_line_limits(line)
+            if not visible_only or line.get_visible():
+                self._update_line_limits(line)
 
         for p in self.patches:
-            self._update_patch_limits(p)
+            if not visible_only or p.get_visible():
+                self._update_patch_limits(p)
 
     def update_datalim(self, xys, updatex=True, updatey=True):
         """

lib/matplotlib/tests/test_axes.py

@@ -1737,6 +1737,32 @@ def test_empty_shared_subplots():
     assert y1 >= 6
 
 
+@cleanup
+def test_relim_visible_only():
+    x1 = (0., 10.)
+    y1 = (0., 10.)
+    x2 = (-10., 20.)
+    y2 = (-10., 30.)
+
+    fig = matplotlib.figure.Figure()
+    ax = fig.add_subplot(111)
+    ax.plot(x1, y1)
+    assert ax.get_xlim() == x1
+    assert ax.get_ylim() == y1
+    l = ax.plot(x2, y2)
+    assert ax.get_xlim() == x2
+    assert ax.get_ylim() == y2
+    l[0].set_visible(False)
+    assert ax.get_xlim() == x2
+    assert ax.get_ylim() == y2
+    
+    ax.relim(visible_only=True)
+    ax.autoscale_view()
+
+    assert ax.get_xlim() == x1
+    assert ax.get_ylim() == y1
+
+
 if __name__ == '__main__':
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)