TempNode.js 1.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. import Node from './Node.js';
  2. class TempNode extends Node {
  3. constructor( type ) {
  4. super( type );
  5. this.isTempNode = true;
  6. }
  7. build( builder, output ) {
  8. const buildStage = builder.getBuildStage();
  9. if ( buildStage === 'generate' ) {
  10. const type = builder.getVectorType( this.getNodeType( builder, output ) );
  11. const nodeData = builder.getDataFromNode( this );
  12. if ( builder.context.tempRead !== false && nodeData.propertyName !== undefined ) {
  13. return builder.format( nodeData.propertyName, type, output );
  14. } else if ( builder.context.tempWrite !== false && type !== 'void ' && output !== 'void' && nodeData.dependenciesCount > 1 ) {
  15. const snippet = super.build( builder, type );
  16. const nodeVar = builder.getVarFromNode( this, type );
  17. const propertyName = builder.getPropertyName( nodeVar );
  18. builder.addFlowCode( `${propertyName} = ${snippet}` );
  19. nodeData.snippet = snippet;
  20. nodeData.propertyName = propertyName;
  21. return builder.format( nodeData.propertyName, type, output );
  22. }
  23. }
  24. return super.build( builder, output );
  25. }
  26. }
  27. export default TempNode;