NormalNode.js 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. import Node from '../core/Node.js';
  2. import AttributeNode from '../core/AttributeNode.js';
  3. import VaryNode from '../core/VaryNode.js';
  4. import ModelNode from '../accessors/ModelNode.js';
  5. import CameraNode from '../accessors/CameraNode.js';
  6. import OperatorNode from '../math/OperatorNode.js';
  7. import MathNode from '../math/MathNode.js';
  8. import { inverseTransformDirection } from '../functions/MathFunctions.js';
  9. class NormalNode extends Node {
  10. static LOCAL = 'local';
  11. static WORLD = 'world';
  12. static VIEW = 'view';
  13. constructor( scope = NormalNode.LOCAL ) {
  14. super( 'vec3' );
  15. this.scope = scope;
  16. }
  17. generate( builder ) {
  18. const scope = this.scope;
  19. let outputNode = null;
  20. if ( scope === NormalNode.LOCAL ) {
  21. outputNode = new AttributeNode( 'normal', 'vec3' );
  22. } else if ( scope === NormalNode.VIEW ) {
  23. const vertexNormalNode = new OperatorNode( '*', new ModelNode( ModelNode.NORMAL_MATRIX ), new NormalNode( NormalNode.LOCAL ) );
  24. outputNode = new MathNode( MathNode.NORMALIZE, new VaryNode( vertexNormalNode ) );
  25. } else if ( scope === NormalNode.WORLD ) {
  26. const vertexNormalNode = inverseTransformDirection.call( { dir: new NormalNode( NormalNode.VIEW ), matrix: new CameraNode( CameraNode.VIEW_MATRIX ) } );
  27. outputNode = new MathNode( MathNode.NORMALIZE, new VaryNode( vertexNormalNode ) );
  28. }
  29. return outputNode.build( builder );
  30. }
  31. }
  32. export default NormalNode;