123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- import TextureNode from './TextureNode.js';
- import UniformNode from '../core/UniformNode.js';
- import { reflectVector } from './ReflectVectorNode.js';
- import { addNodeClass } from '../core/Node.js';
- import { addNodeElement, nodeProxy, vec3 } from '../shadernode/ShaderNode.js';
- class CubeTextureNode extends TextureNode {
- constructor( value, uvNode = null, levelNode = null ) {
- super( value, uvNode, levelNode );
- this.isCubeTextureNode = true;
- }
- getInputType( /*builder*/ ) {
- return 'cubeTexture';
- }
- getDefaultUV() {
- return reflectVector;
- }
- generate( builder, output ) {
- const { uvNode, levelNode } = builder.getNodeProperties( this );
- const texture = this.value;
- if ( ! texture || texture.isCubeTexture !== true ) {
- throw new Error( 'CubeTextureNode: Need a three.js cube texture.' );
- }
- const textureProperty = UniformNode.prototype.generate.call( this, builder, 'cubeTexture' );
- if ( output === 'sampler' ) {
- return textureProperty + '_sampler';
- } else if ( builder.isReference( output ) ) {
- return textureProperty;
- } else {
- const nodeData = builder.getDataFromNode( this );
- let propertyName = nodeData.propertyName;
- if ( propertyName === undefined ) {
- const cubeUV = vec3( uvNode.x.negate(), uvNode.yz );
- const uvSnippet = cubeUV.build( builder, 'vec3' );
- const nodeVar = builder.getVarFromNode( this, 'vec4' );
- propertyName = builder.getPropertyName( nodeVar );
- let snippet = null;
- if ( levelNode && levelNode.isNode === true ) {
- const levelSnippet = levelNode.build( builder, 'float' );
- snippet = builder.getTextureLevel( this, textureProperty, uvSnippet, levelSnippet );
- } else {
- snippet = builder.getTexture( this, textureProperty, uvSnippet );
- }
- builder.addLineFlowCode( `${propertyName} = ${snippet}` );
- nodeData.snippet = snippet;
- nodeData.propertyName = propertyName;
- }
- return builder.format( propertyName, 'vec4', output );
- }
- }
- }
- export default CubeTextureNode;
- export const cubeTexture = nodeProxy( CubeTextureNode );
- addNodeElement( 'cubeTexture', cubeTexture );
- addNodeClass( CubeTextureNode );
|