2
0

ArrayElementNode.js 594 B

123456789101112131415161718192021222324252627282930313233
  1. import Node, { addNodeClass } from '../core/Node.js';
  2. class ArrayElementNode extends Node { // @TODO: If extending from TempNode it breaks webgpu_compute
  3. constructor( node, indexNode ) {
  4. super();
  5. this.node = node;
  6. this.indexNode = indexNode;
  7. }
  8. getNodeType( builder ) {
  9. return this.node.getNodeType( builder );
  10. }
  11. generate( builder ) {
  12. const nodeSnippet = this.node.build( builder );
  13. const indexSnippet = this.indexNode.build( builder, 'uint' );
  14. return `${nodeSnippet}[ ${indexSnippet} ]`;
  15. }
  16. }
  17. export default ArrayElementNode;
  18. addNodeClass( ArrayElementNode );