custom shader stage build Node.buildStage()

mrdoob · Feb 5, 2021 · eef49b1 · eef49b1
1 parent cfa90e0
commit eef49b1
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/examples/jsm/renderers/nodes/core/Node.js b/examples/jsm/renderers/nodes/core/Node.js
@@ -36,7 +36,21 @@ class Node {
 
 	}
 
-	build( builder, output ) {
+	buildStage( builder, shaderStage, output = null ) {
+
+		const oldShaderStage = builder.shaderStage;
+
+		builder.shaderStage = shaderStage;
+
+		const snippet = this.build( builder, output );
+
+		builder.shaderStage = oldShaderStage;
+
+		return snippet;
+
+	}
+
+	build( builder, output = null ) {
 
 		builder.addNode( this );