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 }; |