samien
/
three.js
mirror of https://github.com/mrdoob/three.js.git


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451
							<!DOCTYPE html>
<html lang="en">
<head>
	<title>three.js webgl - mesh - batch</title>
	<meta charset="utf-8">
	<meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0">
	<link type="text/css" rel="stylesheet" href="main.css">
	<style>
		#info {
			background-color: rgba(0,0,0,0.75);
		}
	</style>
</head>
<body>

	<div id="info">

		<a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> webgl - mesh - batch

	</div>

	<script type="importmap">
		{
			"imports": {
				"three": "../build/three.module.js",
				"three/addons/": "./jsm/"
			}
		}
	</script>

	<script type="module">
		import * as THREE from 'three';

		import Stats from 'three/addons/libs/stats.module.js';
		import { GUI } from 'three/addons/libs/lil-gui.module.min.js';

		import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
		import { radixSort } from 'three/addons/utils/SortUtils.js';

		let stats, gui, guiStatsEl;
		let camera, controls, scene, renderer;
		let geometries, mesh;
		const ids = [];
		const matrix = new THREE.Matrix4();

		//

		const position = new THREE.Vector3();
		const rotation = new THREE.Euler();
		const quaternion = new THREE.Quaternion();
		const scale = new THREE.Vector3();

		//

		const MAX_GEOMETRY_COUNT = 20000;

		const Method = {
			BATCHED: 'BATCHED',
			NAIVE: 'NAIVE'
		};

		const api = {
			method: Method.BATCHED,
			count: 256,
			dynamic: 16,

			transparent: true,
			sortObjects: true,
			perObjectFrustumCulled: true,
			useCustomSort: true,
		};

		init();
		initGeometries();
		initMesh();

		//

		function randomizeMatrix( matrix ) {

			position.x = Math.random() * 40 - 20;
			position.y = Math.random() * 40 - 20;
			position.z = Math.random() * 40 - 20;

			rotation.x = Math.random() * 2 * Math.PI;
			rotation.y = Math.random() * 2 * Math.PI;
			rotation.z = Math.random() * 2 * Math.PI;

			quaternion.setFromEuler( rotation );

			scale.x = scale.y = scale.z = 0.5 + ( Math.random() * 0.5 );

			return matrix.compose( position, quaternion, scale );

		}

		function randomizeRotationSpeed( rotation ) {

			rotation.x = Math.random() * 0.01;
			rotation.y = Math.random() * 0.01;
			rotation.z = Math.random() * 0.01;
			return rotation;

		}

		function randomizeColor( color ) {

			return color.setHSL( Math.random() * 0.5, 0.6, 0.5 );

		}

		function randomizeAlpha() {

			// make ~20% of all objects transparent
			return Math.random() > 0.8 ? 0.5 : 1.0;

		}

		function initGeometries() {

			const cone = new THREE.ConeGeometry( 1.0, 2.0 );
			const box = new THREE.BoxGeometry( 2.0, 2.0, 2.0 );
			const sphere = new THREE.SphereGeometry( 1.0, 16, 8 );

			geometries = [ cone, box, sphere ];

			for ( const geometry of geometries ) {

				// add vertex colors for testing
				const count = geometry.getAttribute( 'position' ).count;
				const attribute = new THREE.BufferAttribute( new Float32Array( count * 3 ), 3 );
				geometry.setAttribute( 'color', attribute );

				for ( let i = 0, l = attribute.array.length; i < l; ++ i ) {

					attribute.array[ i ] = 1.0;

				}

			}

		}

		function createMaterial() {

			return new THREE.MeshPhongMaterial( {
				vertexColors: true,
				transparent: api.transparent,
				depthWrite: ! api.transparent
			} );

		}

		function cleanup() {

			if ( mesh ) {

				mesh.traverse( node => {

					if ( node instanceof THREE.Mesh ) {

						node.material.dispose();

					}

				} );

				mesh.parent.remove( mesh );

				if ( mesh.dispose ) {

					mesh.dispose();

				}

			}

		}

		function initMesh() {

			cleanup();

			if ( api.method === Method.BATCHED ) {

				initBatchedMesh();

			} else {

				initRegularMesh();

			}

		}

		function initRegularMesh() {

			mesh = new THREE.Group();

			for ( let i = 0; i < api.count; i ++ ) {

				const material = createMaterial();
				randomizeColor( material.color );
				material.opacity = randomizeAlpha();

				const child = new THREE.Mesh( geometries[ i % geometries.length ], material );
				randomizeMatrix( child.matrix );
				child.matrix.decompose( child.position, child.quaternion, child.scale );
				child.userData.rotationSpeed = randomizeRotationSpeed( new THREE.Euler() );
				mesh.add( child );

			}

			scene.add( mesh );

		}

		function initBatchedMesh() {

			const geometryCount = api.count;
			const vertexCount = geometries.length * 512;
			const indexCount = geometries.length * 1024;

			const euler = new THREE.Euler();
			const matrix = new THREE.Matrix4();
			const color = new THREE.Color();
			const colorWithAlpha = new THREE.Vector4();
			mesh = new THREE.BatchedMesh( geometryCount, vertexCount, indexCount, createMaterial() );
			mesh.userData.rotationSpeeds = [];

			// disable full-object frustum culling since all of the objects can be dynamic.
			mesh.frustumCulled = false;

			ids.length = 0;

			const geometryIds = [
				mesh.addGeometry( geometries[ 0 ] ),
				mesh.addGeometry( geometries[ 1 ] ),
				mesh.addGeometry( geometries[ 2 ] ),
			];

			for ( let i = 0; i < api.count; i ++ ) {

				randomizeColor( color );
				colorWithAlpha.set( color.r, color.g, color.b, randomizeAlpha() );

				const id = mesh.addInstance( geometryIds[ i % geometryIds.length ] );
				mesh.setMatrixAt( id, randomizeMatrix( matrix ) );
				mesh.setColorAt( id, colorWithAlpha );

				const rotationMatrix = new THREE.Matrix4();
				rotationMatrix.makeRotationFromEuler( randomizeRotationSpeed( euler ) );
				mesh.userData.rotationSpeeds.push( rotationMatrix );

				ids.push( id );

			}

			scene.add( mesh );

		}

		function init() {

			const width = window.innerWidth;
			const height = window.innerHeight;

			// camera

			camera = new THREE.PerspectiveCamera( 70, width / height, 1, 100 );
			camera.position.z = 30;

			// renderer

			renderer = new THREE.WebGLRenderer( { antialias: true } );
			renderer.setPixelRatio( window.devicePixelRatio );
			renderer.setSize( width, height );
			renderer.setAnimationLoop( animate );
			document.body.appendChild( renderer.domElement );

			// scene

			scene = new THREE.Scene();
			scene.background = new THREE.Color( 0xffffff );

			// controls

			controls = new OrbitControls( camera, renderer.domElement );
			controls.autoRotate = true;
			controls.autoRotateSpeed = 1.0;

			// light

			const ambientLight = new THREE.AmbientLight( 0xffffff, 2 );
			const directionalLight = new THREE.DirectionalLight( 0xffffff, 2 );
			directionalLight.position.set( 1, 1, 1 );
			scene.add( directionalLight, ambientLight );

			// stats

			stats = new Stats();
			document.body.appendChild( stats.dom );

			// gui

			gui = new GUI();
			gui.add( api, 'count', 1, MAX_GEOMETRY_COUNT ).step( 1 ).onChange( initMesh );
			gui.add( api, 'dynamic', 0, MAX_GEOMETRY_COUNT ).step( 1 );
			gui.add( api, 'method', Method ).onChange( initMesh );

			gui.add( api, 'transparent' ).onChange( v => mesh.traverse( node => {

				if ( node instanceof THREE.Mesh ) {

					const material = node.material;
					material.transparent = v;
					material.depthWrite = ! v;
					material.needsUpdate = true;

				}

			} ) );

			gui.add( api, 'sortObjects' );
			gui.add( api, 'perObjectFrustumCulled' );
			gui.add( api, 'useCustomSort' );

			guiStatsEl = document.createElement( 'li' );
			guiStatsEl.classList.add( 'gui-stats' );

			// listeners

			window.addEventListener( 'resize', onWindowResize );

		}

		//

		function sortFunction( list ) {

			// initialize options
			this._options = this._options || {
				get: el => el.z,
				aux: new Array( this.maxInstanceCount )
			};

			const options = this._options;
			options.reversed = this.material.transparent;

			let minZ = Infinity;
			let maxZ = - Infinity;
			for ( let i = 0, l = list.length; i < l; i ++ ) {

				const z = list[ i ].z;
				if ( z > maxZ ) maxZ = z;
				if ( z < minZ ) minZ = z;

			}

			// convert depth to unsigned 32 bit range
			const depthDelta = maxZ - minZ;
			const factor = ( 2 ** 32 - 1 ) / depthDelta; // UINT32_MAX / z range
			for ( let i = 0, l = list.length; i < l; i ++ ) {

				list[ i ].z -= minZ;
				list[ i ].z *= factor;

			}

			// perform a fast-sort using the hybrid radix sort function
			radixSort( list, options );

		}

		function onWindowResize() {

			const width = window.innerWidth;
			const height = window.innerHeight;

			camera.aspect = width / height;
			camera.updateProjectionMatrix();

			renderer.setSize( width, height );

		}

		function animate() {

			animateMeshes();

			controls.update();
			stats.update();

			render();

		}

		function animateMeshes() {

			const loopNum = Math.min( api.count, api.dynamic );

			if ( api.method === Method.BATCHED ) {

				for ( let i = 0; i < loopNum; i ++ ) {

					const rotationMatrix = mesh.userData.rotationSpeeds[ i ];
					const id = ids[ i ];

					mesh.getMatrixAt( id, matrix );
					matrix.multiply( rotationMatrix );
					mesh.setMatrixAt( id, matrix );

				}

			} else {

				for ( let i = 0; i < loopNum; i ++ ) {

					const child = mesh.children[ i ];
					const rotationSpeed = child.userData.rotationSpeed;

					child.rotation.set(
						child.rotation.x + rotationSpeed.x,
						child.rotation.y + rotationSpeed.y,
						child.rotation.z + rotationSpeed.z
					);

				}

			}

		}

		function render() {

			if ( mesh.isBatchedMesh ) {

				mesh.sortObjects = api.sortObjects;
				mesh.perObjectFrustumCulled = api.perObjectFrustumCulled;
				mesh.setCustomSort( api.useCustomSort ? sortFunction : null );

			}

			renderer.render( scene, camera );

		}

	</script>

</body>
</html>