ENH: add a working 2D example wrapping AxesImage

Author: tacaswell

data_prototype/wrappers.py

@@ -1,4 +1,5 @@
 from matplotlib.lines import Line2D as _Line2D
+from matplotlib.image import AxesImage as _AxesImage
 
 
 class ProxyWrapper:
@@ -50,3 +51,18 @@ def draw(self, renderer):
         data = self._query_and_transform(renderer)
         self._wrapped_instance.set_data(data["x"], data["y"])
         return self._wrapped_instance.draw(renderer)
+
+
+class ImageWrapper(ProxyWrapper):
+    _wrapped_class = _AxesImage
+
+    def __init__(self, data, nus=None, /, **kwargs):
+        super().__init__(data, nus)
+        kwargs.setdefault("origin", "lower")
+        self._wrapped_instance = self._wrapped_class(None, **kwargs)
+
+    def draw(self, renderer):
+        data = self._query_and_transform(renderer)
+        self._wrapped_instance.set_array(data["image"])
+        self._wrapped_instance.set_extent(data["extent"])
+        return self._wrapped_instance.draw(renderer)

examples/2Dfunc.py

@@ -0,0 +1,30 @@
+"""
+=====================
+A functional 2D image
+=====================
+
+
+"""
+
+import matplotlib.pyplot as plt
+import numpy as np
+
+from data_prototype.wrappers import ImageWrapper
+from data_prototype.containers import FuncContainer
+
+from matplotlib.colors import Normalize
+
+fc = FuncContainer(
+    {},
+    xyfuncs={
+        "extent": lambda x, y: [x[0], x[-1], y[0], y[-1]],
+        "image": lambda x, y: np.sin(x).reshape(1, -1) * np.cos(y).reshape(-1, 1),
+    },
+)
+im = ImageWrapper(fc, norm=Normalize(-1, 1))
+
+fig, ax = plt.subplots()
+ax.add_artist(im)
+ax.set_xlim(-5, 5)
+ax.set_ylim(-5, 5)
+fig.colorbar(im)