plotnine 0.15.0.dev2__py3-none-any.whl → 0.15.1__py3-none-any.whl

plotnine/_mpl/layout_manager/_layout_tree.py

@@ -1,14 +1,13 @@
 from __future__ import annotations
 
 import abc
-from contextlib import suppress
 from dataclasses import dataclass
 from functools import cached_property
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, cast
 
 import numpy as np
 
-from plotnine.plot_composition import OR
+from plotnine.composition import Beside
 
 from ._spaces import LayoutSpaces
 
@@ -17,7 +16,7 @@ if TYPE_CHECKING:
 
     from plotnine import ggplot
     from plotnine._mpl.gridspec import p9GridSpec
-    from plotnine.plot_composition import Compose
+    from plotnine.composition import Compose
 
 
 @dataclass
@@ -81,169 +80,181 @@ class LayoutTree:
     represents.
     """
 
-    @cached_property
-    @abc.abstractmethod
-    def lefts(self) -> Sequence[float]:
+    @staticmethod
+    def create(
+        cmp: Compose,
+        lookup_spaces: dict[ggplot, LayoutSpaces],
+    ) -> LayoutTree:
         """
-        Left values [figure space] of nodes in this tree
+        Create a LayoutTree for this composition
+
+        Parameters
+        ----------
+        cmp :
+            Composition
+        lookup_spaces :
+            A table to lookup the LayoutSpaces for each plot.
+
+        Notes
+        -----
+        LayoutTree works by modifying the `.gridspec` of the compositions,
+        and the `LayoutSpaces` of the plots.
+        """
+        from plotnine import ggplot
+
+        nodes: list[LayoutSpaces | LayoutTree] = []
+        for item in cmp:
+            if isinstance(item, ggplot):
+                nodes.append(lookup_spaces[item])
+            else:
+                nodes.append(LayoutTree.create(item, lookup_spaces))
+
+        if isinstance(cmp, Beside):
+            return ColumnsTree(cmp.gridspec, nodes)
+        else:
+            return RowsTree(cmp.gridspec, nodes)
+
+    def harmonise(self):
         """
+        Align and resize plots in composition to look good
+        """
+        self.align_axis_titles()
+        self.align()
+        self.resize()
 
     @abc.abstractmethod
-    def set_left_alignment_margin(self, value: float):
+    def align(self):
         """
-        Set a margin to align the left of the panels in this composition
+        Align all the edges in this composition & contained compositions
 
-        In figure dimenstions
+        This function mutates the layout spaces, specifically the
+        margin_alignments along the sides of the plot.
         """
 
-    @cached_property
     @abc.abstractmethod
-    def bottoms(self) -> Sequence[float]:
+    def resize(self):
         """
-        Bottom values [figure space] of nodes in this tree
+        Resize panels and the entire plots
+
+        This function mutates the composition gridspecs; specifically the
+        width_ratios and height_ratios.
         """
 
+    def align_sub_compositions(self):
+        """
+        Align the compositions contained in this one
+        """
+        # Recurse into the contained compositions
+        for tree in self.sub_compositions:
+            tree.align()
+
     @abc.abstractmethod
-    def set_bottom_alignment_margin(self, value: float):
+    def align_axis_titles(self):
         """
-        Set a margin to align the bottom of the panels in this composition
+        Align the axis titles along the composing dimension
 
-        In figure dimenstions
+        Since the alignment value used to for this purpose is one of
+        the fields in the _side_space, it affects the space created
+        for the panel.
+
+        We could align the titles within self.align but we would have
+        to store the value outside the _side_space and pick it up when
+        setting the position of the texts!
+        """
+
+    def resize_sub_compositions(self):
         """
+        Resize panels in the compositions contained in this one
+        """
+        for tree in self.sub_compositions:
+            tree.resize()
 
     @cached_property
-    @abc.abstractmethod
-    def tops(self) -> Sequence[float]:
+    def sub_compositions(self) -> list[LayoutTree]:
         """
-        Top values [figure space] of nodes in this tree
+        LayoutTrees of the direct sub compositions of this one
         """
+        return [item for item in self.nodes if isinstance(item, LayoutTree)]
 
+    @cached_property
     @abc.abstractmethod
-    def set_top_alignment_margin(self, value: float):
+    def panel_lefts(self) -> Sequence[float]:
         """
