shader_resource_group.hpp

#pragma once
#include <vulkan-cpp/shader_resource.hpp>
#include <core/core.hpp>
#include <unordered_map>
// #include <wtr/watcher.hpp>
 
namespace atlas::vk {
    class shader_resource_group {
    public:
        shader_resource_group() = default;
        shader_resource_group(const VkDevice& p_device,
                              const ::vk::shader_resource_info& p_info);
        ~shader_resource_group() = default;
 
        [[nodiscard]] bool valid() const { return m_resource_valid; }
 
        void vertex_attributes(
          std::span<const ::vk::vertex_attribute> p_attributes);
 
        [[nodiscard]] std::vector<::vk::shader_handle> handles() const {
            return map_to_vector();
        }
 
        [[nodiscard]] std::span<const VkVertexInputAttributeDescription>
        vertex_attributes() const {
            return m_vertex_attributes;
        }
 
        [[nodiscard]] std::span<const VkVertexInputBindingDescription>
        vertex_bind_attributes() const {
            return m_vertex_binding_attributes;
        }
 
        void destroy();
 
        [[nodiscard]] bool reload_requested() const {
            return m_reload_requested;
        }
 
    private:
        [[nodiscard]] std::vector<::vk::shader_handle> map_to_vector() const;
 
        void create_module(std::span<char> p_blob,
                           const ::vk::shader_source& p_source);
 
        void create_module(std::span<uint32_t> p_blob,
                           const ::vk::shader_source& p_source);
 
        void reload_shader(const ::vk::shader_source& p_source);
 
    private:
        VkDevice m_device = nullptr;
        std::vector<VkVertexInputAttributeDescription> m_vertex_attributes;
        std::vector<VkVertexInputBindingDescription>
          m_vertex_binding_attributes;
        bool m_resource_valid = false;
        // shader module handles
        std::unordered_map<std::string, ::vk::shader_handle> m_modules;
        // ref<wtr::watch> m_watcher;
        bool m_reload_requested = false;
    };
};