83 lines
2.2 KiB
Plaintext
83 lines
2.2 KiB
Plaintext
import Common;
|
|
|
|
struct DepthData
|
|
{
|
|
Texture2D<float> texture;
|
|
RWStructuredBuffer<float> buffer;
|
|
}
|
|
ParameterBlock<DepthData> pDepthAttachment;
|
|
|
|
struct MipParam
|
|
{
|
|
uint srcMipOffset;
|
|
uint dstMipOffset;
|
|
uint2 srcMipDim;
|
|
uint2 dstMipDim;
|
|
}
|
|
layout(push_constants)
|
|
ConstantBuffer<MipParam> pMipParam;
|
|
|
|
float getSrcDepth(uint2 pos)
|
|
{
|
|
return pDepthAttachment.buffer[pMipParam.srcMipOffset + pos.x + (pos.y * pMipParam.srcMipDim.x)];
|
|
}
|
|
|
|
void setDstDepth(uint2 pos, float depth)
|
|
{
|
|
pDepthAttachment.buffer[pMipParam.dstMipOffset + pos.x + (pos.y * pMipParam.dstMipDim.x)] = depth;
|
|
}
|
|
|
|
[numthreads(BLOCK_SIZE, BLOCK_SIZE, 1)]
|
|
[shader("compute")]
|
|
void reduceLevel(
|
|
uint3 threadID: SV_GroupThreadID,
|
|
uint3 groupID: SV_GroupID,
|
|
){
|
|
uint2 minCoords = (groupID.xy * BLOCK_SIZE + threadID.xy) * 2;
|
|
|
|
if(minCoords.x >= pMipParam.srcMipDim.x || minCoords.y >= pMipParam.srcMipDim.y)
|
|
{
|
|
return;
|
|
}
|
|
|
|
uint2 maxCoords = uint2(min(minCoords.x + 1, pMipParam.srcMipDim.x - 1), min(minCoords.y + 1, pMipParam.srcMipDim.y - 1));
|
|
|
|
float d0 = getSrcDepth(uint2(minCoords.x, minCoords.y));
|
|
float d1 = getSrcDepth(uint2(maxCoords.x, minCoords.y));
|
|
float d2 = getSrcDepth(uint2(minCoords.x, maxCoords.y));
|
|
float d3 = getSrcDepth(uint2(maxCoords.x, maxCoords.y));
|
|
|
|
float minDepth = min(min(d0, d1), min(d2, d3));
|
|
setDstDepth(minCoords / 2, minDepth);
|
|
}
|
|
|
|
groupshared uint uMinDepth;
|
|
|
|
[numthreads(BLOCK_SIZE, BLOCK_SIZE, 1)]
|
|
[shader("compute")]
|
|
void initialReduce(
|
|
uint3 threadID: SV_GroupThreadID,
|
|
uint3 groupID: SV_GroupID,
|
|
) {
|
|
uint reducedWidth = uint(pViewParams.screenDimensions.x) / BLOCK_SIZE;
|
|
int2 groupOffset = groupID.xy * BLOCK_SIZE;
|
|
int2 texCoord = groupOffset + threadID.xy;
|
|
float fDepth = pDepthAttachment.texture.Load(int3(texCoord, 0)).r;
|
|
uint uDepth = asuint(fDepth);
|
|
if(groupID.x == 0 && groupID.y == 0)
|
|
{
|
|
uMinDepth = 0xffffffff;
|
|
}
|
|
GroupMemoryBarrierWithGroupSync();
|
|
|
|
InterlockedMin(uMinDepth, uDepth);
|
|
|
|
GroupMemoryBarrierWithGroupSync();
|
|
|
|
float fMinDepth = asfloat(uMinDepth);
|
|
if(threadID.x == 0 && threadID.y == 0)
|
|
{
|
|
pDepthAttachment.buffer[groupID.x + (groupID.y * reducedWidth)] = fMinDepth;
|
|
}
|
|
}
|