SelectionBox.js 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. import {
  2. Frustum,
  3. Vector3
  4. } from '../../../build/three.module.js';
  5. /**
  6. * This is a class to check whether objects are in a selection area in 3D space
  7. */
  8. const _frustum = new Frustum();
  9. const _center = new Vector3();
  10. const _tmpPoint = new Vector3();
  11. const _vecNear = new Vector3();
  12. const _vecTopLeft = new Vector3();
  13. const _vecTopRight = new Vector3();
  14. const _vecDownRight = new Vector3();
  15. const _vecDownLeft = new Vector3();
  16. const _vecFarTopLeft = new Vector3();
  17. const _vecFarTopRight = new Vector3();
  18. const _vecFarDownRight = new Vector3();
  19. const _vecFarDownLeft = new Vector3();
  20. const _vectemp1 = new Vector3();
  21. const _vectemp2 = new Vector3();
  22. const _vectemp3 = new Vector3();
  23. class SelectionBox {
  24. constructor( camera, scene, deep = Number.MAX_VALUE ) {
  25. this.camera = camera;
  26. this.scene = scene;
  27. this.startPoint = new Vector3();
  28. this.endPoint = new Vector3();
  29. this.collection = [];
  30. this.deep = deep;
  31. }
  32. select( startPoint, endPoint ) {
  33. this.startPoint = startPoint || this.startPoint;
  34. this.endPoint = endPoint || this.endPoint;
  35. this.collection = [];
  36. this.updateFrustum( this.startPoint, this.endPoint );
  37. this.searchChildInFrustum( _frustum, this.scene );
  38. return this.collection;
  39. }
  40. updateFrustum( startPoint, endPoint ) {
  41. startPoint = startPoint || this.startPoint;
  42. endPoint = endPoint || this.endPoint;
  43. // Avoid invalid frustum
  44. if ( startPoint.x === endPoint.x ) {
  45. endPoint.x += Number.EPSILON;
  46. }
  47. if ( startPoint.y === endPoint.y ) {
  48. endPoint.y += Number.EPSILON;
  49. }
  50. this.camera.updateProjectionMatrix();
  51. this.camera.updateMatrixWorld();
  52. if ( this.camera.isPerspectiveCamera ) {
  53. _tmpPoint.copy( startPoint );
  54. _tmpPoint.x = Math.min( startPoint.x, endPoint.x );
  55. _tmpPoint.y = Math.max( startPoint.y, endPoint.y );
  56. endPoint.x = Math.max( startPoint.x, endPoint.x );
  57. endPoint.y = Math.min( startPoint.y, endPoint.y );
  58. _vecNear.setFromMatrixPosition( this.camera.matrixWorld );
  59. _vecTopLeft.copy( _tmpPoint );
  60. _vecTopRight.set( endPoint.x, _tmpPoint.y, 0 );
  61. _vecDownRight.copy( endPoint );
  62. _vecDownLeft.set( _tmpPoint.x, endPoint.y, 0 );
  63. _vecTopLeft.unproject( this.camera );
  64. _vecTopRight.unproject( this.camera );
  65. _vecDownRight.unproject( this.camera );
  66. _vecDownLeft.unproject( this.camera );
  67. _vectemp1.copy( _vecTopLeft ).sub( _vecNear );
  68. _vectemp2.copy( _vecTopRight ).sub( _vecNear );
  69. _vectemp3.copy( _vecDownRight ).sub( _vecNear );
  70. _vectemp1.normalize();
  71. _vectemp2.normalize();
  72. _vectemp3.normalize();
  73. _vectemp1.multiplyScalar( this.deep );
  74. _vectemp2.multiplyScalar( this.deep );
  75. _vectemp3.multiplyScalar( this.deep );
  76. _vectemp1.add( _vecNear );
  77. _vectemp2.add( _vecNear );
  78. _vectemp3.add( _vecNear );
  79. const planes = _frustum.planes;
  80. planes[ 0 ].setFromCoplanarPoints( _vecNear, _vecTopLeft, _vecTopRight );
  81. planes[ 1 ].setFromCoplanarPoints( _vecNear, _vecTopRight, _vecDownRight );
  82. planes[ 2 ].setFromCoplanarPoints( _vecDownRight, _vecDownLeft, _vecNear );
  83. planes[ 3 ].setFromCoplanarPoints( _vecDownLeft, _vecTopLeft, _vecNear );
  84. planes[ 4 ].setFromCoplanarPoints( _vecTopRight, _vecDownRight, _vecDownLeft );
  85. planes[ 5 ].setFromCoplanarPoints( _vectemp3, _vectemp2, _vectemp1 );
  86. planes[ 5 ].normal.multiplyScalar( - 1 );
  87. } else if ( this.camera.isOrthographicCamera ) {
  88. const left = Math.min( startPoint.x, endPoint.x );
  89. const top = Math.max( startPoint.y, endPoint.y );
  90. const right = Math.max( startPoint.x, endPoint.x );
  91. const down = Math.min( startPoint.y, endPoint.y );
  92. _vecTopLeft.set( left, top, - 1 );
  93. _vecTopRight.set( right, top, - 1 );
  94. _vecDownRight.set( right, down, - 1 );
  95. _vecDownLeft.set( left, down, - 1 );
  96. _vecFarTopLeft.set( left, top, 1 );
  97. _vecFarTopRight.set( right, top, 1 );
  98. _vecFarDownRight.set( right, down, 1 );
  99. _vecFarDownLeft.set( left, down, 1 );
  100. _vecTopLeft.unproject( this.camera );
  101. _vecTopRight.unproject( this.camera );
  102. _vecDownRight.unproject( this.camera );
  103. _vecDownLeft.unproject( this.camera );
  104. _vecFarTopLeft.unproject( this.camera );
  105. _vecFarTopRight.unproject( this.camera );
  106. _vecFarDownRight.unproject( this.camera );
  107. _vecFarDownLeft.unproject( this.camera );
  108. const planes = _frustum.planes;
  109. planes[ 0 ].setFromCoplanarPoints( _vecTopLeft, _vecFarTopLeft, _vecFarTopRight );
  110. planes[ 1 ].setFromCoplanarPoints( _vecTopRight, _vecFarTopRight, _vecFarDownRight );
  111. planes[ 2 ].setFromCoplanarPoints( _vecFarDownRight, _vecFarDownLeft, _vecDownLeft );
  112. planes[ 3 ].setFromCoplanarPoints( _vecFarDownLeft, _vecFarTopLeft, _vecTopLeft );
  113. planes[ 4 ].setFromCoplanarPoints( _vecTopRight, _vecDownRight, _vecDownLeft );
  114. planes[ 5 ].setFromCoplanarPoints( _vecFarDownRight, _vecFarTopRight, _vecFarTopLeft );
  115. planes[ 5 ].normal.multiplyScalar( - 1 );
  116. } else {
  117. console.error( 'THREE.SelectionBox: Unsupported camera type.' );
  118. }
  119. }
  120. searchChildInFrustum( frustum, object ) {
  121. if ( object.isMesh || object.isLine || object.isPoints ) {
  122. if ( object.material !== undefined ) {
  123. if ( object.geometry.boundingSphere === null ) object.geometry.computeBoundingSphere();
  124. _center.copy( object.geometry.boundingSphere.center );
  125. _center.applyMatrix4( object.matrixWorld );
  126. if ( frustum.containsPoint( _center ) ) {
  127. this.collection.push( object );
  128. }
  129. }
  130. }
  131. if ( object.children.length > 0 ) {
  132. for ( let x = 0; x < object.children.length; x ++ ) {
  133. this.searchChildInFrustum( frustum, object.children[ x ] );
  134. }
  135. }
  136. }
  137. }
  138. export { SelectionBox };