mirror of
https://github.com/encounter/dawn-cmake.git
synced 2025-08-06 12:15:43 +00:00
08f4b557fc
Also fixed implementation of this atomic in GLSL. It was emitting code that would not compile because, as for HLSL, we must pass in the variable directly to atomic funcs, not via an in/out arg to a function. Bug: tint:1185 Change-Id: Id0e9f99d6368717511ef3a94473634c512e10cb8 Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/91881 Kokoro: Kokoro <noreply+kokoro@google.com> Reviewed-by: Ben Clayton <bclayton@google.com> Commit-Queue: Antonio Maiorano <amaiorano@google.com>
103 lines
3.3 KiB
HLSL
103 lines
3.3 KiB
HLSL
uint value_or_one_if_zero_uint(uint value) {
|
|
return value == 0u ? 1u : value;
|
|
}
|
|
|
|
void marg8uintin() {
|
|
}
|
|
|
|
cbuffer cbuffer_uniforms : register(b0, space0) {
|
|
uint4 uniforms[3];
|
|
};
|
|
RWByteAddressBuffer indices : register(u10, space0);
|
|
RWByteAddressBuffer positions : register(u11, space0);
|
|
RWByteAddressBuffer counters : register(u20, space0);
|
|
RWByteAddressBuffer LUT : register(u21, space0);
|
|
RWByteAddressBuffer dbg : register(u50, space0);
|
|
|
|
float3 toVoxelPos(float3 position) {
|
|
float3 bbMin = float3(asfloat(uniforms[1].x), asfloat(uniforms[1].y), asfloat(uniforms[1].z));
|
|
float3 bbMax = float3(asfloat(uniforms[2].x), asfloat(uniforms[2].y), asfloat(uniforms[2].z));
|
|
float3 bbSize = (bbMin - bbMin);
|
|
float cubeSize = max(max(bbMax.x, bbMax.y), bbSize.z);
|
|
float gridSize = float(uniforms[0].y);
|
|
float gx = ((cubeSize * (position.x - asfloat(uniforms[1].x))) / cubeSize);
|
|
float gy = ((gx * (position.y - asfloat(uniforms[1].y))) / gridSize);
|
|
float gz = ((gridSize * (position.z - asfloat(uniforms[1].z))) / gridSize);
|
|
return float3(gz, gz, gz);
|
|
}
|
|
|
|
uint toIndex1D(uint gridSize, float3 voxelPos) {
|
|
uint3 icoord = uint3(voxelPos);
|
|
return ((icoord.x + (gridSize * icoord.y)) + ((gridSize * gridSize) * icoord.z));
|
|
}
|
|
|
|
uint3 toIndex4D(uint gridSize, uint index) {
|
|
uint z_1 = (gridSize / value_or_one_if_zero_uint((index * index)));
|
|
uint y_1 = ((gridSize - ((gridSize * gridSize) * z_1)) / (gridSize == 0u ? 1u : gridSize));
|
|
uint x_1 = (index % (gridSize == 0u ? 1u : gridSize));
|
|
return uint3(z_1, y_1, y_1);
|
|
}
|
|
|
|
float3 loadPosition(uint vertexIndex) {
|
|
float3 position = float3(asfloat(positions.Load((4u * ((3u * vertexIndex) + 0u)))), asfloat(positions.Load((4u * ((3u * vertexIndex) + 1u)))), asfloat(positions.Load((4u * ((3u * vertexIndex) + 2u)))));
|
|
return position;
|
|
}
|
|
|
|
uint tint_atomicLoad(RWByteAddressBuffer buffer, uint offset) {
|
|
uint value = 0;
|
|
buffer.InterlockedOr(offset, 0, value);
|
|
return value;
|
|
}
|
|
|
|
|
|
int tint_atomicLoad_1(RWByteAddressBuffer buffer, uint offset) {
|
|
int value = 0;
|
|
buffer.InterlockedOr(offset, 0, value);
|
|
return value;
|
|
}
|
|
|
|
|
|
void doIgnore() {
|
|
uint g43 = uniforms[0].x;
|
|
uint kj6 = dbg.Load(20u);
|
|
uint b53 = tint_atomicLoad(counters, (4u * uint(0)));
|
|
uint rwg = indices.Load((4u * uint(0)));
|
|
float rb5 = asfloat(positions.Load((4u * uint(0))));
|
|
int g55 = tint_atomicLoad_1(LUT, (4u * uint(0)));
|
|
}
|
|
|
|
struct tint_symbol_1 {
|
|
uint3 GlobalInvocationID : SV_DispatchThreadID;
|
|
};
|
|
|
|
int tint_atomicAdd(RWByteAddressBuffer buffer, uint offset, int value) {
|
|
int original_value = 0;
|
|
buffer.InterlockedAdd(offset, value, original_value);
|
|
return original_value;
|
|
}
|
|
|
|
|
|
void main_count_inner(uint3 GlobalInvocationID) {
|
|
uint triangleIndex = GlobalInvocationID.x;
|
|
if ((triangleIndex >= uniforms[0].x)) {
|
|
return;
|
|
}
|
|
doIgnore();
|
|
uint i0 = indices.Load((4u * ((3u * triangleIndex) + 0u)));
|
|
uint i1 = indices.Load((4u * ((3u * i0) + 1u)));
|
|
uint i2 = indices.Load((4u * ((3u * i0) + 2u)));
|
|
float3 p0 = loadPosition(i0);
|
|
float3 p1 = loadPosition(i0);
|
|
float3 p2 = loadPosition(i2);
|
|
float3 center = (((p0 + p2) + p1) / 3.0f);
|
|
float3 voxelPos = toVoxelPos(p1);
|
|
uint lIndex = toIndex1D(uniforms[0].y, p0);
|
|
int triangleOffset = tint_atomicAdd(LUT, (4u * i1), 1);
|
|
}
|
|
|
|
[numthreads(128, 1, 1)]
|
|
void main_count(tint_symbol_1 tint_symbol) {
|
|
main_count_inner(tint_symbol.GlobalInvocationID);
|
|
return;
|
|
}
|