three.js/examples/js/controls/FlyControls.js

( function () {

	var FlyControls = function ( object, domElement ) {

		if ( domElement === undefined ) {

			console.warn( 'THREE.FlyControls: The second parameter "domElement" is now mandatory.' );
			domElement = document;

		}

		this.object = object;
		this.domElement = domElement;
		if ( domElement ) this.domElement.setAttribute( 'tabindex', - 1 ); // API

		this.movementSpeed = 1.0;
		this.rollSpeed = 0.005;
		this.dragToLook = false;
		this.autoForward = false; // disable default target object behavior
		// internals

		var scope = this;
		var changeEvent = {
			type: 'change'
		};
		var EPS = 0.000001;
		this.tmpQuaternion = new THREE.Quaternion();
		this.mouseStatus = 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 THREE.Vector3( 0, 0, 0 );
		this.rotationVector = new THREE.Vector3( 0, 0, 0 );

		this.keydown = function ( event ) {

			if ( event.altKey ) {

				return;

			} //event.preventDefault();


			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 ) {

			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.mousedown = function ( event ) {

			if ( this.domElement !== document ) {

				this.domElement.focus();

			}

			event.preventDefault();

			if ( this.dragToLook ) {

				this.mouseStatus ++;

			} else {

				switch ( event.button ) {

					case 0:
						this.moveState.forward = 1;
						break;

					case 2:
						this.moveState.back = 1;
						break;

				}

				this.updateMovementVector();

			}

		};

		this.mousemove = function ( event ) {

			if ( ! this.dragToLook || this.mouseStatus > 0 ) {

				var container = this.getContainerDimensions();
				var halfWidth = container.size[ 0 ] / 2;
				var 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.mouseup = function ( event ) {

			event.preventDefault();

			if ( this.dragToLook ) {

				this.mouseStatus --;
				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.update = function () {

			var lastQuaternion = new THREE.Quaternion();
			var lastPosition = new THREE.Vector3();
			return function ( delta ) {

				var moveMult = delta * scope.movementSpeed;
				var 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 () {

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

			}

		};

		function bind( scope, fn ) {

			return function () {

				fn.apply( scope, arguments );

			};

		}

		function contextmenu( event ) {

			event.preventDefault();

		}

		this.dispose = function () {

			this.domElement.removeEventListener( 'contextmenu', contextmenu );
			this.domElement.removeEventListener( 'mousedown', _mousedown );
			this.domElement.removeEventListener( 'mousemove', _mousemove );
			this.domElement.removeEventListener( 'mouseup', _mouseup );
			window.removeEventListener( 'keydown', _keydown );
			window.removeEventListener( 'keyup', _keyup );

		};

		var _mousemove = bind( this, this.mousemove );

		var _mousedown = bind( this, this.mousedown );

		var _mouseup = bind( this, this.mouseup );

		var _keydown = bind( this, this.keydown );

		var _keyup = bind( this, this.keyup );

		this.domElement.addEventListener( 'contextmenu', contextmenu );
		this.domElement.addEventListener( 'mousemove', _mousemove );
		this.domElement.addEventListener( 'mousedown', _mousedown );
		this.domElement.addEventListener( 'mouseup', _mouseup );
		window.addEventListener( 'keydown', _keydown );
		window.addEventListener( 'keyup', _keyup );
		this.updateMovementVector();
		this.updateRotationVector();

	};

	FlyControls.prototype = Object.create( THREE.EventDispatcher.prototype );
	FlyControls.prototype.constructor = FlyControls;

	THREE.FlyControls = FlyControls;

} )();