#ifndef EMULATOR_SRC_GRAPHICS_HOST_GPU_RENDERER_FRAMEBUFFERCACHE_H_
#define EMULATOR_SRC_GRAPHICS_HOST_GPU_RENDERER_FRAMEBUFFERCACHE_H_

#include "common/abi.h"
#include "common/assert.h"
#include "common/common.h"
#include "common/threads.h"
#include "graphics/host_gpu/graphicContext.h"
#include "graphics/host_gpu/renderer/renderTarget.h"
#include "graphics/host_gpu/vulkanCommon.h"

#include <vector>

namespace Libs::Graphics {

struct RenderColorInfo;
struct RenderDepthInfo;

static constexpr vk::ImageLayout RENDER_COLOR_IMAGE_LAYOUT = vk::ImageLayout::eGeneral;

struct VulkanFramebuffer {
	vk::RenderPass  render_pass                                = nullptr;
	uint64_t        render_pass_id                             = 0;
	vk::Framebuffer framebuffer                                = nullptr;
	uint32_t        samples                                    = 1;
	vk::ImageLayout color_layout[RENDER_COLOR_ATTACHMENTS_MAX] = {};
	vk::ImageLayout depth_layout = vk::ImageLayout::eDepthStencilAttachmentOptimal;
};

inline uint64_t render_pass_compat_id(uint32_t color_count, const vk::Format* color_formats,
                                      bool with_depth, vk::Format depth_format,
                                      vk::ImageLayout depth_layout, uint32_t samples) {
	uint64_t id  = 0xcbf29ce484222325ull;
	auto     mix = [&id](uint64_t v) {
		id ^= v;
		id *= 0x100000001b3ull;
	};

	mix(color_count);
	for (uint32_t i = 0; i < color_count; i++) {
		mix(static_cast<uint32_t>(color_formats[i]));
	}
	mix(with_depth ? 1u : 0u);
	mix(static_cast<uint32_t>(depth_format));
	mix(static_cast<uint32_t>(depth_layout));
	mix(samples);

	return id;
}

class FramebufferCache {
public:
	explicit FramebufferCache(GraphicContext& graphics): m_graphics(graphics) {
		EXIT_NOT_IMPLEMENTED(!Common::Thread::IsMainThread());
	}
	~FramebufferCache() { KYTY_NOT_IMPLEMENTED; }
	KYTY_CLASS_NO_COPY(FramebufferCache);

	VulkanFramebuffer* CreateFramebuffer(RenderColorInfo* colors, uint32_t color_count,
	                                     RenderDepthInfo& depth);
	void               FreeFramebufferByColor(VulkanImage& image);
	void               FreeFramebufferByDepth(DepthStencilVulkanImage& image);

private:
	struct Framebuffer {
		VulkanFramebuffer* framebuffer                                      = nullptr;
		uint64_t           image_id[RENDER_COLOR_ATTACHMENTS_MAX]           = {};
		vk::ImageView      color_view[RENDER_COLOR_ATTACHMENTS_MAX]         = {};
		uint64_t           depth_id                                         = 0;
		vk::ImageView      depth_view                                       = nullptr;
		bool               color_clear_enable[RENDER_COLOR_ATTACHMENTS_MAX] = {};
		vk::ImageLayout    color_layout[RENDER_COLOR_ATTACHMENTS_MAX]       = {};
		bool               depth_clear_enable                               = false;
		bool               stencil_clear_enable                             = false;
		bool               depth_read_only                                  = false;
	};

	GraphicContext&          m_graphics;
	Common::Mutex            m_mutex;
	std::vector<Framebuffer> m_framebuffers;
};

} // namespace Libs::Graphics

#endif // EMULATOR_SRC_GRAPHICS_HOST_GPU_RENDERER_FRAMEBUFFERCACHE_H_