Export particles and sounds

Author: Yuri Pourre

editor/src/editor/layout/graph/export.ts

@@ -72,33 +72,21 @@ export async function exportNode(editor: Editor, node: Node): Promise<void> {
 		// Store original doNotSerialize values
 		const originalDoNotSerialize = new Map<Node, boolean>();
 
-		// Temporarily set doNotSerialize = true for all nodes except the ones we want to include
-		scene.meshes.forEach((mesh) => {
-			originalDoNotSerialize.set(mesh, mesh.doNotSerialize);
-			mesh.doNotSerialize = !nodesToInclude.has(mesh);
-		});
-
-		scene.lights.forEach((light) => {
-			originalDoNotSerialize.set(light, light.doNotSerialize);
-			light.doNotSerialize = !nodesToInclude.has(light);
-		});
+		exportMeshes(scene, nodesToInclude, originalDoNotSerialize);
+		exportLights(scene, nodesToInclude, originalDoNotSerialize);
+		exportCameras(scene, nodesToInclude, originalDoNotSerialize);
+		exportTransformNodes(scene, nodesToInclude, originalDoNotSerialize);
 
-		scene.cameras.forEach((camera) => {
-			originalDoNotSerialize.set(camera, camera.doNotSerialize);
-			camera.doNotSerialize = !nodesToInclude.has(camera);
-		});
-
-		scene.transformNodes.forEach((transformNode) => {
-			originalDoNotSerialize.set(transformNode, transformNode.doNotSerialize);
-			transformNode.doNotSerialize = !nodesToInclude.has(transformNode);
-		});
+		const originalParticleSystems = exportParticleSystems(scene, nodesToInclude);
+		const originalSoundTracks = exportSounds(scene, nodesToInclude);
 
 		// Serialize the filtered scene
 		const data = await SceneSerializer.SerializeAsync(scene);
 
-		// Restore original doNotSerialize values
-		originalDoNotSerialize.forEach((value, node) => {
-			node.doNotSerialize = value;
+		// Restore original scene state
+		restoreSceneState(scene, originalDoNotSerialize, {
+			originalParticleSystems,
+			originalSoundTracks
 		});
 
 		await writeJSON(filePath, data, JSON_CONFIG);
@@ -112,3 +100,161 @@ export async function exportNode(editor: Editor, node: Node): Promise<void> {
 		}
 	}
 }
+
+/**
+ * Restores the original scene state after export.
+ * @param scene The scene to restore
+ * @param originalDoNotSerialize Map containing original serialization states
+ * @param exportData Data needed for restoration
+ */
+function restoreSceneState(
+	scene: any, 
+	originalDoNotSerialize: Map<Node, boolean>,
+	exportData: { originalParticleSystems: any[]; originalSoundTracks: any[] }
+): void {
+	// Restore original doNotSerialize values
+	originalDoNotSerialize.forEach((value, node) => {
+		node.doNotSerialize = value;
+	});
+	
+	// Restore particle systems
+	scene.particleSystems.length = 0;
+	exportData.originalParticleSystems.forEach((ps: any) => 
+		scene.particleSystems.push(ps)
+	);
+	
+	// Restore soundtracks
+	scene.soundTracks = exportData.originalSoundTracks;
+}
+
+/**
+ * Configures mesh nodes for export.
+ * @param scene The scene containing the meshes
+ * @param nodesToInclude Set of nodes to include in the export
+ * @param originalDoNotSerialize Map to store original serialization state
+ */
+function exportMeshes(
+	scene: any, 
+	nodesToInclude: Set<Node>,
+	originalDoNotSerialize: Map<Node, boolean>
+): void {
+	scene.meshes.forEach((mesh: Node) => {
+		originalDoNotSerialize.set(mesh, mesh.doNotSerialize);
+		mesh.doNotSerialize = !nodesToInclude.has(mesh);
+	});
+}
+
+/**
+ * Configures light nodes for export.
+ * @param scene The scene containing the lights
+ * @param nodesToInclude Set of nodes to include in the export
+ * @param originalDoNotSerialize Map to store original serialization state
+ */
+function exportLights(
+	scene: any, 
+	nodesToInclude: Set<Node>,
+	originalDoNotSerialize: Map<Node, boolean>
+): void {
+	scene.lights.forEach((light: Node) => {
+		originalDoNotSerialize.set(light, light.doNotSerialize);
+		light.doNotSerialize = !nodesToInclude.has(light);
+	});
+}
+
+/**
+ * Configures camera nodes for export.
+ * @param scene The scene containing the cameras
+ * @param nodesToInclude Set of nodes to include in the export
+ * @param originalDoNotSerialize Map to store original serialization state
+ */
+function exportCameras(
+	scene: any, 
+	nodesToInclude: Set<Node>,
+	originalDoNotSerialize: Map<Node, boolean>
+): void {
+	scene.cameras.forEach((camera: Node) => {
+		originalDoNotSerialize.set(camera, camera.doNotSerialize);
+		camera.doNotSerialize = !nodesToInclude.has(camera);
+	});
+}
+
+/**
+ * Configures transform nodes for export.
+ * @param scene The scene containing the transform nodes
+ * @param nodesToInclude Set of nodes to include in the export
+ * @param originalDoNotSerialize Map to store original serialization state
+ */
+function exportTransformNodes(
+	scene: any, 
+	nodesToInclude: Set<Node>,
+	originalDoNotSerialize: Map<Node, boolean>
+): void {
+	scene.transformNodes.forEach((transformNode: Node) => {
+		originalDoNotSerialize.set(transformNode, transformNode.doNotSerialize);
+		transformNode.doNotSerialize = !nodesToInclude.has(transformNode);
+	});
+}
+
+/**
+ * Filters particle systems for export based on their attached nodes.
+ * @param scene The scene containing the particle systems
+ * @param nodesToInclude Set of nodes to include in the export
+ * @returns The original array of particle systems (for restoration)
+ */
+function exportParticleSystems(scene: any, nodesToInclude: Set<Node>): Array<any> {
+	// Save original particle systems
+	const originalParticleSystems = scene.particleSystems.slice();
+	
+	// Filter particle systems to only include those attached to our nodes
+	const particlesToKeep = originalParticleSystems.filter((ps: any) => {
+		const emitter = ps.emitter;
+		return emitter && nodesToInclude.has(emitter as Node);
+	});
+	
+	// Replace the scene's particle systems with only those we want to include
+	scene.particleSystems.length = 0;
+	particlesToKeep.forEach((ps: any) => scene.particleSystems.push(ps));
+	
+	return originalParticleSystems;
+}
+
+/**
+ * Filters sounds for export based on their attached nodes.
+ * @param scene The scene containing the sounds
+ * @param nodesToInclude Set of nodes to include in the export
+ * @returns The original array of soundtracks (for restoration)
+ */
+function exportSounds(scene: any, nodesToInclude: Set<Node>): Array<any> {
+	// Handle sounds - filter out sounds not attached to our nodes
+	let originalSoundTracks: any[] = [];
+	
+	if (scene.soundTracks) {
+		// Store original soundtracks to restore later
+		originalSoundTracks = scene.soundTracks.slice();
+		
+		// Filter each soundtrack to only include sounds attached to our nodes
+		const filteredSoundTracks = scene.soundTracks.map((soundtrack: any) => {
+			// Create a new sound collection with only the sounds attached to our nodes
+			const filteredSoundCollection = soundtrack.soundCollection.filter((sound: any) => {
+				if (sound.spatialSound && sound.metadata && sound.metadata.connectedMeshName) {
+					// Check if the connected mesh name matches any of our nodes
+					for (const meshNode of nodesToInclude) {
+						if (meshNode.name === sound.metadata.connectedMeshName) {
+							return true;
+						}
+					}
+				}
+				return false;
+			});
+			
+			// Replace the original sound collection with our filtered one
+			soundtrack.soundCollection = filteredSoundCollection;
+			return soundtrack;
+		}).filter((st: any) => st.soundCollection.length > 0); // Remove empty soundtracks
+		
+		// Replace scene soundtracks with our filtered ones
+		scene.soundTracks = filteredSoundTracks;
+	}
+	
+	return originalSoundTracks;
+}

editor/src/editor/layout/graph/graph.tsx

@@ -86,6 +86,7 @@ export class EditorGraphContextMenu extends Component<IEditorGraphContextMenuPro
 											<ContextMenuItem onClick={() => this.props.editor.layout.graph.copySelectedNodes()}>
 												Copy <ContextMenuShortcut>{platform() === "darwin" ? "⌘+C" : "CTRL+C"}</ContextMenuShortcut>
 											</ContextMenuItem>
+
 											{isNode(this.props.object) && (
 												<ContextMenuItem onClick={() => this.props.editor.layout.graph.pasteSelectedNodes(this.props.object)}>
 													Paste <ContextMenuShortcut>{platform() === "darwin" ? "⌘+V" : "CTRL+V"}</ContextMenuShortcut>
@@ -98,14 +99,11 @@ export class EditorGraphContextMenu extends Component<IEditorGraphContextMenuPro
 
 									<ContextMenuSeparator />
 
-									<ContextMenuItem onClick={() => this._cloneNode(this.props.object)}>Clone</ContextMenuItem>
-
 									{isNode(this.props.object) && !isScene(this.props.object) && (
 										<ContextMenuItem onClick={() => exportNode(this.props.editor, this.props.object)}>Export Node (.babylon)</ContextMenuItem>
 									)}
 
 									<ContextMenuSeparator />
->>>>>>> 616af57c (Add menu item to export .babylon)
 								</>
 							)}