three.js/examples/jsm/renderers/nodes/core/AttributeNode.js

import Node from './Node.js';

class AttributeNode extends Node {

	constructor( name, type ) {

		super( type );

		this.name = name;

	}

	setAttributeName( name ) {

		this.name = name;

		return this;

	}

	getAttributeName( /*builder*/ ) {

		return this.name;

	}

	generate( builder, output ) {

		const attributeName = this.getAttributeName( builder );
		const attributeType = this.getType( builder );

		const attribute = builder.getAttribute( attributeName, attributeType );

		if ( builder.isShaderStage( 'vertex' ) ) {
		
			return builder.format( attributeName, attributeType, output );
			
		} else {
			
			const nodeData = builder.getDataFromNode( this, builder.shaderStage );
			
			let nodeVary = nodeData.varyNode;
			
			if ( nodeVary === undefined ) {
				
				nodeVary = builder.getVaryFromNode( this, attributeType, attributeName );
				
				nodeData.nodeVary = nodeVary;
				
			}
			
			const varyName = builder.getPropertyName( nodeVary );
			
			return builder.format( varyName, attributeType, output );
			
		}

	}

}

export default AttributeNode;