@woosh/meep-engine 3.11.2 → 3.12.0

package/src/shade/renderer/global_illumination/probe/octahedral/convolve/graph_lvp_atlas_to_sh.js

@@ -1,185 +1,185 @@
-import { BufferResourceDescriptor } from "../../../../../device/graph/BufferResourceDescriptor.js";
-import { u32 } from "../../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { WebGPUArray } from "../../../../shader/type/WebGPUArray.js";
-import { WebGPUStruct } from "../../../../shader/type/WebGPUStruct.js";
-import {
-    chunk_octahedral_solid_angle
-} from "../../../../shader/chunk/geometry/octahedron/chunk_octahedral_solid_angle.js";
-import {
-    chunk_uv_octahedral_unit_decode
-} from "../../../../shader/chunk/geometry/octahedron/chunk_uv_octahedral_unit_decode.js";
-import { chunk_grid2d_to_index } from "../../../../shader/chunk/grid/chunk_grid2d_to_index.js";
-import { PI_DECLARATION } from "../../../../shader/chunk/math/PI_DECLARATION.js";
-import { chunk_sh2_basis_at } from "../../../../shader/chunk/spherical_harmonics/chunk_sh2_basis_at.js";
-import { chunk_sh2_multiply_scalar } from "../../../../shader/chunk/spherical_harmonics/chunk_sh2_multiply_scalar.js";
-import { chunk_texel_coordinate_to_uv } from "../../../../shader/chunk/texture/chunk_texel_coordinate_to_uv.js";
-import { CodeChunk } from "../../../../shader/compiler/CodeChunk.js";
-import { ComputeShader } from "../../../../shader/ComputeShader.js";
-import { ShaderResourceSetDescriptor } from "../../../../shader/resource/ShaderResourceSetDescriptor.js";
-import { chunk_lpv_atlas_read_patch_texel } from "../sample/chunk_lpv_atlas_read_patch_texel.js";
-import { chunk_compute_coefficient_address } from "./chunk_compute_coefficient_address.js";
-
-const PUSH_CONSTANTS = WebGPUStruct.from({
-    probe_index_offset: u32,
-    probe_update_count: u32,
-    probe_resolution: u32,
-    probe_count: u32,
-    probes_per_row: u32, // in g-buffer space
-})
-
-const resources = new ShaderResourceSetDescriptor();
-resources.createGroup()
-    .addUniform("settings", PUSH_CONSTANTS)
-
-resources.createGroup()
-    .addStorageBuffer("coefficients", WebGPUArray.f32, true)
-    .addTexture("tAtlas_radiance", "uint");
-
-const body = CodeChunk.from(
-    //language=WGSL
-    `
-@compute @workgroup_size(16,16,1)
-fn main(
-    @builtin(global_invocation_id) global_id : vec3<u32>,
-){
-   
-    let g_buffer_texel = global_id.xy;
-    
-    let probe_resolution = settings.probe_resolution;
-    
-    let probe_texel = g_buffer_texel % probe_resolution;
-
-    let g_buffer_probe_index = grid2d_to_index(g_buffer_texel / probe_resolution, settings.probes_per_row);
-    
-    if(g_buffer_probe_index >= settings.probe_update_count){
-        // out of bounds
-        return;
-    }
-    
-    let global_probe_index = (g_buffer_probe_index + settings.probe_index_offset) % settings.probe_count;
-    
-    let radiance = lpv_atlas_read_patch_texel(global_probe_index, probe_texel + 1, tAtlas_radiance, probe_resolution + 2);
-    
-    let probe_texel_uv = texel_coordinate_to_uv(vec2<f32>(probe_texel), vec2(probe_resolution));
-    
-    let direction = uv_octahedral_unit_decode(probe_texel_uv);
-        
-    let sh3_basis = sh2_basis_at(direction);
-    
-    let local_texel_index = grid2d_to_index(probe_texel, probe_resolution);
-    
-    let coefficients_address = compute_coefficient_address(g_buffer_probe_index, probe_texel);
-    
-    // adjust by solid angle
-    let solid_angle = octahedral_solid_angle(probe_texel_uv, 1.0 / f32(probe_resolution));
-    
-    let sh_coefficients = sh2_multiply_scalar(sh3_basis, solid_angle);
-    
-    // NOTE: we do not convolve with cosine kernel here, this is done during sampling.
-    // what is stored is radiance projected to SH
-    
-    // store coefficients for this texel.
-    // NOTE: no accumulation here, done in a separate pass.
-    for(var i=0u; i< 4; i++){
-        let coefficient = sh_coefficients[i];
-      
-        let convolved = radiance * coefficient;
-      
-        let offset = coefficients_address + i * 3;
-        
-        coefficients[ offset ] = convolved.r;
-        coefficients[ offset + 1 ] = convolved.g;
-        coefficients[ offset + 2 ] = convolved.b;
-    }
-        
-}
-    `, [
-        chunk_sh2_basis_at,
-        chunk_grid2d_to_index,
-        chunk_uv_octahedral_unit_decode,
-        chunk_texel_coordinate_to_uv,
-        chunk_lpv_atlas_read_patch_texel,
-        chunk_compute_coefficient_address,
-        chunk_octahedral_solid_angle,
-        chunk_sh2_multiply_scalar,
-        PI_DECLARATION,
-    ]
-);
-
-const shader = ComputeShader.from({
-    label: "Radiance Atlas / convert to SH",
-    body,
-    resources,
-});
-
-/**
- * Convolve part of the atlas to Spherical Harmonics
- * @param {FrameGraph} graph
- * @param {GPUProbeAtlas} atlas
- * @param {GPUSceneContext} scene
- * @param {number} probe_index_offset
- * @param {number} probe_update_count
- * @param {number[]} g_buffer_resolution In texels
- * @returns {number} buffer resource with SH3-encoded texels
- */
-export function graph_lvp_atlas_to_sh({
-                                          graph,
-                                          atlas,
-                                          scene,
-                                          probe_index_offset,
-                                          probe_update_count,
-                                          g_buffer_resolution,
-                                      }) {
-
-    // first convert rasterized probes to SH
-    const probe_resolution = atlas.probe_resolution;
-
-    /**
-     *
-     * @param data
-     * @param {FramePassResources} resources
-     * @param {GraphicsExecutionContext} context
-     */
-    function execute(data, resources, context) {
-        /**
-         *
-         * @type {GPUBuffer}
-         */
-        const output = resources.get(data.output);
-
-        const settings = context.encoder.allocateTransientValueBuffer(PUSH_CONSTANTS, {
-            probe_index_offset,
-            probe_update_count,
-            probe_resolution: probe_resolution,
-            probe_count: scene.light_probe_volume.source.probe_count,
-            probes_per_row: Math.round(g_buffer_resolution[0] / probe_resolution)
-        });
-
-        shader.dispatch({
-            encoder: context.encoder,
-            bindings: {
-                settings,
-                coefficients: output,
-                tAtlas_radiance: atlas.texture_radiance.obtainView(),
-            },
-            group_count_x: Math.ceil(g_buffer_resolution[0] / 16),
-            group_count_y: Math.ceil(g_buffer_resolution[1] / 16),
-        });
-    }
-
-    const pass_data = {};
-
-    const pass = graph.add("Atlas to SH", pass_data, execute);
-
-    const probe_texel_count = probe_resolution * probe_resolution;
-    const total_updated_texel_count = probe_texel_count * probe_update_count;
-    const buffer_spec = BufferResourceDescriptor.from(
-        12 * total_updated_texel_count * Float32Array.BYTES_PER_ELEMENT,
-        GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC
-    );
-    const r_sh_buffer = pass.create("sh", buffer_spec);
-
-    pass_data.output = r_sh_buffer;
-
-    return r_sh_buffer;
+import { BufferResourceDescriptor } from "../../../../../device/graph/BufferResourceDescriptor.js";
+import { u32 } from "../../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
+import { WebGPUArray } from "../../../../shader/type/WebGPUArray.js";
+import { WebGPUStruct } from "../../../../shader/type/WebGPUStruct.js";
+import {
+    chunk_octahedral_solid_angle
+} from "../../../../shader/chunk/geometry/octahedron/chunk_octahedral_solid_angle.js";
+import {
+    chunk_uv_octahedral_unit_decode
+} from "../../../../shader/chunk/geometry/octahedron/chunk_uv_octahedral_unit_decode.js";
+import { chunk_grid2d_to_index } from "../../../../shader/chunk/grid/chunk_grid2d_to_index.js";
+import { PI_DECLARATION } from "../../../../shader/chunk/math/PI_DECLARATION.js";
+import { chunk_sh2_basis_at } from "../../../../shader/chunk/spherical_harmonics/chunk_sh2_basis_at.js";
+import { chunk_sh2_multiply_scalar } from "../../../../shader/chunk/spherical_harmonics/chunk_sh2_multiply_scalar.js";
+import { chunk_texel_coordinate_to_uv } from "../../../../shader/chunk/texture/chunk_texel_coordinate_to_uv.js";
+import { CodeChunk } from "../../../../shader/compiler/CodeChunk.js";
+import { ComputeShader } from "../../../../shader/ComputeShader.js";
+import { ShaderResourceSetDescriptor } from "../../../../shader/resource/ShaderResourceSetDescriptor.js";
+import { chunk_lpv_atlas_read_patch_texel } from "../sample/chunk_lpv_atlas_read_patch_texel.js";
+import { chunk_compute_coefficient_address } from "./chunk_compute_coefficient_address.js";
+
+const PUSH_CONSTANTS = WebGPUStruct.from({
+    probe_index_offset: u32,
+    probe_update_count: u32,
+    probe_resolution: u32,
+    probe_count: u32,
+    probes_per_row: u32, // in g-buffer space
+})
+
+const resources = new ShaderResourceSetDescriptor();
+resources.createGroup()
+    .addUniform("settings", PUSH_CONSTANTS)
+
+resources.createGroup()
+    .addStorageBuffer("coefficients", WebGPUArray.f32, true)
+    .addTexture("tAtlas_radiance", "uint");
+
+const body = CodeChunk.from(
+    //language=WGSL
+    `
+@compute @workgroup_size(16,16,1)
+fn main(
+    @builtin(global_invocation_id) global_id : vec3<u32>,
+){
+   
+    let g_buffer_texel = global_id.xy;
+    
+    let probe_resolution = settings.probe_resolution;
+    
+    let probe_texel = g_buffer_texel % probe_resolution;
+
+    let g_buffer_probe_index = grid2d_to_index(g_buffer_texel / probe_resolution, settings.probes_per_row);
+    
+    if(g_buffer_probe_index >= settings.probe_update_count){
+        // out of bounds
+        return;
+    }
+    
+    let global_probe_index = (g_buffer_probe_index + settings.probe_index_offset) % settings.probe_count;
+    
+    let radiance = lpv_atlas_read_patch_texel(global_probe_index, probe_texel + 1, tAtlas_radiance, probe_resolution + 2);
+    
+    let probe_texel_uv = texel_coordinate_to_uv(vec2<f32>(probe_texel), vec2(probe_resolution));
+    
+    let direction = uv_octahedral_unit_decode(probe_texel_uv);
+        
+    let sh3_basis = sh2_basis_at(direction);
+    
+    let local_texel_index = grid2d_to_index(probe_texel, probe_resolution);
+    
+    let coefficients_address = compute_coefficient_address(g_buffer_probe_index, probe_texel);
+    
+    // adjust by solid angle
+    let solid_angle = octahedral_solid_angle(probe_texel_uv, 1.0 / f32(probe_resolution));
+    
+    let sh_coefficients = sh2_multiply_scalar(sh3_basis, solid_angle);
+    
+    // NOTE: we do not convolve with cosine kernel here, this is done during sampling.
+    // what is stored is radiance projected to SH
+    
+    // store coefficients for this texel.
+    // NOTE: no accumulation here, done in a separate pass.
+    for(var i=0u; i< 4; i++){
+        let coefficient = sh_coefficients[i];
+      
+        let convolved = radiance * coefficient;
+      
+        let offset = coefficients_address + i * 3;
+        
+        coefficients[ offset ] = convolved.r;
+        coefficients[ offset + 1 ] = convolved.g;
+        coefficients[ offset + 2 ] = convolved.b;
+    }
+        
+}
+    `, [
+        chunk_sh2_basis_at,
+        chunk_grid2d_to_index,
+        chunk_uv_octahedral_unit_decode,
+        chunk_texel_coordinate_to_uv,
+        chunk_lpv_atlas_read_patch_texel,
+        chunk_compute_coefficient_address,
+        chunk_octahedral_solid_angle,
+        chunk_sh2_multiply_scalar,
+        PI_DECLARATION,
+    ]
+);
+
+const shader = ComputeShader.from({
+    label: "Radiance Atlas / convert to SH",
+    body,
+    resources,
+});
+
+/**
+ * Convolve part of the atlas to Spherical Harmonics
+ * @param {FrameGraph} graph
+ * @param {GPUProbeAtlas} atlas
+ * @param {GPUSceneContext} scene
+ * @param {number} probe_index_offset
+ * @param {number} probe_update_count
+ * @param {number[]} g_buffer_resolution In texels
+ * @returns {number} buffer resource with SH3-encoded texels
+ */
+export function graph_lvp_atlas_to_sh({
+                                          graph,
+                                          atlas,
+                                          scene,
+                                          probe_index_offset,
+                                          probe_update_count,
+                                          g_buffer_resolution,
+                                      }) {
+
+    // first convert rasterized probes to SH
+    const probe_resolution = atlas.probe_resolution;
+
+    /**
+     *
+     * @param data
+     * @param {FramePassResources} resources
+     * @param {GraphicsExecutionContext} context
+     */
+    function execute(data, resources, context) {
+        /**
+         *
+         * @type {GPUBuffer}
+         */
+        const output = resources.get(data.output);
+
+        const settings = context.encoder.allocateTransientValueBuffer(PUSH_CONSTANTS, {
+            probe_index_offset,
+            probe_update_count,
+            probe_resolution: probe_resolution,
+            probe_count: scene.light_probe_volume.source.probe_count,
+            probes_per_row: Math.round(g_buffer_resolution[0] / probe_resolution)
+        });
+
+        shader.dispatchThreads({
+            encoder: context.encoder,
+            bindings: {
+                settings,
+                coefficients: output,
+                tAtlas_radiance: atlas.texture_radiance.obtainView(),
+            },
+            thread_count_x: g_buffer_resolution[0],
+            thread_count_y: g_buffer_resolution[1],
+        });
+    }
+
+    const pass_data = {};
+
+    const pass = graph.add("Atlas to SH", pass_data, execute);
+
+    const probe_texel_count = probe_resolution * probe_resolution;
+    const total_updated_texel_count = probe_texel_count * probe_update_count;
+    const buffer_spec = BufferResourceDescriptor.from(
+        12 * total_updated_texel_count * Float32Array.BYTES_PER_ELEMENT,
+        GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC
+    );
+    const r_sh_buffer = pass.create("sh", buffer_spec);
+
+    pass_data.output = r_sh_buffer;
+
+    return r_sh_buffer;
 }