ENH: Subpanels

jklymak · jklymak · commit 132e42db5581 · 2020-08-25T15:03:05.000-07:00
diff --git a/lib/matplotlib/_constrained_layout.py b/lib/matplotlib/_constrained_layout.py
@@ -208,7 +208,7 @@ def _make_layout_margins(fig, renderer, *, w_pad=0, h_pad=0,
         panel._layoutgrid.parent.edit_outer_margin_mins(margins, ss)
 
     # for ax in [a for a in fig._localaxes if hasattr(a, 'get_subplotspec')]:
-    for ax in fig.get_axes():
+    for ax in fig._localaxes:
         if not hasattr(ax, 'get_subplotspec'):
             continue
 
@@ -223,7 +223,6 @@ def _make_layout_margins(fig, renderer, *, w_pad=0, h_pad=0,
                                            hspace=hspace, wspace=wspace)
         margin0 = margin.copy()
         pos, bbox = _get_pos_and_bbox(ax, renderer)
-
         # the margin is the distance between the bounding box of the axes
         # and its position (plus the padding from above)
         margin['left'] += pos.x0 - bbox.x0
@@ -456,7 +455,7 @@ def _reposition_axes(fig, renderer, *, w_pad=0, h_pad=0, hspace=0, wspace=0):
 
     # for ax in fig._localaxes:
     #     if not hasattr(a, 'get_subplotspec'):
-    for ax in fig.get_axes():
+    for ax in fig._localaxes:
         if not hasattr(ax, 'get_subplotspec'):
             continue
 
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -55,12 +55,12 @@ def __init__(self, bounds, transform):
         self._transform = transform
 
     def __call__(self, ax, renderer):
-        # Subtracting transFigure will typically rely on inverted(), freezing
+        # Subtracting transPanel will typically rely on inverted(), freezing
         # the transform; thus, this needs to be delayed until draw time as
-        # transFigure may otherwise change after this is evaluated.
+        # transPanel may otherwise change after this is evaluated.
         return mtransforms.TransformedBbox(
             mtransforms.Bbox.from_bounds(*self._bounds),
-            self._transform - ax.figure.transFigure)
+            self._transform - ax.figure.transPanel)
 
 
 # The axes module contains all the wrappers to plotting functions.
@@ -579,7 +579,7 @@ def indicate_inset(self, bounds, inset_ax=None, *, transform=None,
 
             # decide which two of the lines to keep visible....
             pos = inset_ax.get_position()
-            bboxins = pos.transformed(self.figure.transFigure)
+            bboxins = pos.transformed(self.figure.transPanel)
             rectbbox = mtransforms.Bbox.from_bounds(
                 *bounds
             ).transformed(transform)
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -629,7 +629,7 @@ def set_figure(self, fig):
         super().set_figure(fig)
 
         self.bbox = mtransforms.TransformedBbox(self._position,
-                                                fig.transFigure)
+                                                fig.transPanel)
         # these will be updated later as data is added
         self.dataLim = mtransforms.Bbox.null()
         self._viewLim = mtransforms.Bbox.unit()
@@ -1607,9 +1607,11 @@ def apply_aspect(self, position=None):
             self._set_position(position, which='active')
             return
 
-        fig_width, fig_height = self.get_figure().get_size_inches()
-        fig_aspect = fig_height / fig_width
-
+        trans = self.get_figure().transPanel
+        bb = mtransforms.Bbox.from_bounds(0, 0, 1, 1).transformed(trans)
+        # this is the physical aspect of the panel (or figure):
+        fig_aspect = bb.height / bb.width
+        
         if self._adjustable == 'box':
             if self in self._twinned_axes:
                 raise RuntimeError("Adjustable 'box' is not allowed in a "
diff --git a/lib/matplotlib/figure.py b/lib/matplotlib/figure.py