-        Set a margin to align the top of the panels in this composition
-
-        In figure dimenstions
+        Left values [figure space] of nodes in this tree
         """
 
     @cached_property
     @abc.abstractmethod
-    def rights(self) -> Sequence[float]:
+    def panel_rights(self) -> Sequence[float]:
         """
         Right values [figure space] of nodes in this tree
         """
 
+    @cached_property
     @abc.abstractmethod
-    def set_right_alignment_margin(self, value: float):
+    def panel_bottoms(self) -> Sequence[float]:
         """
-        Set a margin to align the right of the panels in this composition
-
-        In figure dimenstions
+        Bottom values [figure space] of nodes in this tree
         """
 
-    def align_lefts(self):
+    @cached_property
+    @abc.abstractmethod
+    def panel_tops(self) -> Sequence[float]:
         """
-        Align the immediate left edges in this composition
-
-         -----------        -----------
-        |#          |      |  #        |
-        |#          |      |  #        |
-        |#          |      |  #        |
-        |-----------|  ->  |-----------|
-        |  #        |      |  #        |
-        |  #        |      |  #        |
-        |  #        |      |  #        |
-         -----------        -----------
+        Top values [figure space] of nodes in this tree
         """
 
-    def align_bottoms(self):
+    @property
+    def panel_lefts_align(self) -> bool:
         """
-        Align the immediate bottom edges this composition
-
-         -----------        -----------
-        |     |     |      |     |     |
-        |     |     |      |     |     |
-        |     |     |  ->  |     |     |
-        |     |#####|      |#####|#####|
-        |#####|     |      |     |     |
-         -----------        -----------
+        Return True if panel lefts for the nodes are aligned
         """
+        arr = np.array(self.panel_lefts)
+        return all(arr == arr[0])
 
-    def align_rights(self):
+    @property
+    def panel_rights_align(self) -> bool:
         """
-        Align the immediate right edges in this composition
-
-         -----------        -----------
-        |        #  |      |        #  |
-        |        #  |      |        #  |
-        |        #  |      |        #  |
-        |-----------|  ->  |-----------|
-        |          #|      |        #  |
-        |          #|      |        #  |
-        |          #|      |        #  |
-         -----------        -----------
+        Return True if panel rights for the nodes are aligned
         """
+        arr = np.array(self.panel_rights)
+        return all(arr == arr[0])
 
-    def align_tops(self):
+    @property
+    def panel_bottoms_align(self) -> bool:
         """
-        Align the immediate top edges in this composition
-
-         -----------        -----------
-        |#####|     |      |     |     |
-        |     |#####|      |#####|#####|
-        |     |     |  ->  |     |     |
-        |     |     |      |     |     |
-        |     |     |      |     |     |
-         -----------        -----------
+        Return True if panel bottoms for the nodes are aligned
         """
+        arr = np.array(self.panel_bottoms)
+        return all(arr == arr[0])
 
-    def align(self):
+    @property
+    def panel_tops_align(self) -> bool:
         """
-        Align all the edges in this composition & contained compositions
-
-        This function mutates the layout spaces, specifically the
-        alignment_margins along the sides of the plot.
+        Return True if panel tops for the nodes are aligned
         """
-        self.align_lefts()
-        self.align_bottoms()
-        self.align_rights()
-        self.align_tops()
-
-        for item in self.nodes:
-            if isinstance(item, LayoutTree):
-                item.align()
-
-        with suppress(AttributeError):
-            del self.lefts
-
-        with suppress(AttributeError):
-            del self.bottoms
-
-        with suppress(AttributeError):
-            del self.rights
-
-        with suppress(AttributeError):
-            del self.tops
+        arr = np.array(self.panel_tops)
+        return all(arr == arr[0])
 
     @property
     @abc.abstractmethod
     def panel_width(self) -> float:
         """
-        A representative for width for panels of the nodes
+        A representative width for panels of the nodes
         """
 
     @property
     @abc.abstractmethod
     def panel_height(self) -> float:
         """
-        A representative for height for panels of the nodes
+        A representative height for panels of the nodes
         """
 
     @property
     @abc.abstractmethod
     def plot_width(self) -> float:
         """
-        A representative for width for plots of the nodes
+        A representative width for plots of the nodes
         """
 
     @property
@@ -281,211 +292,494 @@ class LayoutTree:
         """
         return [node.panel_height for node in self.nodes]
 
-    def resize(self):
+    @cached_property
+    @abc.abstractmethod
+    def left_tag_width(self) -> float:
         """
-        Resize panels and the entire plots
-
-        This function mutates the composition gridspecs; specifically the
-        width_ratios and height_ratios.
+        A representative width [figure space] for the left tags of the nodes
         """
-        self.resize_widths()
-        self.resize_heights()
-
-        for item in self.nodes:
-            if isinstance(item, LayoutTree):
-                item.resize()
 
