EquirectUVNode.js 730 B

123456789101112131415161718192021222324252627282930313233
  1. import TempNode from '../core/TempNode.js';
  2. import { positionWorldDirection } from '../accessors/PositionNode.js';
  3. import { nodeProxy, vec2 } from '../shadernode/ShaderNode.js';
  4. import { addNodeClass } from '../core/Node.js';
  5. class EquirectUVNode extends TempNode {
  6. constructor( dirNode = positionWorldDirection ) {
  7. super( 'vec2' );
  8. this.dirNode = dirNode;
  9. }
  10. setup() {
  11. const dir = this.dirNode;
  12. const u = dir.z.atan2( dir.x ).mul( 1 / ( Math.PI * 2 ) ).add( 0.5 );
  13. const v = dir.y.clamp( - 1.0, 1.0 ).asin().mul( 1 / Math.PI ).add( 0.5 );
  14. return vec2( u, v );
  15. }
  16. }
  17. export default EquirectUVNode;
  18. export const equirectUV = nodeProxy( EquirectUVNode );
  19. addNodeClass( 'EquirectUVNode', EquirectUVNode );