#include "scene.hpp"

#include "tree.hpp"
#include "interpolation.hpp"
#include "ball.hpp"
#include "bird.hpp"
#include "bat.hpp"
#include "terrain.hpp"
#include "projectile.hpp"
#include "sapin.hpp"
#include "settings.hpp"

using namespace cgp;

std::vector<vec3> tree_position;
std::vector<vec3> grass_position;

void scene_structure::initialize()
{
	//CAMERA
	camera_control.initialize(inputs, window); // Give access to the inputs and window global state to the camera controler

	//for orbit camera
	//camera_control.set_rotation_axis_z(); 
	//camera_control.look_at({ 20.0f,15.0f,15.0f }, {0,0,0});
	
	//for first person camera
	camera_control.camera_model.position_camera = { 0,0,evaluate_terrain_height(0,0)+10};
	camera_control.camera_model.set_rotation_axis({ 0,0,1 });
	//custum function we added to ..first_person_euler
	global_frame.initialize_data_on_gpu(mesh_primitive_frame());


	//SKYBOX
	//skybox_debug.png for debugging
	image_structure image_skybox_template = image_load_file("assets/skybox2.jpg");
	std::vector<image_structure> image_grid = image_split_grid(image_skybox_template, 4, 3);

	image_structure image_skybox_template2 = image_load_file("assets/skyboxNight.jpeg");
	std::vector<image_structure> image_grid2 = image_split_grid(image_skybox_template2, 4, 3);

	skybox.initialize_data_on_gpu();
	//x neg, x pos, y neg, y pos, z neg, z pos
	skybox.texture.initialize_cubemap_on_gpu(image_grid[7].rotate_90_degrees_counterclockwise(), image_grid[1].rotate_90_degrees_clockwise(), image_grid[4].rotate_90_degrees_clockwise().rotate_90_degrees_clockwise(), image_grid[10], image_grid[3], image_grid[5].rotate_90_degrees_clockwise().rotate_90_degrees_clockwise());
	skybox.texture2.initialize_cubemap_on_gpu(image_grid2[7].rotate_90_degrees_counterclockwise(), image_grid2[1].rotate_90_degrees_clockwise(), image_grid2[4].rotate_90_degrees_clockwise().rotate_90_degrees_clockwise(), image_grid2[10], image_grid2[3], image_grid2[5].rotate_90_degrees_clockwise().rotate_90_degrees_clockwise());




	//TERRAIN
	int N_terrain_samples = 500;
	mesh const terrain_mesh = create_terrain_mesh(N_terrain_samples);
	terrain.initialize_data_on_gpu(terrain_mesh);
	// terrain.material.color = { 0.6f,0.85f,0.5f }; terrain de couleur de base verte
	terrain.material.color = { 1,1,1 };
	terrain.material.phong.specular = 0.0f; // non-specular terrain material

	terrain.shader.load(project::path + "shaders/mesh_multi_texture/mesh_multi_texture.vert.glsl", project::path + "shaders/mesh_multi_texture/mesh_multi_texture.frag.glsl");
	terrain.texture.load_and_initialize_texture_2d_on_gpu(project::path + "assets/texture_grass.jpg",
		GL_REPEAT,
		GL_MIRRORED_REPEAT);
	terrain.supplementary_texture["image_texture_2"].load_and_initialize_texture_2d_on_gpu(project::path + "assets/snow.jpg",
		GL_REPEAT,
		GL_MIRRORED_REPEAT);


	//OBJECTS
	int r = 1;
	int h = 5;
	mesh cyl_mesh = create_cylinder_mesh(r, h);
	cyl.initialize_data_on_gpu(cyl_mesh);
	cyl.material.color = { 0.58f,0.294f,0.0f };
	cyl.material.phong.specular = 0.0f;

	r = 3;
	h = 2;
	int z = 3;
	mesh cone_mesh = create_cone_mesh(r, h, z);
	cone.initialize_data_on_gpu(cone_mesh);
	cone.material.color = { 0.0f,0.8f,0.0f };
	cone.material.phong.specular = 0.0f;


	mesh tree_mesh = create_tree();
	tree.initialize_data_on_gpu(tree_mesh);

	float x = 1.0;
	float y = 2.0;
	tree.model.translation = { x, y, evaluate_terrain_height(x,y) };

	tree_position = generate_positions_on_terrain(num_trees());


	grass_position = generate_positions_on_terrain(num_grass());
	mesh quadrangle_mesh;
	quadrangle_mesh.position = { {0,0,0},{1, 0, 0}, {1,0,1},  {0, 0,1} };
	quadrangle_mesh.uv = { {0,0},{1,0}, {1,1}, {0,1} };
	quadrangle_mesh.connectivity = { {0,1,2}, {0,2,3} };
	quadrangle_mesh.fill_empty_field(); // (fill with some default values the other buffers (colors, normals) that we didn't filled before)
	// Convert the mesh structure into a mesh_drawable structure
	grass.initialize_data_on_gpu(quadrangle_mesh);
	grass.texture.load_and_initialize_texture_2d_on_gpu(project::path + "assets/grass.png",
		GL_CLAMP_TO_BORDER,
		GL_CLAMP_TO_BORDER);

	chain1.initialize();
	bouncing.initialize(10); //10 balls

	bats.initialize_bats();

	vec3 bat_pos1 = { 0,0,evaluate_terrain_height(0,0) + 20 };
	float bat_size1 = 0.2f;
	numarray<vec3> bat_keypos1 = { {-1,1,0}, {0,1,0}, {1,1,0}, {1,2,0}, {2,2,0}, {2,2,1}, {2,0,1.5}, {1.5,-1,1}, {1.5,-1,0}, {1,-1,0}, {0,-0.5,0}, {-1,1,0} };
	numarray<float> bat_keytimes1 = { 0.0f, 1.0f, 2.0f, 2.5f, 3.0f, 3.5f, 3.75f, 4.5f, 5.0f, 6.0f, 7.0f, 8.0f };
	bats.add_bat(bat_pos1, bat_size1, bat_keypos1, bat_keytimes1);
	
	vec3 bat_pos2 = { 5,5,evaluate_terrain_height(0,0) + 10 };
	float bat_size2 = 0.3f;
	numarray<vec3> bat_keypos2 = { {-1,1,0}, {0,1,0}, {1,1,0}, {1,2,0}, {2,2,0}, {2,2,1}, {2,0,1.5}, {1.5,-1,1}, {1.5,-1,0}, {1,-1,0}, {0,-0.5,0}, {-1,1,0} };
	numarray<float> bat_keytimes2 = { 0.0f, 1.0f, 2.0f, 2.5f, 3.0f, 3.5f, 3.75f, 4.5f, 5.0f, 6.0f, 7.0f, 8.0f };
	bats.add_bat(bat_pos2, bat_size2, bat_keypos2, bat_keytimes2);

	projectiles.initialize();
	sapin1.initialize_sapin();
	sapin1.sapin["Trunk"].transform_local.rotation = rotation_transform::from_axis_angle({ 1,0,0 }, Pi / 2);
}