-    def resize_widths(self):
+    @cached_property
+    @abc.abstractmethod
+    def right_tag_width(self) -> float:
         """
-        Resize the widths of gridspec so that panels have equal widths
+        A representative width [figure space] for the right tags of the nodes
         """
 
-    def resize_heights(self):
+    @cached_property
+    @abc.abstractmethod
+    def bottom_tag_height(self) -> float:
         """
-        Resize the heights of gridspec so that panels have equal heights
+        A representative height [figure space] for the top tags of the nodes
         """
 
-    @staticmethod
-    def create(
-        cmp: Compose,
-        lookup_spaces: dict[ggplot, LayoutSpaces],
-    ) -> LayoutTree:
+    @cached_property
+    @abc.abstractmethod
+    def top_tag_height(self) -> float:
+        """
+        A representative height [figure space] for the top tags of the nodes
         """
-        Create a LayoutTree for this composition
 
-        Parameters
-        ----------
-        cmp :
-            Composition
-        lookup_spaces :
-            A table to lookup the LayoutSpaces for each plot.
+    @cached_property
+    def left_tag_widths(self) -> list[float]:
+        """
+        The widths of the left tags in this tree
+        """
+        return [node.left_tag_width for node in self.nodes]
 
-        Notes
-        -----
-        LayoutTree works by modifying the `.gridspec` of the compositions,
-        and the `LayoutSpaces` of the plots.
+    @cached_property
+    def right_tag_widths(self) -> list[float]:
         """
-        from plotnine import ggplot
+        The widths of the right tags in this tree
+        """
+        return [node.right_tag_width for node in self.nodes]
 
-        nodes: list[LayoutSpaces | LayoutTree] = []
-        for item in cmp:
-            if isinstance(item, ggplot):
-                nodes.append(lookup_spaces[item])
-            else:
-                nodes.append(LayoutTree.create(item, lookup_spaces))
+    @cached_property
+    def bottom_tag_heights(self) -> list[float]:
+        """
+        The heights of the bottom tags in this tree
+        """
+        return [node.bottom_tag_height for node in self.nodes]
 
-        if isinstance(cmp, OR):
-            return ColumnsTree(cmp.gridspec, nodes)
-        else:
-            return RowsTree(cmp.gridspec, nodes)
+    @cached_property
+    def top_tag_heights(self) -> list[float]:
+        """
+        The heights of the top tags in this tree
+        """
+        return [node.top_tag_height for node in self.nodes]
 
-    def harmonise(self):
+    @property
+    def left_tags_align(self) -> bool:
         """
-        Align and resize plots in composition to look good
+        Return True if the left tags for the nodes are aligned
         """
-        self.align()
-        self.resize()
+        arr = np.array(self.left_tag_widths)
+        return all(arr == arr[0])
 
+    @property
+    def right_tags_align(self) -> bool:
+        """
+        Return True if the right tags for the nodes are aligned
+        """
+        arr = np.array(self.right_tag_widths)
+        return all(arr == arr[0])
 
-@dataclass
-class ColumnsTree(LayoutTree):
-    """
-    Tree with columns at the outermost level
+    @property
+    def bottom_tags_align(self) -> bool:
+        """
+        Return True if the bottom tags for the nodes are aligned
+        """
+        arr = np.array(self.bottom_tag_heights)
+        return all(arr == arr[0])
 
-    e.g. p1 | (p2 / p3)
+    @property
+    def top_tags_align(self) -> bool:
+        """
+        Return True if the top tags for the nodes are aligned
+        """
+        arr = np.array(self.top_tag_heights)
+        return all(arr == arr[0])
 
+    @property
+    def left_axis_titles_align(self) -> bool:
+        """
+        Return True if the left axis titles align
+        """
+        arr = np.array(self.left_axis_title_clearances)
+        return all(arr == arr[0])
 
-         -------------------
-        |         |         |
-        |         |         |
-        |         |         |
-        |         |         |
-        |         |---------|
-        |         |         |
-        |         |         |
-        |         |         |
-        |         |         |
-         -------------------
-    """
+    @property
+    def bottom_axis_titles_align(self) -> bool:
+        """
+        Return True if the bottom axis titles align
+        """
+        arr = np.array(self.bottom_axis_title_clearances)
+        return all(arr == arr[0])
 
     @cached_property
