TDT4230_final_project/res/shaders/simple.frag

99 lines
2.6 KiB
GLSL

#version 430 core
in layout(location = 0) vec3 vertex;
in layout(location = 1) vec3 normal;
in layout(location = 2) vec2 UV;
in layout(location = 3) mat3 TBN;
layout(binding = 0) uniform sampler2D diffuseTexture;
layout(binding = 1) uniform sampler2D normalTexture;
layout(binding = 2) uniform sampler2D displacementTexture;
uniform float displacementCoefficient;
uniform mat4 MVP;
uniform mat4 MV;
uniform mat4 MVnormal;
uniform float shininess;
uniform bool isIlluminated;
uniform bool isTextured;
uniform bool isNormalMapped;
uniform bool isDisplacementMapped;
uniform bool isInverted;
// lights
struct Light { // point lights, coordinates in MV space
vec3 position;
vec3 attenuation; // 1 / (x + y*l + z*l*l)
vec3 color_emissive;
vec3 color_diffuse;
vec3 color_specular;
bool is_spot; // false means point light
vec3 spot_target;
float spot_cuttof_angle;
};
#define N_LIGHTS 1
uniform Light light[N_LIGHTS];
out vec4 color;
vec4 phong(vec4 basecolor) {
vec3 nnormal; // normalized normal
if (isNormalMapped) {
vec3 tangential = texture(normalTexture, UV).rgb * 2.0 - 1.0;
nnormal = TBN * normalize(tangential);
}
else {
nnormal = normalize(normal);
}
vec3 emmissive_component = vec3(0.0);
vec3 diffuse_component = vec3(0.0);
vec3 specular_component = vec3(0.0);
for (int i = 0; i<N_LIGHTS; i++) {
vec3 L = light[i].position - vertex;
float l = length(L);
L = normalize(L);
if (light[i].is_spot) {
if (dot(normalize(light[i].position - light[i].spot_target), L) < light[i].spot_cuttof_angle) {
continue;
}
}
float attenuation = clamp(1 / (
light[i].attenuation.x +
light[i].attenuation.y * l +
light[i].attenuation.z * l * l
), 0.0, 1.25);
float diffuse_i = dot(nnormal, L);
float specular_i = pow(dot(reflect(-L, nnormal), normalize(-vertex)), shininess);
emmissive_component += light[i].color_emissive;
if (diffuse_i>0) diffuse_component += light[i].color_diffuse * diffuse_i * attenuation;
if (specular_i>0) specular_component += light[i].color_specular * specular_i * attenuation;
}
return vec4(basecolor.rgb * (emmissive_component + diffuse_component) + specular_component, basecolor.a);
}
void main() {
if(isIlluminated) {
if (isTextured) {
color = phong(texture(diffuseTexture, UV));
} else {
color = phong(vec4(1.0));
}
} else {
color = texture(diffuseTexture, UV);
}
if (isInverted) color.rgb = 1 - color.rgb;
}