/*
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;
#ifndef CRAFTER_GRAPHICS_WINDOW_DOM
#include "vulkan/vulkan.h"
#endif // !CRAFTER_GRAPHICS_WINDOW_DOM
export module Crafter.Graphics:ShaderBindingTableVulkan;
#ifndef CRAFTER_GRAPHICS_WINDOW_DOM
import std;
import :Device;
import :VulkanBuffer;
import :ShaderVulkan;
import :Types;

export namespace Crafter {
    class ShaderBindingTableVulkan {
        public:
		std::vector<VkPipelineShaderStageCreateInfo> shaderStages;
        // NVIDIA descriptor-heap AS-read workaround (issue #15 / #7): true when
        // any stage in this table reads an acceleration structure and was
        // rewritten to fetch the TLAS address from a push constant, with the
        // byte offset that stage expects it at. PipelineRTVulkan copies these so
        // RTPass can push the address without consulting a clobber-prone global.
        // Both inert (false/0) on every other driver.
        bool workaroundNeedsTlas = false;
        std::uint32_t workaroundTlasPushOffset = 0;
        void Init(const std::span<const 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(), shader.specilizationInfo);
                if (shader.patchedAS) {
                    workaroundNeedsTlas = true;
                    workaroundTlasPushOffset = shader.tlasPushOffset;
                }
            }
        }
    };
}
#endif // !CRAFTER_GRAPHICS_WINDOW_DOM