-    def lefts(self):
-        left_item = self.nodes[0]
-        if isinstance(left_item, LayoutSpaces):
-            return [left_item.l.left]
-        else:
-            return left_item.lefts
-
-    def set_left_alignment_margin(self, value: float):
-        left_item = self.nodes[0]
-        if isinstance(left_item, LayoutSpaces):
-            left_item.l.alignment_margin = value
-        else:
-            left_item.set_left_alignment_margin(value)
-
-    def align_bottoms(self):
-        values = max(self.bottoms) - np.array(self.bottoms)
-        for item, value in zip(self.nodes, values):
-            if isinstance(item, LayoutSpaces):
-                item.b.alignment_margin = value
-            else:
-                item.set_bottom_alignment_margin(value)
+    @abc.abstractmethod
+    def left_axis_title_clearance(self) -> float:
+        """
+        Distance between the left y-axis title and the panel
+        """
 
     @cached_property
-    def bottoms(self):
-        values = []
-        for item in self.nodes:
-            if isinstance(item, LayoutSpaces):
-                values.append(item.b.bottom)
-            else:
-                values.append(max(item.bottoms))
-        return values
+    @abc.abstractmethod
+    def bottom_axis_title_clearance(self) -> float:
+        """
+        Distance between the left x-axis title and the panel
+        """
 
-    def set_bottom_alignment_margin(self, value: float):
-        for item in self.nodes:
-            if isinstance(item, LayoutSpaces):
-                item.b.alignment_margin = value
-            else:
-                item.set_bottom_alignment_margin(value)
+    @cached_property
+    def left_axis_title_clearances(self) -> list[float]:
+        """
+        Distances between the left y-axis titles and the panels
+        """
+        return [node.left_axis_title_clearance for node in self.nodes]
 
     @cached_property
-    def rights(self):
-        right_item = self.nodes[-1]
-        if isinstance(right_item, LayoutSpaces):
-            return [right_item.r.right]
-        else:
-            return right_item.rights
+    def bottom_axis_title_clearances(self) -> list[float]:
+        """
+        Distances between the bottom x-axis titles and the panels
+        """
+        return [node.bottom_axis_title_clearance for node in self.nodes]
 
-    def set_right_alignment_margin(self, value: float):
-        right_item = self.nodes[-1]
-        if isinstance(right_item, LayoutSpaces):
-            right_item.r.alignment_margin = value
-        else:
-            right_item.set_right_alignment_margin(value)
+    @abc.abstractmethod
+    def set_left_margin_alignment(self, value: float):
+        """
+        Set a margin to align the left of the panels in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_right_margin_alignment(self, value: float):
+        """
+        Set a margin to align the right of the panels in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_bottom_margin_alignment(self, value: float):
+        """
+        Set a margin to align the bottom of the panels in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_top_margin_alignment(self, value: float):
+        """
+        Set a margin to align the top of the panels in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_left_tag_alignment(self, value: float):
+        """
+        Set the space to align the left tags in this composition
+
+        In figure dimenstions
+        """
 
