import Canvas directly

Author: max-models

src/maxplotlib/__init__.py

@@ -0,0 +1,3 @@
+from maxplotlib.canvas.canvas import Canvas
+
+__all__ = ["Canvas"]

src/maxplotlib/canvas/canvas.py

@@ -37,11 +37,15 @@ def __init__(self, **kwargs):
 
         # Dictionary to store lines for each subplot
         # Key: (row, col), Value: list of lines with their data and kwargs
-        self.subplots = {}
+        self._subplots = {}
         self._num_subplots = 0
 
         self._subplot_matrix = [[None] * self.ncols for _ in range(self.nrows)]
 
+    @property
+    def subplots(self):
+        return self._subplots
+
     @property
     def layers(self):
         layers = []
@@ -88,12 +92,12 @@ def add_tikzfigure(self, **kwargs):
 
         # Store the LinePlot instance by its position for easy access
         if label is None:
-            self.subplots[(row, col)] = tikz_figure
+            self._subplots[(row, col)] = tikz_figure
         else:
-            self.subplots[label] = tikz_figure
+            self._subplots[label] = tikz_figure
         return tikz_figure
 
-    def add_subplot(self, **kwargs):
+    def add_subplot(self, col: int | None = None, row: int | None = None, label: str | None = None, **kwargs):
         """
         Adds a subplot to the figure.
 
@@ -103,21 +107,19 @@ def add_subplot(self, **kwargs):
             - row (int): Row index for the subplot.
             - label (str): Label to identify the subplot.
         """
-        col = kwargs.get("col", None)
-        row = kwargs.get("row", None)
-        label = kwargs.get("label", None)
+
 
         row, col = self.generate_new_rowcol(row, col)
 
         # Initialize the LinePlot for the given subplot position
-        line_plot = lp.LinePlot(**kwargs)
+        line_plot = lp.LinePlot(col=col, row=row, label=label, **kwargs)
         self._subplot_matrix[row][col] = line_plot
 
         # Store the LinePlot instance by its position for easy access
         if label is None:
-            self.subplots[(row, col)] = line_plot
+            self._subplots[(row, col)] = line_plot
         else:
