diff --git a/src/maxplotlib/canvas/canvas.py b/src/maxplotlib/canvas/canvas.py index 7872f20..072c665 100644 --- a/src/maxplotlib/canvas/canvas.py +++ b/src/maxplotlib/canvas/canvas.py @@ -1,3 +1,5 @@ +import os + import matplotlib.pyplot as plt import plotly.graph_objects as go from plotly.subplots import make_subplots @@ -120,11 +122,12 @@ def add_subplot(self, **kwargs): def savefig( self, filename, - extension="pdf", backend="matplotlib", layers=None, layer_by_layer=False, + verbose=False, ): + filename_no_extension, extension = os.path.splitext(filename) if backend == "matplotlib": if layer_by_layer: layers = [] @@ -133,17 +136,21 @@ def savefig( fig, axs = self.plot( show=False, backend="matplotlib", savefig=True, layers=layers ) - fig.savefig(f"{filename}_{layers}.{extension}") + _fn = f"{filename_no_extension}_{layers}.{extension}" + fig.savefig(_fn) + print(f"Saved {_fn}") else: if layers is None: layers = self.layers - full_filepath = f"{filename}.{extension}" + full_filepath = filename else: - full_filepath = f"{filename}_{layers}.{extension}" + full_filepath = f"{filename_no_extension}_{layers}.{extension}" fig, axs = self.plot( show=False, backend="matplotlib", savefig=True, layers=layers ) fig.savefig(full_filepath) + if verbose: + print(f"Saved {full_filepath}") def plot(self, backend="matplotlib", show=True, savefig=False, layers=None): if backend == "matplotlib":