-    def align_tops(self):
-        values = np.array(self.tops) - min(self.tops)
+    @abc.abstractmethod
+    def set_right_tag_alignment(self, value: float):
+        """
+        Set the space to align the right tags in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_bottom_tag_alignment(self, value: float):
+        """
+        Set the space to align the bottom tags in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_top_tag_alignment(self, value: float):
+        """
+        Set the space to align the top tags in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_left_axis_title_alignment(self, value: float):
+        """
+        Set the space to align the left axis titles in this composition
+
+        In figure dimenstions
+        """
+
+    @abc.abstractmethod
+    def set_bottom_axis_title_alignment(self, value: float):
+        """
+        Set the space to align the bottom axis titles in this composition
+
+        In figure dimenstions
+        """
+
+
+@dataclass
+class ColumnsTree(LayoutTree):
+    """
+    Tree with columns at the outermost level
+
+    e.g. p1 | (p2 / p3)
+
+
+         -------------------
+        |         |         |
+        |         |         |
+        |         |         |
+        |         |         |
+        |         |---------|
+        |         |         |
+        |         |         |
+        |         |         |
+        |         |         |
+         -------------------
+    """
+
+    def align(self):
+        self.align_top_tags()
+        self.align_bottom_tags()
+        self.align_panel_tops()
+        self.align_panel_bottoms()
+        self.align_sub_compositions()
+
+    def align_axis_titles(self):
+        self.align_bottom_axis_titles()
+        for tree in self.sub_compositions:
+            tree.align_axis_titles()
+
+    def resize(self):
+        """
+        Resize the widths of gridspec so that panels have equal widths
+        """
+        # The new width of each panel is the average width of all
+        # the panels plus all the space to the left and right
+        # of the panels.
+        plot_widths = np.array(self.plot_widths)
+        panel_widths = np.array(self.panel_widths)
+        non_panel_space = plot_widths - panel_widths
+        new_plot_widths = panel_widths.mean() + non_panel_space
+        width_ratios = new_plot_widths / new_plot_widths.min()
+        self.gridspec.set_width_ratios(width_ratios)
+        self.resize_sub_compositions()
+
+    def align_panel_bottoms(self):
+        """
+        Align the immediate bottom edges this composition
+
+         -----------        -----------
+        |     |     |      |     |     |
+        |     |     |      |     |     |
+        |     |     |  ->  |     |     |
+        |     |#####|      |#####|#####|
+        |#####|     |      |     |     |
+         -----------        -----------
+        """
+        # If panels are aligned and have a non-zero margin_alignment,
+        # aligning them again will set that value to zero and undoes
+        # the alignment.
+        if self.panel_bottoms_align:
+            return
+
+        values = max(self.panel_bottoms) - np.array(self.panel_bottoms)
+        for item, value in zip(self.nodes, values):
+            if isinstance(item, LayoutSpaces):
+                item.b.margin_alignment = value
+            else:
+                item.set_bottom_margin_alignment(value)
+
+        del self.panel_bottoms
+
+    def align_panel_tops(self):
+        """
+        Align the immediate top edges in this composition
+
+         -----------        -----------
+        |#####|     |      |     |     |
+        |     |#####|      |#####|#####|
+        |     |     |  ->  |     |     |
+        |     |     |      |     |     |
+        |     |     |      |     |     |
+         -----------        -----------
+        """
+        if self.panel_tops_align:
+            return
+
+        values = np.array(self.panel_tops) - min(self.panel_tops)
+        for item, value in zip(self.nodes, values):
+            if isinstance(item, LayoutSpaces):
+                item.t.margin_alignment = value
+            else:
+                item.set_top_margin_alignment(value)
+
+        del self.panel_tops
+
+    def align_bottom_tags(self):
+        if self.bottom_tags_align:
+            return
+
+        values = cast(
+            "Sequence[float]",
+            max(self.bottom_tag_heights) - np.array(self.bottom_tag_heights),
+        )
+        for item, value in zip(self.nodes, values):
+            if isinstance(item, LayoutSpaces):
+                item.l.tag_alignment = value
+            else:
+                item.set_bottom_tag_alignment(value)
+
+    def align_top_tags(self):
+        if self.top_tags_align:
+            return
+
+        values = cast(
+            "Sequence[float]",
+            max(self.top_tag_heights) - np.array(self.top_tag_heights),
+        )
         for item, value in zip(self.nodes, values):
             if isinstance(item, LayoutSpaces):
-                item.t.alignment_margin = value
+                item.t.tag_alignment = value
             else:
-                item.set_top_alignment_margin(value)
+                item.set_top_tag_alignment(value)
+
+    def align_bottom_axis_titles(self):
+        if self.bottom_axis_titles_align:
+            pass
+
+        values = max(self.bottom_axis_title_clearances) - np.array(
+            self.bottom_axis_title_clearances
+        )
+        # We ignore 0 values since they can undo values
+        # set to align this composition with an outer one.
+        for item, value in zip(self.nodes, values):
+            if value == 0:
+                continue
+            if isinstance(item, LayoutSpaces):
+                item.b.axis_title_alignment = value
+            else:
+                item.set_bottom_axis_title_alignment(value)
+
+    @cached_property
+    def panel_lefts(self):
+        left_item = self.nodes[0]
+        if isinstance(left_item, LayoutSpaces):
+            return [left_item.l.panel_left]
+        else:
+            return left_item.panel_lefts
+
+    @cached_property
+    def panel_rights(self):
+        right_item = self.nodes[-1]
+        if isinstance(right_item, LayoutSpaces):
+            return [right_item.r.panel_right]
+        else:
+            return right_item.panel_rights
 
     @cached_property
-    def tops(self):
+    def panel_bottoms(self):
         values = []
         for item in self.nodes:
             if isinstance(item, LayoutSpaces):
-                values.append(item.t.top)
+                values.append(item.b.panel_bottom)
             else:
-                values.append(min(item.tops))
+                values.append(max(item.panel_bottoms))
         return values
 
-    def set_top_alignment_margin(self, value: float):
+    @cached_property
+    def panel_tops(self):
+        values = []
         for item in self.nodes:
             if isinstance(item, LayoutSpaces):
-                item.t.alignment_margin = value
+                values.append(item.t.panel_top)
             else:
-                item.set_top_alignment_margin(value)
+                values.append(min(item.panel_tops))
+        return values
 
     @property
     def panel_width(self) -> float:
-        """
-        A representative for width for panels of the nodes
-        """
         return sum(self.panel_widths)
 
     @property
     def panel_height(self) -> float:
-        """
-        A representative for height for panels of the nodes
-        """
         return float(np.mean(self.panel_heights))
 
     @property
     def plot_width(self) -> float:
-        """
-        A representative for width for plots of the nodes
-        """
         return sum(self.plot_widths)
 
     @property
     def plot_height(self) -> float:
-        """
-        A representative for height for plots of the nodes
-        """
         return max(self.plot_heights)
 
-    def resize_widths(self):
-        # The new width of each panel is the average width of all
-        # the panels plus all the space to the left and right
-        # of the panels.
-        plot_widths = np.array(self.plot_widths)
-        panel_widths = np.array(self.panel_widths)
-        non_panel_space = plot_widths - panel_widths
-        new_plot_widths = panel_widths.mean() + non_panel_space
-        width_ratios = new_plot_widths / new_plot_widths.min()
-        self.gridspec.set_width_ratios(width_ratios)
+    @cached_property
+    def left_tag_width(self) -> float:
+        return self.left_tag_widths[0]
+
+    @cached_property
+    def right_tag_width(self) -> float:
+        return self.right_tag_widths[-1]
+
+    @cached_property
+    def bottom_tag_height(self) -> float:
+        return max(self.bottom_tag_heights)
+
+    @cached_property
+    def top_tag_height(self) -> float:
+        return max(self.top_tag_heights)
+
+    @cached_property
+    def left_axis_title_clearance(self) -> float:
+        return self.left_axis_title_clearances[0]
+
+    @cached_property
+    def bottom_axis_title_clearance(self) -> float:
+        return max(self.bottom_axis_title_clearances)
+
+    def set_left_margin_alignment(self, value: float):
+        left_item = self.nodes[0]
+        if isinstance(left_item, LayoutSpaces):
+            left_item.l.margin_alignment = value
+        else:
+            left_item.set_left_margin_alignment(value)
+
+    def set_right_margin_alignment(self, value: float):
+        right_item = self.nodes[-1]
+        if isinstance(right_item, LayoutSpaces):
+            right_item.r.margin_alignment = value
+        else:
+            right_item.set_right_margin_alignment(value)
+
+    def set_bottom_margin_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.b.margin_alignment = value
+            else:
+                item.set_bottom_margin_alignment(value)
+
+    def set_top_margin_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.t.margin_alignment = value
+            else:
+                item.set_top_margin_alignment(value)
+
+    def set_bottom_tag_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.l.tag_alignment = value
+            else:
+                item.set_bottom_tag_alignment(value)
+
+    def set_top_tag_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.t.tag_alignment = value
+            else:
+                item.set_top_tag_alignment(value)
+
+    def set_bottom_axis_title_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.b.axis_title_alignment = value
+            else:
+                item.set_bottom_axis_title_alignment(value)
+
+    def set_left_axis_title_alignment(self, value: float):
+        left_item = self.nodes[0]
+        if isinstance(left_item, LayoutSpaces):
+            left_item.l.axis_title_alignment = value
+        else:
+            left_item.set_left_axis_title_alignment(value)
 
 
 @dataclass
@@ -506,120 +800,277 @@ class RowsTree(LayoutTree):
          -------------------
     """
 
