foray_imguistage.hpp

Go to the documentation of this file.

#pragma once
#include "foray_rasterizedRenderStage.hpp"
#include <functional>
#include <sdl2/SDL.h>
 
namespace foray::stages {
    class ImguiStage : public RenderStage
    {
      public:
        ImguiStage() = default;
 
        virtual void Init(core::Context* context, core::ManagedImage* backgroundImage);
        virtual void InitForImage(core::Context* context, core::ManagedImage* backgroundImage);
        virtual void InitForSwapchain(core::Context* context);
        virtual void RecordFrame(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo) override;
        virtual void Destroy() override;
 
        virtual void SetBackgroundImage(core::ManagedImage* backgroundImage);
 
        void AddWindowDraw(std::function<void()> windowDraw) { mWindowDraws.push_back(windowDraw); }
 
        virtual void Resize(const VkExtent2D& extent) override;
 
        void ProcessSdlEvent(const SDL_Event* sdlEvent);
 
        inline virtual ~ImguiStage() { Destroy(); }
 
      protected:
        VkClearValue mClearValue;
 
        core::ManagedImage*                mTargetImage = nullptr;
        VkDescriptorPool                   mImguiPool{};
        std::vector<std::function<void()>> mWindowDraws;
 
        virtual void InitImgui();
        virtual void PrepareRenderpass();
        virtual void DestroyFrameBufferAndRenderPass();
 
 
        std::vector<VkFramebuffer> mFrameBuffers;
        VkRenderPass               mRenderPass = nullptr;
    };
}  // namespace foray::stages