WIP, lines containers decorator

ksunden · ksunden · commit b90572012739 · 2025-10-17T13:26:37.000-05:00
diff --git a/lib/matplotlib/_containers.py b/lib/matplotlib/_containers.py
@@ -0,0 +1,37 @@
+from mpl_data_containers.description import Desc, desc_like
+from mpl_data_containers.graph import Graph
+from mpl_data_containers.conversion_edge import TransformEdge
+
+
+def containerize_draw(draw_func):
+    def draw(self, renderer, *, graph=None):
+        if graph is None:
+            graph = Graph([])
+
+        ax = self.axes
+        if ax is None:
+            implicit_graph = Graph([])
+        else:
+            desc: Desc = Desc(("N",), coordinates="data")
+            xy: dict[str, Desc] = {"x": desc, "y": desc}
+            implicit_graph = Graph(
+                [
+                    TransformEdge(
+                        "data",
+                        xy,
+                        desc_like(xy, coordinates="axes"),
+                        transform=ax.transData - ax.transAxes,
+                    ),
+                    TransformEdge(
+                        "axes",
+                        desc_like(xy, coordinates="axes"),
+                        desc_like(xy, coordinates="display"),
+                        transform=ax.transAxes,
+                    ),
+                ],
+                aliases=(("parent", "axes"),),
+            )
+
+        return draw_func(self, renderer, graph=graph+implicit_graph)
+
+    return draw
diff --git a/lib/matplotlib/lines.py b/lib/matplotlib/lines.py
@@ -18,6 +18,7 @@
 from .path import Path
 from .transforms import Bbox, BboxTransformTo, TransformedPath
 from ._enums import JoinStyle, CapStyle
+from ._containers import containerize_draw
 
 # Imported here for backward compatibility, even though they don't
 # really belong.
@@ -424,6 +425,7 @@ def __init__(self, xdata, ydata, *,
         self._transformed_path = None
         self._subslice = False
         self._x_filled = None  # used in subslicing; only x is needed
+        self._container = None
 
         self.set_data(xdata, ydata)
 
@@ -744,7 +746,8 @@ def set_transform(self, t):
         super().set_transform(t)
 
     @allow_rasterization
-    def draw(self, renderer):
+    @containerize_draw
+    def draw(self, renderer, *, graph=None):
         # docstring inherited
 
         if not self.get_visible():
