/*
Copyright (c) 2013 Daniele Bartolini, Michele Rossi
Copyright (c) 2012 Daniele Bartolini, Simone Boscaratto

Permission is hereby granted, free of charge, to any person
obtaining a copy of this software and associated documentation
files (the "Software"), to deal in the Software without
restriction, including without limitation the rights to use,
copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the
Software is furnished to do so, subject to the following
conditions:

The above copyright notice and this permission notice shall be
included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
OTHER DEALINGS IN THE SOFTWARE.
*/

#include "unit.h"
#include "world.h"
#include "allocator.h"
#include "log.h"
#include "unit_resource.h"
#include "scene_graph_manager.h"
#include "actor.h"
#include "controller.h"
#include "physics_resource.h"
#include "device.h"
#include "resource_manager.h"
#include "sprite.h"
#include "sprite_animation_player.h"

namespace crown
{

using namespace unit_resource;

Unit::Unit(World& w, UnitId unit_id, StringId64 resid, const UnitResource* ur, const Matrix4x4& pose)
	: m_world(w)
	, m_scene_graph(*w.scene_graph_manager()->create_scene_graph())
	, m_sprite_animation(NULL)
	, m_resource_id(resid)
	, m_resource(ur)
	, m_id(unit_id)
	, m_num_cameras(0)
	, m_num_meshes(0)
	, m_num_sprites(0)
	, m_num_actors(0)
	, m_num_materials(0)
	, m_values(NULL)
{
	m_controller.component.id = INVALID_ID;
	create_objects(pose);
}

Unit::~Unit()
{
	destroy_objects();
	m_world.scene_graph_manager()->destroy_scene_graph(&m_scene_graph);
}

void Unit::set_id(const UnitId id)
{
	m_id = id;
}

UnitId Unit::id()
{
	return m_id;
}

const UnitResource*	Unit::resource() const
{
	return m_resource;
}

void Unit::create_objects(const Matrix4x4& pose)
{
	using namespace unit_resource;
	m_scene_graph.create(pose, num_scene_graph_nodes(m_resource), scene_graph_nodes(m_resource));

	create_camera_objects();
	create_renderable_objects();
	create_physics_objects();

	set_default_material();

	m_values = (char*) default_allocator().allocate(values_size(m_resource));
	memcpy(m_values, values(m_resource), values_size(m_resource));

	StringId64 anim_id = sprite_animation(m_resource);
	if (anim_id != 0)
	{
		m_sprite_animation = m_world.sprite_animation_player()->create_sprite_animation((SpriteAnimationResource*) device()->resource_manager()->get(SPRITE_ANIMATION_TYPE, anim_id));
	}
}

void Unit::destroy_objects()
{
	if (m_sprite_animation)
	{
		m_world.sprite_animation_player()->destroy_sprite_animation(m_sprite_animation);
	}

	default_allocator().deallocate(m_values);

	// Destroy cameras
	for (uint32_t i = 0; i < m_num_cameras; i++)
	{
		m_world.destroy_camera(m_cameras[i].component);
	}
	m_num_cameras = 0;

	// Destroy meshes
	for (uint32_t i = 0; i < m_num_meshes; i++)
	{
		m_world.render_world()->destroy_mesh(m_meshes[i].component);
	}
	m_num_meshes = 0;

	// Destroy sprites
	for (uint32_t i = 0; i < m_num_sprites; i++)
	{
		m_world.render_world()->destroy_sprite(m_sprites[i].component);
	}
	m_num_sprites = 0;

	// Destroy actors
	for (uint32_t i = 0; i < m_num_actors; i++)
	{
		m_world.physics_world()->destroy_actor(m_actors[i].component);
	}
	m_num_actors = 0;

	// Destroy controller
	if (m_controller.component.id != INVALID_ID)
	{
		m_world.physics_world()->destroy_controller(m_controller.component);
		m_controller.component.id = INVALID_ID;
	}

	// Destroy materials
	for (uint32_t i = 0; i < m_num_materials; i++)
	{
		material_manager::get()->destroy_material(m_materials[i].component);
	}
	m_num_materials = 0;

	// Destroy scene graph
	m_scene_graph.destroy();
}

void Unit::create_camera_objects()
{
	for (uint32_t i = 0; i < num_cameras(m_resource); i++)
	{
		const UnitCamera* cam = get_camera(m_resource, i);
		const CameraId id = m_world.create_camera(m_scene_graph, cam->node, (ProjectionType::Enum)cam->type, cam->near, cam->far);
		add_camera(cam->name, id);
	}
}

void Unit::create_renderable_objects()
{
	// Create renderables
	for (uint32_t i = 0; i < num_renderables(m_resource); i++)
	{
		const UnitRenderable* ur = get_renderable(m_resource, i);

		if (ur->type == UnitRenderable::MESH)
		{
			MeshResource* mr = (MeshResource*) device()->resource_manager()->get(MESH_TYPE, ur->resource);
			MeshId mesh = m_world.render_world()->create_mesh(mr, m_scene_graph, ur->node);
			add_mesh(ur->name, mesh);
		}
		else if (ur->type == UnitRenderable::SPRITE)
		{
			SpriteResource* sr = (SpriteResource*) device()->resource_manager()->get(SPRITE_TYPE, ur->resource);
			SpriteId sprite = m_world.render_world()->create_sprite(sr, m_scene_graph, ur->node);
			add_sprite(ur->name, sprite);
		}
		else
		{
			CE_FATAL("Oops, bad renderable type");
		}
	}

	for (uint32_t i = 0; i < num_materials(m_resource); i++)
	{
		const UnitMaterial* mat = get_material(m_resource, i);
		add_material(string::murmur2_32("default", string::strlen("default"), 0), material_manager::get()->create_material(mat->id));
	}
}

void Unit::create_physics_objects()
{
	using namespace unit_resource;
	using namespace physics_resource;
	if (unit_resource::physics_resource(m_resource) != 0)
	{
		const PhysicsResource* pr = (PhysicsResource*) device()->resource_manager()->get(PHYSICS_TYPE, unit_resource::physics_resource(m_resource));

		// Create controller if any
		if (has_controller(pr))
		{
			set_controller(physics_resource::controller(pr)->name, m_world.physics_world()->create_controller(pr, m_scene_graph, 0));
		}

		// Create actors if any
		for (uint32_t i = 0; i < num_actors(pr); i++)
		{
			const PhysicsActor* actor = physics_resource::actor(pr, i);

			ActorId id = m_world.physics_world()->create_actor(pr, i, m_scene_graph, m_scene_graph.node(actor->node), m_id);
			add_actor(actor->name, id);
		}

		// Create joints if any
		for (uint32_t i = 0; i < num_joints(pr); i++)
		{
			const PhysicsJoint* joint = physics_resource::joint(pr, i);

			Actor* a1 = actor_by_index(joint->actor_0);
			Actor* a2 = actor_by_index(joint->actor_1);

			m_world.physics_world()->create_joint(pr, i, *a1, *a2);
		}
	}
}

void Unit::set_default_material()
{
	if (m_num_materials == 0) return;

	for (uint32_t i = 0; i < m_num_sprites; i++)
	{
		Sprite* s = m_world.render_world()->get_sprite(m_sprites[i].component);
		s->set_material(m_materials[0].component);
	}
}

int32_t Unit::node(const char* name) const
{
	return m_scene_graph.node(name);
}

bool Unit::has_node(const char* name) const
{
	return m_scene_graph.has_node(name);
}

uint32_t Unit::num_nodes() const
{
	return m_scene_graph.num_nodes();
}

Vector3 Unit::local_position(int32_t node) const
{
	return m_scene_graph.local_position(node);
}

Quaternion Unit::local_rotation(int32_t node) const
{
	return m_scene_graph.local_rotation(node);
}

Matrix4x4 Unit::local_pose(int32_t node) const
{
	return m_scene_graph.local_pose(node);
}

Vector3 Unit::world_position(int32_t node) const
{
	return m_scene_graph.world_position(node);
}

Quaternion Unit::world_rotation(int32_t node) const
{
	return m_scene_graph.world_rotation(node);
}

Matrix4x4 Unit::world_pose(int32_t node) const
{
	return m_scene_graph.world_pose(node);
}

void Unit::set_local_position(int32_t node, const Vector3& pos)
{
	m_scene_graph.set_local_position(node, pos);
}

void Unit::set_local_rotation(int32_t node, const Quaternion& rot)
{
	m_scene_graph.set_local_rotation(node, rot);
}

void Unit::set_local_pose(int32_t node, const Matrix4x4& pose)
{
	m_scene_graph.set_local_pose(node, pose);
}

void Unit::link_node(int32_t child, int32_t parent)
{
	m_scene_graph.link(child, parent);
}

void Unit::unlink_node(int32_t child)
{
	m_scene_graph.unlink(child);
}

void Unit::update()
{
	if (m_sprite_animation)
	{
		sprite(0u)->set_frame(m_sprite_animation->m_cur_frame);
	}
}

void Unit::reload(UnitResource* new_ur)
{
	Matrix4x4 m = m_scene_graph.world_pose(0);
	destroy_objects();
	m_resource = new_ur;
	create_objects(m);
}

void Unit::add_component(StringId32 name, Id component, uint32_t& size, Component* array)
{
	Component comp;
	comp.name = name;
	comp.component = component;

	array[size] = comp;
	size++;
}

Id Unit::find_component(const char* name, uint32_t size, Component* array)
{
	const uint32_t name_hash = string::murmur2_32(name, string::strlen(name), 0);
	return find_component_by_name(name_hash, size, array);
}

Id Unit::find_component_by_index(uint32_t index, uint32_t size, Component* array)
{
	Id comp;
	comp.id = INVALID_ID;

	if (index < size)
	{
		comp = array[index].component;
	}

	return comp;
}

Id Unit::find_component_by_name(StringId32 name, uint32_t size, Component* array)
{
	Id comp;
	comp.id = INVALID_ID;

	for (uint32_t i = 0; i < size; i++)
	{
		if (name == array[i].name)
		{
			comp = array[i].component;
		}
	}

	return comp;
}

void Unit::add_camera(StringId32 name, CameraId camera)
{
	CE_ASSERT(m_num_cameras < CE_MAX_CAMERA_COMPONENTS, "Max camera number reached");

	add_component(name, camera, m_num_cameras, m_cameras);
}

void Unit::add_mesh(StringId32 name, MeshId mesh)
{
	CE_ASSERT(m_num_meshes < CE_MAX_MESH_COMPONENTS, "Max mesh number reached");

	add_component(name, mesh, m_num_meshes, m_meshes);
}

void Unit::add_sprite(StringId32 name, SpriteId sprite)
{
	CE_ASSERT(m_num_sprites < CE_MAX_SPRITE_COMPONENTS, "Max sprite number reached");

	add_component(name, sprite, m_num_sprites, m_sprites);
}

void Unit::add_actor(StringId32 name, ActorId actor)
{
	CE_ASSERT(m_num_actors < CE_MAX_ACTOR_COMPONENTS, "Max actor number reached");

	add_component(name, actor, m_num_actors, m_actors);
}

void Unit::add_material(StringId32 name, MaterialId material)
{
	CE_ASSERT(m_num_materials < CE_MAX_MATERIAL_COMPONENTS, "Max material number reached");

	add_component(name, material, m_num_materials, m_materials);
}

void Unit::set_controller(StringId32 name, ControllerId controller)
{
	m_controller.name = name;
	m_controller.component = controller;
}

Camera* Unit::camera(const char* name)
{
	CameraId cam = find_component(name, m_num_cameras, m_cameras);

	CE_ASSERT(cam.id != INVALID_ID, "Unit does not have camera with name '%s'", name);

	return m_world.get_camera(cam);
}

Camera* Unit::camera(uint32_t i)
{
	CameraId cam = find_component_by_index(i, m_num_cameras, m_cameras);

	CE_ASSERT(cam.id != INVALID_ID, "Unit does not have camera with index '%d'", i);

	return m_world.get_camera(cam);
}

Mesh* Unit::mesh(const char* name)
{
	MeshId mesh = find_component(name, m_num_meshes, m_meshes);

	CE_ASSERT(mesh.id != INVALID_ID, "Unit does not have mesh with name '%s'", name);

	return m_world.render_world()->get_mesh(mesh);
}

Mesh* Unit::mesh(uint32_t i)
{
	MeshId mesh = find_component_by_index(i, m_num_meshes, m_meshes);

	CE_ASSERT(mesh.id != INVALID_ID, "Unit does not have mesh with index '%d'", i);

	return m_world.render_world()->get_mesh(mesh);
}

Sprite*	Unit::sprite(const char* name)
{
	SpriteId sprite = find_component(name, m_num_sprites, m_sprites);

	CE_ASSERT(sprite.id != INVALID_ID, "Unit does not have sprite with name '%s'", name);

	return m_world.render_world()->get_sprite(sprite);
}

Sprite*	Unit::sprite(uint32_t i)
{
	SpriteId sprite = find_component_by_index(i, m_num_sprites, m_sprites);

	CE_ASSERT(sprite.id != INVALID_ID, "Unit does not have sprite with index '%d'", i);

	return m_world.render_world()->get_sprite(sprite);
}

Actor* Unit::actor(const char* name)
{
	ActorId actor = find_component(name, m_num_actors, m_actors);

	CE_ASSERT(actor.id != INVALID_ID, "Unit does not have actor with name '%s'", name);

	return m_world.physics_world()->get_actor(actor);
}

Actor* Unit::actor(uint32_t i)
{
	ActorId actor = find_component_by_index(i, m_num_actors, m_actors);

	CE_ASSERT(actor.id != INVALID_ID, "Unit does not have actor with index '%d'", i);

	return m_world.physics_world()->get_actor(actor);
}

Actor* Unit::actor_by_index(StringId32 name)
{
	ActorId actor = find_component_by_name(name, m_num_actors, m_actors);

	// CE_ASSERT(actor.id != INVALID_ID, "Unit does not have actor with name '%d'", name);

	return m_world.physics_world()->get_actor(actor);
}

Controller* Unit::controller()
{
	if (m_controller.component.id != INVALID_ID)
	{
		return m_world.physics_world()->get_controller(m_controller.component);
	}

	return NULL;
}

Material* Unit::material(const char* name)
{
	MaterialId material = find_component(name, m_num_materials, m_materials);
	CE_ASSERT(material.id != INVALID_ID, "Unit does not have material with name '%s'", name);
	return material_manager::get()->lookup_material(material);
}

Material* Unit::material(uint32_t i)
{
	MaterialId material = find_component_by_index(i, m_num_materials, m_materials);
	CE_ASSERT(material.id != INVALID_ID, "Unit does not have material with name '%d'", i);
	return material_manager::get()->lookup_material(material);
}

bool Unit::is_a(const char* name)
{
	return m_resource_id == ResourceId("unit", name).name;
}

void Unit::play_sprite_animation(const char* name, bool loop)
{
	if (m_sprite_animation)
		m_sprite_animation->play(string::murmur2_32(name, string::strlen(name), 0), loop);
}

void Unit::stop_sprite_animation()
{
	if (m_sprite_animation)
		m_sprite_animation->stop();
}

bool Unit::has_key(const char* k) const
{
	using namespace unit_resource;
	return unit_resource::has_key(m_resource, k);
}

ValueType::Enum Unit::value_type(const char* k)
{
	using namespace unit_resource;
	Key key;
	unit_resource::get_key(m_resource, k, key);
	return (ValueType::Enum) key.type;
}

bool Unit::get_key(const char* k, bool& v) const
{
	using namespace unit_resource;
	Key key;
	bool has = unit_resource::get_key(m_resource, k, key);
	v = *(uint32_t*)(m_values + key.offset);
	return has;
}

bool Unit::get_key(const char* k, float& v) const
{
	using namespace unit_resource;
	Key key;
	bool has = unit_resource::get_key(m_resource, k, key);
	v = *(float*)(m_values + key.offset);
	return has;
}

bool Unit::get_key(const char* k, StringId32& v) const
{
	using namespace unit_resource;
	Key key;
	bool has = unit_resource::get_key(m_resource, k, key);
	v = *(StringId32*)(m_values + key.offset);
	return has;
}

bool Unit::get_key(const char* k, Vector3& v) const
{
	using namespace unit_resource;
	Key key;
	bool has = unit_resource::get_key(m_resource, k, key);
	v = *(Vector3*)(m_values + key.offset);
	return has;
}

void Unit::set_key(const char* k, bool v)
{
	using namespace unit_resource;
	Key key;
	unit_resource::get_key(m_resource, k, key);
	*(uint32_t*)(m_values + key.offset) = v;
}

void Unit::set_key(const char* k, float v)
{
	using namespace unit_resource;
	Key key;
	unit_resource::get_key(m_resource, k, key);
	*(float*)(m_values + key.offset) = v;
}

void Unit::set_key(const char* k, const char* v)
{
	using namespace unit_resource;
	Key key;
	unit_resource::get_key(m_resource, k, key);
	*(StringId32*)(m_values + key.offset) = string::murmur2_32(v, string::strlen(v));
}

void Unit::set_key(const char* k, const Vector3& v)
{
	using namespace unit_resource;
	Key key;
	unit_resource::get_key(m_resource, k, key);
	*(Vector3*)(m_values + key.offset) = v;
}

} // namespace crown