#ifndef __MLX_DRAWABLE__
#define __MLX_DRAWABLE__

#include <Graphics/Enums.h>

namespace mlx
{
	class Drawable
	{
		friend class Render2DPass;

		public:
			inline Drawable(DrawableType type) : m_type(type) {}

			inline void SetColor(Vec4f color) noexcept { m_color = color; }
			inline void SetPosition(Vec2f position) noexcept { m_position = position; }

			inline virtual void Update([[maybe_unused]] VkCommandBuffer cmd) {}

			[[nodiscard]] MLX_FORCEINLINE const Vec4f& GetColor() const noexcept { return m_color; }
			[[nodiscard]] MLX_FORCEINLINE const Vec2f& GetPosition() const noexcept { return m_position; }
			[[nodiscard]] MLX_FORCEINLINE std::shared_ptr<Mesh> GetMesh() const { return p_mesh; }
			[[nodiscard]] MLX_FORCEINLINE DrawableType GetType() const noexcept { return m_type; }

			inline virtual ~Drawable() { if(p_set) p_set->ReturnDescriptorSetToPool(); }

		protected:
			[[nodiscard]] inline bool IsSetInit() const noexcept { return p_set && p_set->IsInit(); }
			[[nodiscard]] inline VkDescriptorSet GetSet(std::size_t frame_index) const noexcept { return p_set ? p_set->GetSet(frame_index) : VK_NULL_HANDLE; }

			inline void UpdateDescriptorSet(std::shared_ptr<DescriptorSet> set)
			{
				p_set = RenderCore::Get().GetDescriptorPoolManager().GetAvailablePool().RequestDescriptorSet(set->GetShaderLayout(), set->GetShaderType());
			}

			virtual void Bind(std::size_t frame_index, VkCommandBuffer cmd) = 0;

		protected:
			std::shared_ptr<DescriptorSet> p_set;
			std::shared_ptr<Mesh> p_mesh;
			Vec4f m_color = Vec4f{ 1.0f, 1.0f, 1.0f, 1.0f };
			Vec2f m_position = Vec2f{ 0.0f, 0.0f };
			DrawableType m_type;
	};
}

#endif