/*
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:RTPass;
#ifndef CRAFTER_GRAPHICS_WINDOW_DOM
import std;
import :RenderPass;
import :Window;
import :Device;
import :PipelineRTVulkan;

export namespace Crafter {
    struct RTPass : RenderPass {
        PipelineRTVulkan* pipeline;

        RTPass(PipelineRTVulkan* p) : pipeline(p) {}

        void Record(VkCommandBuffer cmd, std::uint32_t /*frameIdx*/, Window& window) override {
            vkCmdBindPipeline(cmd, VK_PIPELINE_BIND_POINT_RAY_TRACING_KHR, pipeline->pipeline);
            Device::vkCmdTraceRaysKHR(cmd,
                &pipeline->raygenRegion,
                &pipeline->missRegion,
                &pipeline->hitRegion,
                &pipeline->callableRegion,
                window.width, window.height, 1);
        }
    };
}
#endif // !CRAFTER_GRAPHICS_WINDOW_DOM

#ifdef CRAFTER_GRAPHICS_WINDOW_DOM
import std;
import :RenderPass;
import :Window;
import :WebGPU;
import :PipelineRTWebGPU;
import :RenderingElement3D;

export namespace Crafter {
    // DOM-mode RT pass — dispatches the megakernel pipeline at frame Record
    // time. Picks up the current TLAS for the frame and the application's
    // raygen-side push data (typically empty in v1; pass via window.passes
    // wiring if needed later).
    struct RTPass : RenderPass {
        PipelineRTWebGPU* pipeline;
        // Optional per-dispatch push data forwarded after the standard
        // RTDispatchHeader. Null means "no extra data".
        const void*       pushPtr   = nullptr;
        std::uint32_t     pushBytes = 0;

        RTPass(PipelineRTWebGPU* p) : pipeline(p) {}

        void Record(WebGPUCommandEncoderRef /*cmd*/, std::uint32_t frameIdx, Window& window) override {
            const std::uint32_t gx = (window.width  + 7u) / 8u;
            const std::uint32_t gy = (window.height + 7u) / 8u;
            auto& tlas = RenderingElement3D::tlases[frameIdx];
            WebGPU::wgpuDispatchRT(
                pipeline->pipelineHandle,
                pushPtr,
                static_cast<std::int32_t>(pushBytes),
                tlas.buffer.handle,
                static_cast<std::int32_t>(tlas.builtInstanceCount),
                static_cast<std::int32_t>(gx),
                static_cast<std::int32_t>(gy));
        }
    };
}
#endif // CRAFTER_GRAPHICS_WINDOW_DOM