123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332 |
- import {
- Controls,
- Quaternion,
- Vector3
- } from 'three';
- const _changeEvent = { type: 'change' };
- const _EPS = 0.000001;
- const _tmpQuaternion = new Quaternion();
- class FlyControls extends Controls {
- constructor( object, domElement = null ) {
- super( object, domElement );
- this.movementSpeed = 1.0;
- this.rollSpeed = 0.005;
- this.dragToLook = false;
- this.autoForward = false;
- // internals
- this._moveState = { up: 0, down: 0, left: 0, right: 0, forward: 0, back: 0, pitchUp: 0, pitchDown: 0, yawLeft: 0, yawRight: 0, rollLeft: 0, rollRight: 0 };
- this._moveVector = new Vector3( 0, 0, 0 );
- this._rotationVector = new Vector3( 0, 0, 0 );
- this._lastQuaternion = new Quaternion();
- this._lastPosition = new Vector3();
- this._status = 0;
- // event listeners
- this._onKeyDown = onKeyDown.bind( this );
- this._onKeyUp = onKeyUp.bind( this );
- this._onPointerMove = onPointerMove.bind( this );
- this._onPointerDown = onPointerDown.bind( this );
- this._onPointerUp = onPointerUp.bind( this );
- this._onPointerCancel = onPointerCancel.bind( this );
- this._onContextMenu = onContextMenu.bind( this );
- //
- if ( domElement !== null ) {
- this.connect();
- }
- }
- connect() {
- window.addEventListener( 'keydown', this._onKeyDown );
- window.addEventListener( 'keyup', this._onKeyUp );
- this.domElement.addEventListener( 'pointermove', this._onPointerMove );
- this.domElement.addEventListener( 'pointerdown', this._onPointerDown );
- this.domElement.addEventListener( 'pointerup', this._onPointerUp );
- this.domElement.addEventListener( 'pointercancel', this._onPointerCancel );
- this.domElement.addEventListener( 'contextmenu', this._onContextMenu );
- }
- disconnect() {
- window.removeEventListener( 'keydown', this._onKeyDown );
- window.removeEventListener( 'keyup', this._onKeyUp );
- this.domElement.removeEventListener( 'pointermove', this._onPointerMove );
- this.domElement.removeEventListener( 'pointerdown', this._onPointerDown );
- this.domElement.removeEventListener( 'pointerup', this._onPointerUp );
- this.domElement.removeEventListener( 'pointercancel', this._onPointerCancel );
- this.domElement.removeEventListener( 'contextmenu', this._onContextMenu );
- }
- dispose() {
- this.disconnect();
- }
- update( delta ) {
- if ( this.enabled === false ) return;
- const object = this.object;
- const moveMult = delta * this.movementSpeed;
- const rotMult = delta * this.rollSpeed;
- object.translateX( this._moveVector.x * moveMult );
- object.translateY( this._moveVector.y * moveMult );
- object.translateZ( this._moveVector.z * moveMult );
- _tmpQuaternion.set( this._rotationVector.x * rotMult, this._rotationVector.y * rotMult, this._rotationVector.z * rotMult, 1 ).normalize();
- object.quaternion.multiply( _tmpQuaternion );
- if (
- this._lastPosition.distanceToSquared( object.position ) > _EPS ||
- 8 * ( 1 - this._lastQuaternion.dot( object.quaternion ) ) > _EPS
- ) {
- this.dispatchEvent( _changeEvent );
- this._lastQuaternion.copy( object.quaternion );
- this._lastPosition.copy( object.position );
- }
- }
- // private
- _updateMovementVector() {
- const forward = ( this._moveState.forward || ( this.autoForward && ! this._moveState.back ) ) ? 1 : 0;
- this._moveVector.x = ( - this._moveState.left + this._moveState.right );
- this._moveVector.y = ( - this._moveState.down + this._moveState.up );
- this._moveVector.z = ( - forward + this._moveState.back );
- //console.log( 'move:', [ this._moveVector.x, this._moveVector.y, this._moveVector.z ] );
- }
- _updateRotationVector() {
- this._rotationVector.x = ( - this._moveState.pitchDown + this._moveState.pitchUp );
- this._rotationVector.y = ( - this._moveState.yawRight + this._moveState.yawLeft );
- this._rotationVector.z = ( - this._moveState.rollRight + this._moveState.rollLeft );
- //console.log( 'rotate:', [ this._rotationVector.x, this._rotationVector.y, this._rotationVector.z ] );
- }
- _getContainerDimensions() {
- if ( this.domElement != document ) {
- return {
- size: [ this.domElement.offsetWidth, this.domElement.offsetHeight ],
- offset: [ this.domElement.offsetLeft, this.domElement.offsetTop ]
- };
- } else {
- return {
- size: [ window.innerWidth, window.innerHeight ],
- offset: [ 0, 0 ]
- };
- }
- }
- }
- function onKeyDown( event ) {
- if ( event.altKey || this.enabled === false ) {
- return;
- }
- switch ( event.code ) {
- case 'ShiftLeft':
- case 'ShiftRight': this.movementSpeedMultiplier = .1; break;
- case 'KeyW': this._moveState.forward = 1; break;
- case 'KeyS': this._moveState.back = 1; break;
- case 'KeyA': this._moveState.left = 1; break;
- case 'KeyD': this._moveState.right = 1; break;
- case 'KeyR': this._moveState.up = 1; break;
- case 'KeyF': this._moveState.down = 1; break;
- case 'ArrowUp': this._moveState.pitchUp = 1; break;
- case 'ArrowDown': this._moveState.pitchDown = 1; break;
- case 'ArrowLeft': this._moveState.yawLeft = 1; break;
- case 'ArrowRight': this._moveState.yawRight = 1; break;
- case 'KeyQ': this._moveState.rollLeft = 1; break;
- case 'KeyE': this._moveState.rollRight = 1; break;
- }
- this._updateMovementVector();
- this._updateRotationVector();
- }
- function onKeyUp( event ) {
- if ( this.enabled === false ) return;
- switch ( event.code ) {
- case 'ShiftLeft':
- case 'ShiftRight': this.movementSpeedMultiplier = 1; break;
- case 'KeyW': this._moveState.forward = 0; break;
- case 'KeyS': this._moveState.back = 0; break;
- case 'KeyA': this._moveState.left = 0; break;
- case 'KeyD': this._moveState.right = 0; break;
- case 'KeyR': this._moveState.up = 0; break;
- case 'KeyF': this._moveState.down = 0; break;
- case 'ArrowUp': this._moveState.pitchUp = 0; break;
- case 'ArrowDown': this._moveState.pitchDown = 0; break;
- case 'ArrowLeft': this._moveState.yawLeft = 0; break;
- case 'ArrowRight': this._moveState.yawRight = 0; break;
- case 'KeyQ': this._moveState.rollLeft = 0; break;
- case 'KeyE': this._moveState.rollRight = 0; break;
- }
- this._updateMovementVector();
- this._updateRotationVector();
- }
- function onPointerDown( event ) {
- if ( this.enabled === false ) return;
- if ( this.dragToLook ) {
- this._status ++;
- } else {
- switch ( event.button ) {
- case 0: this._moveState.forward = 1; break;
- case 2: this._moveState.back = 1; break;
- }
- this._updateMovementVector();
- }
- }
- function onPointerMove( event ) {
- if ( this.enabled === false ) return;
- if ( ! this.dragToLook || this._status > 0 ) {
- const container = this._getContainerDimensions();
- const halfWidth = container.size[ 0 ] / 2;
- const halfHeight = container.size[ 1 ] / 2;
- this._moveState.yawLeft = - ( ( event.pageX - container.offset[ 0 ] ) - halfWidth ) / halfWidth;
- this._moveState.pitchDown = ( ( event.pageY - container.offset[ 1 ] ) - halfHeight ) / halfHeight;
- this._updateRotationVector();
- }
- }
- function onPointerUp( event ) {
- if ( this.enabled === false ) return;
- if ( this.dragToLook ) {
- this._status --;
- this._moveState.yawLeft = this._moveState.pitchDown = 0;
- } else {
- switch ( event.button ) {
- case 0: this._moveState.forward = 0; break;
- case 2: this._moveState.back = 0; break;
- }
- this._updateMovementVector();
- }
- this._updateRotationVector();
- }
- function onPointerCancel() {
- if ( this.enabled === false ) return;
- if ( this.dragToLook ) {
- this._status = 0;
- this._moveState.yawLeft = this._moveState.pitchDown = 0;
- } else {
- this._moveState.forward = 0;
- this._moveState.back = 0;
- this._updateMovementVector();
- }
- this._updateRotationVector();
- }
- function onContextMenu( event ) {
- if ( this.enabled === false ) return;
- event.preventDefault();
- }
- export { FlyControls };
|