1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#version 330 core
in vec2 vUV;
out float FragColor;
uniform sampler2D gDepth;
uniform sampler2D gNormal;
uniform sampler2D texNoise;
uniform vec3 samples[64];
uniform mat4 proj;
uniform mat4 invProj;
uniform float radius;
uniform float bias;
uniform float noiseScale;
uniform int kernelSize;
uniform float power;
vec3 getViewPos(vec2 uv)
{
float depth = texture(gDepth, uv).r;
if (depth == 1.0) return vec3(0.0);
float z = depth * 2.0 - 1.0;
vec4 clip = vec4(uv * 2.0 - 1.0, z, 1.0);
vec4 viewPos = invProj * clip;
viewPos /= viewPos.w;
return viewPos.xyz;
}
void main()
{
vec3 fragPos = getViewPos(vUV);
if (fragPos == vec3(0.0))
{
FragColor = 1.0;
return;
}
vec3 normal = texture(gNormal, vUV).rgb;
normal = normalize(normal * 2.0 - 1.0);
vec3 randomVec = normalize(texture(texNoise, vUV * noiseScale).xyz * 2.0 - 1.0);
vec3 tangent = normalize(randomVec - normal * dot(randomVec, normal));
vec3 bitangent = cross(normal, tangent);
mat3 TBN = mat3(tangent, bitangent, normal);
float occlusion = 0.0;
for (int i = 0; i < kernelSize; ++i)
{
vec3 sample = TBN * samples[i];
sample = fragPos + sample * radius;
vec4 offset = proj * vec4(sample, 1.0);
offset.xyz /= offset.w;
vec2 sampleUV = offset.xy * 0.5 + 0.5;
if (sampleUV.x < 0.0 || sampleUV.x > 1.0 || sampleUV.y < 0.0 || sampleUV.y > 1.0)
continue;
vec3 samplePos = getViewPos(sampleUV);
if (samplePos == vec3(0.0))
continue;
float rangeCheck = smoothstep(0.0, 1.0, radius / abs(fragPos.z - samplePos.z));
float diff = samplePos.z - sample.z;
if (diff > bias)
occlusion += rangeCheck;
}
occlusion = 1.0 - (occlusion / float(kernelSize));
occlusion = pow(occlusion, power);
FragColor = occlusion;
}
|
