Improved the tikz plotting, use classes for nodes and paths

max-models · max-models · commit 76101fcc12e5 · 2024-11-24T23:03:44.000+01:00
diff --git a/src/maxplotlib/canvas/canvas.py b/src/maxplotlib/canvas/canvas.py
@@ -142,9 +142,9 @@ def plot_matplotlib(self, show=True, savefig=False):
         print(fig_width, fig_height, fig_width / self._dpi, fig_height / self._dpi)
         fig, axes = plt.subplots(self.nrows, self.ncols, figsize=(fig_width / self._dpi, fig_height / self._dpi), squeeze=False, dpi = self._dpi)
         
-        for (row, col), line_plot in self.subplots.items():
+        for (row, col), subplot in self.subplots.items():
             ax = axes[row][col]
-            line_plot.plot(ax)  # Assuming LinePlot's `plot` method accepts an axis object
+            subplot.plot_matplotlib(ax)
             # ax.set_title(f"Subplot ({row}, {col})")
 
         # Set caption, labels, etc., if needed
diff --git a/src/maxplotlib/subfigure/line_plot.py b/src/maxplotlib/subfigure/line_plot.py
@@ -54,7 +54,7 @@ def add_line(self, x_data, y_data, **kwargs):
             'kwargs': kwargs
         })
 
-    def plot(self, ax):
+    def plot_matplotlib(self, ax):
         """
         Plot all lines on the provided axis.
 
diff --git a/src/maxplotlib/subfigure/tikz_figure.py b/src/maxplotlib/subfigure/tikz_figure.py
@@ -1,6 +1,61 @@
 import subprocess
 import os
 import tempfile
+from matplotlib.image import imread
+
+class Node:
+    def __init__(self, x, y, label="", content="",**kwargs):
+        """
+        Represents a TikZ node.
+
+        Parameters:
+        - x (float): X-coordinate of the node.
+        - y (float): Y-coordinate of the node.
+        - name (str, optional): Name of the node. If None, a default name will be assigned.
+        - **kwargs: Additional TikZ node options (e.g., shape, color).
+        """
+        self.x = x
+        self.y = y
+        self.label = label
+        self.content = content
+        self.options = kwargs
+
+    def to_tikz(self):
+        """
+        Generate the TikZ code for this node.
+
+        Returns:
+        - tikz_str (str): TikZ code string for the node.
+        """
+        options = ', '.join(f"{k.replace('_', ' ')}={{{v}}}" for k, v in self.options.items())
+        if options:
+            options = f"[{options}]"
+        return f"    \\node{options} ({self.label}) at ({self.x}, {self.y}) {{{self.content}}};\n"
+
+class Path:
+    def __init__(self, nodes, **kwargs):
+        """
+        Represents a path (line) connecting multiple nodes.
+
+        Parameters:
+        - nodes (list of str): List of node names to connect.
+        - **kwargs: Additional TikZ path options (e.g., style, color).
+        """
+        self.nodes = nodes
+        self.options = kwargs
+
+    def to_tikz(self):
+        """
+        Generate the TikZ code for this path.
+
+        Returns:
+        - tikz_str (str): TikZ code string for the path.
+        """
+        options = ', '.join(f"{k.replace('_', ' ')}={{{v}}}" for k, v in self.options.items())
+        if options:
+            options = f"[{options}]"
+        path_str = ' -- '.join(f"({node_label})" for node_label in self.nodes)
+        return f"    \\draw{options} {path_str};\n"
 
 class TikzFigure:
     def __init__(self, **kwargs):
@@ -23,48 +78,58 @@ def __init__(self, **kwargs):
         self._label = kwargs.get('label', None)
         self._grid = kwargs.get('grid', False)
 
-        # Initialize nodes and lines
+        # Initialize lists to hold Node and Path objects
         self.nodes = []
-        self.lines = []
+        self.paths = []
+
+        # Counter for unnamed nodes
+        self._node_counter = 0
 
-    def add_node(self, name, x, y, **kwargs):
+    def add_node(self, x, y, label=None, content="", **kwargs):
         """
         Add a node to the TikZ figure.
 
         Parameters:
