90 lines
3.2 KiB
Plaintext
90 lines
3.2 KiB
Plaintext
import Common;
|
|
import LightEnv;
|
|
import MaterialParameter;
|
|
import MATERIAL_FILE_NAME;
|
|
|
|
const static uint64_t NUM_CASCADES = 4;
|
|
|
|
struct ShadowMappingData
|
|
{
|
|
Texture2DArray shadowMaps[NUM_CASCADES];
|
|
|
|
StructuredBuffer<float4x4> lightSpaceMatrices[NUM_CASCADES];
|
|
|
|
SamplerState shadowSampler;
|
|
|
|
ConstantBuffer<float4> cascadeSplits;
|
|
};
|
|
ParameterBlock<ShadowMappingData> pShadowMapping;
|
|
|
|
struct LightCullingData
|
|
{
|
|
RWStructuredBuffer<uint> lightIndexList;
|
|
|
|
RWTexture2D<uint2> lightGrid;
|
|
};
|
|
ParameterBlock<LightCullingData> pLightCullingData;
|
|
|
|
static const float4x4 biasMat = float4x4(
|
|
0.5, 0.0, 0.0, 0.5,
|
|
0.0, -0.5, 0.0, 0.5,
|
|
0.0, 0.0, 1.0, 0.0,
|
|
0.0, 0.0, 0.0, 1.0);
|
|
|
|
[shader("pixel")]
|
|
float4 fragmentMain(in FragmentParameter params) : SV_Target
|
|
{
|
|
LightingParameter lightingParams = params.getLightingParameter();
|
|
MaterialParameter materialParams = params.getMaterialParameter();
|
|
var brdf = Material.prepare(materialParams);
|
|
brdf.setNormal(normalize(params.normal_WS));
|
|
uint2 tileIndex = uint2(floor(params.position_CS.xy / BLOCK_SIZE));
|
|
uint startOffset = pLightCullingData.lightGrid[tileIndex].x;
|
|
uint lightCount = pLightCullingData.lightGrid[tileIndex].y;
|
|
float3 result = float3(0, 0, 0);
|
|
for(int i = 0; i < pLightEnv.numDirectionalLights; ++i)
|
|
{
|
|
uint cascadeIndex = 0;
|
|
for (uint c = 0; c < NUM_CASCADES - 1; ++c) {
|
|
if (params.position_VS.z > pShadowMapping.cascadeSplits[c]) {
|
|
//cascadeIndex = c + 1;
|
|
}
|
|
}
|
|
float4 lightSpacePos = mul(pShadowMapping.lightSpaceMatrices[cascadeIndex][i], float4(params.position_WS, 1));
|
|
lightSpacePos = mul(biasMat, lightSpacePos);
|
|
float4 shadowCoord = lightSpacePos / lightSpacePos.w;
|
|
int3 texDim;
|
|
pShadowMapping.shadowMaps[cascadeIndex].GetDimensions(texDim.x, texDim.y, texDim.z);
|
|
float scale = 1.5f;
|
|
float dx = scale * 1.0 / float(texDim.x);
|
|
float dy = scale * 1.0 / float(texDim.y);
|
|
|
|
float shadowFactor = 0.0f;
|
|
int count = 0;
|
|
int range = 1;
|
|
for (int x = -range; x <= range; ++x) {
|
|
for (int y = -range; y <= range; ++y) {
|
|
float shadow = 1.0f;
|
|
if (shadowCoord.z > 0.0 && shadowCoord.z < 1.0)
|
|
{
|
|
float dist = pShadowMapping.shadowMaps[cascadeIndex].Sample(pShadowMapping.shadowSampler, float3(shadowCoord.xy + float2(dx * x, dy * y), i)).r;
|
|
if (shadowCoord.w > 0 && dist > shadowCoord.z + 0.005f)
|
|
{
|
|
shadow = 0;
|
|
}
|
|
}
|
|
shadowFactor += shadow;
|
|
count++;
|
|
}
|
|
}
|
|
|
|
result = float3(pShadowMapping.shadowMaps[cascadeIndex].Sample(pShadowMapping.shadowSampler, float3(shadowCoord.xy, i)).r, 0, 0);//(shadowFactor / count) * pLightEnv.directionalLights[i].illuminate(lightingParams, brdf);
|
|
}
|
|
for (uint i = 0; i < pLightEnv.numPointLights; ++i)
|
|
{
|
|
//uint lightIndex = pLightCullingData.lightIndexList[startOffset + i];
|
|
//result += pLightEnv.pointLights[i].illuminate(lightingParams, brdf);
|
|
}
|
|
//result += brdf.evaluateAmbient(lightingParams.viewDir_WS);
|
|
return float4(result, brdf.getAlpha());
|
|
} |