Cesium-Examples/libs/three/examples/jsm/nodes/accessors/CubeTextureNode.js
2025-03-19 11:00:22 +08:00

104 lines
2.2 KiB
JavaScript

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;
}
setUpdateMatrix( /*updateMatrix*/ ) { } // Ignore .updateMatrix for CubeTextureNode
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 );