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


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490
							<!DOCTYPE html>
<html lang="en">
	<head>
		<title>three.js webgpu - compute birds</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="example.css">
	</head>
	<body>

		<div id="info">
			<a href="https://threejs.org/" target="_blank" rel="noopener" class="logo-link"></a>

			<div class="title-wrapper">
				<a href="https://threejs.org/" target="_blank" rel="noopener">three.js</a>
				<span>compute birds</span>
			</div>

			<small>
				Move mouse to disturb birds.
			</small>
		</div>

		<script type="importmap">
			{
				"imports": {
					"three": "../build/three.webgpu.js",
					"three/webgpu": "../build/three.webgpu.js",
					"three/tsl": "../build/three.tsl.js",
					"three/addons/": "./jsm/",
					"stats-gl": "https://cdn.jsdelivr.net/npm/stats-gl@3.6.0/dist/main.js"
				}
			}
		</script>

		<script type="module">

			import * as THREE from 'three/webgpu';
			import { uniform, varying, vec4, add, sub, max, dot, sin, mat3, uint, negate, instancedArray, cameraProjectionMatrix, cameraViewMatrix, positionLocal, modelWorldMatrix, sqrt, property, float, Fn, If, cos, Loop, Continue, normalize, instanceIndex, length, vertexIndex } from 'three/tsl';

			import { Inspector } from 'three/addons/inspector/Inspector.js';

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

			import WebGPU from 'three/addons/capabilities/WebGPU.js';

			let container;
			let camera, scene, renderer;

			let last = performance.now();

			let pointer, raycaster;
			let computeVelocity, computePosition, effectController;

			const BIRDS = 16384;
			const SPEED_LIMIT = 9.0;
			const BOUNDS = 800, BOUNDS_HALF = BOUNDS / 2;

			// Custom Geometry - using 3 triangles each. No normals currently.
			class BirdGeometry extends THREE.BufferGeometry {

				constructor() {

					super();

					const points = 3 * 3;

					const vertices = new THREE.BufferAttribute( new Float32Array( points * 3 ), 3 );

					this.setAttribute( 'position', vertices );

					let v = 0;

					function verts_push() {

						for ( let i = 0; i < arguments.length; i ++ ) {

							vertices.array[ v ++ ] = arguments[ i ];

						}

					}

					const wingsSpan = 20;

					// Body
					verts_push(
						0, 0, - 20,
						0, - 8, 10,
						0, 0, 30
					);

					// Left Wing
					verts_push(
						0, 0, - 15,
						- wingsSpan, 0, 5,
						0, 0, 15
					);

					// Right Wing
					verts_push(
						0, 0, 15,
						wingsSpan, 0, 5,
						0, 0, - 15
					);

					this.scale( 0.2, 0.2, 0.2 );

				}

			}

			// TODO: Fix example with WebGL backend
			
			if ( WebGPU.isAvailable() === false ) {

				document.body.appendChild( WebGPU.getErrorMessage() );

				throw new Error( 'No WebGPU support' );

			}

			function init() {

				container = document.createElement( 'div' );
				document.body.appendChild( container );

				camera = new THREE.PerspectiveCamera( 50, window.innerWidth / window.innerHeight, 1, 5000 );
				camera.position.z = 1000;

				scene = new THREE.Scene();
				scene.fog = new THREE.Fog( 0xffffff, 700, 3000 );

				// Pointer

				pointer = new THREE.Vector2();
				raycaster = new THREE.Raycaster();

				// Sky

				const geometry = new THREE.IcosahedronGeometry( 1, 6 );
				const material = new THREE.MeshBasicNodeMaterial( {
					// Use vertex positions to create atmosphere colors
					colorNode: varying(
						vec4(
							sub( 0.25, positionLocal.y ),
							sub( - 0.25, positionLocal.y ),
							add( 1.5, positionLocal.y ),
							1.0
						)
					),
					side: THREE.BackSide
				} );

				const mesh = new THREE.Mesh( geometry, material );
				mesh.rotation.z = 0.75;
				mesh.scale.setScalar( 1200 );
				scene.add( mesh );

				//

				renderer = new THREE.WebGPURenderer( { antialias: true, forceWebGL: false } );
				renderer.setPixelRatio( window.devicePixelRatio );
				renderer.setSize( window.innerWidth, window.innerHeight );
				renderer.setAnimationLoop( render );
				renderer.toneMapping = THREE.NeutralToneMapping;
				renderer.inspector = new Inspector();
				container.appendChild( renderer.domElement );

				const controls = new OrbitControls( camera );
				controls.connect( renderer.domElement );

				// Initialize position, velocity, and phase values

				const positionArray = new Float32Array( BIRDS * 3 );
				const velocityArray = new Float32Array( BIRDS * 3 );
				const phaseArray = new Float32Array( BIRDS );

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

					const posX = Math.random() * BOUNDS - BOUNDS_HALF;
					const posY = Math.random() * BOUNDS - BOUNDS_HALF;
					const posZ = Math.random() * BOUNDS - BOUNDS_HALF;

					positionArray[ i * 3 + 0 ] = posX;
					positionArray[ i * 3 + 1 ] = posY;
					positionArray[ i * 3 + 2 ] = posZ;

					const velX = Math.random() - 0.5;
					const velY = Math.random() - 0.5;
					const velZ = Math.random() - 0.5;

					velocityArray[ i * 3 + 0 ] = velX * 10;
					velocityArray[ i * 3 + 1 ] = velY * 10;
					velocityArray[ i * 3 + 2 ] = velZ * 10;

					phaseArray[ i ] = 1;

				}

				// Labels applied to storage nodes and uniform nodes are reflected within the shader output,
				// and are useful for debugging purposes.

				const positionStorage = instancedArray( positionArray, 'vec3' ).setName( 'positionStorage' );
				const velocityStorage = instancedArray( velocityArray, 'vec3' ).setName( 'velocityStorage' );
				const phaseStorage = instancedArray( phaseArray, 'float' ).setName( 'phaseStorage' );

				// The Pixel Buffer Object (PBO) is required to get the GPU computed data in the WebGL2 fallback.

				positionStorage.setPBO( true );
				velocityStorage.setPBO( true );
				phaseStorage.setPBO( true );

				// Define Uniforms. Uniforms only need to be defined once rather than per shader.

				effectController = {
					separation: uniform( 15.0 ).setName( 'separation' ),
					alignment: uniform( 20.0 ).setName( 'alignment' ),
					cohesion: uniform( 20.0 ).setName( 'cohesion' ),
					freedom: uniform( 0.75 ).setName( 'freedom' ),
					now: uniform( 0.0 ),
					deltaTime: uniform( 0.0 ).setName( 'deltaTime' ),
					rayOrigin: uniform( new THREE.Vector3() ).setName( 'rayOrigin' ),
					rayDirection: uniform( new THREE.Vector3() ).setName( 'rayDirection' )
				};

				// Create geometry

				const birdGeometry = new BirdGeometry();
				const birdMaterial = new THREE.NodeMaterial();

				// Animate bird mesh within vertex shader, then apply position offset to vertices.

				const birdVertexTSL = Fn( () => {

					const position = positionLocal.toVar();
					const newPhase = phaseStorage.element( instanceIndex ).toVar();
					const newVelocity = normalize( velocityStorage.element( instanceIndex ) ).toVar();

					If( vertexIndex.equal( 4 ).or( vertexIndex.equal( 7 ) ), () => {

						// flap wings
						position.y = sin( newPhase ).mul( 5.0 );

					} );

					const newPosition = modelWorldMatrix.mul( position );

					newVelocity.z.mulAssign( - 1.0 );
					const xz = length( newVelocity.xz );
					const xyz = float( 1.0 );
					const x = sqrt( ( newVelocity.y.mul( newVelocity.y ) ).oneMinus() );

					const cosry = newVelocity.x.div( xz ).toVar();
					const sinry = newVelocity.z.div( xz ).toVar();

					const cosrz = x.div( xyz );
					const sinrz = newVelocity.y.div( xyz ).toVar();

					// Nodes must be negated with negate(). Using '-', their values will resolve to NaN.
					const maty = mat3(
						cosry, 0, negate( sinry ),
						0, 1, 0,
						sinry, 0, cosry
					);

					const matz = mat3(
						cosrz, sinrz, 0,
						negate( sinrz ), cosrz, 0,
						0, 0, 1
					);

					const finalVert = maty.mul( matz ).mul( newPosition );
					finalVert.addAssign( positionStorage.element( instanceIndex ) );

					return cameraProjectionMatrix.mul( cameraViewMatrix ).mul( finalVert );

				} );

				birdMaterial.vertexNode = birdVertexTSL();
				birdMaterial.side = THREE.DoubleSide;

				const birdMesh = new THREE.InstancedMesh( birdGeometry, birdMaterial, BIRDS );
				birdMesh.rotation.y = Math.PI / 2;
				birdMesh.matrixAutoUpdate = false;
				birdMesh.frustumCulled = false;
				birdMesh.updateMatrix();

				// Define GPU Compute shaders.
				// Shaders are computationally identical to their GLSL counterparts outside of texture destructuring.

				computeVelocity = Fn( () => {

					// Define consts
					const PI = float( 3.141592653589793 );
					const PI_2 = PI.mul( 2.0 );
					const limit = property( 'float', 'limit' ).assign( SPEED_LIMIT );

					// Destructure uniforms
					const { alignment, separation, cohesion, deltaTime, rayOrigin, rayDirection } = effectController;

					const zoneRadius = separation.add( alignment ).add( cohesion ).toConst();
					const separationThresh = separation.div( zoneRadius ).toConst();
					const alignmentThresh = ( separation.add( alignment ) ).div( zoneRadius ).toConst();
					const zoneRadiusSq = zoneRadius.mul( zoneRadius ).toConst();

					// Cache current bird's position and velocity outside the loop
					const birdIndex = instanceIndex.toConst( 'birdIndex' );
					const position = positionStorage.element( birdIndex ).toVar();
					const velocity = velocityStorage.element( birdIndex ).toVar();

					// Add influence of pointer position to velocity using cached position
					const directionToRay = rayOrigin.sub( position ).toConst();
					const projectionLength = dot( directionToRay, rayDirection ).toConst();
					const closestPoint = rayOrigin.sub( rayDirection.mul( projectionLength ) ).toConst();
					const directionToClosestPoint = closestPoint.sub( position ).toConst();
					const distanceToClosestPoint = length( directionToClosestPoint ).toConst();
					const distanceToClosestPointSq = distanceToClosestPoint.mul( distanceToClosestPoint ).toConst();

					const rayRadius = float( 150.0 ).toConst();
					const rayRadiusSq = rayRadius.mul( rayRadius ).toConst();

					If( distanceToClosestPointSq.lessThan( rayRadiusSq ), () => {

						const velocityAdjust = ( distanceToClosestPointSq.div( rayRadiusSq ).sub( 1.0 ) ).mul( deltaTime ).mul( 100.0 );
						velocity.addAssign( normalize( directionToClosestPoint ).mul( velocityAdjust ) );
						limit.addAssign( 5.0 );

					} );

					// Attract flocks to center
					const dirToCenter = position.toVar();
					dirToCenter.y.mulAssign( 2.5 );
					velocity.subAssign( normalize( dirToCenter ).mul( deltaTime ).mul( 5.0 ) );

					Loop( { start: uint( 0 ), end: uint( BIRDS ), type: 'uint', condition: '<' }, ( { i } ) => {

						If( i.equal( birdIndex ), () => {

							Continue();
			
						} );

						// Cache bird's position and velocity

						const birdPosition = positionStorage.element( i );
						const dirToBird = birdPosition.sub( position );
						const distToBird = length( dirToBird );

						If( distToBird.lessThan( 0.0001 ), () => {

							Continue();

						} );

						const distToBirdSq = distToBird.mul( distToBird );

						// Don't apply any changes to velocity if changes if the bird is outsize the zone's radius.
						If( distToBirdSq.greaterThan( zoneRadiusSq ), () => {

							Continue();

						} );

						// Determine which threshold the bird is flying within and adjust its velocity accordingly

						const percent = distToBirdSq.div( zoneRadiusSq );

						If( percent.lessThan( separationThresh ), () => {

							// Separation - Move apart for comfort
							const velocityAdjust = ( separationThresh.div( percent ).sub( 1.0 ) ).mul( deltaTime );
							velocity.subAssign( normalize( dirToBird ).mul( velocityAdjust ) );

						} ).ElseIf( percent.lessThan( alignmentThresh ), () => {

							// Alignment - fly the same direction
							const threshDelta = alignmentThresh.sub( separationThresh );
							const adjustedPercent = ( percent.sub( separationThresh ) ).div( threshDelta );
							const birdVelocity = velocityStorage.element( i );

							const cosRange = cos( adjustedPercent.mul( PI_2 ) );
							const cosRangeAdjust = float( 0.5 ).sub( cosRange.mul( 0.5 ) ).add( 0.5 );
							const velocityAdjust = cosRangeAdjust.mul( deltaTime );
							velocity.addAssign( normalize( birdVelocity ).mul( velocityAdjust ) );

						} ).Else( () => {

							// Attraction / Cohesion - move closer
							const threshDelta = alignmentThresh.oneMinus();
							const adjustedPercent = threshDelta.equal( 0.0 ).select( 1.0, ( percent.sub( alignmentThresh ) ).div( threshDelta ) );

							const cosRange = cos( adjustedPercent.mul( PI_2 ) );
							const adj1 = cosRange.mul( - 0.5 );
							const adj2 = adj1.add( 0.5 );
							const adj3 = float( 0.5 ).sub( adj2 );

							const velocityAdjust = adj3.mul( deltaTime );
							velocity.addAssign( normalize( dirToBird ).mul( velocityAdjust ) );

						} );

					} );

					If( length( velocity ).greaterThan( limit ), () => {

						velocity.assign( normalize( velocity ).mul( limit ) );

					} );

					// Write back the final velocity to storage
					velocityStorage.element( birdIndex ).assign( velocity );

				} )().compute( BIRDS ).setName( 'Birds Velocity' );

				computePosition = Fn( () => {

					const { deltaTime } = effectController;
					positionStorage.element( instanceIndex ).addAssign( velocityStorage.element( instanceIndex ).mul( deltaTime ).mul( 15.0 ) );

					const velocity = velocityStorage.element( instanceIndex );
					const phase = phaseStorage.element( instanceIndex );

					const modValue = phase.add( deltaTime ).add( length( velocity.xz ).mul( deltaTime ).mul( 3.0 ) ).add( max( velocity.y, 0.0 ).mul( deltaTime ).mul( 6.0 ) );
					phaseStorage.element( instanceIndex ).assign( modValue.mod( 62.83 ) );

				} )().compute( BIRDS ).setName( 'Birds Position' );

				scene.add( birdMesh );

				container.style.touchAction = 'none';
				container.addEventListener( 'pointermove', onPointerMove );

				window.addEventListener( 'resize', onWindowResize );

				const gui = renderer.inspector.createParameters( 'Birds settings' );
				gui.add( effectController.separation, 'value', 0.0, 100.0, 1.0 ).name( 'Separation' );
				gui.add( effectController.alignment, 'value', 0.0, 100, 0.001 ).name( 'Alignment ' );
				gui.add( effectController.cohesion, 'value', 0.0, 100, 0.025 ).name( 'Cohesion' );

			}

			function onWindowResize() {

				camera.aspect = window.innerWidth / window.innerHeight;
				camera.updateProjectionMatrix();

				renderer.setSize( window.innerWidth, window.innerHeight );

			}

			function onPointerMove( event ) {

				if ( event.isPrimary === false ) return;

				pointer.x = ( event.clientX / window.innerWidth ) * 2.0 - 1.0;
				pointer.y = 1.0 - ( event.clientY / window.innerHeight ) * 2.0;

			}

			function render() {

				const now = performance.now();
				let deltaTime = ( now - last ) / 1000;

				if ( deltaTime > 1 ) deltaTime = 1; // safety cap on large deltas
				last = now;

				raycaster.setFromCamera( pointer, camera );

				effectController.now.value = now;
				effectController.deltaTime.value = deltaTime;
				effectController.rayOrigin.value.copy( raycaster.ray.origin );
				effectController.rayDirection.value.copy( raycaster.ray.direction );

				renderer.compute( computeVelocity );
				renderer.compute( computePosition );

				renderer.render( scene, camera );

				// Move pointer away so we only affect birds when moving the mouse
				pointer.y = 10;

			}

			init();

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