-            self.subplots[label] = line_plot
+            self._subplots[label] = line_plot
         return line_plot
 
     def savefig(
@@ -159,19 +161,31 @@ def savefig(
                 if verbose:
                     print(f"Saved {full_filepath}")
 
-    def plot(self, backend="matplotlib", show=True, savefig=False, layers=None):
+    def plot(self, backend="matplotlib", savefig=False, layers=None):
+        if backend == "matplotlib":
+            return self.plot_matplotlib(savefig=savefig, layers=layers)
+        elif backend == "plotly":
+            return self.plot_plotly(savefig=savefig)
+        else:
+            raise ValueError("Invalid backend")
+
+    def show(self, backend="matplotlib"):
         if backend == "matplotlib":
-            return self.plot_matplotlib(show=show, savefig=savefig, layers=layers)
+            fig, axs = self.plot(backend="matplotlib", savefig=False, layers=None)
+            print('hmm')
+            self._matplotlib_fig.show()
+            plt.show()
         elif backend == "plotly":
-            self.plot_plotly(show=show, savefig=savefig)
+            plot = self.plot_plotly(savefig=False)
+        else:
+            raise ValueError("Invalid backend")
 
-    def plot_matplotlib(self, show=True, savefig=False, layers=None, usetex=False):
+    def plot_matplotlib(self, savefig=False, layers=None, usetex=False):
         """
         Generate and optionally display the subplots.
 
         Parameters:
         filename (str, optional): Filename to save the figure.
-        show (bool): Whether to display the plot.
         """
 
         tex_fonts = plt_utils.setup_tex_fonts(fontsize=self.fontsize, usetex=usetex)
@@ -205,17 +219,11 @@ def plot_matplotlib(self, show=True, savefig=False, layers=None, usetex=False):
 
         for (row, col), subplot in self.subplots.items():
             ax = axes[row][col]
-            # print(f"{subplot = }")
             subplot.plot_matplotlib(ax, layers=layers)
             # ax.set_title(f"Subplot ({row}, {col})")
             ax.grid()
-        # Set caption, labels, etc., if needed
-        # plt.tight_layout()
 
-        if show:
-            plt.show()
-        # else:
-        #     plt.close()
+        # Set caption, labels, etc., if needed
         self._plotted = True
         self._matplotlib_fig = fig
         self._matplotlib_axes = axes
@@ -271,8 +279,8 @@ def plot_plotly(self, show=True, savefig=None, usetex=False):
             fig.write_image(savefig)
 
         # Show or return the figure
-        if show:
-            fig.show()
+        # if show:
+        #     fig.show()
         return fig
 
     # Property getters

tutorials/tutorial_01.ipynb

@@ -16,7 +16,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import maxplotlib.canvas.canvas as canvas\n",
+    "from maxplotlib import Canvas\n",
     "\n",
     "%load_ext autoreload\n",
     "%autoreload 2"
@@ -29,14 +29,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=\"17cm\", ratio=0.5, fontsize=12)\n",
+    "c = Canvas(width=\"17cm\", ratio=0.5, fontsize=12)\n",
     "sp = c.add_subplot(\n",
     "    grid=True, xlabel=\"(x - 10) * 0.1\", ylabel=\"10y\", yscale=10, xshift=-10, xscale=0.1\n",
     ")\n",
     "sp.add_line([0, 1, 2, 3], [0, 1, 4, 9], label=\"Line 1\")\n",
     "sp.add_line([0, 1, 2, 3], [0, 2, 3, 4], linestyle=\"dashed\", color=\"red\", label=\"Line 2\")\n",
-    "c.plot()\n",
-    "c.savefig(filename=\"tutorial_01_01.pdf\")"
+    "c.show()"
    ]
   },
   {
@@ -46,17 +45,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=\"17cm\", ncols=2, nrows=2, ratio=0.5)\n",
+    "c = Canvas(width=\"17cm\", ncols=2, nrows=2, ratio=0.5)\n",
     "sp = c.add_subplot(grid=True)\n",
     "c.add_subplot(row=1)\n",
     "sp2 = c.add_subplot(row=1, legend=False)\n",
     "sp.add_line([0, 1, 2, 3], [0, 1, 4, 9], label=\"Line 1\")\n",
     "sp2.add_line(\n",
     "    [0, 1, 2, 3], [0, 2, 3, 4], linestyle=\"dashed\", color=\"red\", label=\"Line 2\"\n",
     ")\n",
-    "c.plot(backend=\"matplotlib\")\n",
-    "c.plot(backend=\"plotly\")\n",
-    "c.savefig(filename=\"tutorial_01_02.pdf\")"
+    "c.show()"
    ]
   },
   {
@@ -67,15 +64,13 @@
    "outputs": [],
    "source": [
     "# Test with plotly backend\n",
-    "c = canvas.Canvas(width=\"17cm\", ratio=0.5)\n",
+    "c = Canvas(width=\"17cm\", ratio=0.5)\n",
     "sp = c.add_subplot(\n",
     "    grid=True, xlabel=\"x (mm)\", ylabel=\"10y\", yscale=10, xshift=-10, xscale=0.1\n",
     ")\n",
     "sp.add_line([0, 1, 2, 3], [0, 1, 4, 9], label=\"Line 1\", linestyle=\"-.\")\n",
     "sp.add_line([0, 1, 2, 3], [0, 2, 3, 4], linestyle=\"dashed\", color=\"red\", label=\"Line 2\")\n",
-    "c.plot(backend=\"matplotlib\")\n",
-    "c.plot(backend=\"plotly\")\n",
-    "c.savefig(filename=\"tutorial_01_03.pdf\")"
+    "c.show()"
    ]
   }
  ],
@@ -95,7 +90,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tutorials/tutorial_02.ipynb

@@ -15,7 +15,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import maxplotlib.canvas.canvas as canvas\n",
+    "from maxplotlib import Canvas\n",
     "\n",
     "%load_ext autoreload\n",
     "%autoreload 2"
@@ -28,7 +28,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=800, ratio=0.5)\n",
+    "c = Canvas(width=800, ratio=0.5)\n",
     "tikz = c.add_tikzfigure(grid=False)\n",
     "\n",
     "# Add nodes\n",
