@woosh/meep-engine 3.3.0 → 3.4.0

package/src/shade/renderer/gpu_primitive/sort/one_sweep/graph_radix_sort.d.ts

@@ -1,15 +0,0 @@
-/**
- * Keys are 32bit unsigned integers
- * Values are 32bit unsigned integers as well
- * Sorting is always in ASCENDING order (0,1,2,3,4...)
- * @param {FrameGraph} graph
- * @param {number} keys
- * @param {number} values
- * @param {number} [element_count] number of input elements
- * @returns {{keys:number, values:number}} sorted output
- */
-export function graph_radix_sort({ graph, keys, values, element_count, }: FrameGraph): {
-    keys: number;
-    values: number;
-};
-//# sourceMappingURL=graph_radix_sort.d.ts.map

package/src/shade/renderer/gpu_primitive/sort/one_sweep/graph_radix_sort.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"graph_radix_sort.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/gpu_primitive/sort/one_sweep/graph_radix_sort.js"],"names":[],"mappings":"AAWA;;;;;;;;;GASG;AACH,uFAFa;IAAC,IAAI,EAAC,MAAM,CAAC;IAAC,MAAM,EAAC,MAAM,CAAA;CAAC,CA4KxC"}

package/src/shade/renderer/gpu_primitive/sort/one_sweep/graph_radix_sort.js