void scene_structure::display_frame()
{
	// Update the current time
	timer.update();

	//  Must be called before drawing the other shapes and without writing in the Depth Buffer
	glDepthMask(GL_FALSE); // disable depth-buffer writing
	draw(skybox, timer.t, environment);
	glDepthMask(GL_TRUE);  // re-activate depth-buffer write

	//Walking on ground
	if (!gui.fly) {
		vec3 eye_level = camera_control.camera_model.position_camera;
		eye_level[2] = evaluate_terrain_height(eye_level[0], eye_level[1]) + 0.75;
		camera_control.camera_model.position_camera = eye_level;
	}

	// Set the light to the current position of the camera
	environment.light = camera_control.camera_model.position();

	if (gui.display_frame)
		draw(global_frame, environment);

	draw(terrain, environment);
	if (gui.display_wireframe)
		draw_wireframe(terrain, environment);

	for (int i = 0; i < num_trees(); i++) {
		//tree.model.translation = tree_position[i];
		//draw(tree, environment);
		sapin1.sapin["Trunk"].transform_local.translation = tree_position[i];
		sapin1.sapin.update_local_to_global_coordinates();
		draw(sapin1.sapin, environment);
	}
	//vec3 p = display_mvt();
	//display_bird(p);
	//display_chain(p);
	display_ball();
	display_bats();
	display_projectiles();


	if (gui.display_wireframe)
		draw_wireframe(tree, environment);

	display_semiTransparent();

}

void scene_structure::display_bats()
{
	bats.update_mvt();

	for (int i = 0; i < bats.N; i++) {
		//Wing animation
		bats.bat_mesh["Bat wing left1"].transform_local.rotation = rotation_transform::from_axis_angle({ 0,1,0 }, 0.5 * cos(5 * bats.bats_prop[i].timer_mvt.t));
		bats.bat_mesh["Bat wing right1"].transform_local.rotation = rotation_transform::from_axis_angle({ 0,1,0 }, -0.5 * cos(5 * bats.bats_prop[i].timer_mvt.t));
		//Rescale
		bats.bat_mesh["Bat base"].drawable.model.scaling = bats.bats_prop[i].size;
		bats.bat_mesh["Bat wing left1"].drawable.model.scaling = bats.bats_prop[i].size;
		bats.bat_mesh["Bat wing right1"].drawable.model.scaling = bats.bats_prop[i].size;
		//Make sure wings are at the right distance
		bats.bat_mesh["Bat wing left1"].transform_local.translation = { -0.25f * bats.bats_prop[i].size, 0, 0 };
		bats.bat_mesh["Bat wing right1"].transform_local.translation = { 0.25f * bats.bats_prop[i].size, 0, 0 };

		//Orientation and position of the bat along the path
		bats.bat_mesh["Bat base"].transform_local.rotation = rotation_transform::from_vector_transform({ 0,-1,0 }, normalize(bats.bats_prop[i].pos_futur - bats.bats_prop[i].pos));
		bats.bat_mesh["Bat base"].transform_local.translation = bats.bats_prop[i].pos;

		bats.bat_mesh.update_local_to_global_coordinates();
		if (!bats.bats_prop[i].isdead)	draw(bats.bat_mesh, environment);
		if (gui.display_wireframe) {
			draw_wireframe(bats.bat_mesh, environment);
		}
	}
}


