CSS2DRenderer.js 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. import {
  2. Matrix4,
  3. CSS2DObject,
  4. Vector3
  5. } from 'three';
  6. //
  7. // class CSS2DObject extends Object3D {
  8. //
  9. // constructor( element = document.createElement( 'div' ) ) {
  10. //
  11. // super();
  12. //
  13. // this.isCSS2DObject = true;
  14. //
  15. // this.element = element;
  16. //
  17. // this.element.style.position = 'absolute';
  18. // this.element.style.userSelect = 'none';
  19. //
  20. // this.element.setAttribute( 'draggable', false );
  21. //
  22. // this.addEventListener( 'removed', function () {
  23. //
  24. // this.traverse( function ( object ) {
  25. //
  26. // if ( object.element instanceof Element && object.element.parentNode !== null ) {
  27. //
  28. // object.element.parentNode.removeChild( object.element );
  29. //
  30. // }
  31. //
  32. // } );
  33. //
  34. // } );
  35. //
  36. // }
  37. //
  38. // copy( source, recursive ) {
  39. //
  40. // super.copy( source, recursive );
  41. //
  42. // this.element = source.element.cloneNode( true );
  43. //
  44. // return this;
  45. //
  46. // }
  47. //
  48. // }
  49. //
  50. const _vector = new Vector3();
  51. const _viewMatrix = new Matrix4();
  52. const _viewProjectionMatrix = new Matrix4();
  53. const _a = new Vector3();
  54. const _b = new Vector3();
  55. class CSS2DRenderer {
  56. constructor( parameters = {} ) {
  57. const _this = this;
  58. let _width, _height;
  59. let _widthHalf, _heightHalf;
  60. const cache = {
  61. objects: new WeakMap()
  62. };
  63. const domElement = parameters.element !== undefined ? parameters.element : document.createElement( 'div' );
  64. domElement.style.overflow = 'hidden';
  65. domElement.style.pointerEvents = 'none';
  66. this.domElement = domElement;
  67. this.getSize = function () {
  68. return {
  69. width: _width,
  70. height: _height
  71. };
  72. };
  73. this.render = function ( scene, camera ) {
  74. if ( scene.matrixWorldAutoUpdate === true ) scene.updateMatrixWorld();
  75. if ( camera.parent === null && camera.matrixWorldAutoUpdate === true ) camera.updateMatrixWorld();
  76. _viewMatrix.copy( camera.matrixWorldInverse );
  77. _viewProjectionMatrix.multiplyMatrices( camera.projectionMatrix, _viewMatrix );
  78. renderObject( scene, scene, camera );
  79. zOrder( scene );
  80. };
  81. this.clean = function () {
  82. domElement.innerHTML = ""
  83. return ;
  84. };
  85. this.setSize = function ( width, height ) {
  86. _width = width;
  87. _height = height;
  88. _widthHalf = _width / 2;
  89. _heightHalf = _height / 2;
  90. domElement.style.width = width + 'px';
  91. domElement.style.height = height + 'px';
  92. };
  93. function renderObject( object, scene, camera ) {
  94. if ( object.isCSS2DObject ) {
  95. _vector.setFromMatrixPosition( object.matrixWorld );
  96. _vector.applyMatrix4( _viewProjectionMatrix );
  97. const visible = ( object.visible === true ) && ( _vector.z >= - 1 && _vector.z <= 1 ) && ( object.layers.test( camera.layers ) === true );
  98. object.element.style.display = ( visible === true ) ? '' : 'none';
  99. if ( visible === true ) {
  100. object.onBeforeRender( _this, scene, camera );
  101. const element = object.element;
  102. element.style.transform = 'translate(-50%,-50%) translate(' + ( _vector.x * _widthHalf + _widthHalf ) + 'px,' + ( - _vector.y * _heightHalf + _heightHalf ) + 'px)';
  103. element.style.pointerEvents = 'initial';
  104. if ( element.parentNode !== domElement ) {
  105. domElement.appendChild( element );
  106. }
  107. object.onAfterRender( _this, scene, camera );
  108. }
  109. const objectData = {
  110. distanceToCameraSquared: getDistanceToSquared( camera, object )
  111. };
  112. cache.objects.set( object, objectData );
  113. }
  114. for ( let i = 0, l = object.children.length; i < l; i ++ ) {
  115. renderObject( object.children[ i ], scene, camera );
  116. }
  117. }
  118. function getDistanceToSquared( object1, object2 ) {
  119. _a.setFromMatrixPosition( object1.matrixWorld );
  120. _b.setFromMatrixPosition( object2.matrixWorld );
  121. return _a.distanceToSquared( _b );
  122. }
  123. function filterAndFlatten( scene ) {
  124. const result = [];
  125. scene.traverse( function ( object ) {
  126. if ( object.isCSS2DObject ) result.push( object );
  127. } );
  128. return result;
  129. }
  130. function zOrder( scene ) {
  131. const sorted = filterAndFlatten( scene ).sort( function ( a, b ) {
  132. if ( a.renderOrder !== b.renderOrder ) {
  133. return b.renderOrder - a.renderOrder;
  134. }
  135. const distanceA = cache.objects.get( a ).distanceToCameraSquared;
  136. const distanceB = cache.objects.get( b ).distanceToCameraSquared;
  137. return distanceA - distanceB;
  138. } );
  139. const zMax = sorted.length;
  140. for ( let i = 0, l = sorted.length; i < l; i ++ ) {
  141. sorted[ i ].element.style.zIndex = zMax - i;
  142. }
  143. }
  144. }
  145. }
  146. export { CSS2DObject, CSS2DRenderer };