Merge pull request #20421 from sunag/nodematerial-webgpu

NodeMaterial: WebGPU

Author: mrdoob

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

@@ -0,0 +1,38 @@
+import AttributeNode from '../core/AttributeNode.js';
+
+class UVNode extends AttributeNode {
+
+	constructor( index = 0 ) {
+
+		super( '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 );
+
+	}
+
+}
+
+export default UVNode;

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

@@ -0,0 +1,54 @@
+import Node from './Node.js';
+
+class AttributeNode extends Node {
+
+	constructor( type, name = null, property = null ) {
+
+		super( type );
+
+		this.name = name;
+		this.property = property;
+
+	}
+
+	setAttributeName( name ) {
+
+		this.name = name;
+
+		return this;
+
+	}
+
+	getAttributeName( /*builder*/ ) {
+
+		return this.name;
+
+	}
+
+	setAttributeProperty( name ) {
+
+		this.property = name;
+
+		return this;
+
+	}
+
+	getAttributeProperty( builder ) {
+
+		const attribute = builder.getAttribute( this.getType( builder ), this.getAttributeName( builder ), this.property );
+
+		return attribute.property;
+
+	}
+
+	generate( builder, output ) {
+
+		const attributeProperty = this.getAttributeProperty( builder );
+
+		return builder.format( attributeProperty, this.getType( builder ), output );
+
+	}
+
+}
+
+export default AttributeNode;

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

@@ -0,0 +1,56 @@
+import Node from './Node.js';
+
+class InputNode extends Node {
+
+	constructor( type ) {
+
+		super( type );
+
+		this.constant = false;
+
+		Object.defineProperty( this, 'isInputNode', { value: true } );
+
+	}
+
+	setConst( value ) {
+
+		this.constant = value;
+
+		return this;
+
+	}
+
+	getConst() {
+
+		return this.constant;
+
+	}
+
+	generateConst( builder ) {
+
+		return builder.getConst( this.getType( builder ), this.value );
+
+	}
+
+	generate( builder, output ) {
+
+		const type = this.getType( builder );
+
+		if ( this.constant === true ) {
+
+			return builder.format( this.generateConst( builder ), type, output );
+
+		} else {
+
+			const nodeUniform = builder.getUniformFromNode( this, builder.shaderStage, this.getType( builder ) );
+			const propertyName = builder.getPropertyName( nodeUniform );
+
+			return builder.format( propertyName, type, output );
+
+		}
+
+	}
+
+}
+
+export default InputNode;

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

@@ -0,0 +1,41 @@
+class Node {
+
+	constructor( type = null ) {
+
+		this.type = type;
+
+		this.needsUpdate = false;
+
+		Object.defineProperty( this, 'isNode', { value: true } );
+
+	}
+
+	getType( /*builder*/ ) {
+
+		return this.type;
+
+	}
+
+	update( /*frame*/ ) {
+
+		console.warn( "Abstract function." );
+
+	}
+
+	generate( /*builder, output*/ ) {
+
+		console.warn( "Abstract function." );
+
+	}
+
+	build( builder, output ) {
+
+		builder.addNode( this );
+
+		return this.generate( builder, output );
+
+	}
+
+}
+
+export default Node;