#include "depth_utils.glslh"

in vec3 vertex_position_in;
in vec3 vertex_color_in;
in vec3 vertex_normal_in;

out float log_z;
out vec3 vertex_color_out;
out vec3 vertex_normal_out;
out vec3 vertex_world_position_out;
out vec4 vertex_position_prev_out;
out vec4 vertex_position_next_out;
out vec3 vertex_object_position_out; // Added: object-space position

uniform mat4 mat_view_proj;
uniform mat4 mat_view_proj_prev;
uniform mat4 mat_view_proj_next;
uniform mat4 mat_world;
uniform mat4 mat_world_prev;
uniform mat4 mat_world_next;

uniform vec4 override_color_1;
uniform vec4 override_color_2;
uniform vec4 override_color_3;
uniform int is_preview;
uniform int is_force_color;
uniform int is_disable_color_replace;

void main()
{
vec3 override_color_1_difference = vertex_color_in - vec3(1.0, 0.494, 0.0);
vec3 override_color_2_difference = vertex_color_in - vec3(0.608, 0.494, 0.0);
vec3 override_color_3_difference = vertex_color_in - vec3(0.216, 0.494, 0.0);
vec3 override_color_4_difference = vertex_color_in - vec3(0.159, 0.275, 0.102);
vec3 preview_color_difference = vertex_color_in - vec3(1.0, 1.0, 1.0);

if((is_disable_color_replace == 0 && (dot(override_color_1_difference, override_color_1_difference) < 0.01 || ( is_preview == 1 && dot(preview_color_difference, preview_color_difference) < 0.01 ))) || (is_force_color == 1))
{
vertex_color_out = override_color_1.rgb;
}
else if((is_disable_color_replace == 0 && (dot(override_color_2_difference, override_color_2_difference) < 0.01 || ( is_preview == 1 && dot(preview_color_difference, preview_color_difference) < 0.01 ))) || (is_force_color == 1))
{
vertex_color_out = override_color_2.rgb;
}
else if((is_disable_color_replace == 0 && (dot(override_color_3_difference, override_color_3_difference) < 0.01 || ( is_preview == 1 && dot(preview_color_difference, preview_color_difference) < 0.01 ))) || (is_force_color == 1))
{
vertex_color_out = override_color_3.rgb;
}
else if((is_disable_color_replace == 0 && (dot(override_color_4_difference, override_color_4_difference) < 0.01) && (dot(vec3(0.0, 1.0, 0.0), vertex_normal_in) >= 0.25)))
{
vertex_color_out = vec3(1.0, 1.0, 1.0);
}
else
{
vertex_color_out = vertex_color_in;
}

vertex_normal_out = (mat_world * vec4(vertex_normal_in, 0.0)).xyz;

vec4 world_pos = mat_world * vec4(vertex_position_in, 1.0);
vertex_world_position_out = world_pos.xyz;
vertex_object_position_out = vertex_position_in; // Pass object-space position

vertex_position_prev_out = mat_view_proj_prev * (mat_world_prev * vec4(vertex_position_in, 1.0));
vertex_position_next_out = mat_view_proj_next * (mat_world_next * vec4(vertex_position_in, 1.0));

gl_Position = mat_view_proj * world_pos;
encode_depth(gl_Position, log_z);
}