//
// Simple passthrough fragment shader
//
varying vec2 v_vTexcoord;
varying vec4 v_vColour;
varying vec3 v_vNormal;

varying vec4 v_worldPosition;

#region ---- light ----
	uniform vec4  light_ambient;
	uniform float shadowBias;
	
	#define LIGHT_DIR_LIMIT 16
	uniform int	  light_dir_count;
	uniform vec3  light_dir_direction[LIGHT_DIR_LIMIT];
	uniform vec4  light_dir_color[LIGHT_DIR_LIMIT];
	uniform float light_dir_intensity[LIGHT_DIR_LIMIT];
	
	uniform mat4  light_dir_view[LIGHT_DIR_LIMIT];
	uniform mat4  light_dir_proj[LIGHT_DIR_LIMIT];
	uniform int   light_dir_shadow_active[LIGHT_DIR_LIMIT];
	uniform sampler2D light_dir_shadowmap_0;
	uniform sampler2D light_dir_shadowmap_1;
	uniform sampler2D light_dir_shadowmap_2;
	uniform sampler2D light_dir_shadowmap_3;
	
	#define LIGHT_PNT_LIMIT 16
	uniform int	  light_pnt_count;
	uniform vec3  light_pnt_position[LIGHT_PNT_LIMIT];
	uniform vec4  light_pnt_color[LIGHT_PNT_LIMIT];
	uniform float light_pnt_intensity[LIGHT_PNT_LIMIT];
	uniform float light_pnt_radius[LIGHT_PNT_LIMIT];
	
	uniform mat4  light_pnt_view[96];
	uniform mat4  light_pnt_proj[LIGHT_PNT_LIMIT];
	uniform int   light_pnt_shadow_active[LIGHT_PNT_LIMIT];
	uniform sampler2D light_pnt_shadowmap_0;
	uniform sampler2D light_pnt_shadowmap_1;
	uniform sampler2D light_pnt_shadowmap_2;
	uniform sampler2D light_pnt_shadowmap_3;
	uniform sampler2D light_pnt_shadowmap_4;
	uniform sampler2D light_pnt_shadowmap_5;
	uniform sampler2D light_pnt_shadowmap_6;
	uniform sampler2D light_pnt_shadowmap_7;
#endregion

#region ---- rendering ----
	uniform int gammaCorrection;
#endregion

#region ---- matrix ----
	float matrixGet(mat4 matrix, int index) { 
		if(index < 0 || index > 15) return 0.;
		
		int _x = int(floor(float(index) / 4.));
		int _y = int(mod(float(index), 4.));
		return matrix[_x][_y]; 
	}

	mat4 matrixSet(mat4 matrix, int index, float value) { 
		if(index < 0 || index > 15) return matrix;
		
		int _x = int(floor(float(index) / 4.));
		int _y = int(mod(float(index), 4.));
		matrix[_x][_y] = value; 
		return matrix;
	}
#endregion

float sampleDirShadowMap(int index, vec2 position) {
	if(index == 0) return texture2D(light_dir_shadowmap_0, position).r;
	if(index == 1) return texture2D(light_dir_shadowmap_1, position).r;
	if(index == 2) return texture2D(light_dir_shadowmap_2, position).r;
	if(index == 3) return texture2D(light_dir_shadowmap_3, position).r;
	return 0.;
}

float samplePntShadowMap(int index, vec2 position, int side) {
	position.x /= 2.;
	if(side >= 3) {
		position.x += 0.5;
		side -= 3;
	}
	
	if(index == 0) return texture2D(light_pnt_shadowmap_0, position)[side];
	if(index == 1) return texture2D(light_pnt_shadowmap_1, position)[side];
	if(index == 2) return texture2D(light_pnt_shadowmap_2, position)[side];
	if(index == 3) return texture2D(light_pnt_shadowmap_3, position)[side];
	return 0.;
}

