CubeTextureNode.js 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. import TextureNode from './TextureNode.js';
  2. import UniformNode from '../core/UniformNode.js';
  3. import ReflectVectorNode from './ReflectVectorNode.js';
  4. import { negate, vec3, nodeObject } from '../shadernode/ShaderNodeBaseElements.js';
  5. let defaultUV;
  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. defaultUV ||= new ReflectVectorNode();
  16. return defaultUV;
  17. }
  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 uvNodeObject = nodeObject( uvNode );
  34. const cubeUV = vec3( negate( uvNodeObject.x ), uvNodeObject.yz );
  35. const uvSnippet = cubeUV.build( builder, 'vec3' );
  36. const nodeVar = builder.getVarFromNode( this, 'vec4' );
  37. propertyName = builder.getPropertyName( nodeVar );
  38. let snippet = null;
  39. if ( levelNode && levelNode.isNode === true ) {
  40. const levelSnippet = levelNode.build( builder, 'float' );
  41. snippet = builder.getCubeTextureLevel( textureProperty, uvSnippet, levelSnippet );
  42. } else {
  43. snippet = builder.getCubeTexture( textureProperty, uvSnippet );
  44. }
  45. builder.addFlowCode( `${propertyName} = ${snippet}` );
  46. nodeData.snippet = snippet;
  47. nodeData.propertyName = propertyName;
  48. }
  49. return builder.format( propertyName, 'vec4', output );
  50. }
  51. }
  52. }
  53. export default CubeTextureNode;