40 lines
1.1 KiB
HLSL
40 lines
1.1 KiB
HLSL
#include "ibl_common.hlsli"
|
|
|
|
struct Block
|
|
{
|
|
uint HdrEnvHandle;
|
|
uint OutputTextureHandle;
|
|
int CubeSide;
|
|
};
|
|
|
|
[[vk::push_constant]]
|
|
Block pcb;
|
|
|
|
float2 SampleSphericalMap(float3 v)
|
|
{
|
|
const float2 InvTan = float2(0.1591f, 0.3183f); // (1/2PI, 1/PI)
|
|
float2 UV = float2(atan2(-v.x, v.z), asin(-v.y)); // (-PI, -PI/2) to (PI, PI/2)
|
|
UV *= InvTan; // (-1/2, -1/2) to (1/2, 1/2)
|
|
UV += 0.5f.xx; // (0, 0) to (1, 1)
|
|
return UV;
|
|
}
|
|
|
|
/*
|
|
| Axis | Layer | Up |
|
|
|:----:|:-----:|:--:|
|
|
| +x | 0 | -y |
|
|
| -x | 1 | -y |
|
|
| +y | 2 | +z |
|
|
| -y | 3 | -z |
|
|
| -z | 4 | -y |
|
|
| +z | 5 | -y |
|
|
*/
|
|
|
|
[numthreads(16, 16, 1)]
|
|
void main(uint3 GlobalInvocationID : SV_DispatchThreadID)
|
|
{
|
|
float3 LocalDir = GetLocalDir(pcb.CubeSide, GlobalInvocationID);
|
|
|
|
float2 UV = SampleSphericalMap(LocalDir);
|
|
StorageTextureArrays[pcb.OutputTextureHandle][GlobalInvocationID.xyz] = Textures[pcb.HdrEnvHandle].SampleLevel(ImmutableSamplers[pcb.HdrEnvHandle], UV, 0);
|
|
} |