313 lines
		
	
	
		
			7.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			313 lines
		
	
	
		
			7.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| import {
 | |
| 	Mesh,
 | |
| 	MeshBasicMaterial,
 | |
| 	Object3D,
 | |
| 	SphereGeometry,
 | |
| } from 'three';
 | |
| 
 | |
| import { GLTFLoader } from '../loaders/GLTFLoader.js';
 | |
| 
 | |
| import {
 | |
| 	Constants as MotionControllerConstants,
 | |
| 	fetchProfile,
 | |
| 	MotionController
 | |
| } from '../libs/motion-controllers.module.js';
 | |
| 
 | |
| const DEFAULT_PROFILES_PATH = 'https://cdn.jsdelivr.net/npm/@webxr-input-profiles/assets@1.0/dist/profiles';
 | |
| const DEFAULT_PROFILE = 'generic-trigger';
 | |
| 
 | |
| class XRControllerModel extends Object3D {
 | |
| 
 | |
| 	constructor() {
 | |
| 
 | |
| 		super();
 | |
| 
 | |
| 		this.motionController = null;
 | |
| 		this.envMap = null;
 | |
| 
 | |
| 	}
 | |
| 
 | |
| 	setEnvironmentMap( envMap ) {
 | |
| 
 | |
| 		if ( this.envMap == envMap ) {
 | |
| 
 | |
| 			return this;
 | |
| 
 | |
| 		}
 | |
| 
 | |
| 		this.envMap = envMap;
 | |
| 		this.traverse( ( child ) => {
 | |
| 
 | |
| 			if ( child.isMesh ) {
 | |
| 
 | |
| 				child.material.envMap = this.envMap;
 | |
| 				child.material.needsUpdate = true;
 | |
| 
 | |
| 			}
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 		return this;
 | |
| 
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Polls data from the XRInputSource and updates the model's components to match
 | |
| 	 * the real world data
 | |
| 	 */
 | |
| 	updateMatrixWorld( force ) {
 | |
| 
 | |
| 		super.updateMatrixWorld( force );
 | |
| 
 | |
| 		if ( ! this.motionController ) return;
 | |
| 
 | |
| 		// Cause the MotionController to poll the Gamepad for data
 | |
| 		this.motionController.updateFromGamepad();
 | |
| 
 | |
| 		// Update the 3D model to reflect the button, thumbstick, and touchpad state
 | |
| 		Object.values( this.motionController.components ).forEach( ( component ) => {
 | |
| 
 | |
| 			// Update node data based on the visual responses' current states
 | |
| 			Object.values( component.visualResponses ).forEach( ( visualResponse ) => {
 | |
| 
 | |
| 				const { valueNode, minNode, maxNode, value, valueNodeProperty } = visualResponse;
 | |
| 
 | |
| 				// Skip if the visual response node is not found. No error is needed,
 | |
| 				// because it will have been reported at load time.
 | |
| 				if ( ! valueNode ) return;
 | |
| 
 | |
| 				// Calculate the new properties based on the weight supplied
 | |
| 				if ( valueNodeProperty === MotionControllerConstants.VisualResponseProperty.VISIBILITY ) {
 | |
| 
 | |
| 					valueNode.visible = value;
 | |
| 
 | |
| 				} else if ( valueNodeProperty === MotionControllerConstants.VisualResponseProperty.TRANSFORM ) {
 | |
| 
 | |
| 					valueNode.quaternion.slerpQuaternions(
 | |
| 						minNode.quaternion,
 | |
| 						maxNode.quaternion,
 | |
| 						value
 | |
| 					);
 | |
| 
 | |
| 					valueNode.position.lerpVectors(
 | |
| 						minNode.position,
 | |
| 						maxNode.position,
 | |
| 						value
 | |
| 					);
 | |
| 
 | |
| 				}
 | |
| 
 | |
| 			} );
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| /**
 | |
|  * Walks the model's tree to find the nodes needed to animate the components and
 | |
|  * saves them to the motionContoller components for use in the frame loop. When
 | |
|  * touchpads are found, attaches a touch dot to them.
 | |
|  */
 | |
| function findNodes( motionController, scene ) {
 | |
| 
 | |
| 	// Loop through the components and find the nodes needed for each components' visual responses
 | |
| 	Object.values( motionController.components ).forEach( ( component ) => {
 | |
| 
 | |
| 		const { type, touchPointNodeName, visualResponses } = component;
 | |
| 
 | |
| 		if ( type === MotionControllerConstants.ComponentType.TOUCHPAD ) {
 | |
| 
 | |
| 			component.touchPointNode = scene.getObjectByName( touchPointNodeName );
 | |
| 			if ( component.touchPointNode ) {
 | |
| 
 | |
| 				// Attach a touch dot to the touchpad.
 | |
| 				const sphereGeometry = new SphereGeometry( 0.001 );
 | |
| 				const material = new MeshBasicMaterial( { color: 0x0000FF } );
 | |
| 				const sphere = new Mesh( sphereGeometry, material );
 | |
| 				component.touchPointNode.add( sphere );
 | |
| 
 | |
| 			} else {
 | |
| 
 | |
| 				console.warn( `Could not find touch dot, ${component.touchPointNodeName}, in touchpad component ${component.id}` );
 | |
| 
 | |
| 			}
 | |
| 
 | |
| 		}
 | |
| 
 | |
| 		// Loop through all the visual responses to be applied to this component
 | |
| 		Object.values( visualResponses ).forEach( ( visualResponse ) => {
 | |
| 
 | |
| 			const { valueNodeName, minNodeName, maxNodeName, valueNodeProperty } = visualResponse;
 | |
| 
 | |
| 			// If animating a transform, find the two nodes to be interpolated between.
 | |
| 			if ( valueNodeProperty === MotionControllerConstants.VisualResponseProperty.TRANSFORM ) {
 | |
| 
 | |
| 				visualResponse.minNode = scene.getObjectByName( minNodeName );
 | |
| 				visualResponse.maxNode = scene.getObjectByName( maxNodeName );
 | |
| 
 | |
| 				// If the extents cannot be found, skip this animation
 | |
| 				if ( ! visualResponse.minNode ) {
 | |
| 
 | |
| 					console.warn( `Could not find ${minNodeName} in the model` );
 | |
| 					return;
 | |
| 
 | |
| 				}
 | |
| 
 | |
| 				if ( ! visualResponse.maxNode ) {
 | |
| 
 | |
| 					console.warn( `Could not find ${maxNodeName} in the model` );
 | |
| 					return;
 | |
| 
 | |
| 				}
 | |
| 
 | |
| 			}
 | |
| 
 | |
| 			// If the target node cannot be found, skip this animation
 | |
| 			visualResponse.valueNode = scene.getObjectByName( valueNodeName );
 | |
| 			if ( ! visualResponse.valueNode ) {
 | |
| 
 | |
| 				console.warn( `Could not find ${valueNodeName} in the model` );
 | |
| 
 | |
| 			}
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 	} );
 | |
| 
 | |
| }
 | |
| 
 | |
| function addAssetSceneToControllerModel( controllerModel, scene ) {
 | |
| 
 | |
| 	// Find the nodes needed for animation and cache them on the motionController.
 | |
| 	findNodes( controllerModel.motionController, scene );
 | |
| 
 | |
| 	// Apply any environment map that the mesh already has set.
 | |
| 	if ( controllerModel.envMap ) {
 | |
| 
 | |
| 		scene.traverse( ( child ) => {
 | |
| 
 | |
| 			if ( child.isMesh ) {
 | |
| 
 | |
| 				child.material.envMap = controllerModel.envMap;
 | |
| 				child.material.needsUpdate = true;
 | |
| 
 | |
| 			}
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 	}
 | |
| 
 | |
| 	// Add the glTF scene to the controllerModel.
 | |
| 	controllerModel.add( scene );
 | |
| 
 | |
| }
 | |
| 
 | |
| class XRControllerModelFactory {
 | |
| 
 | |
| 	constructor( gltfLoader = null, onLoad = null ) {
 | |
| 
 | |
| 		this.gltfLoader = gltfLoader;
 | |
| 		this.path = DEFAULT_PROFILES_PATH;
 | |
| 		this._assetCache = {};
 | |
| 		this.onLoad = onLoad;
 | |
| 
 | |
| 		// If a GLTFLoader wasn't supplied to the constructor create a new one.
 | |
| 		if ( ! this.gltfLoader ) {
 | |
| 
 | |
| 			this.gltfLoader = new GLTFLoader();
 | |
| 
 | |
| 		}
 | |
| 
 | |
| 	}
 | |
| 
 | |
| 	setPath( path ) {
 | |
| 
 | |
| 		this.path = path;
 | |
| 
 | |
| 		return this;
 | |
| 
 | |
| 	}
 | |
| 
 | |
| 	createControllerModel( controller ) {
 | |
| 
 | |
| 		const controllerModel = new XRControllerModel();
 | |
| 		let scene = null;
 | |
| 
 | |
| 		controller.addEventListener( 'connected', ( event ) => {
 | |
| 
 | |
| 			const xrInputSource = event.data;
 | |
| 
 | |
| 			if ( xrInputSource.targetRayMode !== 'tracked-pointer' || ! xrInputSource.gamepad ) return;
 | |
| 
 | |
| 			fetchProfile( xrInputSource, this.path, DEFAULT_PROFILE ).then( ( { profile, assetPath } ) => {
 | |
| 
 | |
| 				controllerModel.motionController = new MotionController(
 | |
| 					xrInputSource,
 | |
| 					profile,
 | |
| 					assetPath
 | |
| 				);
 | |
| 
 | |
| 				const cachedAsset = this._assetCache[ controllerModel.motionController.assetUrl ];
 | |
| 				if ( cachedAsset ) {
 | |
| 
 | |
| 					scene = cachedAsset.scene.clone();
 | |
| 
 | |
| 					addAssetSceneToControllerModel( controllerModel, scene );
 | |
| 
 | |
| 					if ( this.onLoad ) this.onLoad( scene );
 | |
| 
 | |
| 				} else {
 | |
| 
 | |
| 					if ( ! this.gltfLoader ) {
 | |
| 
 | |
| 						throw new Error( 'GLTFLoader not set.' );
 | |
| 
 | |
| 					}
 | |
| 
 | |
| 					this.gltfLoader.setPath( '' );
 | |
| 					this.gltfLoader.load( controllerModel.motionController.assetUrl, ( asset ) => {
 | |
| 
 | |
| 						this._assetCache[ controllerModel.motionController.assetUrl ] = asset;
 | |
| 
 | |
| 						scene = asset.scene.clone();
 | |
| 
 | |
| 						addAssetSceneToControllerModel( controllerModel, scene );
 | |
| 
 | |
| 						if ( this.onLoad ) this.onLoad( scene );
 | |
| 
 | |
| 					},
 | |
| 					null,
 | |
| 					() => {
 | |
| 
 | |
| 						throw new Error( `Asset ${controllerModel.motionController.assetUrl} missing or malformed.` );
 | |
| 
 | |
| 					} );
 | |
| 
 | |
| 				}
 | |
| 
 | |
| 			} ).catch( ( err ) => {
 | |
| 
 | |
| 				console.warn( err );
 | |
| 
 | |
| 			} );
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 		controller.addEventListener( 'disconnected', () => {
 | |
| 
 | |
| 			controllerModel.motionController = null;
 | |
| 			controllerModel.remove( scene );
 | |
| 			scene = null;
 | |
| 
 | |
| 		} );
 | |
| 
 | |
| 		return controllerModel;
 | |
| 
 | |
| 	}
 | |
| 
 | |
| }
 | |
| 
 | |
| export { XRControllerModelFactory };
 |