@openhps/core
Version:
Open Hybrid Positioning System - Core component
79 lines (74 loc) • 2.13 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);