334 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			334 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | import { | ||
|  | 	BoxGeometry, | ||
|  | 	CanvasTexture, | ||
|  | 	Color, | ||
|  | 	Euler, | ||
|  | 	Mesh, | ||
|  | 	MeshBasicMaterial, | ||
|  | 	Object3D, | ||
|  | 	OrthographicCamera, | ||
|  | 	Quaternion, | ||
|  | 	Raycaster, | ||
|  | 	Sprite, | ||
|  | 	SpriteMaterial, | ||
|  | 	Vector2, | ||
|  | 	Vector3, | ||
|  | 	Vector4 | ||
|  | } from 'three'; | ||
|  | 
 | ||
|  | class ViewHelper extends Object3D { | ||
|  | 
 | ||
|  | 	constructor( camera, domElement ) { | ||
|  | 
 | ||
|  | 		super(); | ||
|  | 
 | ||
|  | 		this.isViewHelper = true; | ||
|  | 
 | ||
|  | 		this.animating = false; | ||
|  | 		this.center = new Vector3(); | ||
|  | 
 | ||
|  | 		const color1 = new Color( '#ff3653' ); | ||
|  | 		const color2 = new Color( '#8adb00' ); | ||
|  | 		const color3 = new Color( '#2c8fff' ); | ||
|  | 
 | ||
|  | 		const interactiveObjects = []; | ||
|  | 		const raycaster = new Raycaster(); | ||
|  | 		const mouse = new Vector2(); | ||
|  | 		const dummy = new Object3D(); | ||
|  | 
 | ||
|  | 		const orthoCamera = new OrthographicCamera( - 2, 2, 2, - 2, 0, 4 ); | ||
|  | 		orthoCamera.position.set( 0, 0, 2 ); | ||
|  | 
 | ||
|  | 		const geometry = new BoxGeometry( 0.8, 0.05, 0.05 ).translate( 0.4, 0, 0 ); | ||
|  | 
 | ||
|  | 		const xAxis = new Mesh( geometry, getAxisMaterial( color1 ) ); | ||
|  | 		const yAxis = new Mesh( geometry, getAxisMaterial( color2 ) ); | ||
|  | 		const zAxis = new Mesh( geometry, getAxisMaterial( color3 ) ); | ||
|  | 
 | ||
|  | 		yAxis.rotation.z = Math.PI / 2; | ||
|  | 		zAxis.rotation.y = - Math.PI / 2; | ||
|  | 
 | ||
|  | 		this.add( xAxis ); | ||
|  | 		this.add( zAxis ); | ||
|  | 		this.add( yAxis ); | ||
|  | 
 | ||
|  | 		const posXAxisHelper = new Sprite( getSpriteMaterial( color1, 'X' ) ); | ||
|  | 		posXAxisHelper.userData.type = 'posX'; | ||
|  | 		const posYAxisHelper = new Sprite( getSpriteMaterial( color2, 'Y' ) ); | ||
|  | 		posYAxisHelper.userData.type = 'posY'; | ||
|  | 		const posZAxisHelper = new Sprite( getSpriteMaterial( color3, 'Z' ) ); | ||
|  | 		posZAxisHelper.userData.type = 'posZ'; | ||
|  | 		const negXAxisHelper = new Sprite( getSpriteMaterial( color1 ) ); | ||
|  | 		negXAxisHelper.userData.type = 'negX'; | ||
|  | 		const negYAxisHelper = new Sprite( getSpriteMaterial( color2 ) ); | ||
|  | 		negYAxisHelper.userData.type = 'negY'; | ||
|  | 		const negZAxisHelper = new Sprite( getSpriteMaterial( color3 ) ); | ||
|  | 		negZAxisHelper.userData.type = 'negZ'; | ||
|  | 
 | ||
|  | 		posXAxisHelper.position.x = 1; | ||
|  | 		posYAxisHelper.position.y = 1; | ||
|  | 		posZAxisHelper.position.z = 1; | ||
|  | 		negXAxisHelper.position.x = - 1; | ||
|  | 		negXAxisHelper.scale.setScalar( 0.8 ); | ||
|  | 		negYAxisHelper.position.y = - 1; | ||
|  | 		negYAxisHelper.scale.setScalar( 0.8 ); | ||
|  | 		negZAxisHelper.position.z = - 1; | ||
|  | 		negZAxisHelper.scale.setScalar( 0.8 ); | ||
|  | 
 | ||
|  | 		this.add( posXAxisHelper ); | ||
|  | 		this.add( posYAxisHelper ); | ||
|  | 		this.add( posZAxisHelper ); | ||
|  | 		this.add( negXAxisHelper ); | ||
|  | 		this.add( negYAxisHelper ); | ||
|  | 		this.add( negZAxisHelper ); | ||
|  | 
 | ||
|  | 		interactiveObjects.push( posXAxisHelper ); | ||
|  | 		interactiveObjects.push( posYAxisHelper ); | ||
|  | 		interactiveObjects.push( posZAxisHelper ); | ||
|  | 		interactiveObjects.push( negXAxisHelper ); | ||
|  | 		interactiveObjects.push( negYAxisHelper ); | ||
|  | 		interactiveObjects.push( negZAxisHelper ); | ||
|  | 
 | ||
|  | 		const point = new Vector3(); | ||
|  | 		const dim = 128; | ||
|  | 		const turnRate = 2 * Math.PI; // turn rate in angles per second
 | ||
|  | 
 | ||
|  | 		this.render = function ( renderer ) { | ||
|  | 
 | ||
|  | 			this.quaternion.copy( camera.quaternion ).invert(); | ||
|  | 			this.updateMatrixWorld(); | ||
|  | 
 | ||
|  | 			point.set( 0, 0, 1 ); | ||
|  | 			point.applyQuaternion( camera.quaternion ); | ||
|  | 
 | ||
|  | 			if ( point.x >= 0 ) { | ||
|  | 
 | ||
|  | 				posXAxisHelper.material.opacity = 1; | ||
|  | 				negXAxisHelper.material.opacity = 0.5; | ||
|  | 
 | ||
|  | 			} else { | ||
|  | 
 | ||
|  | 				posXAxisHelper.material.opacity = 0.5; | ||
|  | 				negXAxisHelper.material.opacity = 1; | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			if ( point.y >= 0 ) { | ||
|  | 
 | ||
|  | 				posYAxisHelper.material.opacity = 1; | ||
|  | 				negYAxisHelper.material.opacity = 0.5; | ||
|  | 
 | ||
|  | 			} else { | ||
|  | 
 | ||
|  | 				posYAxisHelper.material.opacity = 0.5; | ||
|  | 				negYAxisHelper.material.opacity = 1; | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			if ( point.z >= 0 ) { | ||
|  | 
 | ||
|  | 				posZAxisHelper.material.opacity = 1; | ||
|  | 				negZAxisHelper.material.opacity = 0.5; | ||
|  | 
 | ||
|  | 			} else { | ||
|  | 
 | ||
|  | 				posZAxisHelper.material.opacity = 0.5; | ||
|  | 				negZAxisHelper.material.opacity = 1; | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			//
 | ||
|  | 
 | ||
|  | 			const x = domElement.offsetWidth - dim; | ||
|  | 
 | ||
|  | 			renderer.clearDepth(); | ||
|  | 
 | ||
|  | 			renderer.getViewport( viewport ); | ||
|  | 			renderer.setViewport( x, 0, dim, dim ); | ||
|  | 
 | ||
|  | 			renderer.render( this, orthoCamera ); | ||
|  | 
 | ||
|  | 			renderer.setViewport( viewport.x, viewport.y, viewport.z, viewport.w ); | ||
|  | 
 | ||
|  | 		}; | ||
|  | 
 | ||
|  | 		const targetPosition = new Vector3(); | ||
|  | 		const targetQuaternion = new Quaternion(); | ||
|  | 
 | ||
|  | 		const q1 = new Quaternion(); | ||
|  | 		const q2 = new Quaternion(); | ||
|  | 		const viewport = new Vector4(); | ||
|  | 		let radius = 0; | ||
|  | 
 | ||
|  | 		this.handleClick = function ( event ) { | ||
|  | 
 | ||
|  | 			if ( this.animating === true ) return false; | ||
|  | 
 | ||
|  | 			const rect = domElement.getBoundingClientRect(); | ||
|  | 			const offsetX = rect.left + ( domElement.offsetWidth - dim ); | ||
|  | 			const offsetY = rect.top + ( domElement.offsetHeight - dim ); | ||
|  | 			mouse.x = ( ( event.clientX - offsetX ) / ( rect.right - offsetX ) ) * 2 - 1; | ||
|  | 			mouse.y = - ( ( event.clientY - offsetY ) / ( rect.bottom - offsetY ) ) * 2 + 1; | ||
|  | 
 | ||
|  | 			raycaster.setFromCamera( mouse, orthoCamera ); | ||
|  | 
 | ||
|  | 			const intersects = raycaster.intersectObjects( interactiveObjects ); | ||
|  | 
 | ||
|  | 			if ( intersects.length > 0 ) { | ||
|  | 
 | ||
|  | 				const intersection = intersects[ 0 ]; | ||
|  | 				const object = intersection.object; | ||
|  | 
 | ||
|  | 				prepareAnimationData( object, this.center ); | ||
|  | 
 | ||
|  | 				this.animating = true; | ||
|  | 
 | ||
|  | 				return true; | ||
|  | 
 | ||
|  | 			} else { | ||
|  | 
 | ||
|  | 				return false; | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		}; | ||
|  | 
 | ||
|  | 		this.update = function ( delta ) { | ||
|  | 
 | ||
|  | 			const step = delta * turnRate; | ||
|  | 
 | ||
|  | 			// animate position by doing a slerp and then scaling the position on the unit sphere
 | ||
|  | 
 | ||
|  | 			q1.rotateTowards( q2, step ); | ||
|  | 			camera.position.set( 0, 0, 1 ).applyQuaternion( q1 ).multiplyScalar( radius ).add( this.center ); | ||
|  | 
 | ||
|  | 			// animate orientation
 | ||
|  | 
 | ||
|  | 			camera.quaternion.rotateTowards( targetQuaternion, step ); | ||
|  | 
 | ||
|  | 			if ( q1.angleTo( q2 ) === 0 ) { | ||
|  | 
 | ||
|  | 				this.animating = false; | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 		}; | ||
|  | 
 | ||
|  | 		this.dispose = function () { | ||
|  | 
 | ||
|  | 			geometry.dispose(); | ||
|  | 
 | ||
|  | 			xAxis.material.dispose(); | ||
|  | 			yAxis.material.dispose(); | ||
|  | 			zAxis.material.dispose(); | ||
|  | 
 | ||
|  | 			posXAxisHelper.material.map.dispose(); | ||
|  | 			posYAxisHelper.material.map.dispose(); | ||
|  | 			posZAxisHelper.material.map.dispose(); | ||
|  | 			negXAxisHelper.material.map.dispose(); | ||
|  | 			negYAxisHelper.material.map.dispose(); | ||
|  | 			negZAxisHelper.material.map.dispose(); | ||
|  | 
 | ||
|  | 			posXAxisHelper.material.dispose(); | ||
|  | 			posYAxisHelper.material.dispose(); | ||
|  | 			posZAxisHelper.material.dispose(); | ||
|  | 			negXAxisHelper.material.dispose(); | ||
|  | 			negYAxisHelper.material.dispose(); | ||
|  | 			negZAxisHelper.material.dispose(); | ||
|  | 
 | ||
|  | 		}; | ||
|  | 
 | ||
|  | 		function prepareAnimationData( object, focusPoint ) { | ||
|  | 
 | ||
|  | 			switch ( object.userData.type ) { | ||
|  | 
 | ||
|  | 				case 'posX': | ||
|  | 					targetPosition.set( 1, 0, 0 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler( 0, Math.PI * 0.5, 0 ) ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				case 'posY': | ||
|  | 					targetPosition.set( 0, 1, 0 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler( - Math.PI * 0.5, 0, 0 ) ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				case 'posZ': | ||
|  | 					targetPosition.set( 0, 0, 1 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler() ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				case 'negX': | ||
|  | 					targetPosition.set( - 1, 0, 0 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler( 0, - Math.PI * 0.5, 0 ) ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				case 'negY': | ||
|  | 					targetPosition.set( 0, - 1, 0 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler( Math.PI * 0.5, 0, 0 ) ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				case 'negZ': | ||
|  | 					targetPosition.set( 0, 0, - 1 ); | ||
|  | 					targetQuaternion.setFromEuler( new Euler( 0, Math.PI, 0 ) ); | ||
|  | 					break; | ||
|  | 
 | ||
|  | 				default: | ||
|  | 					console.error( 'ViewHelper: Invalid axis.' ); | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			//
 | ||
|  | 
 | ||
|  | 			radius = camera.position.distanceTo( focusPoint ); | ||
|  | 			targetPosition.multiplyScalar( radius ).add( focusPoint ); | ||
|  | 
 | ||
|  | 			dummy.position.copy( focusPoint ); | ||
|  | 
 | ||
|  | 			dummy.lookAt( camera.position ); | ||
|  | 			q1.copy( dummy.quaternion ); | ||
|  | 
 | ||
|  | 			dummy.lookAt( targetPosition ); | ||
|  | 			q2.copy( dummy.quaternion ); | ||
|  | 
 | ||
|  | 		} | ||
|  | 
 | ||
|  | 		function getAxisMaterial( color ) { | ||
|  | 
 | ||
|  | 			return new MeshBasicMaterial( { color: color, toneMapped: false } ); | ||
|  | 
 | ||
|  | 		} | ||
|  | 
 | ||
|  | 		function getSpriteMaterial( color, text = null ) { | ||
|  | 
 | ||
|  | 			const canvas = document.createElement( 'canvas' ); | ||
|  | 			canvas.width = 64; | ||
|  | 			canvas.height = 64; | ||
|  | 
 | ||
|  | 			const context = canvas.getContext( '2d' ); | ||
|  | 			context.beginPath(); | ||
|  | 			context.arc( 32, 32, 16, 0, 2 * Math.PI ); | ||
|  | 			context.closePath(); | ||
|  | 			context.fillStyle = color.getStyle(); | ||
|  | 			context.fill(); | ||
|  | 
 | ||
|  | 			if ( text !== null ) { | ||
|  | 
 | ||
|  | 				context.font = '24px Arial'; | ||
|  | 				context.textAlign = 'center'; | ||
|  | 				context.fillStyle = '#000000'; | ||
|  | 				context.fillText( text, 32, 41 ); | ||
|  | 
 | ||
|  | 			} | ||
|  | 
 | ||
|  | 			const texture = new CanvasTexture( canvas ); | ||
|  | 
 | ||
|  | 			return new SpriteMaterial( { map: texture, toneMapped: false } ); | ||
|  | 
 | ||
|  | 		} | ||
|  | 
 | ||
|  | 	} | ||
|  | 
 | ||
|  | } | ||
|  | 
 | ||
|  | export { ViewHelper }; |