216 lines
4.3 KiB
JavaScript
216 lines
4.3 KiB
JavaScript
import {
|
|
Matrix4,
|
|
Object3D,
|
|
Vector2,
|
|
Vector3
|
|
} from 'three';
|
|
|
|
class CSS2DObject extends Object3D {
|
|
|
|
constructor( element = document.createElement( 'div' ) ) {
|
|
|
|
super();
|
|
|
|
this.isCSS2DObject = true;
|
|
|
|
this.element = element;
|
|
|
|
this.element.style.position = 'absolute';
|
|
this.element.style.userSelect = 'none';
|
|
|
|
this.element.setAttribute( 'draggable', false );
|
|
|
|
this.center = new Vector2( 0.5, 0.5 ); // ( 0, 0 ) is the lower left; ( 1, 1 ) is the top right
|
|
|
|
this.addEventListener( 'removed', function () {
|
|
|
|
this.traverse( function ( object ) {
|
|
|
|
if ( object.element instanceof Element && object.element.parentNode !== null ) {
|
|
|
|
object.element.parentNode.removeChild( object.element );
|
|
|
|
}
|
|
|
|
} );
|
|
|
|
} );
|
|
|
|
}
|
|
|
|
copy( source, recursive ) {
|
|
|
|
super.copy( source, recursive );
|
|
|
|
this.element = source.element.cloneNode( true );
|
|
|
|
this.center = source.center;
|
|
|
|
return this;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//
|
|
|
|
const _vector = new Vector3();
|
|
const _viewMatrix = new Matrix4();
|
|
const _viewProjectionMatrix = new Matrix4();
|
|
const _a = new Vector3();
|
|
const _b = new Vector3();
|
|
|
|
class CSS2DRenderer {
|
|
|
|
constructor( parameters = {} ) {
|
|
|
|
const _this = this;
|
|
|
|
let _width, _height;
|
|
let _widthHalf, _heightHalf;
|
|
|
|
const cache = {
|
|
objects: new WeakMap()
|
|
};
|
|
|
|
const domElement = parameters.element !== undefined ? parameters.element : document.createElement( 'div' );
|
|
|
|
domElement.style.overflow = 'hidden';
|
|
|
|
this.domElement = domElement;
|
|
|
|
this.getSize = function () {
|
|
|
|
return {
|
|
width: _width,
|
|
height: _height
|
|
};
|
|
|
|
};
|
|
|
|
this.render = function ( scene, camera ) {
|
|
|
|
if ( scene.matrixWorldAutoUpdate === true ) scene.updateMatrixWorld();
|
|
if ( camera.parent === null && camera.matrixWorldAutoUpdate === true ) camera.updateMatrixWorld();
|
|
|
|
_viewMatrix.copy( camera.matrixWorldInverse );
|
|
_viewProjectionMatrix.multiplyMatrices( camera.projectionMatrix, _viewMatrix );
|
|
|
|
renderObject( scene, scene, camera );
|
|
zOrder( scene );
|
|
|
|
};
|
|
|
|
this.setSize = function ( width, height ) {
|
|
|
|
_width = width;
|
|
_height = height;
|
|
|
|
_widthHalf = _width / 2;
|
|
_heightHalf = _height / 2;
|
|
|
|
domElement.style.width = width + 'px';
|
|
domElement.style.height = height + 'px';
|
|
|
|
};
|
|
|
|
function renderObject( object, scene, camera ) {
|
|
|
|
if ( object.isCSS2DObject ) {
|
|
|
|
_vector.setFromMatrixPosition( object.matrixWorld );
|
|
_vector.applyMatrix4( _viewProjectionMatrix );
|
|
|
|
const visible = ( object.visible === true ) && ( _vector.z >= - 1 && _vector.z <= 1 ) && ( object.layers.test( camera.layers ) === true );
|
|
object.element.style.display = ( visible === true ) ? '' : 'none';
|
|
|
|
if ( visible === true ) {
|
|
|
|
object.onBeforeRender( _this, scene, camera );
|
|
|
|
const element = object.element;
|
|
|
|
element.style.transform = 'translate(' + ( - 100 * object.center.x ) + '%,' + ( - 100 * object.center.y ) + '%)' + 'translate(' + ( _vector.x * _widthHalf + _widthHalf ) + 'px,' + ( - _vector.y * _heightHalf + _heightHalf ) + 'px)';
|
|
|
|
if ( element.parentNode !== domElement ) {
|
|
|
|
domElement.appendChild( element );
|
|
|
|
}
|
|
|
|
object.onAfterRender( _this, scene, camera );
|
|
|
|
}
|
|
|
|
const objectData = {
|
|
distanceToCameraSquared: getDistanceToSquared( camera, object )
|
|
};
|
|
|
|
cache.objects.set( object, objectData );
|
|
|
|
}
|
|
|
|
for ( let i = 0, l = object.children.length; i < l; i ++ ) {
|
|
|
|
renderObject( object.children[ i ], scene, camera );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
function getDistanceToSquared( object1, object2 ) {
|
|
|
|
_a.setFromMatrixPosition( object1.matrixWorld );
|
|
_b.setFromMatrixPosition( object2.matrixWorld );
|
|
|
|
return _a.distanceToSquared( _b );
|
|
|
|
}
|
|
|
|
function filterAndFlatten( scene ) {
|
|
|
|
const result = [];
|
|
|
|
scene.traverse( function ( object ) {
|
|
|
|
if ( object.isCSS2DObject ) result.push( object );
|
|
|
|
} );
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
function zOrder( scene ) {
|
|
|
|
const sorted = filterAndFlatten( scene ).sort( function ( a, b ) {
|
|
|
|
if ( a.renderOrder !== b.renderOrder ) {
|
|
|
|
return b.renderOrder - a.renderOrder;
|
|
|
|
}
|
|
|
|
const distanceA = cache.objects.get( a ).distanceToCameraSquared;
|
|
const distanceB = cache.objects.get( b ).distanceToCameraSquared;
|
|
|
|
return distanceA - distanceB;
|
|
|
|
} );
|
|
|
|
const zMax = sorted.length;
|
|
|
|
for ( let i = 0, l = sorted.length; i < l; i ++ ) {
|
|
|
|
sorted[ i ].element.style.zIndex = zMax - i;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
export { CSS2DObject, CSS2DRenderer };
|