void main() {
	vec4 final_color = texture2D( gm_BaseTexture, v_vTexcoord );
	final_color *= v_vColour;
	gl_FragColor = vec4(0.);
	
	#region ++++ light ++++
		int shadow_map_index = 0;
		vec3 light_effect = light_ambient.rgb;
		float val = 0.;
		
		#region ---- directional ----
			float light_dir_strength;
			float light_map_depth;
			float lightDistance;
			float shadow_culled;
		
			shadow_map_index = 0;
			for(int i = 0; i < light_dir_count; i++) {
				light_dir_strength = dot(normalize(v_vNormal), normalize(light_dir_direction[i]));
				if(light_dir_strength < 0.)
					continue;
			
				if(light_dir_shadow_active[i] == 1) {
					vec4 cameraSpace = light_dir_view[i] * v_worldPosition;
					vec4 screenSpace = light_dir_proj[i] * cameraSpace;
	
					float v_lightDistance = screenSpace.z / screenSpace.w;
					vec2 lightMapPosition = (screenSpace.xy / screenSpace.w * 0.5) + 0.5;
				
					light_map_depth = sampleDirShadowMap(shadow_map_index, lightMapPosition);
					shadow_map_index++;
					lightDistance = v_lightDistance;
				
					if(lightDistance > light_map_depth + shadowBias)
						continue;
				} 
			
				light_dir_strength = max(light_dir_strength * light_dir_intensity[i], 0.);
				light_effect += light_dir_color[i].rgb * light_dir_strength;
			}
		#endregion
		#region ---- point ----
			float light_pnt_strength;
			float light_distance;
			float light_attenuation;
		
			shadow_map_index = 0;
			for(int i = 0; i < light_pnt_count; i++) {
				light_pnt_strength = dot(normalize(v_vNormal), normalize(light_pnt_position[i] - v_worldPosition.xyz));
				if(light_pnt_strength < 0.)
					continue;
			
				light_distance = distance(light_pnt_position[i], v_worldPosition.xyz);
				if(light_distance > light_pnt_radius[i])
					continue;
			
				if(light_pnt_shadow_active[i] == 1) {
					vec3 dir	= v_worldPosition.xyz - light_pnt_position[i];
					vec3 dirAbs = abs(dir);
					int side    = dirAbs.x > dirAbs.y ?
								  (dirAbs.x > dirAbs.z ? 0 : 2) :
								  (dirAbs.y > dirAbs.z ? 1 : 2);
					side *= 2;
					if(side == 0 && dir.x < 0.)	     side += 1;
					else if(side == 2 && dir.y < 0.) side += 1;
					else if(side == 4 && dir.z < 0.) side += 1;
					
					vec4 cameraSpace = light_pnt_view[i * 6 + side] * v_worldPosition;
					vec4 screenSpace = light_pnt_proj[i] * cameraSpace;
					float v_lightDistance = screenSpace.z / screenSpace.w;
					vec2 lightMapPosition = (screenSpace.xy / screenSpace.w * 0.5) + 0.5;
					
					light_map_depth = samplePntShadowMap(shadow_map_index, lightMapPosition, side);
					shadow_map_index++;
					
					if(v_lightDistance > light_map_depth + shadowBias)
						continue;
				} 
			
				light_attenuation = 1. - pow(light_distance / light_pnt_radius[i], 2.);
			
				light_pnt_strength = max(light_pnt_strength * light_pnt_intensity[i] * light_attenuation, 0.);
				light_effect += light_pnt_color[i].rgb * light_pnt_strength;
			}
		#endregion
	
		light_effect = max(light_effect, 0.);
		final_color.rgb *= light_effect;
	#endregion
	
	if(gammaCorrection == 1) {
		final_color.r = pow(final_color.r, 1. / 2.2);
		final_color.g = pow(final_color.g, 1. / 2.2);
		final_color.b = pow(final_color.b, 1. / 2.2);
	}
	
	gl_FragColor = final_color;
}