// Copyright (C) 2013 Jérôme Leclercq
// This file is part of the "Nazara Engine - Graphics module"
// For conditions of distribution and use, see copyright notice in Config.hpp

#include <Nazara/Graphics/Light.hpp>
#include <Nazara/Core/Error.hpp>
#include <Nazara/Graphics/AbstractRenderQueue.hpp>
#include <Nazara/Math/Basic.hpp>
#include <Nazara/Math/Sphere.hpp>
#include <Nazara/Renderer/Renderer.hpp>
#include <Nazara/Renderer/Shader.hpp>
#include <cstring>
#include <Nazara/Graphics/Debug.hpp>

///TODO: Utilisation des UBOs

NzLight::NzLight(nzLightType type) :
m_type(type),
m_ambientColor((type == nzLightType_Directional) ? NzColor(50, 50, 50) : NzColor::Black),
m_diffuseColor(NzColor::White),
m_specularColor(NzColor::White),
m_boundingBoxUpdated(false),
m_attenuation(0.9f),
m_innerAngle(15.f),
m_outerAngle(45.f),
m_radius(500.f)
{
}

NzLight::NzLight(const NzLight& light) :
NzSceneNode(light)
{
	std::memcpy(this, &light, sizeof(NzLight)); // Aussi simple que ça
}

void NzLight::AddToRenderQueue(NzAbstractRenderQueue* renderQueue) const
{
	renderQueue->AddLight(this);
}

void NzLight::Apply(const NzShader* shader, unsigned int lightUnit) const
{
	/*
	struct Light
	{
		int type;
		vec4 ambient;
		vec4 diffuse;
		vec4 specular;

		vec4 parameters1;
		vec4 parameters2;
		vec2 parameters3;
	};

	Directional
	-P1: vec3 direction

	Point
	-P1: vec3 position + float attenuation
	-P2: float invRadius

	Spot
	-P1: vec3 position + float attenuation
	-P2: vec3 direction + float invRadius
	-P3: float cosInnerAngle + float cosOuterAngle
	*/

	int typeLocation = shader->GetUniformLocation("Lights[0].type");
	int ambientLocation = shader->GetUniformLocation("Lights[0].ambient");
	int diffuseLocation = shader->GetUniformLocation("Lights[0].diffuse");
	int specularLocation = shader->GetUniformLocation("Lights[0].specular");
	int parameters1Location = shader->GetUniformLocation("Lights[0].parameters1");
	int parameters2Location = shader->GetUniformLocation("Lights[0].parameters2");
	int parameters3Location = shader->GetUniformLocation("Lights[0].parameters3");

	if (lightUnit > 0)
	{
		int type2Location = shader->GetUniformLocation("Lights[1].type");
		int offset = lightUnit * (type2Location - typeLocation); // type2Location - typeLocation donne la taille de la structure

		// On applique cet offset
		typeLocation += offset;
		ambientLocation += offset;
		diffuseLocation += offset;
		specularLocation += offset;
		parameters1Location += offset;
		parameters2Location += offset;
		parameters3Location += offset;
	}

	shader->SendInteger(typeLocation, m_type);
	shader->SendColor(ambientLocation, m_ambientColor);
	shader->SendColor(diffuseLocation, m_diffuseColor);
	shader->SendColor(specularLocation, m_specularColor);

	if (!m_derivedUpdated)
		UpdateDerived();

	switch (m_type)
	{
		case nzLightType_Directional:
			shader->SendVector(parameters1Location, NzVector4f(m_derivedRotation * NzVector3f::Forward()));
			break;

		case nzLightType_Point:
			shader->SendVector(parameters1Location, NzVector4f(m_derivedPosition, m_attenuation));
			shader->SendVector(parameters2Location, NzVector4f(1.f/m_radius, 0.f, 0.f, 0.f));
			break;

		case nzLightType_Spot:
			shader->SendVector(parameters1Location, NzVector4f(m_derivedPosition, m_attenuation));
			shader->SendVector(parameters2Location, NzVector4f(m_derivedRotation * NzVector3f::Forward(), 1.f/m_radius));
			shader->SendVector(parameters3Location, NzVector2f(std::cos(NzDegreeToRadian(m_innerAngle)), std::cos(NzDegreeToRadian(m_outerAngle))));
			break;
	}
}

const NzBoundingBoxf& NzLight::GetBoundingBox() const
{
	if (!m_boundingBoxUpdated)
		UpdateBoundingBox();

	return m_boundingBox;
}

NzColor NzLight::GetAmbientColor() const
{
	return m_ambientColor;
}

