81 lines
2.2 KiB
Plaintext
81 lines
2.2 KiB
Plaintext
|
import InputGeometry;
|
||
|
|
import LightEnv;
|
||
|
|
import BRDF;
|
||
|
|
import Material;
|
||
|
|
import TexturedMaterial;
|
||
|
|
import FlatColorMaterial;
|
||
|
|
|
||
|
|
struct ViewParameter
|
||
|
|
{
|
||
|
|
float4x4 viewMatrix;
|
||
|
|
float4x4 projectionMatrix;
|
||
|
|
float4 cameraPos_WS;
|
||
|
|
}
|
||
|
|
layout(set = 0, binding = 0, std430)
|
||
|
|
ConstantBuffer<ViewParameter> gViewParams;
|
||
|
|
|
||
|
|
layout(set = 0, binding = 1, std430)
|
||
|
|
ConstantBuffer<Lights> gLightEnv;
|
||
|
|
|
||
|
|
layout(set = 1, std430)
|
||
|
|
type_param TMaterial : IMaterial;
|
||
|
|
ParameterBlock<TMaterial> gMaterial;
|
||
|
|
|
||
|
|
struct ModelParameter
|
||
|
|
{
|
||
|
|
float4x4 modelMatrix;
|
||
|
|
}
|
||
|
|
|
||
|
|
[[vk::push_constant]]
|
||
|
|
ConstantBuffer<ModelParameter> gModelParams;
|
||
|
|
|
||
|
|
struct VertexStageOutput
|
||
|
|
{
|
||
|
|
MaterialPixelParameter materialParameter : MaterialParameter;
|
||
|
|
float4 sv_position : SV_Position;
|
||
|
|
};
|
||
|
|
|
||
|
|
[shader("vertex")]
|
||
|
|
VertexStageOutput vertexMain(
|
||
|
|
InputGeometry inputGeometry)
|
||
|
|
{
|
||
|
|
VertexStageOutput output;
|
||
|
|
MaterialPixelParameter pixelParams;
|
||
|
|
float4 worldPosition = mul(gModelParams.modelMatrix, float4(inputGeometry.getVertexPosition(), 1));
|
||
|
|
pixelParams.position = worldPosition.xyz;
|
||
|
|
pixelParams.texCoord = inputGeometry.texCoord;
|
||
|
|
|
||
|
|
float4 viewPosition = mul(gViewParams.viewMatrix, worldPosition);
|
||
|
|
pixelParams.viewDir = gViewParams.cameraPos_WS.xyz - worldPosition.xyz;
|
||
|
|
|
||
|
|
pixelParams.normal = mul(gModelParams.modelMatrix, float4(inputGeometry.getNormal(), 0)).xyz;
|
||
|
|
pixelParams.tangent = mul(gModelParams.modelMatrix, float4(inputGeometry.getTangent(), 0)).xyz;
|
||
|
|
pixelParams.biTangent = mul(gModelParams.modelMatrix, float4(inputGeometry.getBiTangent(), 0)).xyz;
|
||
|
|
pixelParams.clipPosition = mul(gViewParams.projectionMatrix, viewPosition);
|
||
|
|
|
||
|
|
output.materialParameter = pixelParams;
|
||
|
|
output.sv_position = pixelParams.clipPosition;
|
||
|
|
|
||
|
|
return output;
|
||
|
|
}
|
||
|
|
|
||
|
|
[shader("fragment")]
|
||
|
|
float4 fragmentMain(MaterialPixelParameter input : MaterialParameter) : SV_Target
|
||
|
|
{
|
||
|
|
TMaterial.BRDF brdf = gMaterial.prepare(input);
|
||
|
|
|
||
|
|
float3 viewDir = normalize(input.viewDir);
|
||
|
|
|
||
|
|
float3 result = float3(0, 0, 0);
|
||
|
|
for (int i = 0; i < gLightEnv.numDirectionalLights; ++i)
|
||
|
|
{
|
||
|
|
result += gLightEnv.directionalLights[i].illuminate(input, brdf, viewDir);
|
||
|
|
}
|
||
|
|
for (int i = 0; i < gLightEnv.numPointLights; ++i)
|
||
|
|
{
|
||
|
|
result += gLightEnv.pointLights[i].illuminate(input, brdf, viewDir);
|
||
|
|
}
|
||
|
|
|
||
|
|
return float4(result, 0);
|
||
|
|
}
|