ShaderNode: Share nodeObjects (#23226)

mrdoob · Jan 13, 2022 · 09eca6d · 09eca6d
1 parent c9daf11
commit 09eca6d
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/examples/jsm/renderers/nodes/ShaderNode.js b/examples/jsm/renderers/nodes/ShaderNode.js
@@ -69,6 +69,8 @@ const NodeHandler = {
 
 };
 
+const nodeObjects = new WeakMap();
+
 const ShaderNodeObject = ( obj ) => {
 
 	const type = typeof obj;
@@ -81,16 +83,17 @@ const ShaderNodeObject = ( obj ) => {
 
 		if ( obj.isNode === true ) {
 
-			const node = obj;
-
-			if ( node.isProxyNode !== true ) {
+			let nodeObject = nodeObjects.get( obj );
 
-				node.isProxyNode = true;
+			if ( nodeObject === undefined ) {
 
-				return new Proxy( node, NodeHandler );
+				nodeObject = new Proxy( obj, NodeHandler );
+				nodeObjects.set( obj, nodeObject );
 
 			}
 
+			return nodeObject;
+
 		}
 
 	}