You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
372 lines
7.5 KiB
HTML
372 lines
7.5 KiB
HTML
2 months ago
|
<!DOCTYPE html>
|
||
|
|
<html lang="en">
|
||
|
|
<head>
|
||
|
|
<title>three.js webgpu - 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> webgpu - mesh - batch
|
||
|
|
</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/"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
</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';
|
||
|
|
|
||
|
|
import { normalView, directionToColor, diffuseColor } from 'three/tsl';
|
||
|
|
|
||
|
|
let camera, scene, renderer;
|
||
|
|
let controls, stats;
|
||
|
|
let gui;
|
||
|
|
let geometries, mesh, material;
|
||
|
|
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 api = {
|
||
|
|
webgpu: true,
|
||
|
|
count: 512,
|
||
|
|
dynamic: 16,
|
||
|
|
|
||
|
|
sortObjects: true,
|
||
|
|
perObjectFrustumCulled: true,
|
||
|
|
opacity: 1,
|
||
|
|
useCustomSort: true,
|
||
|
|
randomizeGeometry: ()=>{
|
||
|
|
|
||
|
|
for ( let i = 0; i < api.count; i ++ ) {
|
||
|
|
|
||
|
|
mesh.setGeometryIdAt( i, Math.floor( Math.random() * geometries.length ) );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
init();
|
||
|
|
|
||
|
|
//
|
||
|
|
|
||
|
|
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 initGeometries() {
|
||
|
|
|
||
|
|
geometries = [
|
||
|
|
new THREE.ConeGeometry( 1.0, 2.0 ),
|
||
|
|
new THREE.BoxGeometry( 2.0, 2.0, 2.0 ),
|
||
|
|
new THREE.SphereGeometry( 1.0, 16, 8 ),
|
||
|
|
];
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
function createMaterial() {
|
||
|
|
|
||
|
|
if ( ! material ) {
|
||
|
|
|
||
|
|
material = new THREE.MeshBasicNodeMaterial();
|
||
|
|
material.outputNode = diffuseColor.mul( directionToColor( normalView ).y.add( 0.5 ) );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return material;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
function cleanup() {
|
||
|
|
|
||
|
|
if ( mesh ) {
|
||
|
|
|
||
|
|
mesh.parent.remove( mesh );
|
||
|
|
|
||
|
|
if ( mesh.dispose ) {
|
||
|
|
|
||
|
|
mesh.dispose();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
function initMesh() {
|
||
|
|
|
||
|
|
cleanup();
|
||
|
|
initBatchedMesh();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
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();
|
||
|
|
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 ++ ) {
|
||
|
|
|
||
|
|
const id = mesh.addInstance( geometryIds[ i % geometryIds.length ] );
|
||
|
|
mesh.setMatrixAt( id, randomizeMatrix( matrix ) );
|
||
|
|
mesh.setColorAt( id, new THREE.Color( Math.random() * 0xffffff ) );
|
||
|
|
|
||
|
|
const rotationMatrix = new THREE.Matrix4();
|
||
|
|
rotationMatrix.makeRotationFromEuler( randomizeRotationSpeed( euler ) );
|
||
|
|
mesh.userData.rotationSpeeds.push( rotationMatrix );
|
||
|
|
|
||
|
|
ids.push( id );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
scene.add( mesh );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
function init( forceWebGL = false ) {
|
||
|
|
|
||
|
|
if ( renderer ) {
|
||
|
|
|
||
|
|
renderer.dispose();
|
||
|
|
controls.dispose();
|
||
|
|
document.body.removeChild( stats.dom );
|
||
|
|
document.body.removeChild( renderer.domElement );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// camera
|
||
|
|
|
||
|
|
const aspect = window.innerWidth / window.innerHeight;
|
||
|
|
|
||
|
|
camera = new THREE.PerspectiveCamera( 70, aspect, 1, 100 );
|
||
|
|
camera.position.z = 30;
|
||
|
|
|
||
|
|
// renderer
|
||
|
|
|
||
|
|
renderer = new THREE.WebGPURenderer( { antialias: true, forceWebGL } );
|
||
|
|
renderer.setPixelRatio( window.devicePixelRatio );
|
||
|
|
renderer.setSize( window.innerWidth, window.innerHeight );
|
||
|
|
|
||
|
|
renderer.setAnimationLoop( animate );
|
||
|
|
|
||
|
|
// scene
|
||
|
|
|
||
|
|
scene = new THREE.Scene();
|
||
|
|
scene.background = forceWebGL ? new THREE.Color( 0xffc1c1 ) : new THREE.Color( 0xc1c1ff );
|
||
|
|
|
||
|
|
document.body.appendChild( renderer.domElement );
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
initGeometries();
|
||
|
|
initMesh();
|
||
|
|
|
||
|
|
// controls
|
||
|
|
|
||
|
|
controls = new OrbitControls( camera, renderer.domElement );
|
||
|
|
controls.autoRotate = true;
|
||
|
|
controls.autoRotateSpeed = 1.0;
|
||
|
|
|
||
|
|
// stats
|
||
|
|
|
||
|
|
stats = new Stats();
|
||
|
|
document.body.appendChild( stats.dom );
|
||
|
|
|
||
|
|
// gui
|
||
|
|
|
||
|
|
gui = new GUI();
|
||
|
|
gui.add( api, 'webgpu' ).onChange( () => {
|
||
|
|
|
||
|
|
init( ! api.webgpu );
|
||
|
|
|
||
|
|
} );
|
||
|
|
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, 'opacity', 0, 1 ).onChange( v => {
|
||
|
|
|
||
|
|
if ( v < 1 ) {
|
||
|
|
|
||
|
|
material.transparent = true;
|
||
|
|
material.depthWrite = false;
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
material.transparent = false;
|
||
|
|
material.depthWrite = true;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
material.opacity = v;
|
||
|
|
material.needsUpdate = true;
|
||
|
|
|
||
|
|
} );
|
||
|
|
gui.add( api, 'sortObjects' );
|
||
|
|
gui.add( api, 'perObjectFrustumCulled' );
|
||
|
|
gui.add( api, 'useCustomSort' );
|
||
|
|
gui.add( api, 'randomizeGeometry' );
|
||
|
|
|
||
|
|
|
||
|
|
// listeners
|
||
|
|
|
||
|
|
window.addEventListener( 'resize', onWindowResize );
|
||
|
|
|
||
|
|
function onWindowResize() {
|
||
|
|
|
||
|
|
const width = window.innerWidth;
|
||
|
|
const height = window.innerHeight;
|
||
|
|
|
||
|
|
camera.aspect = width / height;
|
||
|
|
camera.updateProjectionMatrix();
|
||
|
|
|
||
|
|
renderer.setSize( width, height );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
async function animate() {
|
||
|
|
|
||
|
|
animateMeshes();
|
||
|
|
|
||
|
|
controls.update();
|
||
|
|
|
||
|
|
|
||
|
|
if ( mesh.isBatchedMesh ) {
|
||
|
|
|
||
|
|
mesh.sortObjects = api.sortObjects;
|
||
|
|
mesh.perObjectFrustumCulled = api.perObjectFrustumCulled;
|
||
|
|
mesh.setCustomSort( api.useCustomSort ? sortFunction : null );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
await renderer.renderAsync( scene, camera );
|
||
|
|
|
||
|
|
stats.update();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
function animateMeshes() {
|
||
|
|
|
||
|
|
const loopNum = Math.min( api.count, api.dynamic );
|
||
|
|
|
||
|
|
|
||
|
|
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 );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
|
||
|
|
function sortFunction( list, camera ) {
|
||
|
|
|
||
|
|
// initialize options
|
||
|
|
this._options = this._options || {
|
||
|
|
get: el => el.z,
|
||
|
|
aux: new Array( this.maxInstanceCount )
|
||
|
|
};
|
||
|
|
|
||
|
|
const options = this._options;
|
||
|
|
options.reversed = this.material.transparent;
|
||
|
|
|
||
|
|
// convert depth to unsigned 32 bit range
|
||
|
|
const factor = ( 2 ** 32 - 1 ) / camera.far; // UINT32_MAX / max_depth
|
||
|
|
for ( let i = 0, l = list.length; i < l; i ++ ) {
|
||
|
|
|
||
|
|
list[ i ].z *= factor;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
// perform a fast-sort using the hybrid radix sort function
|
||
|
|
radixSort( list, options );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
</script>
|
||
|
|
|
||
|
|
</body>
|
||
|
|
</html>
|