WGSLNodeBuilder: Add polyfill for storage pointer when not available

src/nodes/code/FunctionCallNode.js

@@ -104,15 +104,38 @@ class FunctionCallNode extends TempNode {
 		const inputs = functionNode.getInputs( builder );
 		const parameters = this.parameters;
 
+		// Track storage pointer bindings for this function
+		const storageBindingMap = {};
+
 		const generateInput = ( node, inputNode ) => {
 
 			const type = inputNode.type;
 			const pointer = type === 'pointer';
+			const storagePointer = type === 'storagePointer';
 
 			let output;
 
-			if ( pointer ) output = '&' + node.build( builder );
-			else output = node.build( builder, type );
+			if ( storagePointer ) {
+
+				// Build the storage buffer node - this registers it as a uniform/binding
+				// and returns the property name (e.g., "nodeU1")
+				output = node.build( builder );
+
+				// Store the mapping from parameter name to the generated binding name
+				storageBindingMap[ inputNode.name ] = output;
+
+				// Return null to indicate this parameter should not be in the call
+				return null;
+
+			} else if ( pointer ) {
+
+				output = '&' + node.build( builder );
+
+			} else {
+
+				output = node.build( builder, type );
+
+			}
 
 			return output;
 
@@ -140,7 +163,14 @@ class FunctionCallNode extends TempNode {
 
 			for ( let i = 0; i < parameters.length; i ++ ) {
 
-				params.push( generateInput( parameters[ i ], inputs[ i ] ) );
+				const result = generateInput( parameters[ i ], inputs[ i ] );
+
+				// Only add non-storage pointer parameters to the call
+				if ( result !== null ) {
+
+					params.push( result );
+
+				}
 
 			}
 
@@ -152,20 +182,40 @@ class FunctionCallNode extends TempNode {
 
 				if ( node !== undefined ) {
 
-					params.push( generateInput( node, inputNode ) );
+					const result = generateInput( node, inputNode );
+
+					// Only add non-storage pointer parameters to the call
+					if ( result !== null ) {
+
+						params.push( result );
+
+					}
 
 				} else {
 
-					error( `TSL: Input '${ inputNode.name }' not found in \'Fn()\'.` );
+					// Only error for non-storage pointer parameters
+					if ( inputNode.type !== 'storagePointer' ) {
+
+						error( `TSL: Input '${ inputNode.name }' not found in \'Fn()\'.` );
 
-					params.push( generateInput( float( 0 ), inputNode ) );
+						params.push( generateInput( float( 0 ), inputNode ) );
+
+					}
 
 				}
 
 			}
 
 		}
 
+		// Store storage binding map in builder's node data for FunctionNode to use
+		if ( Object.keys( storageBindingMap ).length > 0 ) {
+
+			const functionNodeData = builder.getDataFromNode( functionNode );
+			functionNodeData.storageBindingMap = storageBindingMap;
+
+		}
+
 		const functionName = functionNode.build( builder, 'property' );
 
 		return `${ functionName }( ${ params.join( ', ' ) } )`;

src/nodes/code/FunctionNode.js

@@ -1,4 +1,5 @@
 import CodeNode from './CodeNode.js';
+import { nodeObject } from '../tsl/TSLBase.js';
 
 /**
  * This class represents a native shader function. It can be used to implement
@@ -45,6 +46,15 @@
 
 		super( code, includes, language );
 
+		/**
+		 * This flag can be used for type testing.
+		 *
+		 * @type {boolean}
+		 * @readonly
+		 * @default true
+		 */
+		this.isFunctionNode = true;
+
 	}
 
 	/**
@@ -114,6 +124,35 @@
 
 	generate( builder, output ) {
 
+		// Get storage binding map if set by FunctionCallNode (before building includes)
+		const nodeData = builder.getDataFromNode( this );
+		const storageBindingMap = nodeData.storageBindingMap || null;
+
+		// Propagate storage binding map to included functions before building them
+		if ( storageBindingMap !== null ) {
+
+			const includes = this.getIncludes( builder );
+
+			for ( const include of includes ) {
+
+				if ( include.isFunctionNode || ( include.value && include.value.isFunctionNode ) ) {
+
+					const includeNode = include.isFunctionNode ? include : include.value;
+					const includeData = builder.getDataFromNode( includeNode );
+
+					// Merge storage binding maps (don't overwrite if already set)
+					if ( ! includeData.storageBindingMap ) {
+
+						includeData.storageBindingMap = storageBindingMap;
+
+					}
+
+				}
+
+			}
+
+		}
+
 		super.generate( builder );
 
 		const nodeFunction = this.getNodeFunction( builder );
@@ -133,7 +172,7 @@
 
 		const propertyName = builder.getPropertyName( nodeCode );
 
-		const code = this.getNodeFunction( builder ).getCode( propertyName );
+		const code = this.getNodeFunction( builder ).getCode( propertyName, storageBindingMap );
 
 		nodeCode.code = code + '\n';