Add toneMapping extended mode to Particles sample

webgpu · Jul 11, 2024 · c738625 · c738625
1 parent 464b104
commit c738625
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/sample/particles/main.ts b/sample/particles/main.ts
@@ -24,11 +24,12 @@ const context = canvas.getContext('webgpu') as GPUCanvasContext;
 const devicePixelRatio = window.devicePixelRatio;
 canvas.width = canvas.clientWidth * devicePixelRatio;
 canvas.height = canvas.clientHeight * devicePixelRatio;
-const presentationFormat = navigator.gpu.getPreferredCanvasFormat();
+const presentationFormat = 'rgba16float';
 
 context.configure({
   device,
   format: presentationFormat,
+  toneMapping: { mode: 'extended' },
   alphaMode: 'premultiplied',
 });
 
@@ -315,10 +316,12 @@ let numMipLevels = 1;
 const simulationParams = {
   simulate: true,
   deltaTime: 0.04,
+  brightness: 1.0,
 };
 
 const simulationUBOBufferSize =
   1 * 4 + // deltaTime
+  1 * 4 + // brightness
   3 * 4 + // padding
   4 * 4 + // seed
   0;
@@ -330,6 +333,7 @@ const simulationUBOBuffer = device.createBuffer({
 const gui = new GUI();
 gui.add(simulationParams, 'simulate');
 gui.add(simulationParams, 'deltaTime');
+gui.add(simulationParams, 'brightness', 0, 4, 0.1);
 
 const computePipeline = device.createComputePipeline({
   layout: 'auto',
@@ -375,6 +379,7 @@ function frame() {
     0,
     new Float32Array([
       simulationParams.simulate ? simulationParams.deltaTime : 0.0,
+      simulationParams.brightness,
       0.0,
       0.0,
       0.0, // padding

diff --git a/sample/particles/particle.wgsl b/sample/particles/particle.wgsl
@@ -64,6 +64,7 @@ fn fs_main(in : VertexOutput) -> @location(0) vec4f {
 ////////////////////////////////////////////////////////////////////////////////
 struct SimulationParams {
   deltaTime : f32,
+  brightness : f32,
   seed : vec4f,
 }
 
@@ -125,6 +126,9 @@ fn simulate(@builtin(global_invocation_id) global_invocation_id : vec3u) {
     let uv = vec2f(coord) / vec2f(textureDimensions(texture));
     particle.position = vec3f((uv - 0.5) * 3.0 * vec2f(1.0, -1.0), 0.0);
     particle.color = textureLoad(texture, coord, 0);
+    particle.color.r *= sim_params.brightness;
+    particle.color.g *= sim_params.brightness;
+    particle.color.b *= sim_params.brightness;
     particle.velocity.x = (rand() - 0.5) * 0.1;
     particle.velocity.y = (rand() - 0.5) * 0.1;
     particle.velocity.z = rand() * 0.3;