basic svg export

fastplotlib/layouts/_subplot.py

@@ -6,9 +6,10 @@
 from inspect import signature, getfullargspec
 from warnings import warn
 import traceback
+from pathlib import Path
 
 from pygfx import Scene, OrthographicCamera, PanZoomController, OrbitController, \
-    AxesHelper, GridHelper, WgpuRenderer, Texture
+    AxesHelper, GridHelper, WgpuRenderer, Texture, SvgRenderer
 from wgpu.gui.auto import WgpuCanvas
 from wgpu.gui.base import WgpuCanvasBase
 
@@ -327,6 +328,31 @@ def set_grid_visibility(self, visible: bool):
         else:
             self.scene.remove(self._grid)
 
+    def export_svg(self, path: [str, Path], overwrite: bool = False):
+        """
+        CURRENTLY BUGGY. Export the plot area to an svg file.
+
+        Parameters
+        ----------
+        path: str
+            path to save svg file
+
+        overwrite: bool, default False
+            overwrite file if exists
+
+        """
+
+        if not overwrite and Path(path).exists():
+            raise FileExistsError(
+                f"The file at the specific path exists: {path}\n"
+                f"provide a path to a new file or use `overwrite=True`"
+            )
+
+        w, h = map(int, self.canvas.get_logical_size())
+
+        renderer = SvgRenderer(w, h, str(path))
+        renderer.render(self.scene, self.camera)
+
 
 class _DockedViewport(PlotArea):
     _valid_positions = [