Add CompatibilityAxes (as opposed to compatibilityArtist)

Author: ksunden

data_prototype/artist.py

@@ -91,3 +91,83 @@ def draw(self, renderer, edges=None):
             )
 
         self._artist.draw(renderer, edges)
+
+
+class CompatibilityAxes:
+    """A compatibility shim to add to traditional matplotlib axes.
+
+    At this time features are implemented on an "as needed" basis, and many
+    are only implemented insofar as they do not fail, not necessarily providing
+    full functionality of a full MPL Artist.
+
+    The idea is to keep the new Artist class as minimal as possible.
+    As features are added this may shrink.
+
+    The main thing we are trying to avoid is the reliance on the axes/figure
+
+    Ultimately for useability, whatever remains shimmed out here may be rolled in as
+    some form of gaurded option to ``Artist`` itself, but a firm dividing line is
+    useful for avoiding accidental dependency.
+    """
+
+    def __init__(self, axes):
+        self.axes = axes
+        self.figure = None
+        self._clippath = None
+        self.zorder = 2
+        self._children = []
+
+    def set_figure(self, fig):
+        self.figure = fig
+
+    def is_transform_set(self):
+        return True
+
+    def get_mouseover(self):
+        return False
+
+    def get_clip_path(self):
+        self._clippath
+
+    def set_clip_path(self, path):
+        self._clippath = path
+
+    def get_animated(self):
+        return False
+
+    def draw(self, renderer, edges=None):
+        if edges is None:
+            edges = []
+
+        if self.axes is not None:
+            desc: Desc = Desc(("N",), coordinates="data")
+            xy: dict[str, Desc] = {"x": desc, "y": desc}
+            edges.append(
+                TransformEdge(
+                    "data",
+                    xy,
+                    desc_like(xy, coordinates="axes"),
+                    transform=self.axes.transData - self.axes.transAxes,
+                )
+            )
+            edges.append(
+                TransformEdge(
+                    "axes",
+                    desc_like(xy, coordinates="axes"),
+                    desc_like(xy, coordinates="display"),
+                    transform=self.axes.transAxes,
+                )
+            )
+
+        # TODO independent zorder
+        for c in self._children:
+            c.draw(renderer, edges)
+
+    def add_artist(self, artist):
+        self._children.append(artist)
+
+    def set_xlim(self, min_=None, max_=None):
+        self.axes.set_xlim(min_, max_)
+
+    def set_ylim(self, min_=None, max_=None):
+        self.axes.set_ylim(min_, max_)

examples/first.py

@@ -11,7 +11,7 @@
 import numpy as np
 import pandas as pd
 
-from data_prototype.artist import CompatibilityArtist
+from data_prototype.artist import CompatibilityAxes
 from data_prototype.line import Line
 from data_prototype.containers import FuncContainer, SeriesContainer
 
@@ -30,9 +30,11 @@
     markersize=12,
 )
 
-fig, ax = plt.subplots()
-ax.add_artist(CompatibilityArtist(lw))
-ax.add_artist(CompatibilityArtist(lw2))
+fig, nax = plt.subplots()
+ax = CompatibilityAxes(nax)
+nax.add_artist(ax)
+ax.add_artist(lw)
+ax.add_artist(lw2)
 ax.set_xlim(0, np.pi * 4)
 ax.set_ylim(-1.1, 1.1)