// Copyright (C) 2023 Jérôme "Lynix" Leclercq (lynix680@gmail.com)
// This file is part of the "Nazara Engine - JoltPhysics3D module"
// For conditions of distribution and use, see copyright notice in Config.hpp

#pragma once

#ifndef NAZARA_JOLTPHYSICS3D_JOLTPHYSWORLD3D_HPP
#define NAZARA_JOLTPHYSICS3D_JOLTPHYSWORLD3D_HPP

#include <NazaraUtils/Prerequisites.hpp>
#include <Nazara/Core/Time.hpp>
#include <Nazara/JoltPhysics3D/Config.hpp>
#include <Nazara/Math/Box.hpp>
#include <Nazara/Math/Vector3.hpp>
#include <NazaraUtils/FunctionRef.hpp>
#include <NazaraUtils/MovablePtr.hpp>
#include <atomic>
#include <optional>
#include <vector>

namespace JPH
{
	class BodyID;
	class PhysicsSystem;
	class Shape;
}

namespace Nz
{
	class JoltAbstractBody;
	class JoltCharacter;
	class JoltCharacterImpl;
	class JoltCollider3D;
	class JoltPhysicsStepListener;
	class JoltRigidBody3D;

	class NAZARA_JOLTPHYSICS3D_API JoltPhysWorld3D
	{
		friend JoltCharacter;
		friend JoltRigidBody3D;

		public:
			struct PointCollisionInfo;
			struct RaycastHit;
			struct ShapeCollisionInfo;

			JoltPhysWorld3D();
			JoltPhysWorld3D(const JoltPhysWorld3D&) = delete;
			JoltPhysWorld3D(JoltPhysWorld3D&& ph) = delete;
			~JoltPhysWorld3D();

			bool CollisionQuery(const Vector3f& point, const FunctionRef<std::optional<float>(const PointCollisionInfo& collisionInfo)>& callback);
			bool CollisionQuery(const JoltCollider3D& collider, const Matrix4f& colliderTransform, const FunctionRef<std::optional<float>(const ShapeCollisionInfo& hitInfo)>& callback);
			bool CollisionQuery(const JoltCollider3D& collider, const Matrix4f& colliderTransform, const Vector3f& colliderScale, const FunctionRef<std::optional<float>(const ShapeCollisionInfo& hitInfo)>& callback);

			UInt32 GetActiveBodyCount() const;
			Vector3f GetGravity() const;
			std::size_t GetMaxStepCount() const;
			JPH::PhysicsSystem* GetPhysicsSystem();
			Time GetStepSize() const;

			inline bool IsBodyActive(UInt32 bodyIndex) const;
			inline bool IsBodyRegistered(UInt32 bodyIndex) const;

			bool RaycastQuery(const Vector3f& from, const Vector3f& to, const FunctionRef<std::optional<float>(const RaycastHit& hitInfo)>& callback);
			bool RaycastQueryFirst(const Vector3f& from, const Vector3f& to, const FunctionRef<void(const RaycastHit& hitInfo)>& callback);

			void RefreshBodies();

			inline void RegisterStepListener(JoltPhysicsStepListener* stepListener);

			void SetGravity(const Vector3f& gravity);
			void SetMaxStepCount(std::size_t maxStepCount);
			void SetStepSize(Time stepSize);

			bool Step(Time timestep);

			inline void UnregisterStepListener(JoltPhysicsStepListener* stepListener);

			JoltPhysWorld3D& operator=(const JoltPhysWorld3D&) = delete;
			JoltPhysWorld3D& operator=(JoltPhysWorld3D&&) = delete;

			struct PointCollisionInfo
			{
				JoltAbstractBody* hitBody = nullptr;
			};

			struct RaycastHit
			{
				float fraction;
				JoltAbstractBody* hitBody = nullptr;
				Vector3f hitNormal;
				Vector3f hitPosition;
			};

			struct ShapeCollisionInfo
			{
				JoltAbstractBody* hitBody = nullptr;
				Vector3f collisionPosition1;
				Vector3f collisionPosition2;
				Vector3f penetrationAxis;
				float penetrationDepth;
			};

		private:
			class BodyActivationListener;
			friend BodyActivationListener;

			class StepListener;
			friend StepListener;

			struct JoltWorld;

			std::shared_ptr<JoltCharacterImpl> GetDefaultCharacterImpl();
			const JPH::Shape* GetNullShape() const;

			void OnPreStep(float deltatime);

			void RegisterBody(const JPH::BodyID& bodyID, bool activate, bool removeFromDeactivationList);

			void UnregisterBody(const JPH::BodyID& bodyID, bool destroy, bool removeFromRegisterList);

			std::size_t m_maxStepCount;
			std::shared_ptr<JoltCharacterImpl> m_defaultCharacterImpl;
			std::unique_ptr<std::atomic_uint64_t[]> m_activeBodies;
			std::unique_ptr<std::uint64_t[]> m_registeredBodies;
			std::unique_ptr<JoltWorld> m_world;
			std::vector<JoltPhysicsStepListener*> m_stepListeners;
			Vector3f m_gravity;
			Time m_stepSize;
			Time m_timestepAccumulator;
	};
}

#include <Nazara/JoltPhysics3D/JoltPhysWorld3D.inl>

#endif // NAZARA_JOLTPHYSICS3D_JOLTPHYSWORLD3D_HPP