Merge pull request #21170 from sunag/nodematerial-webgpu

WebGPU: All UBOs and varys is generate by nodes

Author: Mugen87

examples/jsm/renderers/nodes/accessors/CameraNode.js

@@ -0,0 +1,96 @@
+import Node from '../core/Node.js';
+import Vector3Node from '../inputs/Vector3Node.js';
+import Matrix4Node from '../inputs/Matrix4Node.js';
+import { NodeUpdateType } from '../core/constants.js';
+
+class CameraNode extends Node {
+
+	static POSITION = 'position';
+	static PROJECTION = 'projection';
+	static VIEW = 'view';
+
+	constructor( scope = CameraNode.POSITION ) {
+
+		super();
+
+		this.updateType = NodeUpdateType.Frame;
+
+		this.scope = scope;
+
+		this._inputNode = null;
+
+	}
+
+	getType() {
+
+		const scope = this.scope;
+
+		if ( scope === CameraNode.PROJECTION || scope === CameraNode.VIEW ) {
+
+			return 'mat4';
+
+		}
+
+		return 'vec3';
+
+	}
+
+	update( frame ) {
+
+		const camera = frame.camera;
+		const inputNode = this._inputNode;
+		const scope = this.scope;
+
+		if ( scope === CameraNode.PROJECTION ) {
+
+			inputNode.value = camera.projectionMatrix;
+
+		} else if ( scope === CameraNode.VIEW ) {
+
+			inputNode.value = camera.matrixWorldInverse;
+
+		} else if ( scope === CameraNode.POSITION ) {
+
+			camera.getWorldPosition( inputNode.value );
+
+		}
+
+	}
+
+	generate( builder, output ) {
+
+		const nodeData = builder.getDataFromNode( this );
+
+		let inputNode = this._inputNode;
+
+		if ( nodeData.inputNode === undefined ) {
+
+			const scope = this.scope;
+
+			if ( scope === CameraNode.PROJECTION || scope === CameraNode.VIEW ) {
+
+				if ( inputNode === null || inputNode.isMatrix4Node !== true ) {
+
+					inputNode = new Matrix4Node( null );
+
+				}
+
+			} else if ( inputNode === null || inputNode.isVector3Node !== true ) {
+
+				inputNode = new Vector3Node();
+
+			}
+
+			this._inputNode = inputNode;
+
+			nodeData.inputNode = inputNode;
+
+		}
+
+		return inputNode.build( builder, output );
+
+	}
+
+}
+
+export default CameraNode;

examples/jsm/renderers/nodes/accessors/GLPositionNode.js

@@ -0,0 +1,32 @@
+import Node from '../core/Node.js';
+import CameraNode from '../accessors/CameraNode.js';
+import ModelNode from '../accessors/ModelNode.js';
+import OperatorNode from '../math/OperatorNode.js';
+import PositionNode from '../accessors/PositionNode.js';
+
+class GLPositionNode extends Node {
+
+	constructor( position = new PositionNode() ) {
+
+		super( 'vec4' );
+
+		this.position = position;
+
+		this._mvpMatrix = new OperatorNode( '*', new CameraNode( CameraNode.PROJECTION ), new ModelNode( ModelNode.VIEW ) );
+
+	}
+
+	generate( builder, output ) {
+
+		const type = this.getType( builder );
+
+		const mvpSnipped = this._mvpMatrix.build( builder );
+		const positionSnipped = this.position.build( builder, 'vec3' );
+
+		return builder.format( `( ${mvpSnipped} * vec4( ${positionSnipped}, 1.0 ) )`, type, output );
+
+	}
+
+}
+
+export default GLPositionNode;

examples/jsm/renderers/nodes/accessors/ModelNode.js

@@ -0,0 +1,39 @@
+import Node from '../core/Node.js';
+import Vector3Node from '../inputs/Vector3Node.js';
+import Matrix4Node from '../inputs/Matrix4Node.js';
+import { NodeUpdateType } from '../core/constants.js';
+
+class ModelNode extends Node {
+
+	static VIEW = 'view';
+
+	constructor( scope = ModelNode.VIEW ) {
+
+		super( 'mat4' );
+
+		this.scope = scope;
+
+		this.updateType = NodeUpdateType.Object;
+
+		this._inputNode = new Matrix4Node( null );
+
+	}
+
+	update( frame ) {
+
+		const object = frame.object;
+		const inputNode = this._inputNode;
+
+		inputNode.value = object.modelViewMatrix;
+
+	}
+
+	generate( builder, output ) {
+
+		return this._inputNode.build( builder, output );
+
+	}
+
+}
+
+export default ModelNode;