@@ -61,7 +61,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(ncols=2, width=\"20cm\", ratio=0.5)\n",
+    "c = Canvas(ncols=2, width=\"20cm\", ratio=0.5)\n",
     "tikz = c.add_tikzfigure(grid=False)\n",
     "\n",
     "# Add nodes\n",
@@ -111,7 +111,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=800, ratio=0.5)\n",
+    "c = Canvas(width=800, ratio=0.5)\n",
     "tikz = c.add_tikzfigure(grid=False)\n",
     "\n",
     "# Add nodes\n",
@@ -146,7 +146,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tutorials/tutorial_03.ipynb

@@ -15,7 +15,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import maxplotlib.canvas.canvas as canvas"
+    "from maxplotlib import Canvas"
    ]
   },
   {
@@ -25,16 +25,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=\"17cm\", ratio=0.5)\n",
+    "c = Canvas(width=\"17cm\", ratio=0.5)\n",
     "sp = c.add_subplot(\n",
     "    grid=False, xlabel=\"(x - 10) * 0.1\", ylabel=\"10y\", yscale=10, xshift=-10, xscale=0.1\n",
     ")\n",
     "sp.add_line([0, 1, 2, 3], [0, 1, 4, 9], label=\"Line 1\", layer=0)\n",
     "sp.add_line(\n",
     "    [0, 1, 2, 3], [0, 2, 3, 4], linestyle=\"dashed\", color=\"red\", label=\"Line 2\", layer=1\n",
     ")\n",
-    "# c.plot()\n",
-    "c.savefig(layer_by_layer=True, filename=\"tutorial_03_01.pdf\")"
+    "c.show()"
    ]
   }
  ],
@@ -59,7 +58,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tutorials/tutorial_04.ipynb

@@ -29,7 +29,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import maxplotlib.canvas.canvas as canvas"
+    "from maxplotlib import Canvas"
    ]
   },
   {
@@ -39,7 +39,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=800, ratio=0.5)\n",
+    "c = Canvas(width=800, ratio=0.5)\n",
     "tikz = c.add_tikzfigure(grid=False)"
    ]
   },
@@ -154,7 +154,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tutorials/tutorial_05.ipynb

@@ -17,7 +17,7 @@
    "source": [
     "import numpy as np\n",
     "\n",
-    "import maxplotlib.canvas.canvas as canvas\n",
+    "from maxplotlib import Canvas\n",
     "\n",
     "%load_ext autoreload\n",
     "%autoreload 2"
@@ -30,7 +30,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=\"17cm\", ratio=0.5)\n",
+    "c = Canvas(width=\"17cm\", ratio=0.5)\n",
     "sp = c.add_subplot(grid=True, xlabel=\"x\", ylabel=\"y\")\n",
     "\n",
     "# node_a = sp.add_node(\n",
@@ -49,8 +49,8 @@
     "x = np.arange(0, 2 * np.pi, 0.01)\n",
     "y = np.sin(x)\n",
     "sp.add_line(x, y, label=r\"$\\sin(x)$\")\n",
-    "# c.plot()\n",
-    "c.savefig(filename=\"tutorial_05_01.pdf\")"
+    "\n",
+    "c.show()"
    ]
   },
   {
@@ -78,7 +78,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,

tutorials/tutorial_06.ipynb

@@ -15,7 +15,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import maxplotlib.canvas.canvas as canvas\n",
+    "from maxplotlib import Canvas\n",
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "\n",
@@ -30,13 +30,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "c = canvas.Canvas(width=\"17cm\", ratio=0.5)\n",
+    "c = Canvas(width=\"17cm\", ratio=0.5)\n",
     "sp = c.add_subplot(grid=False, xlabel=\"x\", ylabel=\"y\")\n",
     "# sp.add_line([0, 1, 2, 3], [0, 1, 4, 9], label=\"Line 1\",layer=1)\n",
     "data = np.random.random((10, 10))\n",
     "sp.add_imshow(data, extent=[1, 10, 1, 20], layer=1)\n",
-    "# c.plot()\n",
-    "c.savefig(layer_by_layer=True, filename=\"tutorial_06_01.pdf\")"
+    "\n",
+    "c.show()"
    ]
   },
   {
@@ -64,7 +64,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.13.3"
+   "version": "3.12.3"
   }
  },
  "nbformat": 4,