Add forward shadow receiving support

engine/assets/builtin/shaders/forward-lit/forward-lit.frag.vk.glsl

@@ -2,6 +2,8 @@
 #version 450
 layout(set = 3, binding = 0) uniform texture2D uBaseColorTexture;
 layout(set = 4, binding = 0) uniform sampler uLinearSampler;
+layout(set = 6, binding = 0) uniform texture2D uShadowMapTexture;
+layout(set = 7, binding = 0) uniform sampler uShadowMapSampler;
 
 layout(set = 1, binding = 0, std140) uniform PerObjectConstants {
     mat4 gProjectionMatrix;
@@ -16,11 +18,44 @@ layout(set = 2, binding = 0, std140) uniform MaterialConstants {
     vec4 gBaseColorFactor;
 };
 
+layout(set = 5, binding = 0, std140) uniform ShadowReceiverConstants {
+    mat4 gWorldToShadowMatrix;
+    vec4 gShadowBiasAndTexelSize;
+    vec4 gShadowOptions;
+};
+
 layout(location = 0) in vec3 vNormalWS;
 layout(location = 1) in vec2 vTexCoord;
+layout(location = 2) in vec3 vPositionWS;
 
 layout(location = 0) out vec4 fragColor;
 
+float ComputeShadowAttenuation(vec3 positionWS) {
+    if (gShadowOptions.x < 0.5) {
+        return 1.0;
+    }
+
+    vec4 shadowClip = gWorldToShadowMatrix * vec4(positionWS, 1.0);
+    if (shadowClip.w <= 0.0) {
+        return 1.0;
+    }
+
+    vec3 shadowNdc = shadowClip.xyz / shadowClip.w;
+    vec2 shadowUv = vec2(
+        shadowNdc.x * 0.5 + 0.5,
+        shadowNdc.y * -0.5 + 0.5);
+    if (shadowUv.x < 0.0 || shadowUv.x > 1.0 ||
+        shadowUv.y < 0.0 || shadowUv.y > 1.0 ||
+        shadowNdc.z < 0.0 || shadowNdc.z > 1.0) {
+        return 1.0;
+    }
+
+    float shadowDepth = texture(sampler2D(uShadowMapTexture, uShadowMapSampler), shadowUv).r;
+    float receiverDepth = shadowNdc.z - gShadowBiasAndTexelSize.x;
+    float shadowStrength = clamp(gShadowBiasAndTexelSize.w, 0.0, 1.0);
+    return receiverDepth <= shadowDepth ? 1.0 : (1.0 - shadowStrength);
+}
+
 void main() {
     vec4 baseColor = texture(sampler2D(uBaseColorTexture, uLinearSampler), vTexCoord) * gBaseColorFactor;
     if (gMainLightColorAndFlags.w < 0.5) {
@@ -31,7 +66,10 @@ void main() {
     vec3 normalWS = normalize(vNormalWS);
     vec3 directionToLightWS = normalize(gMainLightDirectionAndIntensity.xyz);
     float diffuse = max(dot(normalWS, directionToLightWS), 0.0);
+    float shadowAttenuation = diffuse > 0.0
+        ? ComputeShadowAttenuation(vPositionWS)
+        : 1.0;
     vec3 lighting = vec3(0.28) +
-        gMainLightColorAndFlags.rgb * (diffuse * gMainLightDirectionAndIntensity.w);
+        gMainLightColorAndFlags.rgb * (diffuse * gMainLightDirectionAndIntensity.w * shadowAttenuation);
     fragColor = vec4(baseColor.rgb * lighting, baseColor.a);
 }