327 lines
7.8 KiB
JavaScript
327 lines
7.8 KiB
JavaScript
|
import {
|
||
|
EventDispatcher,
|
||
|
Quaternion,
|
||
|
Vector3
|
||
|
} from 'three';
|
||
|
|
||
|
const _changeEvent = { type: 'change' };
|
||
|
|
||
|
class FlyControls extends EventDispatcher {
|
||
|
|
||
|
constructor( object, domElement ) {
|
||
|
|
||
|
super();
|
||
|
|
||
|
this.object = object;
|
||
|
this.domElement = domElement;
|
||
|
|
||
|
// API
|
||
|
|
||
|
// Set to false to disable this control
|
||
|
this.enabled = true;
|
||
|
|
||
|
this.movementSpeed = 1.0;
|
||
|
this.rollSpeed = 0.005;
|
||
|
|
||
|
this.dragToLook = false;
|
||
|
this.autoForward = false;
|
||
|
|
||
|
// disable default target object behavior
|
||
|
|
||
|
// internals
|
||
|
|
||
|
const scope = this;
|
||
|
|
||
|
const EPS = 0.000001;
|
||
|
|
||
|
const lastQuaternion = new Quaternion();
|
||
|
const lastPosition = new Vector3();
|
||
|
|
||
|
this.tmpQuaternion = new Quaternion();
|
||
|
|
||
|
this.status = 0;
|
||
|
|
||
|
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.keydown = function ( 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();
|
||
|
|
||
|
};
|
||
|
|
||
|
this.keyup = function ( 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();
|
||
|
|
||
|
};
|
||
|
|
||
|
this.pointerdown = function ( 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();
|
||
|
|
||
|
}
|
||
|
|
||
|
};
|
||
|
|
||
|
this.pointermove = function ( 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();
|
||
|
|
||
|
}
|
||
|
|
||
|
};
|
||
|
|
||
|
this.pointerup = function ( 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();
|
||
|
|
||
|
};
|
||
|
|
||
|
this.pointercancel = function () {
|
||
|
|
||
|
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();
|
||
|
|
||
|
};
|
||
|
|
||
|
this.contextMenu = function ( event ) {
|
||
|
|
||
|
if ( this.enabled === false ) return;
|
||
|
|
||
|
event.preventDefault();
|
||
|
|
||
|
};
|
||
|
|
||
|
this.update = function ( delta ) {
|
||
|
|
||
|
if ( this.enabled === false ) return;
|
||
|
|
||
|
const moveMult = delta * scope.movementSpeed;
|
||
|
const rotMult = delta * scope.rollSpeed;
|
||
|
|
||
|
scope.object.translateX( scope.moveVector.x * moveMult );
|
||
|
scope.object.translateY( scope.moveVector.y * moveMult );
|
||
|
scope.object.translateZ( scope.moveVector.z * moveMult );
|
||
|
|
||
|
scope.tmpQuaternion.set( scope.rotationVector.x * rotMult, scope.rotationVector.y * rotMult, scope.rotationVector.z * rotMult, 1 ).normalize();
|
||
|
scope.object.quaternion.multiply( scope.tmpQuaternion );
|
||
|
|
||
|
if (
|
||
|
lastPosition.distanceToSquared( scope.object.position ) > EPS ||
|
||
|
8 * ( 1 - lastQuaternion.dot( scope.object.quaternion ) ) > EPS
|
||
|
) {
|
||
|
|
||
|
scope.dispatchEvent( _changeEvent );
|
||
|
lastQuaternion.copy( scope.object.quaternion );
|
||
|
lastPosition.copy( scope.object.position );
|
||
|
|
||
|
}
|
||
|
|
||
|
};
|
||
|
|
||
|
this.updateMovementVector = function () {
|
||
|
|
||
|
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 ] );
|
||
|
|
||
|
};
|
||
|
|
||
|
this.updateRotationVector = function () {
|
||
|
|
||
|
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 ] );
|
||
|
|
||
|
};
|
||
|
|
||
|
this.getContainerDimensions = function () {
|
||
|
|
||
|
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 ]
|
||
|
};
|
||
|
|
||
|
}
|
||
|
|
||
|
};
|
||
|
|
||
|
this.dispose = function () {
|
||
|
|
||
|
this.domElement.removeEventListener( 'contextmenu', _contextmenu );
|
||
|
this.domElement.removeEventListener( 'pointerdown', _pointerdown );
|
||
|
this.domElement.removeEventListener( 'pointermove', _pointermove );
|
||
|
this.domElement.removeEventListener( 'pointerup', _pointerup );
|
||
|
this.domElement.removeEventListener( 'pointercancel', _pointercancel );
|
||
|
|
||
|
window.removeEventListener( 'keydown', _keydown );
|
||
|
window.removeEventListener( 'keyup', _keyup );
|
||
|
|
||
|
};
|
||
|
|
||
|
const _contextmenu = this.contextMenu.bind( this );
|
||
|
const _pointermove = this.pointermove.bind( this );
|
||
|
const _pointerdown = this.pointerdown.bind( this );
|
||
|
const _pointerup = this.pointerup.bind( this );
|
||
|
const _pointercancel = this.pointercancel.bind( this );
|
||
|
const _keydown = this.keydown.bind( this );
|
||
|
const _keyup = this.keyup.bind( this );
|
||
|
|
||
|
this.domElement.addEventListener( 'contextmenu', _contextmenu );
|
||
|
this.domElement.addEventListener( 'pointerdown', _pointerdown );
|
||
|
this.domElement.addEventListener( 'pointermove', _pointermove );
|
||
|
this.domElement.addEventListener( 'pointerup', _pointerup );
|
||
|
this.domElement.addEventListener( 'pointercancel', _pointercancel );
|
||
|
|
||
|
window.addEventListener( 'keydown', _keydown );
|
||
|
window.addEventListener( 'keyup', _keyup );
|
||
|
|
||
|
this.updateMovementVector();
|
||
|
this.updateRotationVector();
|
||
|
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
export { FlyControls };
|