Passing Arguments to TSL Compute Functions

In TSL, there are a few approaches to create reusable compute functions that can accept different storage nodes across scenes. Based on your example, here's the recommended pattern:

Create a Function Factory

The best approach is to create a function factory that returns a compute function configured with your specific arguments:

JAVASCRIPTCopied
// Create a function factory that returns a compute function
const createInitFunction = (currentElements) => {
  return Fn(() => {
    // Access the storage passed in from the factory closure
    currentElements.element(instanceIndex).assign(1)
  }).compute(numElements)
}

// In your scenes
initScene1() {
  const currentElementsStorage1 = ...
  const runFn1 = Fn(() => {
    // Scene 1 specific code
  }).compute(numElements)
  
  // Create a compute function specifically for this scene's storage
  renderer.compute(createInitFunction(currentElementsStorage1))
  renderer.compute(runFn1)
}

initScene2() {
  const currentElementsStorage2 = ...
  const runFn2 = Fn(() => {
    // Scene 2 specific code
  }).compute(numElements)
  
  // Create a compute function for scene 2's storage
  renderer.compute(createInitFunction(currentElementsStorage2))
  renderer.compute(runFn2)
}

This pattern allows you to:

1. Define the compute function logic once
2. Pass different storage nodes to each instance
3. Keep the code clean and maintainable

The key is that the factory function creates a closure that captures the storage node, making it available to the compute function when it runs.

Note that you cannot directly pass arguments to a compute function at execution time - you need to capture them in the closure when creating the function.