-    def align_lefts(self):
-        values = max(self.lefts) - np.array(self.lefts)
+    def align(self):
+        self.align_left_tags()
+        self.align_right_tags()
+        self.align_panel_lefts()
+        self.align_panel_rights()
+        self.align_sub_compositions()
+
+    def align_axis_titles(self):
+        self.align_left_axis_titles()
+        for tree in self.sub_compositions:
+            tree.align_axis_titles()
+
+    def resize(self):
+        """
+        Resize the heights of gridspec so that panels have equal heights
+
+        This method resizes (recursively) the contained compositions
+        """
+        # The new height of each panel is the average width of all
+        # the panels plus all the space above and below the panels.
+        plot_heights = np.array(self.plot_heights)
+        panel_heights = np.array(self.panel_heights)
+        non_panel_space = plot_heights - panel_heights
+        new_plot_heights = panel_heights.mean() + non_panel_space
+        height_ratios = new_plot_heights / new_plot_heights.max()
+        self.gridspec.set_height_ratios(height_ratios)
+        self.resize_sub_compositions()
+
+    def align_panel_lefts(self):
+        """
+        Align the immediate left edges in this composition
+
+         -----------        -----------
+        |#          |      |  #        |
+        |#          |      |  #        |
+        |#          |      |  #        |
+        |-----------|  ->  |-----------|
+        |  #        |      |  #        |
+        |  #        |      |  #        |
+        |  #        |      |  #        |
+         -----------        -----------
+        """
+        if self.panel_lefts_align:
+            return
+
+        values = max(self.panel_lefts) - np.array(self.panel_lefts)
         for item, value in zip(self.nodes, values):
             if isinstance(item, LayoutSpaces):
