CubeTextureNode.js 2.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  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. generate( builder, output ) {
  18. const { uvNode, levelNode } = builder.getNodeProperties( this );
  19. const texture = this.value;
  20. if ( ! texture || texture.isCubeTexture !== true ) {
  21. throw new Error( 'CubeTextureNode: Need a three.js cube texture.' );
  22. }
  23. const textureProperty = UniformNode.prototype.generate.call( this, builder, 'cubeTexture' );
  24. if ( output === 'sampler' ) {
  25. return textureProperty + '_sampler';
  26. } else if ( builder.isReference( output ) ) {
  27. return textureProperty;
  28. } else {
  29. const nodeData = builder.getDataFromNode( this );
  30. let propertyName = nodeData.propertyName;
  31. if ( propertyName === undefined ) {
  32. const cubeUV = vec3( uvNode.x.negate(), uvNode.yz );
  33. const uvSnippet = cubeUV.build( builder, 'vec3' );
  34. const nodeVar = builder.getVarFromNode( this, 'vec4' );
  35. propertyName = builder.getPropertyName( nodeVar );
  36. let snippet = null;
  37. if ( levelNode && levelNode.isNode === true ) {
  38. const levelSnippet = levelNode.build( builder, 'float' );
  39. snippet = builder.getTextureLevel( this, textureProperty, uvSnippet, levelSnippet );
  40. } else {
  41. snippet = builder.getTexture( this, textureProperty, uvSnippet );
  42. }
  43. builder.addLineFlowCode( `${propertyName} = ${snippet}` );
  44. nodeData.snippet = snippet;
  45. nodeData.propertyName = propertyName;
  46. }
  47. return builder.format( propertyName, 'vec4', output );
  48. }
  49. }
  50. }
  51. export default CubeTextureNode;
  52. export const cubeTexture = nodeProxy( CubeTextureNode );
  53. addNodeElement( 'cubeTexture', cubeTexture );
  54. addNodeClass( CubeTextureNode );