TempNode.js 1.2 KB

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