@@ -1,192 +0,0 @@
-import { assert } from "../../../../../core/assert.js";
-import { round_up } from "../../../../../core/math/round_up.js";
-import { BufferResourceDescriptor } from "../../../../device/graph/BufferResourceDescriptor.js";
-import { u32 } from "../../../shader/type/WEBGPU_WGSL_PRIMITIVE_TYPES.js";
-import { k_globalHistPartitionSize, k_partitionSize, k_radix, k_radixPasses } from "./defs.js";
-import { PUSH_CONSTANTS, shader_digit_binning_pass_pairs } from "./shader_digit_binning_pass_pairs.js";
-import { GLOBAL_HISTOGRAM_PUSH_CONSTANTS, shader_global_histogram } from "./shader_global_histogram.js";
-import { shader_init_sweep } from "./shader_init_sweep.js";
-import { SCAN_PUSH_CONSTANTS, shader_scan } from "./shader_scan.js";
-
-
-/**
- * Keys are 32bit unsigned integers
- * Values are 32bit unsigned integers as well
- * Sorting is always in ASCENDING order (0,1,2,3,4...)
- * @param {FrameGraph} graph
- * @param {number} keys
- * @param {number} values
- * @param {number} [element_count] number of input elements
- * @returns {{keys:number, values:number}} sorted output
- */
-export function graph_radix_sort({
-                                     graph,
-                                     keys,
-                                     values,
-                                     element_count,
-                                 }) {
-
-    const globalHistThreadBlocks = Math.ceil(element_count / k_globalHistPartitionSize);
-    const threadBlocks = Math.ceil(element_count / k_partitionSize);
-
-    const data = {};
-
-    /**
-     *
-     * @param data
-     * @param {FramePassResources} resources
-     * @param {GraphicsExecutionContext} context
-     */
-    function execute(data, resources, context) {
-        // see https://github.com/b0nes164/GPUSorting/blob/09a6081d964b682bbb58838b83ab75aebe7f4e05/GPUSortingUnity/Runtime/OneSweep.cs#L241
-
-        const histogram_global = resources.get(data.histogram_global);
-
-        const histogram_pass = resources.get(data.histogram_pass);
-
-        /**
-         * @type {GPUBuffer}
-         */
-        const keys = resources.get(data.keys);
-        /**
-         * @type {GPUBuffer}
-         */
-        const alt_keys = resources.get(data.alt_keys);
-
-        /**
-         * @type {GPUBuffer}
-         */
-        const payload = resources.get(data.payload);
-        /**
-         * @type {GPUBuffer}
-         */
-        const alt_payload = resources.get(data.alt_payload);
-
-        const index = resources.get(data.index);
-
-        assert.greaterThanOrEqual(keys.size, element_count * Uint32Array.BYTES_PER_ELEMENT, 'key buffer underflow');
-        assert.greaterThanOrEqual(payload.size, element_count * Uint32Array.BYTES_PER_ELEMENT, 'value buffer underflow');
-
-        const encoder = context.encoder;
-
-        /**
-         * TODO we can do this by clearing buffers directly
-         */
-        shader_init_sweep.dispatch({
-            encoder,
-            group_count_x: 256,
-            bindings: {
-                b_passHist: histogram_pass,
-                b_globalHist: histogram_global,
-                b_index: index,
-                e_threadBlocks: encoder.allocateTransientValueBuffer(u32, threadBlocks)
-            }
-        });
-
-        shader_global_histogram.dispatch({
-            encoder,
-            group_count_x: globalHistThreadBlocks,
-            bindings: {
-                b_sort: keys,
-                b_globalHist: histogram_global,
-                settings: encoder.allocateTransientValueBuffer(GLOBAL_HISTOGRAM_PUSH_CONSTANTS, {
-                    e_threadBlocks: globalHistThreadBlocks,
-                    e_numKeys: element_count,
-                })
-            }
-        });
-
-        shader_scan.dispatch({
-            encoder,
-            group_count_x: k_radixPasses,
-            bindings: {
-                histogram_global,
-                histogram_pass,
-                settings: encoder.allocateTransientValueBuffer(SCAN_PUSH_CONSTANTS, {
-                    e_threadBlocks: threadBlocks,
-                    e_numKeys: element_count,
-                })
-            }
-        });
-
-        let input_keys = keys;
-        let input_payload = payload;
-        let output_keys = alt_keys;
-        let output_payload = alt_payload;
-
-        for (let radixShift = 0; radixShift < 32; radixShift += 8) {
-
-            const settings = encoder.allocateTransientValueBuffer(PUSH_CONSTANTS, {
-                e_numKeys: element_count,
-                e_radixShift: radixShift,
-                e_threadBlocks: threadBlocks
-            });
-
-            shader_digit_binning_pass_pairs.dispatch({
-                encoder,
-                group_count_x: threadBlocks,
-                bindings: {
-                    b_index: index,
-                    b_sort: input_keys,
-                    b_sortPayload: input_payload,
-                    b_alt: output_keys,
-                    b_altPayload: output_payload,
-                    b_passHist: histogram_pass,
-                    settings,
-                }
-            });
-
-            // swap keys and payload ping-pong buffers independently (keys[i]/payload[i] must stay
-            // paired). Destructured to avoid the copy-paste slip that assigned the payload temp
-            // back into output_keys.
-            [input_keys, output_keys] = [output_keys, input_keys];
-            [input_payload, output_payload] = [output_payload, input_payload];
-        }
-
-
-    }
-
-    const graph_pass = graph.add(`radix sort`, data, execute);
-
-    data.keys = graph_pass.write(keys);
-    data.payload = graph_pass.write(values);
-
-    // these are temporary ping-pong buffers
-    // aligned on 1k boundary to lower memory fragmentation
-    const temporary_buffer_size = round_up(element_count * Uint32Array.BYTES_PER_ELEMENT, 1024);
-
-    data.alt_keys = graph_pass.create("alt_keys", BufferResourceDescriptor.from(
-        temporary_buffer_size,
-        GPUBufferUsage.STORAGE
-    ));
-    data.alt_payload = graph_pass.create("alt_payload", BufferResourceDescriptor.from(
-        temporary_buffer_size,
-        GPUBufferUsage.STORAGE
-    ));
-
-    data.index = graph_pass.create("index", BufferResourceDescriptor.from(
-        k_radixPasses * Uint32Array.BYTES_PER_ELEMENT,
-        GPUBufferUsage.STORAGE,
-    ));
-
-    data.histogram_global = graph_pass.create("histogram_global", BufferResourceDescriptor.from(
-        k_radix * k_radixPasses * Uint32Array.BYTES_PER_ELEMENT,
-        GPUBufferUsage.STORAGE,
-    ));
-
-
-    const PASS_HISTOGRAM_MEMORY_SIZE = threadBlocks * k_radix * k_radixPasses * Uint32Array.BYTES_PER_ELEMENT;
-    const PASS_HISTOGRAM_DESCRIPTOR = BufferResourceDescriptor.from(
-        round_up(PASS_HISTOGRAM_MEMORY_SIZE, 1024),
-        GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_DST,
-        0, PASS_HISTOGRAM_MEMORY_SIZE
-    );
-
-    data.histogram_pass = graph_pass.create("histogram_pass", PASS_HISTOGRAM_DESCRIPTOR);
-
-    // sorted output
-    return {
-        keys: data.keys,
-        values: data.payload,
-    }
-}

package/src/shade/renderer/gpu_primitive/sort/one_sweep/shader_digit_binning_pass_pairs.d.ts

@@ -1,5 +0,0 @@
-export const PUSH_CONSTANTS: WebGPUStruct;
-export const shader_digit_binning_pass_pairs: ComputeShader;
-import { WebGPUStruct } from "../../../shader/type/WebGPUStruct.js";
-import { ComputeShader } from "../../../shader/ComputeShader.js";
-//# sourceMappingURL=shader_digit_binning_pass_pairs.d.ts.map

package/src/shade/renderer/gpu_primitive/sort/one_sweep/shader_digit_binning_pass_pairs.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"shader_digit_binning_pass_pairs.d.ts","sourceRoot":"","sources":["../../../../../../../src/shade/renderer/gpu_primitive/sort/one_sweep/shader_digit_binning_pass_pairs.js"],"names":[],"mappings":"AASA,0CAIG;AA48BH,4DAIG;6BA19B0B,sCAAsC;8BAErC,kCAAkC"}