55 lines
910 B
JavaScript
55 lines
910 B
JavaScript
import Node from './Node.js';
|
|
import { NodeShaderStage } from './constants.js';
|
|
|
|
class VaryNode extends Node {
|
|
|
|
constructor( node, name = null ) {
|
|
|
|
super();
|
|
|
|
this.node = node;
|
|
this.name = name;
|
|
|
|
}
|
|
|
|
getHash( builder ) {
|
|
|
|
return this.name || super.getHash( builder );
|
|
|
|
}
|
|
|
|
getNodeType( builder ) {
|
|
|
|
// VaryNode is auto type
|
|
|
|
return this.node.getNodeType( builder );
|
|
|
|
}
|
|
|
|
generate( builder ) {
|
|
|
|
const type = this.getNodeType( builder );
|
|
const node = this.node;
|
|
const name = this.name;
|
|
|
|
const nodeVary = builder.getVaryFromNode( this, type );
|
|
|
|
if ( name !== null ) {
|
|
|
|
nodeVary.name = name;
|
|
|
|
}
|
|
|
|
const propertyName = builder.getPropertyName( nodeVary, NodeShaderStage.Vertex );
|
|
|
|
// force node run in vertex stage
|
|
builder.flowNodeFromShaderStage( NodeShaderStage.Vertex, node, type, propertyName );
|
|
|
|
return builder.getPropertyName( nodeVary );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export default VaryNode;
|