Don't plot if plotted

Author: max-models

src/maxplotlib/backends/matplotlib/utils_old.py

@@ -419,7 +419,7 @@ def scale_axis(
 
         if not axs_in == None:
             ax = axs_in
-        print("precision", precision, precision, precision)
+        # print("precision", precision, precision, precision)
         plt.sca(ax)
         if axis == "x":
             if locs_labels == None:
@@ -520,7 +520,7 @@ def add_figure_label(
         ax=None,
     ):
         limits = self.get_limits(ax)
-        print(limits)
+        # print(limits)
         lx = limits[1] - limits[0]
         ly = limits[3] - limits[2]
 

src/maxplotlib/canvas/canvas.py

@@ -33,6 +33,7 @@ def __init__(self, **kwargs):
         self._width = kwargs.get("width", "17cm")
         self._ratio = kwargs.get("ratio", "golden")
         self._gridspec_kw = kwargs.get("gridspec_kw", {"wspace": 0.08, "hspace": 0.1})
+        self._plotted = False
 
         # Dictionary to store lines for each subplot
         # Key: (row, col), Value: list of lines with their data and kwargs
@@ -126,6 +127,7 @@ def savefig(
         layers=None,
         layer_by_layer=False,
         verbose=False,
+        plot=True,
     ):
         filename_no_extension, extension = os.path.splitext(filename)
         if backend == "matplotlib":
@@ -145,10 +147,16 @@ def savefig(
                     full_filepath = filename
                 else:
                     full_filepath = f"{filename_no_extension}_{layers}.{extension}"
-                fig, axs = self.plot(
-                    show=False, backend="matplotlib", savefig=True, layers=layers
-                )
-                fig.savefig(full_filepath)
+                print(f"Save to {full_filepath}")
+                if self._plotted:
+                    self._matplotlib_fig.savefig(full_filepath)
+                else:
+                        
+                    fig, axs = self.plot(
+                        show=False, backend="matplotlib", savefig=True, layers=layers
+                    )
+                    print('done plotting')
+                    fig.savefig(full_filepath)
                 if verbose:
                     print(f"Saved {full_filepath}")
 
@@ -166,7 +174,7 @@ def plot_matplotlib(self, show=True, savefig=False, layers=None, usetex=False):
         filename (str, optional): Filename to save the figure.
         show (bool): Whether to display the plot.
         """
-
+        print(f"Plotting with maxplotlib ")
         tex_fonts = plt_utils.setup_tex_fonts(fontsize=self.fontsize, usetex=usetex)
 
         plt_utils.setup_plotstyle(
@@ -198,6 +206,7 @@ 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.plot_matplotlib(ax, layers=layers)")
             subplot.plot_matplotlib(ax, layers=layers)
             # ax.set_title(f"Subplot ({row}, {col})")
             ax.grid()
@@ -208,6 +217,9 @@ def plot_matplotlib(self, show=True, savefig=False, layers=None, usetex=False):
             plt.show()
         # else:
         #     plt.close()
+        self._plotted = True
+        self._matplotlib_fig = fig
+        self._matplotlib_axes = axes
         return fig, axes
 
     def plot_plotly(self, show=True, savefig=None, usetex=False):