DragControls.js 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. import {
  2. EventDispatcher,
  3. Matrix4,
  4. Plane,
  5. Raycaster,
  6. Vector2,
  7. Vector3
  8. } from 'three';
  9. const _plane = new Plane();
  10. const _raycaster = new Raycaster();
  11. const _pointer = new Vector2();
  12. const _offset = new Vector3();
  13. const _intersection = new Vector3();
  14. const _worldPosition = new Vector3();
  15. const _inverseMatrix = new Matrix4();
  16. class DragControls extends EventDispatcher {
  17. constructor( _objects, _camera, _domElement ) {
  18. super();
  19. _domElement.style.touchAction = 'none'; // disable touch scroll
  20. let _selected = null, _hovered = null;
  21. const _intersections = [];
  22. //
  23. const scope = this;
  24. function activate() {
  25. _domElement.addEventListener( 'pointermove', onPointerMove );
  26. _domElement.addEventListener( 'pointerdown', onPointerDown );
  27. _domElement.addEventListener( 'pointerup', onPointerCancel );
  28. _domElement.addEventListener( 'pointerleave', onPointerCancel );
  29. }
  30. function deactivate() {
  31. _domElement.removeEventListener( 'pointermove', onPointerMove );
  32. _domElement.removeEventListener( 'pointerdown', onPointerDown );
  33. _domElement.removeEventListener( 'pointerup', onPointerCancel );
  34. _domElement.removeEventListener( 'pointerleave', onPointerCancel );
  35. _domElement.style.cursor = '';
  36. }
  37. function dispose() {
  38. deactivate();
  39. }
  40. function getObjects() {
  41. return _objects;
  42. }
  43. function onPointerMove( event ) {
  44. if ( scope.enabled === false ) return;
  45. updatePointer( event );
  46. _raycaster.setFromCamera( _pointer, _camera );
  47. if ( _selected ) {
  48. if ( _raycaster.ray.intersectPlane( _plane, _intersection ) ) {
  49. _selected.position.copy( _intersection.sub( _offset ).applyMatrix4( _inverseMatrix ) );
  50. }
  51. scope.dispatchEvent( { type: 'drag', object: _selected } );
  52. return;
  53. }
  54. // hover support
  55. if ( event.pointerType === 'mouse' || event.pointerType === 'pen' ) {
  56. _intersections.length = 0;
  57. _raycaster.setFromCamera( _pointer, _camera );
  58. _raycaster.intersectObjects( _objects, true, _intersections );
  59. if ( _intersections.length > 0 ) {
  60. const object = _intersections[ 0 ].object;
  61. _plane.setFromNormalAndCoplanarPoint( _camera.getWorldDirection( _plane.normal ), _worldPosition.setFromMatrixPosition( object.matrixWorld ) );
  62. if ( _hovered !== object && _hovered !== null ) {
  63. scope.dispatchEvent( { type: 'hoveroff', object: _hovered } );
  64. _domElement.style.cursor = 'auto';
  65. _hovered = null;
  66. }
  67. if ( _hovered !== object ) {
  68. scope.dispatchEvent( { type: 'hoveron', object: object } );
  69. _domElement.style.cursor = 'pointer';
  70. _hovered = object;
  71. }
  72. } else {
  73. if ( _hovered !== null ) {
  74. scope.dispatchEvent( { type: 'hoveroff', object: _hovered } );
  75. _domElement.style.cursor = 'auto';
  76. _hovered = null;
  77. }
  78. }
  79. }
  80. }
  81. function onPointerDown( event ) {
  82. if ( scope.enabled === false ) return;
  83. updatePointer( event );
  84. _intersections.length = 0;
  85. _raycaster.setFromCamera( _pointer, _camera );
  86. _raycaster.intersectObjects( _objects, true, _intersections );
  87. if ( _intersections.length > 0 ) {
  88. _selected = ( scope.transformGroup === true ) ? _objects[ 0 ] : _intersections[ 0 ].object;
  89. _plane.setFromNormalAndCoplanarPoint( _camera.getWorldDirection( _plane.normal ), _worldPosition.setFromMatrixPosition( _selected.matrixWorld ) );
  90. if ( _raycaster.ray.intersectPlane( _plane, _intersection ) ) {
  91. _inverseMatrix.copy( _selected.parent.matrixWorld ).invert();
  92. _offset.copy( _intersection ).sub( _worldPosition.setFromMatrixPosition( _selected.matrixWorld ) );
  93. }
  94. _domElement.style.cursor = 'move';
  95. scope.dispatchEvent( { type: 'dragstart', object: _selected } );
  96. }
  97. }
  98. function onPointerCancel() {
  99. if ( scope.enabled === false ) return;
  100. if ( _selected ) {
  101. scope.dispatchEvent( { type: 'dragend', object: _selected } );
  102. _selected = null;
  103. }
  104. _domElement.style.cursor = _hovered ? 'pointer' : 'auto';
  105. }
  106. function updatePointer( event ) {
  107. const rect = _domElement.getBoundingClientRect();
  108. _pointer.x = ( event.clientX - rect.left ) / rect.width * 2 - 1;
  109. _pointer.y = - ( event.clientY - rect.top ) / rect.height * 2 + 1;
  110. }
  111. activate();
  112. // API
  113. this.enabled = true;
  114. this.transformGroup = false;
  115. this.activate = activate;
  116. this.deactivate = deactivate;
  117. this.dispose = dispose;
  118. this.getObjects = getObjects;
  119. }
  120. }
  121. export { DragControls };