#ifndef __SCOP_RENDERER_MATERIAL__
#define __SCOP_RENDERER_MATERIAL__

#include <memory>

#include <Core/EventBus.h>
#include <Renderer/Image.h>
#include <Renderer/Buffer.h>
#include <Renderer/Descriptor.h>

namespace Scop
{
	struct MaterialTextures
	{
		std::shared_ptr<Texture> albedo;
	};

	struct MaterialData
	{
		float dissolve_texture_factor = 1.0f;
		float dissolve_black_white_colors_factor = 1.0f;
		float dissolve_normals_colors_factor = 0.0f;
		std::uint8_t __padding[4];
	};

	class Material
	{
		friend class Model;

		public:
			Material() { m_data_buffer.Init(sizeof(m_data)); SetupEventListener(); }
			Material(const MaterialTextures& textures) : m_textures(textures) { m_data_buffer.Init(sizeof(m_data)); SetupEventListener(); }

			inline void SetMaterialData(const MaterialData& data) noexcept { m_data = data; }

			~Material() { m_data_buffer.Destroy(); }

		private:
			[[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->GetSet(frame_index); }

			inline void SetupEventListener()
			{
				std::function<void(const EventBase&)> functor = [this](const EventBase& event)
				{
					if(event.What() == Event::FrameBeginEventCode)
						m_have_been_updated_this_frame = false;
				};
				EventBus::RegisterListener({ functor, "__ScopMaterial" + std::to_string(reinterpret_cast<std::uintptr_t>(this)) });
			}

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

			inline void Bind(std::size_t frame_index, VkCommandBuffer cmd)
			{
				if(m_have_been_updated_this_frame)
					return;
				p_set->SetImage(frame_index, 0, *m_textures.albedo);
				p_set->SetUniformBuffer(frame_index, 1, m_data_buffer.Get(frame_index));
				p_set->Update(frame_index, cmd);

				static CPUBuffer buffer(sizeof(MaterialData));
				std::memcpy(buffer.GetData(), &m_data, buffer.GetSize());
				m_data_buffer.SetData(buffer, frame_index);

				m_have_been_updated_this_frame = true;
			}

		private:
			UniformBuffer m_data_buffer;
			MaterialTextures m_textures;
			MaterialData m_data;
			std::shared_ptr<DescriptorSet> p_set;
			bool m_have_been_updated_this_frame = false;
	};
}

#endif