CubeTextureNode.js 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. import TextureNode from './TextureNode.js';
  2. import UniformNode from '../core/UniformNode.js';
  3. import { reflectVector } from './ReflectVectorNode.js';
  4. import { addNodeClass } from '../core/Node.js';
  5. import { addNodeElement, nodeProxy, vec3 } from '../shadernode/ShaderNode.js';
  6. class CubeTextureNode extends TextureNode {
  7. constructor( value, uvNode = null, levelNode = null ) {
  8. super( value, uvNode, levelNode );
  9. this.isCubeTextureNode = true;
  10. }
  11. getInputType( /*builder*/ ) {
  12. return 'cubeTexture';
  13. }
  14. getDefaultUV() {
  15. return reflectVector;
  16. }
  17. setUpdateMatrix( /*updateMatrix*/ ) { } // Ignore .updateMatrix for CubeTextureNode
  18. generate( builder, output ) {
  19. const { uvNode, levelNode } = builder.getNodeProperties( this );
  20. const texture = this.value;
  21. if ( ! texture || texture.isCubeTexture !== true ) {
  22. throw new Error( 'CubeTextureNode: Need a three.js cube texture.' );
  23. }
  24. const textureProperty = UniformNode.prototype.generate.call( this, builder, 'cubeTexture' );
  25. if ( output === 'sampler' ) {
  26. return textureProperty + '_sampler';
  27. } else if ( builder.isReference( output ) ) {
  28. return textureProperty;
  29. } else {
  30. const nodeData = builder.getDataFromNode( this );
  31. let propertyName = nodeData.propertyName;
  32. if ( propertyName === undefined ) {
  33. const cubeUV = vec3( uvNode.x.negate(), uvNode.yz );
  34. const uvSnippet = cubeUV.build( builder, 'vec3' );
  35. const nodeVar = builder.getVarFromNode( this, 'vec4' );
  36. propertyName = builder.getPropertyName( nodeVar );
  37. let snippet = null;
  38. if ( levelNode && levelNode.isNode === true ) {
  39. const levelSnippet = levelNode.build( builder, 'float' );
  40. snippet = builder.getTextureLevel( this, textureProperty, uvSnippet, levelSnippet );
  41. } else {
  42. snippet = builder.getTexture( this, textureProperty, uvSnippet );
  43. }
  44. builder.addLineFlowCode( `${propertyName} = ${snippet}` );
  45. nodeData.snippet = snippet;
  46. nodeData.propertyName = propertyName;
  47. }
  48. return builder.format( propertyName, 'vec4', output );
  49. }
  50. }
  51. }
  52. export default CubeTextureNode;
  53. export const cubeTexture = nodeProxy( CubeTextureNode );
  54. addNodeElement( 'cubeTexture', cubeTexture );
  55. addNodeClass( CubeTextureNode );