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.
259 lines
6.7 KiB
HTML
259 lines
6.7 KiB
HTML
2 months ago
|
<html lang="en">
|
||
|
|
<head>
|
||
|
|
<title>three.js webgpu - storage pbo external element</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>
|
||
|
|
<br />This example demonstrates fetching an external element from a StorageBuffer.
|
||
|
|
<br />The left canvas uses the WebGPU Backend, while the right uses the WebGL Backend.
|
||
|
|
<div id="timestamps" style="
|
||
|
|
position: absolute;
|
||
|
|
top: 60px;
|
||
|
|
left: 0;
|
||
|
|
padding: 10px;
|
||
|
|
background: rgba( 0, 0, 0, 0.5 );
|
||
|
|
color: #fff;
|
||
|
|
font-family: monospace;
|
||
|
|
font-size: 12px;
|
||
|
|
line-height: 1.5;
|
||
|
|
pointer-events: none;
|
||
|
|
text-align: left;
|
||
|
|
"></div>
|
||
|
|
<div id="timestamps_webgl" style="
|
||
|
|
position: absolute;
|
||
|
|
top: 60px;
|
||
|
|
right: 0;
|
||
|
|
padding: 10px;
|
||
|
|
background: rgba( 0, 0, 0, 0.5 );
|
||
|
|
color: #fff;
|
||
|
|
font-family: monospace;
|
||
|
|
font-size: 12px;
|
||
|
|
line-height: 1.5;
|
||
|
|
pointer-events: none;
|
||
|
|
text-align: left;
|
||
|
|
"></div>
|
||
|
|
</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 { storage, If, vec3, uv, uint, float, Fn, instanceIndex, workgroupBarrier } from 'three/tsl';
|
||
|
|
|
||
|
|
const timestamps = {
|
||
|
|
webgpu: document.getElementById( 'timestamps' ),
|
||
|
|
webgl: document.getElementById( 'timestamps_webgl' )
|
||
|
|
};
|
||
|
|
|
||
|
|
// WebGPU Backend
|
||
|
|
init();
|
||
|
|
|
||
|
|
// WebGL Backend
|
||
|
|
init( true );
|
||
|
|
|
||
|
|
async function init( forceWebGL = false ) {
|
||
|
|
|
||
|
|
const aspect = ( window.innerWidth / 2 ) / window.innerHeight;
|
||
|
|
const camera = new THREE.OrthographicCamera( - aspect, aspect, 1, - 1, 0, 2 );
|
||
|
|
camera.position.z = 1;
|
||
|
|
|
||
|
|
const scene = new THREE.Scene();
|
||
|
|
|
||
|
|
// texture
|
||
|
|
|
||
|
|
const size = 32; // non power of two buffer size is not well supported in WebGPU
|
||
|
|
const barCount = 32;
|
||
|
|
|
||
|
|
const type = [ 'float', 'vec2', 'vec3', 'vec4' ];
|
||
|
|
|
||
|
|
const arrayBufferNodes = [];
|
||
|
|
|
||
|
|
for ( let i = 0; i < type.length; i ++ ) {
|
||
|
|
|
||
|
|
const typeSize = i + 1;
|
||
|
|
const array = new Array( size * typeSize ).fill( 0 );
|
||
|
|
|
||
|
|
const arrayBuffer = new THREE.StorageInstancedBufferAttribute( new Float32Array( array ), typeSize );
|
||
|
|
|
||
|
|
arrayBufferNodes.push( storage( arrayBuffer, type[ i ], size ).setPBO( true ) );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
const computeInitOrder = Fn( () => {
|
||
|
|
|
||
|
|
for ( let i = 0; i < type.length; i ++ ) {
|
||
|
|
|
||
|
|
arrayBufferNodes[ i ].element( instanceIndex ).assign( instanceIndex );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
const computeInvertOrder = Fn( () => {
|
||
|
|
|
||
|
|
for ( let i = 0; i < type.length; i ++ ) {
|
||
|
|
|
||
|
|
const invertIndex = arrayBufferNodes[ i ].element( uint( size - 1 ).sub( instanceIndex ) ).toVar();
|
||
|
|
workgroupBarrier();
|
||
|
|
arrayBufferNodes[ i ].element( instanceIndex ).assign( invertIndex );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
// compute
|
||
|
|
|
||
|
|
const computeInit = computeInitOrder().compute( size );
|
||
|
|
|
||
|
|
const compute = computeInvertOrder().compute( size );
|
||
|
|
|
||
|
|
const material = new THREE.MeshBasicNodeMaterial( { color: 0x00ff00 } );
|
||
|
|
|
||
|
|
material.colorNode = Fn( () => {
|
||
|
|
|
||
|
|
const index = uint( uv().x.mul( size ).floor() ).toVar();
|
||
|
|
|
||
|
|
If( index.greaterThanEqual( size ), () => {
|
||
|
|
|
||
|
|
index.assign( uint( size ).sub( 1 ) );
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
const color = vec3( 0, 0, 0 ).toVar();
|
||
|
|
|
||
|
|
If( uv().y.greaterThan( 0.0 ), () => {
|
||
|
|
|
||
|
|
const indexValue = arrayBufferNodes[ 0 ].element( index ).toVar();
|
||
|
|
const value = float( indexValue ).div( float( size ) ).mul( barCount ).floor().div( barCount );
|
||
|
|
|
||
|
|
color.assign( vec3( value, 0, 0 ) );
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
If( uv().y.greaterThan( 0.25 ), () => {
|
||
|
|
|
||
|
|
const indexValue = arrayBufferNodes[ 1 ].element( index ).toVar();
|
||
|
|
const value = float( indexValue ).div( float( size ) ).mul( barCount ).floor().div( barCount );
|
||
|
|
|
||
|
|
color.assign( vec3( 0, value, 0 ) );
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
If( uv().y.greaterThan( 0.5 ), () => {
|
||
|
|
|
||
|
|
const indexValue = arrayBufferNodes[ 2 ].element( index ).toVar();
|
||
|
|
const value = float( indexValue ).div( float( size ) ).mul( barCount ).floor().div( barCount );
|
||
|
|
|
||
|
|
color.assign( vec3( 0, 0, value ) );
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
If( uv().y.greaterThan( 0.75 ), () => {
|
||
|
|
|
||
|
|
const indexValue = arrayBufferNodes[ 3 ].element( index ).toVar();
|
||
|
|
const value = float( indexValue ).div( float( size ) ).mul( barCount ).floor().div( barCount );
|
||
|
|
|
||
|
|
color.assign( vec3( value, value, value ) );
|
||
|
|
|
||
|
|
} );
|
||
|
|
|
||
|
|
return color;
|
||
|
|
|
||
|
|
} )();
|
||
|
|
|
||
|
|
//
|
||
|
|
|
||
|
|
const plane = new THREE.Mesh( new THREE.PlaneGeometry( 1, 1 ), material );
|
||
|
|
scene.add( plane );
|
||
|
|
|
||
|
|
const renderer = new THREE.WebGPURenderer( { antialias: false, forceWebGL: forceWebGL, trackTimestamp: true } );
|
||
|
|
renderer.setPixelRatio( window.devicePixelRatio );
|
||
|
|
renderer.setSize( window.innerWidth / 2, window.innerHeight );
|
||
|
|
|
||
|
|
document.body.appendChild( renderer.domElement );
|
||
|
|
renderer.domElement.style.position = 'absolute';
|
||
|
|
renderer.domElement.style.top = '0';
|
||
|
|
renderer.domElement.style.left = '0';
|
||
|
|
renderer.domElement.style.width = '50%';
|
||
|
|
renderer.domElement.style.height = '100%';
|
||
|
|
|
||
|
|
if ( forceWebGL ) {
|
||
|
|
|
||
|
|
renderer.domElement.style.left = '50%';
|
||
|
|
|
||
|
|
scene.background = new THREE.Color( 0x212121 );
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
scene.background = new THREE.Color( 0x313131 );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
await renderer.computeAsync( computeInit );
|
||
|
|
|
||
|
|
//
|
||
|
|
|
||
|
|
renderer.info.autoReset = false;
|
||
|
|
|
||
|
|
const stepAnimation = async function () {
|
||
|
|
|
||
|
|
renderer.info.reset();
|
||
|
|
|
||
|
|
await renderer.computeAsync( compute );
|
||
|
|
await renderer.renderAsync( scene, camera );
|
||
|
|
|
||
|
|
renderer.resolveTimestampsAsync( THREE.TimestampQuery.COMPUTE );
|
||
|
|
renderer.resolveTimestampsAsync( THREE.TimestampQuery.RENDER );
|
||
|
|
|
||
|
|
timestamps[ forceWebGL ? 'webgl' : 'webgpu' ].innerHTML = `
|
||
|
|
|
||
|
|
Compute ${renderer.info.compute.frameCalls} pass in ${renderer.info.compute.timestamp.toFixed( 6 )}ms<br>
|
||
|
|
Draw ${renderer.info.render.drawCalls} pass in ${renderer.info.render.timestamp.toFixed( 6 )}ms`;
|
||
|
|
|
||
|
|
setTimeout( stepAnimation, 1000 );
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
stepAnimation();
|
||
|
|
|
||
|
|
window.addEventListener( 'resize', onWindowResize );
|
||
|
|
|
||
|
|
function onWindowResize() {
|
||
|
|
|
||
|
|
renderer.setSize( window.innerWidth / 2, window.innerHeight );
|
||
|
|
|
||
|
|
const aspect = ( window.innerWidth / 2 ) / window.innerHeight;
|
||
|
|
|
||
|
|
const frustumHeight = camera.top - camera.bottom;
|
||
|
|
|
||
|
|
camera.left = - frustumHeight * aspect / 2;
|
||
|
|
camera.right = frustumHeight * aspect / 2;
|
||
|
|
|
||
|
|
camera.updateProjectionMatrix();
|
||
|
|
|
||
|
|
renderer.render( scene, camera );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
</script>
|
||
|
|
</body>
|
||
|
|
</html>
|