CondNode.js 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. import Node, { addNodeClass } from '../core/Node.js';
  2. import { property } from '../core/PropertyNode.js';
  3. import { addNodeElement, nodeProxy } from '../shadernode/ShaderNode.js';
  4. class CondNode extends Node {
  5. constructor( condNode, ifNode, elseNode = null ) {
  6. super();
  7. this.condNode = condNode;
  8. this.ifNode = ifNode;
  9. this.elseNode = elseNode;
  10. }
  11. getNodeType( builder ) {
  12. const ifType = this.ifNode.getNodeType( builder );
  13. if ( this.elseNode !== null ) {
  14. const elseType = this.elseNode.getNodeType( builder );
  15. if ( builder.getTypeLength( elseType ) > builder.getTypeLength( ifType ) ) {
  16. return elseType;
  17. }
  18. }
  19. return ifType;
  20. }
  21. setup( builder ) {
  22. const properties = builder.getNodeProperties( this );
  23. properties.condNode = this.condNode.cache();
  24. properties.ifNode = this.ifNode.cache();
  25. properties.elseNode = this.elseNode ? this.elseNode.cache() : null;
  26. }
  27. generate( builder, output ) {
  28. const type = this.getNodeType( builder );
  29. const nodeData = builder.getDataFromNode( this );
  30. if ( nodeData.nodeProperty !== undefined ) {
  31. return nodeData.nodeProperty;
  32. }
  33. const { condNode, ifNode, elseNode } = builder.getNodeProperties( this );
  34. const needsOutput = output !== 'void';
  35. const nodeProperty = needsOutput ? property( type ).build( builder ) : '';
  36. nodeData.nodeProperty = nodeProperty;
  37. const nodeSnippet = condNode.build( builder, 'bool' );
  38. builder.addFlowCode( `\n${ builder.tab }if ( ${ nodeSnippet } ) {\n\n` ).addFlowTab();
  39. let ifSnippet = ifNode.build( builder, type );
  40. if ( ifSnippet ) {
  41. if ( needsOutput ) {
  42. ifSnippet = nodeProperty + ' = ' + ifSnippet + ';';
  43. } else {
  44. ifSnippet = 'return ' + ifSnippet + ';';
  45. }
  46. }
  47. builder.removeFlowTab().addFlowCode( builder.tab + '\t' + ifSnippet + '\n\n' + builder.tab + '}' );
  48. if ( elseNode !== null ) {
  49. builder.addFlowCode( ' else {\n\n' ).addFlowTab();
  50. let elseSnippet = elseNode.build( builder, type );
  51. if ( elseSnippet ) {
  52. if ( needsOutput ) {
  53. elseSnippet = nodeProperty + ' = ' + elseSnippet + ';';
  54. } else {
  55. elseSnippet = 'return ' + elseSnippet + ';';
  56. }
  57. }
  58. builder.removeFlowTab().addFlowCode( builder.tab + '\t' + elseSnippet + '\n\n' + builder.tab + '}\n\n' );
  59. } else {
  60. builder.addFlowCode( '\n\n' );
  61. }
  62. return builder.format( nodeProperty, type, output );
  63. }
  64. }
  65. export default CondNode;
  66. export const cond = nodeProxy( CondNode );
  67. addNodeElement( 'cond', cond );
  68. addNodeClass( 'CondNode', CondNode );