foray_computestage.hpp

Go to the documentation of this file.

#pragma once
#include "../core/foray_shadermodule.hpp"
#include "../foray_glm.hpp"
#include "../util/foray_pipelinelayout.hpp"
#include "foray_renderstage.hpp"
 
namespace foray::stages {
    class ComputeStageBase : public RenderStage
    {
      public:
        void Init(core::Context* context);
 
        virtual void RecordFrame(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo) override;
 
        virtual void Destroy() override;
      protected:
        core::ShaderModule mShader;
 
        core::DescriptorSet  mDescriptorSet;
        util::PipelineLayout mPipelineLayout;
 
        VkPipeline mPipeline = nullptr;
 
        inline virtual void ApiInitShader(){};
        inline virtual void ApiCreateDescriptorSet(){};
        inline virtual void ApiCreatePipelineLayout(){};
        inline virtual void ApiBeforeFrame(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo){};
        inline virtual void ApiBeforeDispatch(VkCommandBuffer cmdBuffer, base::FrameRenderInfo& renderInfo, glm::uvec3& groupSize){};
 
        virtual void CreatePipeline();
 
        virtual void ReloadShaders() override;
    };
}  // namespace foray::stages