float NzLight::GetAttenuation() const
{
	return m_attenuation;
}

NzColor NzLight::GetDiffuseColor() const
{
	return m_diffuseColor;
}

float NzLight::GetInnerAngle() const
{
	return m_innerAngle;
}

nzLightType NzLight::GetLightType() const
{
	return m_type;
}

float NzLight::GetOuterAngle() const
{
	return m_outerAngle;
}

float NzLight::GetRadius() const
{
	return m_radius;
}

nzSceneNodeType NzLight::GetSceneNodeType() const
{
	return nzSceneNodeType_Light;
}

NzColor NzLight::GetSpecularColor() const
{
	return m_specularColor;
}

void NzLight::SetAmbientColor(const NzColor& ambient)
{
	m_ambientColor = ambient;
}

void NzLight::SetAttenuation(float attenuation)
{
	m_attenuation = attenuation;
}

void NzLight::SetDiffuseColor(const NzColor& diffuse)
{
	m_diffuseColor = diffuse;
}

void NzLight::SetInnerAngle(float innerAngle)
{
	m_innerAngle = innerAngle;
}

void NzLight::SetOuterAngle(float outerAngle)
{
	m_outerAngle = outerAngle;

	m_boundingBox.MakeNull();
	m_boundingBoxUpdated = false;
}

void NzLight::SetRadius(float radius)
{
	m_radius = radius;

	m_boundingBox.MakeNull();
	m_boundingBoxUpdated = false;
}

void NzLight::SetSpecularColor(const NzColor& specular)
{
	m_specularColor = specular;
}

NzLight& NzLight::operator=(const NzLight& light)
{
	std::memcpy(this, &light, sizeof(NzLight));

	return *this;
}

void NzLight::Invalidate()
{
	NzSceneNode::Invalidate();

	m_boundingBoxUpdated = false;
}

void NzLight::Register()
{
}

void NzLight::Unregister()
{
}

void NzLight::UpdateBoundingBox() const
{
	if (m_boundingBox.IsNull())
	{
		switch (m_type)
		{
			case nzLightType_Directional:
				m_boundingBox.MakeInfinite();
				m_boundingBoxUpdated = true;
				return; // Rien d'autre à faire

			case nzLightType_Point:
			{
				NzVector3f radius(m_radius);
				m_boundingBox.Set(-radius, radius);
				break;
			}

			case nzLightType_Spot:
			{
				// On forme un cube sur l'origine
				NzCubef cube(NzVector3f::Zero());

				// On calcule le reste des points
				float height = m_radius;
				NzVector3f base(NzVector3f::Forward()*height);

				// Il nous faut maintenant le rayon du cercle projeté à cette distance
				// Tangente = Opposé/Adjaçent <=> Opposé = Adjaçent*Tangente
				float radius = height*std::tan(NzDegreeToRadian(m_outerAngle));
				NzVector3f lExtend = NzVector3f::Left()*radius;
				NzVector3f uExtend = NzVector3f::Up()*radius;

				// Et on ajoute ensuite les quatres extrêmités de la pyramide
				cube.ExtendTo(base + lExtend + uExtend);
				cube.ExtendTo(base + lExtend - uExtend);
				cube.ExtendTo(base - lExtend + uExtend);
				cube.ExtendTo(base - lExtend - uExtend);

				m_boundingBox.Set(cube);
				break;
			}
		}
	}

	switch (m_type)
	{
		case nzLightType_Directional:
			break;

		case nzLightType_Point:
			if (!m_derivedUpdated)
				UpdateDerived();

			m_boundingBox.Update(NzMatrix4f::Translate(m_derivedPosition)); // Notre BoundingBox ne changera que selon la position
			break;

		case nzLightType_Spot:
			if (!m_transformMatrixUpdated)
				UpdateTransformMatrix();

			m_boundingBox.Update(m_transformMatrix);
			break;
	}

	m_boundingBoxUpdated = true;
}

bool NzLight::VisibilityTest(const NzFrustumf& frustum)
{
	switch (m_type)
	{
		case nzLightType_Directional:
			return true; // Toujours visible

		case nzLightType_Point:
			if (!m_derivedUpdated)
				UpdateDerived();

			// Un test sphérique est bien plus rapide et précis que celui de la bounding box
			return frustum.Contains(NzSpheref(m_derivedPosition, m_radius));

		case nzLightType_Spot:
			if (!m_boundingBoxUpdated)
				UpdateBoundingBox();

			return frustum.Contains(m_boundingBox);
	}

	NazaraError("Invalid light type (0x" + NzString::Number(m_type, 16) + ')');
	return false;
}