96 lines
2.6 KiB
GLSL
96 lines
2.6 KiB
GLSL
#version 450
|
|
|
|
#include "shaders://_builtins/constants.glsl"
|
|
#include "shaders://_builtins/PBR.glsl"
|
|
|
|
struct Light {
|
|
mat4 tranform;
|
|
vec4 color;
|
|
float intensity;
|
|
};
|
|
|
|
const int MAX_LIGHT_COUNT = 128;
|
|
|
|
layout( push_constant ) uniform constants
|
|
{
|
|
uint lightCount;
|
|
} PushConstants;
|
|
|
|
layout(set = 0, binding = 0) uniform sampler2D positionTexture;
|
|
layout(set = 0, binding = 1) uniform sampler2D normalTexture;
|
|
layout(set = 0, binding = 2) uniform sampler2D albedoTexture;
|
|
layout(set = 0, binding = 3) uniform sampler2D specularTexture;
|
|
|
|
layout(set = 1, binding = 1) uniform LightBuffer {
|
|
layout(align = 16) Light lights[MAX_LIGHT_COUNT];
|
|
} LightsBuffer;
|
|
|
|
layout (location = 0) in vec2 inUV;
|
|
layout(location = 1) in vec3 cameraPos;
|
|
|
|
layout(location = 0) out vec4 outColor;
|
|
|
|
void main()
|
|
{
|
|
vec4 gposition = texture(positionTexture, inUV);
|
|
vec4 gnormal = texture(normalTexture, inUV);
|
|
vec4 galbedo = texture(albedoTexture, inUV);
|
|
vec4 gspec = texture(specularTexture, inUV);
|
|
|
|
vec3 position = gposition.xyz;
|
|
vec3 normal = gnormal.xyz;
|
|
vec3 albedo = galbedo.xyz;
|
|
vec3 spec = gspec.xyz;
|
|
float specFactor = gspec.w;
|
|
|
|
float metallicFactor = gposition.w;
|
|
float roughnessFactor = galbedo.w;
|
|
float ao = gnormal.w;
|
|
|
|
vec3 V = normalize(cameraPos - position);
|
|
|
|
vec3 F0 = mix(vec3(0.04), albedo, metallicFactor);
|
|
vec3 F = fresnelSchlickRoughness(max(dot(normal, V), 0.0), F0, roughnessFactor);
|
|
|
|
vec3 Lo = vec3(0);
|
|
for(int i = 0; i < PushConstants.lightCount; ++i)
|
|
{
|
|
Light l = LightsBuffer.lights[i];
|
|
|
|
vec3 L = normalize(l.tranform[3].xyz - position);
|
|
vec3 H = normalize(V + L);
|
|
float distance = length(l.tranform[3].xyz - position);
|
|
float attenuation = 1.0 / (distance * distance);
|
|
vec3 radiance = l.color.xyz * attenuation * l.intensity;
|
|
|
|
float NDF = DistributionGGX(normal, H, roughnessFactor);
|
|
float G = GeometrySmith(normal, V, L, roughnessFactor);
|
|
|
|
vec3 numerator = NDF * G * F;
|
|
float denominator = 4.0 * max(dot(normal, V), 0.0) * max(dot(normal, L), 0.0);
|
|
|
|
vec3 specular = numerator / max(denominator, 0.001);
|
|
|
|
vec3 kS = F;
|
|
vec3 kD = vec3(1.0) - kS;
|
|
kD *= 1.0 - metallicFactor;
|
|
|
|
float NdotL = max(dot(normal, L), 0.0);
|
|
Lo += clamp((kD * albedo / PI + specular) * radiance * NdotL, vec3(0), vec3(1));
|
|
}
|
|
|
|
vec3 ambient = vec3(0.2) * albedo;
|
|
|
|
vec3 color = ambient + Lo;
|
|
|
|
color += spec * specFactor;
|
|
// color = mix(color, spec, specFactor);
|
|
|
|
//color = color / (color + vec3(1.0));
|
|
// float reflectance = max(max(F.x, F.y), F.z);
|
|
// float reflectance = max(max(F0.x, F0.y), F0.z);
|
|
|
|
outColor = vec4(color, 1.0);
|
|
// outColor = vec4(vec3(reflectance), 1.0);
|
|
}
|