Added add_imshow

Author: max-models

src/maxplotlib/subfigure/line_plot.py

@@ -91,6 +91,19 @@ def add_line(self, x_data, y_data, layer=0, plot_type='plot', **kwargs):
             self.layered_line_data[layer].append(ld)
         else:
             self.layered_line_data[layer] = [ld]
+    
+    def add_imshow(self, data, layer=0, plot_type='imshow', **kwargs):
+        ld = {
+            "data": np.array(data),
+            "layer": layer,
+            "plot_type": plot_type,
+            "kwargs": kwargs,
+        }
+        self.line_data.append(ld)
+        if layer in self.layered_line_data:
+            self.layered_line_data[layer].append(ld)
+        else:
+            self.layered_line_data[layer] = [ld]
 
     @property
     def layers(self):
@@ -122,6 +135,11 @@ def plot_matplotlib(self, ax, layers=None):
                         (line["y"] + self._yshift) * self._yscale,
                         **line["kwargs"],
                     )
+                elif line["plot_type"] == "imshow":
+                    ax.imshow(
+                        line["data"],
+                        **line["kwargs"],
+                    )
             # if self._caption:
             #     ax.set_title(self._caption)
             if self._title: