update filename with the current workspace

MetaCell · Oct 2, 2024 · 605c595 · 605c595
1 parent d551e27
commit 605c595
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/applications/visualizer/frontend/src/components/viewers/ThreeD/SceneControls.tsx b/applications/visualizer/frontend/src/components/viewers/ThreeD/SceneControls.tsx
@@ -4,13 +4,15 @@ import ZoomOutIcon from "@mui/icons-material/ZoomOut";
 import { Box, Divider, IconButton, Popover, Typography } from "@mui/material";
 import Tooltip from "@mui/material/Tooltip";
 import { useRef, useState } from "react";
+import { useSelectedWorkspace } from "../../../hooks/useSelectedWorkspace.ts";
 import { vars } from "../../../theme/variables.ts";
 import CustomFormControlLabel from "./CustomFormControlLabel.tsx";
 import { Recorder } from "./Recorder.ts";
 
 const { gray500 } = vars;
 
 function SceneControls({ cameraControlRef, isWireframe, setIsWireframe, recorderRef, handleScreenshot }) {
+  const workspace = useSelectedWorkspace();
   const [anchorEl, setAnchorEl] = useState<HTMLElement | null>(null);
   const rotateAnimationRef = useRef<number | null>(null);
   const [isRotating, setIsRotating] = useState(false);
@@ -69,7 +71,7 @@ function SceneControls({ cameraControlRef, isWireframe, setIsWireframe, recorder
   const stopRecording = async () => {
     if (recorderRef.current) {
       recorderRef.current.stopRecording({ type: "video/webm" });
-      recorderRef.current.download("CanvasRecording.webm", { type: "video/webm" });
+      recorderRef.current.download(workspace.name + ".webm", { type: "video/webm" });
       recorderRef.current = null;
     }
   };

diff --git a/applications/visualizer/frontend/src/components/viewers/ThreeD/Screenshoter.ts b/applications/visualizer/frontend/src/components/viewers/ThreeD/Screenshoter.ts
@@ -31,6 +31,7 @@ export function downloadScreenshot(
   canvasRef: React.RefObject<HTMLCanvasElement>,
   sceneRef: React.RefObject<THREE.Scene>,
   cameraRef: React.RefObject<THREE.PerspectiveCamera>,
+  filename?: string,
 ) {
   if (!sceneRef.current || !cameraRef.current || !canvasRef.current) return;
 
@@ -50,7 +51,7 @@ export function downloadScreenshot(
       if (blob) {
         const link = document.createElement("a");
         link.href = URL.createObjectURL(blob);
-        link.download = "screenshot.png";
+        link.download = filename || "screenshot.png";
         link.click();
         URL.revokeObjectURL(link.href);
       }

diff --git a/applications/visualizer/frontend/src/components/viewers/ThreeD/ThreeDViewer.tsx b/applications/visualizer/frontend/src/components/viewers/ThreeD/ThreeDViewer.tsx
@@ -72,7 +72,7 @@ function ThreeDViewer() {
   }, [selectedDataset, workspace.availableNeurons, workspace.visibilities]);
 
   const handleScreenshot = () => {
-    downloadScreenshot(canvasRef, sceneRef, cameraRef);
+    downloadScreenshot(canvasRef, sceneRef, cameraRef, workspace.name);
   };
 
   const onCreated = (state) => {