foray_denoiserstage.hpp

Go to the documentation of this file.

#pragma once
#include "../bench/foray_bench_declares.hpp"
#include "../util/foray_externalsemaphore.hpp"
#include "foray_gbuffer.hpp"
#include "foray_renderstage.hpp"
 
namespace foray::stages {
 
    struct DenoiserConfig
    {
        core::ManagedImage* PrimaryInput = nullptr;
        std::array<core::ManagedImage*, (size_t)GBufferStage::EOutput::MaxEnum> GBufferOutputs;
        std::unordered_map<std::string, core::ManagedImage*> AuxiliaryInputs;
        core::ManagedImage* PrimaryOutput = nullptr;
        void* AuxiliaryData = nullptr;
        util::ExternalSemaphore* Semaphore = nullptr;
        bench::DeviceBenchmark*  Benchmark = nullptr;
 
        inline DenoiserConfig() {}
        inline DenoiserConfig(core::ManagedImage* primaryIn, core::ManagedImage* primaryOut, GBufferStage* gbuffer) : PrimaryInput(primaryIn), PrimaryOutput(primaryOut)
        {
            if(!!gbuffer)
            {
                GBufferOutputs = {gbuffer->GetImageOutput(GBufferStage::PositionOutputName),    gbuffer->GetImageOutput(GBufferStage::NormalOutputName),
                                  gbuffer->GetImageOutput(GBufferStage::AlbedoOutputName),      gbuffer->GetImageOutput(GBufferStage::MotionOutputName),
                                  gbuffer->GetImageOutput(GBufferStage::MaterialIdxOutputName), gbuffer->GetImageOutput(GBufferStage::MeshInstanceIdOutputName),
                                  gbuffer->GetImageOutput(GBufferStage::LinearZOutputName),     gbuffer->GetImageOutput(GBufferStage::DepthOutputName)};
            }
        }
    };
 
    class DenoiserStage : public RenderStage
    {
      public:
        virtual void Init(core::Context* context, const DenoiserConfig& config){};
        virtual void UpdateConfig(const DenoiserConfig& config){};
        virtual std::string GetUILabel() { return "Unnamed Denoiser"; }
        virtual void DisplayImguiConfiguration(){};
        virtual void IgnoreHistoryNextFrame(){};
    };
 
    class ExternalDenoiserStage : public DenoiserStage
    {
      public:
        virtual void BeforeDenoise(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo){};
        virtual void AfterDenoise(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo){};
        virtual void DispatchDenoise(uint64_t timelineSemaphoreValueBefore, uint64_t timelineSemaphoreValueAfter){};
    };
}  // namespace foray::stages