ht_vkmaterial.h

#pragma once

#include <ht_material.h>
#include <ht_material_base.h>
#include <ht_vkrenderpass.h>
#include <ht_vkpipeline.h>
#include <ht_vktexture.h>
#include <ht_material_resource.h>
#include <ht_refcounted.h>

namespace Hatchit {

    namespace Graphics {

        namespace Vulkan {

            class HT_API VKMaterial :  public MaterialBase
            {
            public:
                VKMaterial();
                ~VKMaterial();

                //Required function for RefCounted class
                bool Initialize(Resource::MaterialHandle handle, const VkDevice& device, const VkDescriptorPool& descriptorPool);

                bool VBindTexture(std::string name, TextureHandle texture)      override;
                bool VUnbindTexture(std::string name, TextureHandle texture)    override;

                bool VUpdate()                                              override;

                const void BindMaterial(const VkCommandBuffer& commandBuffer, const VkPipelineLayout& pipelineLayout) const;
                
                PipelineHandle const VGetPipeline() const override;
                const VKPipeline* GetVKPipeline() const;

            private:
                const VkDevice* m_device;
                const VkDescriptorPool* m_descriptorPool;

                bool setupDescriptorSet();

                PipelineHandle m_pipelineHandle;
                VKPipeline* m_pipeline;

                std::vector<VkDescriptorSetLayout> m_descriptorSetLayouts;
                std::vector<VkDescriptorSetLayout> m_materialLayouts;
                std::vector<VkDescriptorSet> m_materialSets;

                std::map<std::string, Graphics::TextureHandle> m_textureHandles;

                UniformBlock_vk m_uniformVSBuffer;
                //UniformBlock m_uniformFSBuffer;
                std::vector<UniformBlock_vk> m_fragmentTextures;
                
                std::vector<LayoutLocation> m_textureLocations;
                std::vector<VKTexture*> m_textures;
            };
        }
    }
}