[3d] improve storing Camera State logic

Comfy-Org · Jan 23, 2025 · b45b317 · b45b317
1 parent b54e270
commit b45b317
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 30 deletions.
diff --git a/src/extensions/core/load3d.ts b/src/extensions/core/load3d.ts
@@ -354,20 +354,7 @@ app.registerExtension({
 
     const fov = node.widgets.find((w: IWidget) => w.name === 'fov')
 
-    let cameraState
-    try {
-      const cameraInfo = node.properties['Camera Info']
-      if (
-        cameraInfo &&
-        typeof cameraInfo === 'string' &&
-        cameraInfo.trim() !== ''
-      ) {
-        cameraState = JSON.parse(cameraInfo)
-      }
-    } catch (error) {
-      console.warn('Failed to parse camera state:', error)
-      cameraState = undefined
-    }
+    let cameraState = node.properties['Camera Info']
 
     configureLoad3D(
       load3d,
@@ -386,7 +373,7 @@ app.registerExtension({
 
     // @ts-expect-error hacky override
     sceneWidget.serializeValue = async () => {
-      node.properties['Camera Info'] = JSON.stringify(load3d.getCameraState())
+      node.properties['Camera Info'] = load3d.getCameraState()
 
       const { scene: imageData, mask: maskData } = await load3d.captureScene(
         w.value,
@@ -598,20 +585,7 @@ app.registerExtension({
 
     const fov = node.widgets.find((w: IWidget) => w.name === 'fov')
 
-    let cameraState
-    try {
-      const cameraInfo = node.properties['Camera Info']
-      if (
-        cameraInfo &&
-        typeof cameraInfo === 'string' &&
-        cameraInfo.trim() !== ''
-      ) {
-        cameraState = JSON.parse(cameraInfo)
-      }
-    } catch (error) {
-      console.warn('Failed to parse camera state:', error)
-      cameraState = undefined
-    }
+    let cameraState = node.properties['Camera Info']
 
     configureLoad3D(
       load3d,
@@ -643,7 +617,7 @@ app.registerExtension({
 
     // @ts-expect-error hacky override
     sceneWidget.serializeValue = async () => {
-      node.properties['Camera Info'] = JSON.stringify(load3d.getCameraState())
+      node.properties['Camera Info'] = load3d.getCameraState()
 
       load3d.toggleAnimation(false)
 

diff --git a/src/extensions/core/load3d/Load3d.ts b/src/extensions/core/load3d/Load3d.ts
@@ -82,6 +82,10 @@ class Load3d {
     )
     this.controls.enableDamping = true
 
+    this.controls.addEventListener('end', () => {
+      this.storeNodeProperty('Camera Info', this.getCameraState())
+    })
+
     this.gltfLoader = new GLTFLoader()
     this.objLoader = new OBJLoader()
     this.mtlLoader = new MTLLoader()