-        - name (str): Name of the node.
         - x (float): X-coordinate of the node.
         - y (float): Y-coordinate of the node.
+        - label (str, optional): Label of the node. If None, a default label will be assigned.
         - **kwargs: Additional TikZ node options (e.g., shape, color).
+
+        Returns:
+        - node (Node): The Node object that was added.
         """
-        node = {
-            'name': name,
-            'x': x,
-            'y': y,
-            'options': kwargs
-        }
+        if label is None:
+            label = f"node{self._node_counter}"
+        node = Node(x=x, y=y, label=label, content=content, **kwargs)
         self.nodes.append(node)
+        self._node_counter += 1
+        return node
 
-    def add_line(self, nodes, **kwargs):
+    def add_path(self, nodes, **kwargs):
         """
         Add a line or path connecting multiple nodes.
 
         Parameters:
         - nodes (list of str): List of node names to connect.
-        - **kwargs: Additional TikZ line options (e.g., style, color).
+        - **kwargs: Additional TikZ path options (e.g., style, color).
 
         Examples:
-        - add_line(['A', 'B', 'C'], color='blue')
+        - add_path(['A', 'B', 'C'], color='blue')
           Connects nodes A -> B -> C with a blue line.
         """
         if not isinstance(nodes, list):
             raise ValueError("nodes parameter must be a list of node names.")
-
-        line = {
-            'nodes': nodes,
-            'options': kwargs
-        }
-        self.lines.append(line)
+        
+        nodes = [
+            node.label if isinstance(node, Node)
+            else node if isinstance(node, str)
+            else ValueError(f"Invalid node type: {type(node)}")
+            for node in nodes
+        ]
+
+        path = Path(nodes, **kwargs)
+        self.paths.append(path)
+        return path
 
     def generate_tikz(self):
         """
@@ -81,19 +146,11 @@ def generate_tikz(self):
 
         # Add nodes
         for node in self.nodes:
-            options = ', '.join(f"{k.replace('_',' ')}={{{v}}}" for k, v in node['options'].items())
-            if options:
-                options = f"[{options}]"
-            tikz_script += f"    \\node{options} ({node['name']}) at ({node['x']}, {node['y']}) {{{node['name']}}};\n"
-
-        # Add lines
-        for line in self.lines:
-            options = ', '.join(f"{k.replace('_',' ')}={{{v}}}" for k, v in line['options'].items())
-            if options:
-                options = f"[{options}]"
-            # Create the path by connecting all nodes in the list
-            path = ' -- '.join(f"({node_name})" for node_name in line['nodes'])
-            tikz_script += f"    \\draw{options} {path};\n"
+            tikz_script += node.to_tikz()
+
+        # Add paths
+        for path in self.paths:
+            tikz_script += path.to_tikz()
 
         tikz_script += "\\end{tikzpicture}"
 
@@ -157,3 +214,11 @@ def compile_pdf(self, filename='output.pdf'):
                 print(f"PDF successfully compiled and saved as '{filename}'.")
             else:
                 print("PDF compilation failed. Please check the LaTeX log for details.")
