12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- import { addNodeClass } from '../core/Node.js';
- import TempNode from '../core/TempNode.js';
- class JoinNode extends TempNode {
- constructor( nodes = [], nodeType = null ) {
- super( nodeType );
- this.nodes = nodes;
- }
- getNodeType( builder ) {
- if ( this.nodeType !== null ) {
- return builder.getVectorType( this.nodeType );
- }
- return builder.getTypeFromLength( this.nodes.reduce( ( count, cur ) => count + builder.getTypeLength( cur.getNodeType( builder ) ), 0 ) );
- }
- generate( builder, output ) {
- const type = this.getNodeType( builder );
- const nodes = this.nodes;
- const primitiveType = builder.getComponentType( type );
- const snippetValues = [];
- for ( const input of nodes ) {
- let inputSnippet = input.build( builder );
- const inputPrimitiveType = builder.getComponentType( input.getNodeType( builder ) );
- if ( inputPrimitiveType !== primitiveType ) {
- inputSnippet = builder.format( inputSnippet, inputPrimitiveType, primitiveType );
- }
- snippetValues.push( inputSnippet );
- }
- const snippet = `${ builder.getType( type ) }( ${ snippetValues.join( ', ' ) } )`;
- return builder.format( snippet, type, output );
- }
- }
- export default JoinNode;
- addNodeClass( 'JoinNode', JoinNode );
|