TangentNode.js 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. import Node, { addNodeClass } from '../core/Node.js';
  2. import { attribute } from '../core/AttributeNode.js';
  3. import { label } from '../core/VarNode.js';
  4. import { varying } from '../core/VaryingNode.js';
  5. import { normalize } from '../math/MathNode.js';
  6. import { cameraViewMatrix } from './CameraNode.js';
  7. import { modelViewMatrix } from './ModelNode.js';
  8. import { nodeImmutable } from '../shadernode/ShaderNode.js';
  9. class TangentNode extends Node {
  10. constructor( scope = TangentNode.LOCAL ) {
  11. super();
  12. this.scope = scope;
  13. }
  14. getHash( /*builder*/ ) {
  15. return `tangent-${this.scope}`;
  16. }
  17. getNodeType() {
  18. const scope = this.scope;
  19. if ( scope === TangentNode.GEOMETRY ) {
  20. return 'vec4';
  21. }
  22. return 'vec3';
  23. }
  24. generate( builder ) {
  25. const scope = this.scope;
  26. let outputNode = null;
  27. if ( scope === TangentNode.GEOMETRY ) {
  28. outputNode = attribute( 'tangent', 'vec4' );
  29. } else if ( scope === TangentNode.LOCAL ) {
  30. outputNode = varying( tangentGeometry.xyz );
  31. } else if ( scope === TangentNode.VIEW ) {
  32. const vertexNode = modelViewMatrix.mul( tangentLocal ).xyz;
  33. outputNode = normalize( varying( vertexNode ) );
  34. } else if ( scope === TangentNode.WORLD ) {
  35. const vertexNode = tangentView.transformDirection( cameraViewMatrix );
  36. outputNode = normalize( varying( vertexNode ) );
  37. }
  38. return outputNode.build( builder, this.getNodeType( builder ) );
  39. }
  40. serialize( data ) {
  41. super.serialize( data );
  42. data.scope = this.scope;
  43. }
  44. deserialize( data ) {
  45. super.deserialize( data );
  46. this.scope = data.scope;
  47. }
  48. }
  49. TangentNode.GEOMETRY = 'geometry';
  50. TangentNode.LOCAL = 'local';
  51. TangentNode.VIEW = 'view';
  52. TangentNode.WORLD = 'world';
  53. export default TangentNode;
  54. export const tangentGeometry = nodeImmutable( TangentNode, TangentNode.GEOMETRY );
  55. export const tangentLocal = nodeImmutable( TangentNode, TangentNode.LOCAL );
  56. export const tangentView = nodeImmutable( TangentNode, TangentNode.VIEW );
  57. export const tangentWorld = nodeImmutable( TangentNode, TangentNode.WORLD );
  58. export const transformedTangentView = label( tangentView, 'TransformedTangentView' );
  59. export const transformedTangentWorld = normalize( transformedTangentView.transformDirection( cameraViewMatrix ) );
  60. addNodeClass( TangentNode );