+    def plot_matplotlib(self, ax):
+        """
+        Plot all lines on the provided axis.
+
+        Parameters:
+        ax (matplotlib.axes.Axes): Axis on which to plot the lines.
+        """
+        
diff --git a/tutorials/tutorial_02.ipynb b/tutorials/tutorial_02.ipynb
@@ -2,36 +2,45 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 37,
    "id": "3816e8ed-5107-4d8c-9e78-f801af118811",
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "The autoreload extension is already loaded. To reload it, use:\n",
+      "  %reload_ext autoreload\n"
+     ]
+    }
+   ],
    "source": [
     "import maxplotlib.canvas.canvas as canvas\n",
-    "\n",
+    "from IPython.display import IFrame\n",
     "%load_ext autoreload\n",
     "%autoreload 2"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 34,
+   "execution_count": 62,
    "id": "650f0afa-f719-4495-b020-fc2b2df36bd7",
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "A\n",
       "\\begin{tikzpicture}\n",
-      "    \\node[shape={circle}, draw={black}, fill={blue!20}] (A) at (0, 0) {A};\n",
-      "    \\node[shape={rectangle}, draw={red}] (B) at (2, 2) {B};\n",
-      "    \\node[shape={rectangle}, draw={red}] (C) at (2, 5) {C};\n",
-      "    \\node[shape={rectangle}, draw={red}] (D) at (1, 5) {D};\n",
-      "    \\draw[color={darkgreen}, style={dashed}, line width={2pt}] (A) -- (B) -- (C) -- (A) -- (D);\n",
+      "    \\node[shape={circle}, draw={black}, fill={blue!20}] (A) at (0, 0) {Origin node};\n",
+      "    \\node[shape={rectangle}, draw={red}] (B) at (2, 2) {};\n",
+      "    \\node[shape={rectangle}, draw={red}] (C) at (2, 5) {};\n",
+      "    \\node[shape={rectangle}, draw={red}] (node3) at (1, 5) {};\n",
+      "    \\draw[color={green}, style={solid}, line width={2pt}] (A) -- (B) -- (C) -- (A) -- (node3);\n",
       "\\end{tikzpicture}\n",
-      "An error occurred while compiling the LaTeX document:\n",
-      "\n"
+      "PDF successfully compiled and saved as 'my_figure.pdf'.\n"
      ]
     }
    ],
@@ -40,32 +49,94 @@
     "tikz = c.add_tikzfigure(grid=False)\n",
     "\n",
     "# Add nodes\n",
-    "tikz.add_node('A', 0, 0, shape='circle', draw='black', fill='blue!20')\n",
-    "tikz.add_node('B', 2, 2, shape='rectangle', draw='red')\n",
-    "tikz.add_node('C', 2, 5, shape='rectangle', draw='red')\n",
-    "tikz.add_node('D', 1, 5, shape='rectangle', draw='red')\n",
+    "node_a = tikz.add_node(0, 0, 'A', content='Origin node', shape='circle', draw='black', fill='blue!20')\n",
+    "tikz.add_node(2, 2, 'B', shape='rectangle', draw='red')\n",
+    "tikz.add_node(2, 5, 'C', shape='rectangle', draw='red')\n",
+    "last_node = tikz.add_node(1, 5, shape='rectangle', draw='red')\n",
+    "\n",
+    "print(node_a.label)\n",
     "\n",
     "# Add a line between nodes \n",
-    "tikz.add_line(['A', 'B', 'C', 'A', 'D'], color='darkgreen', style='dashed', line_width='2pt')\n",
+    "tikz.add_path([node_a, 'B', 'C', 'A', last_node], color='green', style='solid', line_width='2pt')\n",
     "\n",
     "# Generate the TikZ script\n",
     "script = tikz.generate_tikz()\n",
     "print(script)\n",
-    "tikz.compile_pdf('my_figure.pdf')"
+    "tikz.compile_pdf('my_figure.pdf')\n",
+    "# Replace 'path_to_pdf' with the path to your PDF file\n",
+    "\n"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 73,
    "id": "344589d9-c60f-4d01-8171-e35db585f1ad",
    "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "A\n",
