three
Version:
JavaScript 3D library
104 lines (70 loc) • 2.06 kB
JavaScript
import Node from './Node.js';
import { nodeProxy } from '../tsl/TSLBase.js';
/**
* This node can be used to define multiple outputs in a shader programs.
*
* @augments Node
*/
class OutputStructNode extends Node {
static get type() {
return 'OutputStructNode';
}
/**
* Constructs a new output struct node. The constructor can be invoked with an
* arbitrary number of nodes representing the members.
*
* @param {...Node} members - A parameter list of nodes.
*/
constructor( ...members ) {
super();
/**
* An array of nodes which defines the output.
*
* @type {Array<Node>}
*/
this.members = members;
/**
* This flag can be used for type testing.
*
* @type {boolean}
* @readonly
* @default true
*/
this.isOutputStructNode = true;
}
getNodeType( builder ) {
const properties = builder.getNodeProperties( this );
if ( properties.membersLayout === undefined ) {
const members = this.members;
const membersLayout = [];
for ( let i = 0; i < members.length; i ++ ) {
const name = 'm' + i;
const type = members[ i ].getNodeType( builder );
membersLayout.push( { name, type, index: i } );
}
properties.membersLayout = membersLayout;
properties.structType = builder.getOutputStructTypeFromNode( this, properties.membersLayout );
}
return properties.structType.name;
}
generate( builder ) {
const propertyName = builder.getOutputStructName();
const members = this.members;
const structPrefix = propertyName !== '' ? propertyName + '.' : '';
for ( let i = 0; i < members.length; i ++ ) {
const snippet = members[ i ].build( builder );
builder.addLineFlowCode( `${ structPrefix }m${ i } = ${ snippet }`, this );
}
return propertyName;
}
}
export default OutputStructNode;
/**
* TSL function for creating an output struct node.
*
* @tsl
* @function
* @param {...Node} members - A parameter list of nodes.
* @returns {OutputStructNode}
*/
export const outputStruct = /*@__PURE__*/ nodeProxy( OutputStructNode );