VertexTangentsHelper.js 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  1. import {
  2. BufferGeometry,
  3. Float32BufferAttribute,
  4. LineSegments,
  5. LineBasicMaterial,
  6. Vector3
  7. } from '../../../build/three.module.js';
  8. var _v1 = new Vector3();
  9. var _v2 = new Vector3();
  10. function VertexTangentsHelper( object, size, hex ) {
  11. this.object = object;
  12. this.size = ( size !== undefined ) ? size : 1;
  13. var color = ( hex !== undefined ) ? hex : 0x00ffff;
  14. //
  15. var objGeometry = this.object.geometry;
  16. if ( ! ( objGeometry && objGeometry.isBufferGeometry ) ) {
  17. console.error( 'THREE.VertexTangentsHelper: geometry not an instance of THREE.BufferGeometry.', objGeometry );
  18. return;
  19. }
  20. var nTangents = objGeometry.attributes.tangent.count;
  21. //
  22. var geometry = new BufferGeometry();
  23. var positions = new Float32BufferAttribute( nTangents * 2 * 3, 3 );
  24. geometry.setAttribute( 'position', positions );
  25. LineSegments.call( this, geometry, new LineBasicMaterial( { color: color, toneMapped: false } ) );
  26. this.type = 'VertexTangentsHelper';
  27. //
  28. this.matrixAutoUpdate = false;
  29. this.update();
  30. }
  31. VertexTangentsHelper.prototype = Object.create( LineSegments.prototype );
  32. VertexTangentsHelper.prototype.constructor = VertexTangentsHelper;
  33. VertexTangentsHelper.prototype.update = function () {
  34. this.object.updateMatrixWorld( true );
  35. var matrixWorld = this.object.matrixWorld;
  36. var position = this.geometry.attributes.position;
  37. //
  38. var objGeometry = this.object.geometry;
  39. var objPos = objGeometry.attributes.position;
  40. var objTan = objGeometry.attributes.tangent;
  41. var idx = 0;
  42. // for simplicity, ignore index and drawcalls, and render every tangent
  43. for ( var j = 0, jl = objPos.count; j < jl; j ++ ) {
  44. _v1.set( objPos.getX( j ), objPos.getY( j ), objPos.getZ( j ) ).applyMatrix4( matrixWorld );
  45. _v2.set( objTan.getX( j ), objTan.getY( j ), objTan.getZ( j ) );
  46. _v2.transformDirection( matrixWorld ).multiplyScalar( this.size ).add( _v1 );
  47. position.setXYZ( idx, _v1.x, _v1.y, _v1.z );
  48. idx = idx + 1;
  49. position.setXYZ( idx, _v2.x, _v2.y, _v2.z );
  50. idx = idx + 1;
  51. }
  52. position.needsUpdate = true;
  53. };
  54. export { VertexTangentsHelper };