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


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273
							<html lang="en">
	<head>
		<title>three.js - WebGPU - Compute Particles</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">
	</head>
	<body>

		<div id="info">
			<a href="https://threejs.org" target="_blank" rel="noopener">three.js</a> WebGPU - GPU Compute Particles / 100000 Particles
		</div>

		<script async src="https://unpkg.com/[email protected]/dist/es-module-shims.js"></script>

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

		<script type="module">

			import * as THREE from 'three';
			import { ShaderNode, uniform, texture, instanceIndex, float, vec3, storage, SpriteNodeMaterial } from 'three/nodes';

			import WebGPU from 'three/addons/capabilities/WebGPU.js';
			import WebGPURenderer from 'three/addons/renderers/webgpu/WebGPURenderer.js';

			import { OrbitControls } from 'three/addons/controls/OrbitControls.js';
			import Stats from 'three/addons/libs/stats.module.js';

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

			const particleCount = 100000;

			const gravity = uniform( - .0098 );
			const bounce = uniform( .8 );
			const friction = uniform( .99 );
			const size = uniform( .12 );

			const clickPosition = uniform( new THREE.Vector3() );

			let camera, scene, renderer;
			let controls, stats;
			let computeParticles;

			init();

			function init() {

				if ( WebGPU.isAvailable() === false ) {

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

					throw new Error( 'No WebGPU support' );

				}

				const { innerWidth, innerHeight } = window;

				camera = new THREE.PerspectiveCamera( 50, innerWidth / innerHeight, .1, 1000 );
				camera.position.set( 40, 15, 40 );

				scene = new THREE.Scene();

				// textures

				const textureLoader = new THREE.TextureLoader();
				const map = textureLoader.load( 'textures/sprite1.png' );

				//

				const createBuffer = () => storage( new THREE.InstancedBufferAttribute( new Float32Array( particleCount * 4 ), 4 ), 'vec3', particleCount );

				const positionBuffer = createBuffer();
				const velocityBuffer = createBuffer();
				const colorBuffer = createBuffer();

				// compute

				const computeInit = new ShaderNode( ( stack ) => {

					const position = positionBuffer.element( instanceIndex );
					const color = colorBuffer.element( instanceIndex );

					const randX = instanceIndex.hash();
					const randY = instanceIndex.add( 2 ).hash();
					const randZ = instanceIndex.add( 3 ).hash();

					stack.assign( position.x, randX.mul( 60 ).add( - 30 ) );
					stack.assign( position.y, randY.mul( 10 ) );
					stack.assign( position.z, randZ.mul( 60 ).add( - 30 ) );

					stack.assign( color, vec3( randX, randY, randZ ) );

				} ).compute( particleCount );

				//

				const computeUpdate = new ShaderNode( ( stack ) => {

					const position = positionBuffer.element( instanceIndex );
					const velocity = velocityBuffer.element( instanceIndex );

					stack.assign( velocity, velocity.add( vec3( 0.00, gravity, 0.00 ) ) );
					stack.assign( position, position.add( velocity ) );

					stack.assign( velocity, velocity.mul( friction ) );

					// floor

					stack.if( position.y.lessThan( 0 ), ( stack ) => {

						stack.assign( position.y, 0 );
						stack.assign( velocity.y, velocity.y.negate().mul( bounce ) );

						// floor friction

						stack.assign( velocity.x, velocity.x.mul( .9 ) );
						stack.assign( velocity.z, velocity.z.mul( .9 ) );

					} );

				} );

				computeParticles = computeUpdate.compute( particleCount );

				// create nodes

				const textureNode = texture( map );

				// create particles

				const particleMaterial = new SpriteNodeMaterial();
				particleMaterial.colorNode = textureNode.mul( colorBuffer.element( instanceIndex ) );
				particleMaterial.positionNode = positionBuffer.toAttribute();
				particleMaterial.scaleNode = size;
				particleMaterial.depthWrite = false;
				particleMaterial.depthTest = true;
				particleMaterial.transparent = true;

				const particles = new THREE.Mesh( new THREE.PlaneGeometry( 1, 1 ), particleMaterial );
				particles.isInstancedMesh = true;
				particles.count = particleCount;
				scene.add( particles );

				//

				const helper = new THREE.GridHelper( 60, 40, 0x303030, 0x303030 );
				scene.add( helper );

				const geometry = new THREE.PlaneGeometry( 1000, 1000 );
				geometry.rotateX( - Math.PI / 2 );

				const plane = new THREE.Mesh( geometry, new THREE.MeshBasicMaterial( { visible: false } ) );
				scene.add( plane );

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

				//

				renderer = new WebGPURenderer( { antialias: true } );
				renderer.setPixelRatio( window.devicePixelRatio );
				renderer.setSize( window.innerWidth, window.innerHeight );
				renderer.setAnimationLoop( animate );
				document.body.appendChild( renderer.domElement );
				stats = new Stats();
				document.body.appendChild( stats.dom );

				//

				renderer.compute( computeInit );

				// click event

				const computeHit = new ShaderNode( ( stack ) => {

					const position = positionBuffer.element( instanceIndex );
					const velocity = velocityBuffer.element( instanceIndex );

					const dist = position.distance( clickPosition );
					const direction = position.sub( clickPosition ).normalize();
					const distArea = float( 7 ).sub( dist ).max( 0 );

					const power = distArea.mul( .1 );
					const relativePower = power.mul( instanceIndex.hash().mul( .5 ).add( .5 ) );

					stack.assign( velocity, velocity.add( direction.mul( relativePower ) ) );

				} ).compute( particleCount );

				//

				function onHit( event ) {

					pointer.set( ( event.clientX / window.innerWidth ) * 2 - 1, - ( event.clientY / window.innerHeight ) * 2 + 1 );

					raycaster.setFromCamera( pointer, camera );

					const intersects = raycaster.intersectObjects( [ plane ], false );

					if ( intersects.length > 0 ) {

						const { point } = intersects[ 0 ];

						// move to uniform

						clickPosition.value.copy( point );
						clickPosition.value.y = - 1;

						// compute

						renderer.compute( computeHit );

					}

				}

				// events

				renderer.domElement.addEventListener( 'pointerdown', onHit );

				//

				controls = new OrbitControls( camera, renderer.domElement );
				controls.minDistance = 5;
				controls.maxDistance = 70;
				controls.target.set( 0, - 1, 0 );
				controls.update();

				//

				window.addEventListener( 'resize', onWindowResize );

				// gui

				const gui = new GUI();

				gui.add( gravity, 'value', - .0098, 0, 0.0001 ).name( 'gravity' );
				gui.add( bounce, 'value', .1, 1, 0.01 ).name( 'bounce' );
				gui.add( friction, 'value', .96, .99, 0.01 ).name( 'friction' );
				gui.add( size, 'value', .12, .5, 0.01 ).name( 'size' );

			}

			function onWindowResize() {

				const { innerWidth, innerHeight } = window;

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

				renderer.setSize( innerWidth, innerHeight );

			}

			function animate() {

				stats.update();
				renderer.compute( computeParticles );
				renderer.render( scene, camera );

			}

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