VaryNode.js 715 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. import Node from './Node.js';
  2. import { NodeShaderStage } from './constants.js';
  3. class VaryNode extends Node {
  4. constructor( value ) {
  5. super();
  6. this.value = value;
  7. }
  8. getType( builder ) {
  9. // VaryNode is auto type
  10. return this.value.getType( builder );
  11. }
  12. generate( builder, output ) {
  13. const type = this.getType( builder );
  14. // force nodeVary.snippet work in vertex stage
  15. const snippet = this.value.buildStage( builder, NodeShaderStage.Vertex, type );
  16. const nodeVary = builder.getVaryFromNode( this, type );
  17. nodeVary.snippet = snippet;
  18. const propertyName = builder.getPropertyName( nodeVary );
  19. return builder.format( propertyName, type, output );
  20. }
  21. }
  22. export default VaryNode;