void scene_structure::display_semiTransparent()
{
	// Enable use of alpha component as color blending for transparent elements
	//  alpha = current_color.alpha
	//  new color = previous_color * alpha + current_color * (1-alpha)
	glEnable(GL_BLEND);
	glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);

	// Disable depth buffer writing
	//  - Transparent elements cannot use depth buffer
	//  - They are supposed to be display from furest to nearest elements
	glDepthMask(false);

	auto const& camera = camera_control.camera_model;

	// Re-orient the grass shape to always face the camera direction
	vec3 const right = camera.right();
	// Rotation such that the grass follows the right-vector of the camera, while pointing toward the z-direction
	
	for (int i = 0; i < num_grass(); i++) {
		grass.model.translation = grass_position[i];
		rotation_transform R = rotation_transform::from_frame_transform({ 1,0,0 }, { 0,0,1 }, right, { 0,0,1 });
		grass.model.rotation = R;
		draw(grass, environment);
	}

	// Don't forget to re-activate the depth-buffer write
	glDepthMask(true);
	glDisable(GL_BLEND);
}

void scene_structure::draw_segment(vec3 const& a, vec3 const& b)
{
	chain1.segment.vbo_position.update(numarray<vec3>{ a, b });
	draw(chain1.segment, environment);
}

void scene_structure::display_chain(vec3 bird_pos)
{
	
	// Update the current time
	timer_chain.update();


	chain1.simulation_step(timer_chain.scale * 0.01f, bird_pos);

	for (int i = 0; i < chain1.n; i++) {
		chain1.particle_sphere.model.translation = chain1.p[i];
		chain1.particle_sphere.material.color = { (float)i / chain1.n,(float)i / chain1.n,0 };
		draw(chain1.particle_sphere, environment);
	}

	for (int i = 0; i < chain1.n - 1; i++) {
		draw_segment(chain1.p[i], chain1.p[i + 1]);
	}
}


void scene_structure::display_gui()
{
	ImGui::Checkbox("Frame", &gui.display_frame);
	ImGui::Checkbox("Wireframe", &gui.display_wireframe);
	ImGui::Checkbox("Fly", &gui.fly);

	//ImGui::SliderFloat("Time", &timer_mvt.t, timer_mvt.t_min, timer_mvt.t_max);
	//ImGui::SliderFloat("Time scale", &timer_mvt.scale, 0.0f, 2.0f);
	ImGui::SliderFloat("K", &gui.k, 0.0f, 10.0f);
	ImGui::SliderFloat("Speed", &gui.speed, 0.0f, 10.0f);

	// Display the GUI associated to the key position
	//keyframe.display_gui();
}

void scene_structure::display_ball() {

	bouncing.simulate(timer.scale * 0.01f);
	for (int i = 0; i < bouncing.N; i++) {	
		bouncing.mesh.model.translation = bouncing.pos[i];
		bouncing.mesh.material.color = bouncing.color[i];
		draw(bouncing.mesh, environment);
	}
}

void scene_structure::display_projectiles() {

	projectiles.simulate(timer.scale * 0.03f);
	for (int i = 0; i < projectiles.N; i++) {
		projectiles.mesh.model.translation = projectiles.projectiles_prop[i].pos;
		projectiles.mesh.material.color = projectiles.projectiles_prop[i].color;
		draw(projectiles.mesh, environment);
	}
}

void scene_structure::mouse_move_event()
{
	if (!inputs.keyboard.shift)
		camera_control.action_mouse_move(environment.camera_view);
}
void scene_structure::mouse_click_event()
{
	if (camera_control.inputs->mouse.click.left){
		//std::cout << "ball created";
		projectiles.add_ball(camera_control.camera_model.position_camera + 2*camera_control.camera_model.front(), 20*camera_control.camera_model.front());
	}
	camera_control.action_mouse_click(environment.camera_view);
}
void scene_structure::keyboard_event()
{
	camera_control.action_keyboard(environment.camera_view);
	if(camera_control.inputs->keyboard.shift) {
		projectiles.reset();
	}
}
void scene_structure::idle_frame()
{
	if (camera_control.inputs->keyboard.ctrl) {
		camera_control.idle_frame(environment.camera_view, 10);
	}
	else {
		camera_control.idle_frame(environment.camera_view, gui.speed);
	}
}