foray_defaultraytracingstage.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_core_declares.hpp"
#include "../rtpipe/foray_rtpipeline.hpp"
#include "../scene/foray_scene_declares.hpp"
#include "../util/foray_pipelinelayout.hpp"
#include "foray_raytracingshared.hpp"
#include "foray_renderstage.hpp"
 
namespace foray::stages {
    class DefaultRaytracingStageBase : public RenderStage
    {
      public:
        void Init(core::Context* context, scene::Scene* scene, core::CombinedImageSampler* envMap = nullptr, core::ManagedImage* noiseImage = nullptr);
 
        virtual void RecordFrame(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo) override;
        virtual void Resize(const VkExtent2D& extent) override;
 
        virtual void Destroy() override;
 
        inline static constexpr std::string_view OutputName = "Rt.Output";
 
        inline core::ManagedImage* GetRtOutput() { return &mOutput; }
 
      protected:
        virtual void CreateOutputImages();
 
        virtual void CreatePipelineLayout();
        virtual void ApiCreateRtPipeline() = 0;
        virtual void ApiDestroyRtPipeline() = 0;
 
        virtual void ApiCustomObjectsCreate() {}
        virtual void ApiCustomObjectsDestroy() {}
 
        virtual void CreateOrUpdateDescriptors();
        virtual void DestroyDescriptors();
 
        virtual void ReloadShaders() override;
 
        virtual void RecordFramePrepare(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo);
        virtual void RecordFrameBind(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo);
        virtual void RecordFrameTraceRays(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo);
 
        scene::Scene* mScene = nullptr;
        core::CombinedImageSampler* mEnvironmentMap = nullptr;
        core::ManagedImage* mNoiseTexture = nullptr;
 
        core::ManagedImage mOutput;
        core::DescriptorSet mDescriptorSet;
        VkWriteDescriptorSetAccelerationStructureKHR mDescriptorAccelerationStructureInfo{};
 
        util::PipelineLayout mPipelineLayout;
        rtpipe::RtPipeline mPipeline;
 
        uint32_t mRngSeedPushCOffset = 0;
    };
    class DefaultRaytracingStageBase : public RenderStage {…};
}  // namespace foray::stages