Added add_raw via a TikzWrapper to the tikzfigure

Author: max-models

src/maxplotlib/subfigure/tikz_figure.py

@@ -36,6 +36,16 @@ def generate_tikz(self):
         tikz_script += f"\\end{{pgfonlayer}}{{{self.label}}}\n"
         return tikz_script
 
+class TikzWrapper:
+    def __init__(self, raw_tikz, label="", content="", layer=0, **kwargs):
+        self.raw_tikz = raw_tikz
+        self.label = label
+        self.content = content
+        self.layer = layer
+        self.options = kwargs
+
+    def to_tikz(self):
+        return self.raw_tikz
 
 class Node:
     def __init__(self, x, y, label="", content="", layer=0, **kwargs):
@@ -198,6 +208,15 @@ def add_path(self, nodes, layer=0, **kwargs):
             self.layers[layer].add(path)
         return path
 
+    def add_raw(self, raw_tikz, layer=0, **kwargs):
+        tikz = TikzWrapper(raw_tikz)
+        if layer in self.layers:
+            self.layers[layer].add(tikz)
+        else:
+            self.layers[layer] = Tikzlayer(layer)
+            self.layers[layer].add(tikz)
+        return tikz
+
     def get_node(self, node_label):
         for node in self.nodes:
             if node.label == node_label:

tutorials/tutorial_04.py

@@ -0,0 +1,43 @@
+"""
+Tutorial 4.
+
+Add raw tikz code to the tikz subplot.
+"""
+
+import maxplotlib.canvas.canvas as canvas
+
+c = canvas.Canvas(width=800, ratio=0.5)
+tikz = c.add_tikzfigure(grid=False)
+
+# Add nodes
+tikz.add_node(0, 0, 'A', shape='circle', draw='black', fill='blue', layer=0)
+tikz.add_node(10, 0, 'B', shape='circle', draw='black', fill='blue', layer=0)
+tikz.add_node(10, 10, 'C', shape='circle', draw='black', fill='blue', layer=0)
+tikz.add_node(0, 10, 'D', shape='circle', draw='black', fill='blue', layer=2)
+
+
+# Add a line between nodes 
+tikz.add_path(['A', 'B', 'C', 'D'], path_actions=['draw', 'rounded corners'], fill='red', opacity=0.5,cycle=True, layer=1)
+
+raw_tikz = r"""
+\foreach \i in {0, 45, 90, 135, 180, 225, 270, 315} {
+    % Place a node at angle \i
+    \node[circle, draw, fill=green] at (\i:3) (N\i) {};
+}
+
+% Draw lines connecting the nodes
+\foreach \i/\j in {0/45, 45/90, 90/135, 135/180, 180/225, 225/270, 270/315, 315/0} {
+    \draw (N\i) -- (N\j);
+}
+"""
+
+tikz.add_raw(raw_tikz)
+
+tikz.add_node(0.5, 0.5, content='Cube', layer=10)
+
+# Generate the TikZ script
+script = tikz.generate_tikz()
+print(script)
+#print(tikz.generate_standalone())
+tikz.compile_pdf('figures/tutorial_04_figure_01.pdf')
+#