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

#pragma once

#ifndef NAZARA_NODE_HPP
#define NAZARA_NODE_HPP

#include <Nazara/Prerequesites.hpp>
#include <Nazara/Math/Matrix4.hpp>
#include <Nazara/Math/Quaternion.hpp>
#include <Nazara/Math/Vector3.hpp>
#include <Nazara/Utility/Enums.hpp>
#include <unordered_map>
#include <vector>

class NAZARA_API NzNode
{
	public:
		class Listener;

		NzNode();
		NzNode(const NzNode& node);
		virtual ~NzNode();

		void AddListener(Listener* listener, void* userdata = nullptr) const;

		void EnsureDerivedUpdate() const;
		void EnsureTransformMatrixUpdate() const;

		virtual NzVector3f GetBackward() const;
		const std::vector<NzNode*>& GetChilds() const;
		virtual NzVector3f GetDown() const;
		virtual NzVector3f GetForward() const;
		bool GetInheritPosition() const;
		bool GetInheritRotation() const;
		bool GetInheritScale() const;
		NzVector3f GetInitialPosition() const;
		NzQuaternionf GetInitialRotation() const;
		NzVector3f GetInitialScale() const;
		virtual NzVector3f GetLeft() const;
		virtual nzNodeType GetNodeType() const;
		const NzNode* GetParent() const;
		NzVector3f GetPosition(nzCoordSys coordSys = nzCoordSys_Global) const;
		virtual NzVector3f GetRight() const;
		NzQuaternionf GetRotation(nzCoordSys coordSys = nzCoordSys_Global) const;
		NzVector3f GetScale(nzCoordSys coordSys = nzCoordSys_Global) const;
		const NzMatrix4f& GetTransformMatrix() const;
		virtual NzVector3f GetUp() const;

		bool HasChilds() const;

		NzNode& Interpolate(const NzNode& nodeA, const NzNode& nodeB, float interpolation, nzCoordSys coordSys = nzCoordSys_Global);

		NzNode& Move(const NzVector3f& movement, nzCoordSys coordSys = nzCoordSys_Local);
		NzNode& Move(float movementX, float movementY, float movementZ = 0.f, nzCoordSys coordSys = nzCoordSys_Local);

		void RemoveListener(Listener* listener) const;
		NzNode& Rotate(const NzQuaternionf& rotation, nzCoordSys coordSys = nzCoordSys_Local);

		NzNode& Scale(const NzVector3f& scale);
		NzNode& Scale(float scale);
		NzNode& Scale(float scaleX, float scaleY, float scaleZ = 1.f);

		void SetInheritRotation(bool inheritRotation);
		void SetInheritScale(bool inheritScale);
		void SetInheritPosition(bool inheritPosition);
		void SetInitialRotation(const NzQuaternionf& quat);
		void SetInitialScale(const NzVector3f& scale);
		void SetInitialScale(float scale);
		void SetInitialScale(float scaleX, float scaleY, float scaleZ = 1.f);
		void SetInitialPosition(const NzVector3f& translation);
		void SetInitialPosition(float translationX, float translationXY, float translationZ = 0.f);
		void SetParent(const NzNode* node = nullptr, bool keepDerived = false);
		void SetParent(const NzNode& node, bool keepDerived = false);
		void SetPosition(const NzVector3f& translation, nzCoordSys coordSys = nzCoordSys_Local);
		void SetPosition(float translationX, float translationY, float translationZ = 0.f, nzCoordSys coordSys = nzCoordSys_Local);
		void SetRotation(const NzQuaternionf& quat, nzCoordSys coordSys = nzCoordSys_Local);
		void SetScale(const NzVector3f& scale, nzCoordSys coordSys = nzCoordSys_Local);
		void SetScale(float scale, nzCoordSys coordSys = nzCoordSys_Local);
		void SetScale(float scaleX, float scaleY, float scaleZ = 1.f, nzCoordSys coordSys = nzCoordSys_Local);
		void SetTransformMatrix(const NzMatrix4f& matrix);

		// Local -> global
		NzVector3f ToGlobalPosition(const NzVector3f& localPosition) const;
		NzQuaternionf ToGlobalRotation(const NzQuaternionf& localRotation) const;
		NzVector3f ToGlobalScale(const NzVector3f& localScale) const;

		// Global -> local
		NzVector3f ToLocalPosition(const NzVector3f& globalPosition) const;
		NzQuaternionf ToLocalRotation(const NzQuaternionf& globalRotation) const;
		NzVector3f ToLocalScale(const NzVector3f& globalScale) const;

		NzNode& operator=(const NzNode& node);

		class NAZARA_API Listener
		{
			public:
				Listener() = default;
				virtual ~Listener();

				virtual bool OnNodeInvalidated(const NzNode* node, void* userdata);
				virtual bool OnNodeParented(const NzNode* node, const NzNode* parent, void* userdata);
				virtual void OnNodeReleased(const NzNode* node, void* userdata);
		};

	protected:
		void AddChild(NzNode* node) const;
		virtual void InvalidateNode();
		virtual void OnParenting(const NzNode* parent);
		void RemoveChild(NzNode* node) const;
		virtual void UpdateDerived() const;
		virtual void UpdateTransformMatrix() const;

		mutable std::vector<NzNode*> m_childs;
		mutable NzMatrix4f m_transformMatrix;
		mutable NzQuaternionf m_derivedRotation;
		NzQuaternionf m_initialRotation;
		NzQuaternionf m_rotation;
		mutable NzVector3f m_derivedPosition;
		mutable NzVector3f m_derivedScale;
		NzVector3f m_initialPosition;
		NzVector3f m_initialScale;
		NzVector3f m_position;
		NzVector3f m_scale;
		const NzNode* m_parent;
		mutable bool m_derivedUpdated;
		bool m_inheritPosition;
		bool m_inheritRotation;
		bool m_inheritScale;
		mutable bool m_transformMatrixUpdated;

	private:
		void NotifyInvalidation();
		void NotifyParented(const NzNode* parent);

	private:
		mutable std::unordered_map<Listener*, void*> m_listeners;
		bool m_listenersLocked;

};

#endif // NAZARA_NODE_HPP