#pragma once

#include <vulkan/vulkan_core.h>
#include <vector>
#include <map>

class Buffer;

class Image;

enum class DescriptorSet {
	WORLD,
	SIMULATION,
	MESH
};

class DescriptorPool {
public:
	DescriptorPool();
	~DescriptorPool();

	void bindBuffer(const Buffer &buffer, VkDescriptorType type, DescriptorSet set, uint32_t binding);
	void bindImage(const Image &image, VkDescriptorType type, DescriptorSet set, uint32_t binding);
	std::map<DescriptorSet, VkDescriptorSet> sets;
	std::map<DescriptorSet, VkDescriptorSetLayout> layouts;
	VkDescriptorPool handle = VK_NULL_HANDLE;
private:

	void createLayout(DescriptorSet set, const std::vector<VkDescriptorSetLayoutBinding> &bindings);
};