#pragma once

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

class Buffer;

enum class DescriptorSet {
	WORLD,
	SIMULATION,
	MESH

};

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

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

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