#ifndef PKE_COMPONENTS_HPP
#define PKE_COMPONENTS_HPP

#include "pk.h"
#include "physics.hpp"
#include "plugin-types.hpp"

#include <BulletDynamics/Dynamics/btRigidBody.h>
#include <LinearMath/btDefaultMotionState.h>
#include <vulkan/vulkan_core.h>

const uint64_t ECS_UNSET_VAL = 0xFFFFFFFFFFFFFFFF;
const uint32_t ECS_UNSET_VAL_32 = 0xFFFFFFFF;

struct EntityHandle : public pk_handle { };
struct GrBindsHandle : public pk_handle { };
struct InstanceHandle : public pk_handle { };

constexpr EntityHandle EntityHandle_MAX = EntityHandle{ pk_handle_MAX_constexpr };
constexpr GrBindsHandle GrBindsHandle_MAX = GrBindsHandle{ pk_handle_MAX_constexpr };
constexpr InstanceHandle InstanceHandle_MAX = InstanceHandle{ pk_handle_MAX_constexpr };

struct Entity_Base {
	EntityHandle handle = EntityHandle_MAX;
	EntityHandle parentHandle = EntityHandle_MAX;
	bool isMarkedForRemoval = false;
};

struct BufferBindingDetails {
	VkBuffer buffer = VK_NULL_HANDLE;
	uint32_t firstBinding = 0;
	uint32_t bindingCount = 0;
	VkDeviceSize offsets[1] = {0};
};
struct CompGrBinds {
	EntityHandle entHandle = EntityHandle_MAX;
	GrBindsHandle grBindsHandle = GrBindsHandle_MAX;
	VkPipelineLayout vkPipelineLayout = VK_NULL_HANDLE;
	VkPipeline graphicsPipeline = VK_NULL_HANDLE;
	VkDescriptorSet *vkDescriptorSets = nullptr;
	BufferBindingDetails vertexBD;
	BufferBindingDetails normalsBD;
	BufferBindingDetails uvBD;
	BufferBindingDetails indexBD;
	BufferBindingDetails physVertBD;
	BufferBindingDetails physNormBD;
	BufferBindingDetails physUvBD;
	BufferBindingDetails physIndxBD;
	VkBuffer instanceBuffer = VK_NULL_HANDLE;
	uint32_t indexCount;
	uint32_t instanceFirstBinding = 0;
	uint32_t instanceBindingCount = 0;
	uint32_t instanceCounter = 0;
	uint32_t instanceBufferMaxCount = 0;
	VkDeviceSize instanceOffsets = 0;
	PkeCallback collisionCallback{};
};

struct InstPos {
	btTransform posRot;
	btVector3 scale;
	btScalar mass;
};
struct InstBt {
	btDefaultMotionState *motionState = nullptr;
	btRigidBody *rigidBody = nullptr;
};
struct CompInstance {
	EntityHandle entHandle = EntityHandle_MAX;
	GrBindsHandle grBindsHandle = GrBindsHandle_MAX;
	InstanceHandle instanceHandle = InstanceHandle_MAX;
	uint32_t index = ECS_UNSET_VAL_32;
	PhysicsCollision physicsLayer = PhysicsCollision{1};
	PhysicsCollision physicsMask = PhysicsCollision{1};
	InstBt bt{};
	PkeCallback collisionCallback{};
	bool isNeedingUpdated = false;
};

#endif /* PKE_COMPONENTS_HPP */