#include "debug_sphere3d.h"

#include "defines.h"
#include "math/geometry.h"
#include "math/kmath.h"
#include "math/math_types.h"
#include "renderer/renderer_frontend.h"
#include "systems/ktransform_system.h"

static void update_vert_colour (debug_sphere3d *sphere);

b8 debug_sphere3d_create (f32 radius, vec4 colour, ktransform parent_ktransform, debug_sphere3d *out_sphere) {
	if (!out_sphere) {
		return false;
	}
	out_sphere->ktransform = ktransform_create(0);
	out_sphere->parent_ktransform = parent_ktransform;
	// out_sphere->name // TODO: name?
	out_sphere->radius = radius;
	out_sphere->colour = colour;

	out_sphere->geometry.type = KGEOMETRY_TYPE_3D_STATIC_COLOUR;
	out_sphere->geometry.generation = INVALID_ID_U16;
	out_sphere->is_dirty = true;

	return true;
}

void debug_sphere3d_destroy (debug_sphere3d *sphere) {
	// TODO: zero out, etc.
}

void debug_sphere3d_parent_set (debug_sphere3d *sphere, ktransform parent_ktransform) {
	if (sphere) {
		sphere->parent_ktransform = parent_ktransform;
	}
}

void debug_sphere3d_colour_set (debug_sphere3d *sphere, vec4 colour) {
	if (sphere) {
		if (colour.a == 0) {
			colour.a = 1.0f;
		}
		sphere->colour = colour;
		vec4_clamp_scalar(&sphere->colour, 0.0f, 1.0f);
		if (sphere->geometry.generation != INVALID_ID_U16 && sphere->geometry.vertex_count && sphere->geometry.vertices) {
			update_vert_colour(sphere);
			sphere->is_dirty = true;
		}
	}
}

void debug_sphere3d_render_frame_prepare (debug_sphere3d *sphere, const struct frame_data *p_frame_data) {
	if (!sphere || !sphere->is_dirty) {
		return;
	}

	// Upload the new vertex data.
	renderer_geometry_vertex_update(&sphere->geometry, 0, sphere->geometry.vertex_count, sphere->geometry.vertices, true);

	sphere->geometry.generation++;

	// Roll this over to zero so we don't lock ourselves out of updating.
	if (sphere->geometry.generation == INVALID_ID_U16) {
		sphere->geometry.generation = 0;
	}

	sphere->is_dirty = false;
}

b8 debug_sphere3d_initialize (debug_sphere3d *sphere) {
	if (!sphere) {
		return false;
	}

	sphere->geometry = geometry_generate_line_sphere3d(sphere->radius, 32, sphere->name);
	for (u32 i = 0; i < sphere->geometry.vertex_count; ++i) {
		((colour_vertex_3d *)sphere->geometry.vertices)[i].colour = sphere->colour;
	}

	return true;
}

b8 debug_sphere3d_load (debug_sphere3d *sphere) {
	// Send the geometry off to the renderer to be uploaded to the GPU.
	if (!renderer_geometry_upload(&sphere->geometry)) {
		return false;
	}
	if (sphere->geometry.generation == INVALID_ID_U16) {
		sphere->geometry.generation = 0;
	} else {
		sphere->geometry.generation++;
	}
	return true;
}

b8 debug_sphere3d_unload (debug_sphere3d *sphere) {
	renderer_geometry_destroy(&sphere->geometry);

	return true;
}

b8 debug_sphere3d_update (debug_sphere3d *sphere) {
	return true;
}

static void update_vert_colour (debug_sphere3d *sphere) {
	if (sphere) {
		if (sphere->geometry.vertex_count && sphere->geometry.vertices) {
			colour_vertex_3d *verts = (colour_vertex_3d *)sphere->geometry.vertices;
			for (u32 i = 0; i < sphere->geometry.vertex_count; ++i) {
				verts[i].colour = sphere->colour;
			}
			sphere->is_dirty = true;
		}
	}
}