-                item.l.alignment_margin = value
+                item.l.margin_alignment = value
             else:
-                item.set_left_alignment_margin(value)
+                item.set_left_margin_alignment(value)
 
-    @cached_property
-    def lefts(self):
-        values = []
-        for item in self.nodes:
+        del self.panel_lefts
+
+    def align_panel_rights(self):
+        """
+        Align the immediate right edges in this composition
+
+         -----------        -----------
+        |        #  |      |        #  |
+        |        #  |      |        #  |
+        |        #  |      |        #  |
+        |-----------|  ->  |-----------|
+        |          #|      |        #  |
+        |          #|      |        #  |
+        |          #|      |        #  |
+         -----------        -----------
+        """
+        if self.panel_rights_align:
+            return
+
+        values = np.array(self.panel_rights) - min(self.panel_rights)
+        for item, value in zip(self.nodes, values):
             if isinstance(item, LayoutSpaces):
-                values.append(item.l.left)
+                item.r.margin_alignment = value
             else:
-                values.append(max(item.lefts))
-        return values
+                item.set_right_margin_alignment(value)
 
-    def set_left_alignment_margin(self, value: float):
-        for item in self.nodes:
+        del self.panel_rights
+
+    def align_left_tags(self):
+        """
+        Make all the left tags takeup the same amount of space
+
+
+        Given
+
+                      V
+         ------------------------------------
+        | plot_margin | tag | artists        |
+        |------------------------------------|
+        | plot_margin | A long tag | artists |
+         ------------------------------------
+
+                      V
+         ------------------------------------
+        | plot_margin | #######tag | artists |
+        |------------------------------------|
+        | plot_margin | A long tag | artists |
+         ------------------------------------
+        """
+        if self.left_tags_align:
+            return
+
+        values = cast(
+            "Sequence[float]",
+            max(self.left_tag_widths) - np.array(self.left_tag_widths),
+        )
+        for item, value in zip(self.nodes, values):
             if isinstance(item, LayoutSpaces):
-                item.l.alignment_margin = value
+                item.l.tag_alignment = value
             else:
-                item.set_left_alignment_margin(value)
+                item.set_left_tag_alignment(value)
 
-    @cached_property
-    def bottoms(self):
-        bottom_item = self.nodes[-1]
-        if isinstance(bottom_item, LayoutSpaces):
-            return [bottom_item.b.bottom]
-        else:
-            return bottom_item.bottoms
+    def align_right_tags(self):
+        if self.right_tags_align:
+            return
 
-    def set_bottom_alignment_margin(self, value: float):
-        bottom_item = self.nodes[-1]
-        if isinstance(bottom_item, LayoutSpaces):
-            bottom_item.b.alignment_margin = value
-        else:
-            bottom_item.set_bottom_alignment_margin(value)
+        values = cast(
+            "Sequence[float]",
+            max(self.right_tag_widths) - np.array(self.right_tag_widths),
+        )
+        for item, value in zip(self.nodes, values):
+            if isinstance(item, LayoutSpaces):
+                item.r.tag_alignment = value
+            else:
+                item.set_right_tag_alignment(value)
 
-    def align_rights(self):
-        values = np.array(self.rights) - min(self.rights)
+    def align_left_axis_titles(self):
+        if self.left_axis_titles_align:
+            pass
+
+        values = max(self.left_axis_title_clearances) - np.array(
+            self.left_axis_title_clearances
+        )
         for item, value in zip(self.nodes, values):
+            if value == 0:
+                continue
             if isinstance(item, LayoutSpaces):
-                item.r.alignment_margin = value
+                item.l.axis_title_alignment = value
             else:
-                item.set_right_alignment_margin(value)
+                item.set_left_axis_title_alignment(value)
 
     @cached_property
-    def rights(self):
+    def panel_lefts(self):
         values = []
         for item in self.nodes:
             if isinstance(item, LayoutSpaces):
-                values.append(item.r.right)
+                values.append(item.l.panel_left)
             else:
-                values.append(min(item.rights))
+                values.append(max(item.panel_lefts))
         return values
 
