56 lines
1.4 KiB
GLSL
56 lines
1.4 KiB
GLSL
|
// Copyright 2020 Google LLC
|
||
|
|
|
||
|
|
Texture2D textureposition : register(t0);
|
||
|
|
SamplerState samplerposition : register(s0);
|
||
|
|
Texture2D textureNormal : register(t1);
|
||
|
|
SamplerState samplerNormal : register(s1);
|
||
|
|
Texture2D textureAlbedo : register(t2);
|
||
|
|
SamplerState samplerAlbedo : register(s2);
|
||
|
|
Texture2D textureSSAO : register(t3);
|
||
|
|
SamplerState samplerSSAO : register(s3);
|
||
|
|
Texture2D textureSSAOBlur : register(t4);
|
||
|
|
SamplerState samplerSSAOBlur : register(s4);
|
||
|
|
struct UBO
|
||
|
|
{
|
||
|
|
float4x4 _dummy;
|
||
|
|
int ssao;
|
||
|
|
int ssaoOnly;
|
||
|
|
int ssaoBlur;
|
||
|
|
};
|
||
|
|
cbuffer uboParams : register(b5) { UBO uboParams; };
|
||
|
|
|
||
|
|
float4 main([[vk::location(0)]] float2 inUV : TEXCOORD0) : SV_TARGET
|
||
|
|
{
|
||
|
|
float3 fragPos = textureposition.Sample(samplerposition, inUV).rgb;
|
||
|
|
float3 normal = normalize(textureNormal.Sample(samplerNormal, inUV).rgb * 2.0 - 1.0);
|
||
|
|
float4 albedo = textureAlbedo.Sample(samplerAlbedo, inUV);
|
||
|
|
|
||
|
|
float ssao = (uboParams.ssaoBlur == 1) ? textureSSAOBlur.Sample(samplerSSAOBlur, inUV).r : textureSSAO.Sample(samplerSSAO, inUV).r;
|
||
|
|
|
||
|
|
float3 lightPos = float3(0.0, 0.0, 0.0);
|
||
|
|
float3 L = normalize(lightPos - fragPos);
|
||
|
|
float NdotL = max(0.5, dot(normal, L));
|
||
|
|
|
||
|
|
float4 outFragColor;
|
||
|
|
if (uboParams.ssaoOnly == 1)
|
||
|
|
{
|
||
|
|
outFragColor.rgb = ssao.rrr;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
float3 baseColor = albedo.rgb * NdotL;
|
||
|
|
|
||
|
|
if (uboParams.ssao == 1)
|
||
|
|
{
|
||
|
|
outFragColor.rgb = ssao.rrr;
|
||
|
|
|
||
|
|
if (uboParams.ssaoOnly != 1)
|
||
|
|
outFragColor.rgb *= baseColor;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
outFragColor.rgb = baseColor;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return outFragColor;
|
||
|
|
}
|