UNPKG

@openhps/core

Version:

Open Hybrid Positioning System - Core component

79 lines (74 loc) 2.13 kB
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);