Class StreamCompactIDs

src/stream_compact_ids.ts:16–205

}

export class StreamCompactIDs
{
    #device: GPUDevice;

    // Should be at least 64 so that we process elements
    // in 256b blocks with each WG. This will ensure that our
    // dynamic offsets meet the 256b alignment requirement
    readonly WORKGROUP_SIZE: number = 64;

    readonly #maxDispatchSize: number;

    #computePipeline: GPUComputePipeline;

    private constructor(device: GPUDevice)
    {
        this.#device = device;

        this.#maxDispatchSize = device.limits.maxComputeWorkgroupsPerDimension;
    }

    static async create(device: GPUDevice)
    {
        let self = new StreamCompactIDs(device);

        let paramsBGLayout = device.createBindGroupLayout({
            entries: [
                {
                    binding: 0,
                    visibility: GPUShaderStage.COMPUTE,
                    buffer: {
                        type: "storage",
                        hasDynamicOffset: true,
                    }
                },
                {
                    binding: 1,
                    visibility: GPUShaderStage.COMPUTE,
                    buffer: {
                        type: "storage",
                        hasDynamicOffset: true,
                    }
                },
                {
                    binding: 2,
                    visibility: GPUShaderStage.COMPUTE,
                    buffer: {
                        type: "storage",
                    }
                },
            ],
        });

        let pushConstantsBGLayout = device.createBindGroupLayout({
            entries: [
                {
                    binding: 0,
                    visibility: GPUShaderStage.COMPUTE,
                    buffer: {type: "uniform", hasDynamicOffset: true}