-    def set_right_alignment_margin(self, value: float):
+    @cached_property
+    def panel_rights(self):
+        values = []
         for item in self.nodes:
             if isinstance(item, LayoutSpaces):
-                item.r.alignment_margin = value
+                values.append(item.r.panel_right)
             else:
-                item.set_right_alignment_margin(value)
+                values.append(min(item.panel_rights))
+        return values
 
     @cached_property
-    def tops(self):
-        top_item = self.nodes[0]
-        if isinstance(top_item, LayoutSpaces):
-            return [top_item.t.top]
+    def panel_bottoms(self):
+        bottom_item = self.nodes[-1]
+        if isinstance(bottom_item, LayoutSpaces):
+            return [bottom_item.b.panel_bottom]
         else:
-            return top_item.tops
+            return bottom_item.panel_bottoms
 
-    def set_top_alignment_margin(self, value: float):
+    @cached_property
+    def panel_tops(self):
         top_item = self.nodes[0]
         if isinstance(top_item, LayoutSpaces):
-            top_item.t.alignment_margin = value
+            return [top_item.t.panel_top]
         else:
-            top_item.set_top_alignment_margin(value)
+            return top_item.panel_tops
 
     @property
     def panel_width(self) -> float:
-        """
-        A representative for width for panels of the nodes
-        """
         return float(np.mean(self.panel_widths))
 
     @property
     def panel_height(self) -> float:
-        """
-        A representative for height for panels of the nodes
-        """
         return sum(self.panel_heights)
 
     @property
     def plot_width(self) -> float:
-        """
-        A representative for width for plots of the nodes
-        """
         return max(self.plot_widths)
 
     @property
     def plot_height(self) -> float:
-        """
-        A representative for height for plots of the nodes
-        """
         return sum(self.plot_heights)
 
-    def resize_heights(self):
-        # The new width of each panel is the average width of all
-        # the panels plus all the space above and below the panels.
-        plot_heights = np.array(self.plot_heights)
-        panel_heights = np.array(self.panel_heights)
-        non_panel_space = plot_heights - panel_heights
-        new_plot_heights = panel_heights.mean() + non_panel_space
-        height_ratios = new_plot_heights / new_plot_heights.max()
-        self.gridspec.set_height_ratios(height_ratios)
+    @cached_property
+    def left_tag_width(self) -> float:
+        return max(self.left_tag_widths)
+
+    @cached_property
+    def right_tag_width(self) -> float:
+        return max(self.right_tag_widths)
+
+    @cached_property
+    def top_tag_height(self) -> float:
+        return self.top_tag_heights[0]
+
+    @cached_property
+    def bottom_tag_height(self) -> float:
+        return self.bottom_tag_heights[-1]
+
+    @cached_property
+    def left_axis_title_clearance(self) -> float:
+        return max(self.left_axis_title_clearances)
+
+    @cached_property
+    def bottom_axis_title_clearance(self) -> float:
+        return self.bottom_axis_title_clearances[-1]
+
+    def set_left_margin_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.l.margin_alignment = value
+            else:
+                item.set_left_margin_alignment(value)
+
+    def set_right_margin_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.r.margin_alignment = value
+            else:
+                item.set_right_margin_alignment(value)
+
+    def set_bottom_margin_alignment(self, value: float):
+        bottom_item = self.nodes[-1]
+        if isinstance(bottom_item, LayoutSpaces):
+            bottom_item.b.margin_alignment = value
+        else:
+            bottom_item.set_bottom_margin_alignment(value)
+
+    def set_top_margin_alignment(self, value: float):
+        top_item = self.nodes[0]
+        if isinstance(top_item, LayoutSpaces):
+            top_item.t.margin_alignment = value
+        else:
+            top_item.set_top_margin_alignment(value)
+
+    def set_left_tag_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.l.tag_alignment = value
+            else:
+                item.set_left_tag_alignment(value)
+
+    def set_right_tag_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.r.tag_alignment = value
+            else:
+                item.set_right_tag_alignment(value)
+
+    def set_left_axis_title_alignment(self, value: float):
+        for item in self.nodes:
+            if isinstance(item, LayoutSpaces):
+                item.l.axis_title_alignment = value
+            else:
+                item.set_left_axis_title_alignment(value)
+
+    def set_bottom_axis_title_alignment(self, value: float):
+        bottom_item = self.nodes[-1]
+        if isinstance(bottom_item, LayoutSpaces):
+            bottom_item.b.axis_title_alignment = value
+        else:
+            bottom_item.set_bottom_axis_title_alignment(value)