examples/jsm/renderers/nodes/accessors/PositionNode.js

@@ -0,0 +1,39 @@
+import Node from '../core/Node.js';
+import AttributeNode from '../core/AttributeNode.js';
+
+class PositionNode extends Node {
+
+	static LOCAL = 'position';
+
+	constructor( scope = PositionNode.POSITION ) {
+
+		super( 'vec3' );
+
+		this.scope = scope;
+
+	}
+
+	generate( builder, output ) {
+
+		const type = this.getType( builder );
+		const nodeData = builder.getDataFromNode( this, builder.shaderStage );
+
+		let positionNode = nodeData.positionNode;
+
+		if ( positionNode === undefined ) {
+
+			positionNode = new AttributeNode( 'position', 'vec3' );
+
+			nodeData.positionNode = positionNode;
+
+		}
+
+		const positionSnipped = positionNode.build( builder, type );
+
+		return builder.format( positionSnipped, type, output );
+
+	}
+
+}
+
+export default PositionNode;

examples/jsm/renderers/nodes/accessors/UVNode.js

@@ -4,32 +4,15 @@ class UVNode extends AttributeNode {
 
 	constructor( index = 0 ) {
 
-		super( 'vec2' );
+		super( null, 'vec2' );
 
 		this.index = index;
 
 	}
 
-	getIndexProperty( prefix ) {
-
-		return prefix + ( this.index > 0 ? this.index + 1 : '' );
-
-	}
-
 	getAttributeName( /*builder*/ ) {
 
-		return this.getIndexProperty( 'uv' );
-
-	}
-
-	getAttributeProperty( builder ) {
-
-		// customize 'uv' property
-		const property = this.getIndexProperty( 'vUv' );
-
-		this.setAttributeProperty( property );
-
-		return super.getAttributeProperty( builder );
+		return 'uv' + ( this.index > 0 ? this.index + 1 : '' );
 
 	}
 

examples/jsm/renderers/nodes/core/AttributeNode.js

@@ -2,12 +2,11 @@ import Node from './Node.js';
 
 class AttributeNode extends Node {
 
-	constructor( type, name = null, property = null ) {
+	constructor( name, type ) {
 
 		super( type );
 
 		this.name = name;
-		this.property = property;
 
 	}
 
@@ -25,27 +24,36 @@ class AttributeNode extends Node {
 
 	}
 
-	setAttributeProperty( name ) {
+	generate( builder, output ) {
 
-		this.property = name;
+		const attributeName = this.getAttributeName( builder );
+		const attributeType = this.getType( builder );
 
-		return this;
+		const attribute = builder.getAttribute( attributeName, attributeType );
 
-	}
+		if ( builder.isShaderStage( 'vertex' ) ) {
 
-	getAttributeProperty( builder ) {
+			return builder.format( attribute.name, attribute.type, output );
 
-		const attribute = builder.getAttribute( this.getType( builder ), this.getAttributeName( builder ), this.property );
+		} else {
 
-		return attribute.property;
+			const nodeData = builder.getDataFromNode( this, builder.shaderStage );
 
-	}
+			let nodeVary = nodeData.varyNode;
 
-	generate( builder, output ) {
+			if ( nodeVary === undefined ) {
+
+				nodeVary = builder.getVaryFromNode( this, attribute.type, attributeName );
+
+				nodeData.nodeVary = nodeVary;
+
+			}
+
+			const varyName = builder.getPropertyName( nodeVary );
 
-		const attributeProperty = this.getAttributeProperty( builder );
+			return builder.format( varyName, attribute.type, output );
 
-		return builder.format( attributeProperty, this.getType( builder ), output );
+		}
 
 	}
 

examples/jsm/renderers/nodes/core/Node.js

@@ -1,15 +1,23 @@
+import { NodeUpdateType } from './constants.js';
+
 class Node {
 
 	constructor( type = null ) {
 
 		this.type = type;
 
-		this.needsUpdate = false;
+		this.updateType = NodeUpdateType.None;
 
 		Object.defineProperty( this, 'isNode', { value: true } );
 
 	}
 
+	getUpdateType( /*builder*/ ) {
+
+		return this.updateType;
+
+	}
+
 	getType( /*builder*/ ) {
 
 		return this.type;

examples/jsm/renderers/nodes/core/NodeAttribute.js

@@ -0,0 +1,14 @@
+class NodeAttribute {
+
+	constructor( name, type ) {
+
+		this.name = name;
+		this.type = type;
+
+		Object.defineProperty( this, 'isNodeAttribute', { value: true } );
+
+	}
+
+}
+
+export default NodeAttribute;