ht_vkrenderpass.h

#pragma once

#include <ht_renderpass_base.h>
#include <ht_vulkan.h>
#include <ht_vktexture.h>
#include <ht_vkrendertarget.h>
#include <ht_vkrootlayout.h>
#include <ht_rootlayout.h>      //RootLayoutHandle
#include <ht_vkcommandpool.h>   //VKCommandPool

namespace Hatchit {

    namespace Graphics {

        namespace Vulkan {

            class HT_API VKRenderPass : public RenderPassBase
            {
            public:
                VKRenderPass();
                ~VKRenderPass();

                //Required function for RefCounted classes
                bool Initialize(const Resource::RenderPassHandle& handle, const VkDevice& device,
                    const VkDescriptorPool& descriptorPool, const VKSwapChain* swapchain);

                //Will this be sent the Objects that it needs to render?
                void VUpdate() override;

                bool VBuildCommandList(const ICommandPool* commandPool) override;

                const VkRenderPass& GetVkRenderPass() const;
                const VkCommandBuffer& GetVkCommandBuffer() const;
                const VKRootLayout* GetVKRootLayout() const;

                const std::vector<RenderTargetHandle>& GetOutputRenderTargets() const;

            private:
                //Input
                uint32_t m_firstInputTargetSetIndex;
                std::vector<VkDescriptorSet> m_inputTargetDescriptorSets;

                std::vector<RenderTargetHandle> m_renderTargets;

                const VKSwapChain* m_swapchain;

                bool setupRenderPass();
                bool setupAttachmentImages();
                bool setupFramebuffer();

                bool allocateCommandBuffer(const VKCommandPool* commandPool);
                //Mapping set index to maps of binding indicies and render targets
                bool setupDescriptorSets(std::map < uint32_t, std::map < uint32_t, VKRenderTarget* >> inputTargets);

                VkDevice m_device;
                VkDescriptorPool m_descriptorPool;

                VkRenderPass m_renderPass;
                VkCommandBuffer m_commandBuffer;
                
                Graphics::RootLayoutHandle m_rootLayoutHandle; //To keep this referenced
                VKRootLayout* m_rootLayout;
                
                //For instance data
                std::map<MeshHandle, UniformBlock_vk> m_instanceBlocks;

                std::vector<Image_vk> m_colorImages;
                Image_vk m_depthImage;

                VkFramebuffer m_framebuffer;
            };
        }
    }
}