javascript
/
three.js
mirror of https://github.com/tazdij/three.js.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
							<!DOCTYPE html>
<html lang="en">
	<head>
		<title>three.js webgl - octree</title>
		<meta charset="utf-8">
		<meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
		<style>
			body {
				background-color: #ffffff;
				margin: 0px;
				overflow: hidden;
			}
		</style>
	</head>
	<body>

		<script type="text/javascript" src="../build/three.min.js"></script>
		<script type="text/javascript" src="js/Octree.js"></script>
		<script>

			var camera, 
				scene, 
				renderer,
				octree,
				geometry, 
				material, 
				mesh,
				meshes = [],
				meshesSearch = [],
				meshCountMax = 1000,
				radius = 500,
				radiusMax = radius * 10,
				radiusMaxHalf = radiusMax * 0.5,
				radiusSearch = 400,
				searchMesh,
				baseR = 255, baseG = 0, baseB = 255,
				foundR = 0, foundG = 255, foundB = 0,
				adding = true;

			init();
			animate();

			function init() {
			
				// standard three scene, camera, renderer

				scene = new THREE.Scene();

				camera = new THREE.PerspectiveCamera( 75, window.innerWidth / window.innerHeight, 1, radius * 100 );
				scene.add( camera );

				renderer = new THREE.WebGLRenderer();
				renderer.setSize( window.innerWidth, window.innerHeight );

				document.body.appendChild( renderer.domElement );
			
				// create octree
			
				octree = new THREE.Octree( {
					scene: scene
				} );
			
				// create object to show search radius and add to scene
			
				searchMesh = new THREE.Mesh(
					new THREE.SphereGeometry( radiusSearch ),
					new THREE.MeshBasicMaterial( { color: 0x00FF00, transparent: true, opacity: 0.4 } )
				);
				scene.add( searchMesh );

			}

			function animate() {

				// note: three.js includes requestAnimationFrame shim
				requestAnimationFrame( animate );
			
				// modify octree structure by adding/removing objects
			
				modifyOctree();
			
				// search octree at random location
			
				searchOctree();
			
				// render results
			
				render();

			}
		
			function modifyOctree() {
			
				// if is adding objects to octree
			
				if ( adding === true ) {
				
					// create new object
				
					geometry = new THREE.CubeGeometry( 50, 50, 50 );
					material = new THREE.MeshBasicMaterial();
					material.color.setRGB( baseR, baseG, baseB );
				
					mesh = new THREE.Mesh( geometry, material );
				
					// give new object a random position in radius
				
					mesh.position.set(
						Math.random() * radiusMax - radiusMaxHalf,
						Math.random() * radiusMax - radiusMaxHalf,
						Math.random() * radiusMax - radiusMaxHalf
					);
				
					// add new object to octree and scene
				
					octree.add( mesh );
					scene.add( mesh );
				
					// store object for later
				
					meshes.push( mesh );
				
					// if at max, stop adding
				
					if ( meshes.length === meshCountMax ) {
					
						adding = false;
					
					}
				
				}
				// else remove objects from octree
				else {
				
					// get object
				
					mesh = meshes.shift();
				
					// remove from scene and octree
				
					scene.remove( mesh );
					octree.remove( mesh );
				
					// if no more objects, start adding
				
					if ( meshes.length === 0 ) {
					
						adding = true;
					
					}
				
				}
			
				/*
			
				// octree details to console
			
				console.log( ' OCTREE: ', octree );
				console.log( ' ... depth ', octree.depth, ' vs depth end?', octree.depth_end() );
				console.log( ' ... num nodes: ', octree.node_count_end() );
				console.log( ' ... total objects: ', octree.object_count_end(), ' vs tree objects length: ', octree.objects.length );
			
				// print full octree structure to console
			
				octree.to_console();
			
				*/
			
			}
		
			function searchOctree() {
			
				var i, il;
			
				// revert previous search objects to base color
			
				for ( i = 0, il = meshesSearch.length; i < il; i++ ) {
				
					meshesSearch[ i ].object.material.color.setRGB( baseR, baseG, baseB );
				
				}
			
				// new search position
				searchMesh.position.set(
					Math.random() * radiusMax - radiusMaxHalf,
					Math.random() * radiusMax - radiusMaxHalf,
					Math.random() * radiusMax - radiusMaxHalf
				);
			
				// record start time
			
				var timeStart = Date.now();
			
				// search octree from search mesh position with search radius
				// optional third parameter: boolean, if should sort results by object when using faces in octree
				// optional fourth parameter: vector3, direction of search when using ray (assumes radius is distance/far of ray)
			
				var rayCaster = new THREE.Raycaster( new THREE.Vector3().copy( searchMesh.position ), new THREE.Vector3( Math.random() * 2 - 1, Math.random() * 2 - 1, Math.random() * 2 - 1 ).normalize() );
				meshesSearch = octree.search( rayCaster.ray.origin, radiusSearch, true, rayCaster.ray.direction );
				var intersections = rayCaster.intersectOctreeObjects( meshesSearch );
			
				// record end time
			
				var timeEnd = Date.now();
			
				// set color of all meshes found in search
			
				for ( i = 0, il = meshesSearch.length; i < il; i++ ) {
				
					meshesSearch[ i ].object.material.color.setRGB( foundR, foundG, foundB );
				
				}
			
				/*
			
				// results to console
			
				console.log( 'OCTREE: ', octree );
				console.log( '... searched ', meshes.length, ' and found ', meshesSearch.length, ' with intersections ', intersections.length, ' and took ', ( timeEnd - timeStart ), ' ms ' );
			
				*/
			
			}
		
			function render() {
	
				var timer = - Date.now() / 5000;

				camera.position.x = Math.cos( timer ) * 10000;
				camera.position.z = Math.sin( timer ) * 10000;
				camera.lookAt( scene.position );
		
				renderer.render( scene, camera );

			}

		</script>
	</body>
</html>