CSS2DRenderer.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192
  1. THREE.CSS2DObject = function ( element ) {
  2. THREE.Object3D.call( this );
  3. this.element = element || document.createElement( 'div' );
  4. this.element.style.position = 'absolute';
  5. this.addEventListener( 'removed', function () {
  6. this.traverse( function ( object ) {
  7. if ( object.element instanceof Element && object.element.parentNode !== null ) {
  8. object.element.parentNode.removeChild( object.element );
  9. }
  10. } );
  11. } );
  12. };
  13. THREE.CSS2DObject.prototype = Object.assign( Object.create( THREE.Object3D.prototype ), {
  14. constructor: THREE.CSS2DObject,
  15. copy: function ( source, recursive ) {
  16. THREE.Object3D.prototype.copy.call( this, source, recursive );
  17. this.element = source.element.cloneNode( true );
  18. return this;
  19. }
  20. } );
  21. //
  22. THREE.CSS2DRenderer = function () {
  23. var _this = this;
  24. var _width, _height;
  25. var _widthHalf, _heightHalf;
  26. var vector = new THREE.Vector3();
  27. var viewMatrix = new THREE.Matrix4();
  28. var viewProjectionMatrix = new THREE.Matrix4();
  29. var cache = {
  30. objects: new WeakMap()
  31. };
  32. var domElement = document.createElement( 'div' );
  33. domElement.style.overflow = 'hidden';
  34. this.domElement = domElement;
  35. this.getSize = function () {
  36. return {
  37. width: _width,
  38. height: _height
  39. };
  40. };
  41. this.setSize = function ( width, height ) {
  42. _width = width;
  43. _height = height;
  44. _widthHalf = _width / 2;
  45. _heightHalf = _height / 2;
  46. domElement.style.width = width + 'px';
  47. domElement.style.height = height + 'px';
  48. };
  49. var renderObject = function ( object, scene, camera ) {
  50. if ( object instanceof THREE.CSS2DObject ) {
  51. object.onBeforeRender( _this, scene, camera );
  52. vector.setFromMatrixPosition( object.matrixWorld );
  53. vector.applyMatrix4( viewProjectionMatrix );
  54. var element = object.element;
  55. var style = 'translate(-50%,-50%) translate(' + ( vector.x * _widthHalf + _widthHalf ) + 'px,' + ( - vector.y * _heightHalf + _heightHalf ) + 'px)';
  56. element.style.WebkitTransform = style;
  57. element.style.MozTransform = style;
  58. element.style.oTransform = style;
  59. element.style.transform = style;
  60. element.style.display = ( object.visible && vector.z >= - 1 && vector.z <= 1 ) ? '' : 'none';
  61. var objectData = {
  62. distanceToCameraSquared: getDistanceToSquared( camera, object )
  63. };
  64. cache.objects.set( object, objectData );
  65. if ( element.parentNode !== domElement ) {
  66. domElement.appendChild( element );
  67. }
  68. object.onAfterRender( _this, scene, camera );
  69. }
  70. for ( var i = 0, l = object.children.length; i < l; i ++ ) {
  71. renderObject( object.children[ i ], scene, camera );
  72. }
  73. };
  74. var getDistanceToSquared = function () {
  75. var a = new THREE.Vector3();
  76. var b = new THREE.Vector3();
  77. return function ( object1, object2 ) {
  78. a.setFromMatrixPosition( object1.matrixWorld );
  79. b.setFromMatrixPosition( object2.matrixWorld );
  80. return a.distanceToSquared( b );
  81. };
  82. }();
  83. var filterAndFlatten = function ( scene ) {
  84. var result = [];
  85. scene.traverse( function ( object ) {
  86. if ( object instanceof THREE.CSS2DObject ) result.push( object );
  87. } );
  88. return result;
  89. };
  90. var zOrder = function ( scene ) {
  91. var sorted = filterAndFlatten( scene ).sort( function ( a, b ) {
  92. var distanceA = cache.objects.get( a ).distanceToCameraSquared;
  93. var distanceB = cache.objects.get( b ).distanceToCameraSquared;
  94. return distanceA - distanceB;
  95. } );
  96. var zMax = sorted.length;
  97. for ( var i = 0, l = sorted.length; i < l; i ++ ) {
  98. sorted[ i ].element.style.zIndex = zMax - i;
  99. }
  100. };
  101. this.render = function ( scene, camera ) {
  102. if ( scene.autoUpdate === true ) scene.updateMatrixWorld();
  103. if ( camera.parent === null ) camera.updateMatrixWorld();
  104. viewMatrix.copy( camera.matrixWorldInverse );
  105. viewProjectionMatrix.multiplyMatrices( camera.projectionMatrix, viewMatrix );
  106. renderObject( scene, scene, camera );
  107. zOrder( scene );
  108. };
  109. };