playcanvas

Version:
Open-source WebGL/WebGPU 3D engine for the web
29 lines (28 loc) • 919 B
JavaScript
const computeGsplatLocalChunkSortSource = `
#include "gsplatLocalBitonicCS"
@group(0) @binding(0) var<storage, read_write> tileEntries: array<u32>;
@group(0) @binding(1) var<storage, read> depthBuffer: array<u32>;
@group(0) @binding(2) var<storage, read> chunkRanges: array<u32>;
@group(0) @binding(3) var<storage, read> totalChunks: array<u32>;
struct Uniforms {
	maxChunks: u32,
}
@group(0) @binding(4) var<uniform> uniforms: Uniforms;
@compute @workgroup_size(256)
fn main(
	@builtin(local_invocation_index) localIdx: u32,
	@builtin(workgroup_id) wid: vec3u,
	@builtin(num_workgroups) numWorkgroups: vec3u
) {
	let chunkIdx = wid.y * numWorkgroups.x + wid.x;
	if (chunkIdx >= min(totalChunks[0], uniforms.maxChunks)) {
		return;
	}
	let tStart = chunkRanges[chunkIdx * 2u];
	let count = chunkRanges[chunkIdx * 2u + 1u];
	bitonicSortRange(localIdx, tStart, count);
}
`;
export {
	computeGsplatLocalChunkSortSource
};