/*
Crafter®.Graphics
Copyright (C) 2026 Catcrafts®
catcrafts.net

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License version 3.0 as published by the Free Software Foundation;

This library is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
*/

module;
#ifdef CRAFTER_GRAPHICS_VULKAN
#include "vulkan/vulkan.h"
#endif
export module Crafter.Graphics:ShaderBindingTableVulkan;
#ifdef CRAFTER_GRAPHICS_VULKAN
import std;
import :VulkanDevice;
import :VulkanBuffer;
import :ShaderVulkan;
import :Types;

export namespace Crafter {
    class ShaderBindingTableVulkan {
        public:
		std::vector<VkPipelineShaderStageCreateInfo> shaderStages;
        void Init(std::span<VulkanShader> shaders) {
            shaderStages.reserve(shaders.size());
            for(const VulkanShader& shader: shaders) {
                shaderStages.emplace_back(VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO, nullptr, 0, shader.stage, shader.shader, shader.entrypoint.c_str(), nullptr);
            }
        }
    };

    template <typename Shaders>
    class ShaderBindingTableVulkanConst {
        public:
		inline static std::array<VkPipelineShaderStageCreateInfo, std::tuple_size_v<Shaders>> shaderStages;
        static void Init() {
            AddAllToSBT(std::make_index_sequence<std::tuple_size_v<Shaders>>{});
        }
        private:
        template<std::size_t index>
        static void AddToSBT() {
            shaderStages[index] = {
                .sType  = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
                .stage  = std::tuple_element_t<index, Shaders>::_stage,
                .module = std::tuple_element_t<index, Shaders>::shader,
                .pName  = std::tuple_element_t<index, Shaders>::_entrypoint.value
            };
        }
        template<std::size_t... Is>
        static void AddAllToSBT(std::index_sequence<Is...>) {
            (AddToSBT<Is>(), ...);
        }   
    };
}

#endif