+      "800 400.0 2.6666666666666665 1.3333333333333333\n"
+     ]
+    },
+    {
+     "data": {
+      "image/png": "iVBORw0KGgoAAAANSUhEUgAAArEAAAFOCAYAAACR9kV8AAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjguMywgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/H5lhTAAAACXBIWXMAAC4jAAAuIwF4pT92AAAQ7UlEQVR4nO3dvW4bd7rA4VcHQbYLRnKTZlOMynSUXRvYUHW2oOA7MO+AQq5gIZXbyb4ChyzSi03KAOJUKTdTJE2axES6pJlT+JhHoj9W4tfwJZ8HCMAxyNE/wODFb8gZ8qBpmiYAACCR/2l7AQAA8FAiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIZ6sjdjqdxvHxcfT7/baXAsADmeHAOm1txFZVFScnJ1HXddR13fZyAHgAMxxYt62L2Lquo9/vz4YfAHmY4cCmfNL2At6epVdVFVdXV4YeQCJmONCW1iL29PQ0xuPxO//e6/WiLMu4vLxsYVUA3IcZDrSt1YgtiiKOjo7i+Pg4ut1udDqdiIgYj8cGIMAWM8OBtrUWsYPBoK0/DcCSzHCgbVt3YxcAAPw3IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdFr72dl9M51O4/vvv4/pdBp//PFHfP755/Hpp58+aB+Hh4fx6NGjNa0QlvPnn3/GL7/8Mtt++vRpFEXR3oJgRcxv9kHGGS5iN+T777+Pf/7zn20vAzbmu+++i6+//rrtZcDSzG/2UYYZ7nICAADSEbEAAKTjcoIN+fvf/35n+9///nd88cUXD9qHa6rYZv/5z3/ufOQ6f8xDVuY3+yDjDBexG/K3v/3tzvY//vGP+PLLL1taDazf/DEPWZnf7KMMM9zlBAAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgna2M2Ol0eme7rut2FgLAg5nhwCZsXcROp9M4Pz+/8291XcdoNGppRQDclxkObMonbf3h8XgcV1dXEfH/Z+11XX/wjP3s7CyKooiyLOPo6CiKooiIiLIs4+LiYhNLBuD/mOFA21qL2Ol0OjszfzvM5h+/z+0hOZ1Oo9PprGuJAHyAGQ60rbWI7fV60TRNW38egCWY4UDbtu6aWAAA+G9ELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDpbG7FVVUW/34+Tk5M4PDyMg4ODOD4+jtPT0xiNRm0vD4CPMMOBddu6iK2qKk5OTuLk5CS+/fbbePbsWQyHw5hMJnF+fh4REWdnZ3FwcDDbBmA7mOHAxjRbZDgcNhHRREQzGAw++Lzr6+umKIomIppOp9O8fv16c4tc0I8//jj7f4uI5scff2x7SbBSjnF2dYY7ttkHGY/zrXkn9vLyMs7OziIiYjAYxMXFxQef2+12YzKZRFEUs7N+ANpjhgObthUROx6PZx8rdbvdjw6/t8qyjJcvX0ZERF3Xs+EJwGaZ4UAbtiJi+/3+7PFDrpHq9XpRlmVERIxGoxiPxytf26r89ttvH92G7Bzj+2vXZ7hjm32Q8ThvPWJfvHgRdV1HxJsz8263+6DXLzo8N+3169cf3YbsHOP7aR9muGObfZDxOG89Yq+urmaPHzr8It6cyb9VVdVsmAKwfmY40JZWI7au66iqara9yMX9bz+Keuv2QAVgfcxwoE2tRuz8F14/fvx4of3cHoK+RBtgM8xwoE2tRuz19fWd7fkz8vu6/bq6rmM6nS6zLADuwQwH2tRqxN7c3NzZLopiof3MD875/QKwemY40KbWInb+bHvR4fe+186/OwDAapnhQNtajdhVefTo0dr2DcC7zHCgba1F7DqveTIAAdbLDAfatjXvxB4dHa1s324KAFgvMxxoW2sRu8qfM5u/nspZPMB6meFA21r/xS4AAHioT9r6w/MfFy1zZ2sGf/31153tH3744cH7ODw8fOcGCNgWP//8853t+WOe3bJPM9z8Zh9knOGtRey++fXXX+9s/+tf/2ppJbAZ88c8ZGV+s48yzHCXE2zIZ5991vYSYKMc8+wKxzL7KMNxvzURu+t3o+7yR23wPo75/bLLM9yxzD7KcNy7nGBDnj59Gt99911Mp9P4448/4vPPP49PP/30QftwTRXb7M8//4xffvlltv306dMWVwOrY36zDzLO8NYidpWFn+EdgKIo4uuvv257GbBWnU6n7SWwIfs0w81v9kW2Gd7a5QTrPCMty3Jt+wbADAfa11rErvNaiwzXcQBkZoYDbWstYufPtH///feF9/XTTz/d2V7lzx8C8C4zHGjb1kTsMtdEzQ/PbNd0AGRjhgNt25qIXcb872w/efJkZfsG4F1mONC2Vr8ndv5se36Q3df8OwDO4gHWzwwH2tRqxHa73Tvbiw7A268risKdrQAbYIYDbWo1Yp89e3Znu6qqB+9j/gz++fPnyywJgHsyw4E2tX45we2Pja6vrx+8j/F4fGd7fqgCsB5mONCmViM2IqLf788e39zcPPj1t4dmWZaupQLYIDMcaMtB0zRN24s4PDycfaQ0HA6j1+vd+7UHBwezx9fX1+9cowXAepnhQBtafyc2IuLly5ezx+fn5/d+3eXl5exxr9cz/ABaYIYDbdiKiO31ejEYDCLizV2q9xmCVVXNntfpdGI4HK51jQC8nxkOtGErIjYi4uLiIi4uLiLizdn5x4ZgVVXx1VdfRcSb4TeZTDayRgDezwwHNm0rrom9raqqODs7i7quoyzL6Pf70el04ujoKOq6jlevXsVoNIqiKOKbb76Znf0D0D4zHNiUrYvYt6qqiqurq7i5uYm6rmM6nUZRFPH48eM4PT01+AC2mBkOrNvWRiwAAHzI1lwTCwAA9yViAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6InZNqqqKfr8fJycncXh4GAcHB3F8fBynp6cxGo3aXh4AH2B+Qw4idsWqqoqTk5M4OTmJb7/9Np49exbD4TAmk0mcn59HRMTZ2VkcHBzMtiGr6XQax8fH0e/3214KLM38Zp/sxPxuWJnhcNhERBMRzWAw+ODzrq+vm6IomohoOp1O8/r1680tElZkMpk0ZVk2EdF0u922lwNLMb/ZJ7syv70TuyKXl5dxdnYWERGDwSAuLi4++NxutxuTySSKopid+UMWdV3PPmqt67rt5cDSzG/2xa7N74OmaZq2F5HdeDyO09PTiHgz4K6vr+/1utFoNBucvV4vhsPh2tYIi6rrOuq6jqqq4urq6r2D7yHHPWwT85tdtuvzW8SuwPHx8ezAuL6+jm63u5HXwjqdnp7GeDx+5997vV6UZRmXl5ezf8s8BNlv5je7aF/mt8sJlvTixYvZECvL8sFD7PYF1W4UYJucnp5Gr9eL58+fx8XFRUwmk2iaJobD4eydK8jM/GZX7cv8/qTtBWR3dXU1e7zIWXiv15sNv6qqoq7rKMtyZeuDRQ0Gg7aXAGtlfrOr9mV+eyd2CW+vM3lrkQv85wfe7aEKwHqY35CfiF3C/JdeP378eKH93B6EvkgbYP3Mb8hPxC5h/kLoRT9Guv26uq5jOp0usywA/gvzG/ITsUu4ubm5s10UxUL7mR+e8/sFYLXMb8hPxC5o/ox70QH4vtdm/aoLgAzMb9gNInZBq/yli0ePHq1t3wDcZX7DbhCxC1rndU+GIMD6mN+wG0TsguYH1dHR0cr27cYAgPUxv2E3iNgF/fbbbyvb1/w1Vc7kAdbH/IbdIGIBAEhHxC5o/iOjZe5uBWBzzG/YDSIWAIB0RCwAAOmI2BVxRypATuY35CRiAQBIR8QuaJU3AngXAGBzzG/YDSJ2QfM/NbhKZVmubd8A+878ht0gYhe0zq9k8XUvAOtjfsNuELELmj/b/v333xfe108//XRne5U/gQjAXeY37AYRu6D5IbjMdVHzA7TT6Sy8LwA+zvyG3SBiF7TK657mf2v7yZMnK9s3AHeZ37AbROwS5s+454fZfc2/C+BMHmC9zG/IT8Quodvt3tledAjefl1RFO5uBVgz8xvyE7FLePbs2Z3tqqoevI/5s/jnz58vsyQA7sH8hvxE7BI6nc6dj46ur68fvI/xeHxne36wArB65jfkJ2KX1O/3Z49vbm4e/Prbg7MsS9dTAWyI+Q25idglPX/+fPbl1tPpNEaj0YNe/+LFi9njq6urVS4NgI8wvyE3EbsCL1++nD0+Pz+/9+suLy9nj3u93js3GgCwXuY35CViV6DX68VgMIiIN3eq3mcQVlU1e16n04nhcLjWNcIqzd/Qsuid3dA285t9s1Pzu2FlLi4umohoIqIZDAYffN5kMmmKomgioul0OhtcISzv9evXTVmWs2P97X/D4bDtpcHCzG/2wa7N74OmaZo24nlXVVUVZ2dnUdd1lGUZ/X4/Op1OHB0dRV3X8erVqxiNRlEURXzzzTezdwBg24zH49l1fm/P3Ou6/uhZ+9vvyTw6Oppda1iWZVxcXKx7ubA085tdsS/zW8SuSVVVcXV1FTc3N1HXdUyn0yiKIh4/fhynp6eGH1tvNBrF2dlZRMRsoD3UdDqNTqcTk8lkhSuD9TK/yW5f5reIBQAgHTd2AQCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEhHxAIAkI6IBQAgHRELAEA6IhYAgHRELAAA6YhYAADSEbEAAKQjYgEASEfEAgCQjogFACAdEQsAQDoiFgCAdEQsAADpiFgAANIRsQAApCNiAQBIR8QCAJCOiAUAIB0RCwBAOiIWAIB0RCwAAOmIWAAA0hGxAACkI2IBAEjnfwGPDrCOb4JtYAAAAABJRU5ErkJggg==",
+      "text/plain": [
+       "<Figure size 800x400 with 2 Axes>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": [
+       "(<Figure size 800x400 with 2 Axes>,\n",
+       " array([[<Axes: >, <Axes: >]], dtype=object))"
+      ]
+     },
+     "execution_count": 73,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "c = canvas.Canvas(ncols=2,width=800, ratio=0.5)\n",
+    "tikz = c.add_tikzfigure(grid=False)\n",
+    "\n",
+    "# Add nodes\n",
+    "node_a = tikz.add_node(0, 0, 'A', content='Origin node', shape='circle', draw='black', fill='blue!20')\n",
+    "tikz.add_node(2, 2, 'B', shape='rectangle', draw='red')\n",
+    "tikz.add_node(2, 5, 'C', shape='rectangle', draw='red')\n",
+    "last_node = tikz.add_node(1, 5, shape='rectangle', draw='red')\n",
+    "\n",
+    "print(node_a.label)\n",
+    "\n",
+    "# Add a line between nodes \n",
+    "tikz.add_path([node_a, 'B', 'C', 'A', last_node], color='green', style='solid', line_width='2pt')\n",
+    "\n",
+    "# Generate the TikZ script\n",
+    "#script = tikz.generate_tikz()\n",
+    "\n",
+    "# Replace 'path_to_pdf' with the path to your PDF file\n",
+    "c.plot(backend='matplotlib')"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7c37d9b8-256b-4907-9c95-288b18cb396d",
+   "metadata": {},
    "outputs": [],
    "source": []
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7c37d9b8-256b-4907-9c95-288b18cb396d",
+   "id": "3081683f-8f1e-47ad-83bd-5a38899dde33",
    "metadata": {},
    "outputs": [],
    "source": []
