render_stage_runtime.h

#pragma once
 
#include "pixelbullet/graphics/render_stage.h"
 
#include <glm/glm.hpp>
#include <volk.h>
 
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <vector>
 
namespace pixelbullet
{
class Descriptor;
class Framebuffers;
class ImageDepth;
class RenderDevice;
class Renderpass;
class RenderTarget2D;
class Swapchain;
 
class RenderStageRuntime
{
public:
    explicit RenderStageRuntime(const RenderStage& specification);
    ~RenderStageRuntime();
 
    void Update(const RenderDevice& render_device, const std::map<std::string, std::unique_ptr<RenderTarget2D>>& render_targets);
    void Rebuild(const RenderDevice& render_device, const std::map<std::string, std::unique_ptr<RenderTarget2D>>& render_targets,
                 const Swapchain& swapchain);
 
    const Descriptor* GetDescriptor(const std::string& name) const;
    const Descriptor* GetDescriptor(uint32_t binding) const;
    const VkFramebuffer& GetActiveFramebuffer(uint32_t active_swapchain_image) const;
    const Renderpass* GetRenderpass() const
    {
        return renderpass_.get();
    }
    const ImageDepth* GetDepthStencil() const
    {
        return depth_stencil_.get();
    }
 
    const std::vector<VkClearValue>& GetClearValues() const
    {
        return clear_values_;
    }
    uint32_t GetAttachmentCount(uint32_t subpass) const
    {
        return subpass_attachment_count_[subpass];
    }
    bool HasSwapchain() const
    {
        return swapchain_attachment_.has_value();
    }
    bool IsOutOfDate() const
    {
        return out_of_date_;
    }
    const glm::uvec2& GetExtent() const
    {
        return extent_;
    }
    const glm::ivec2& GetOffset() const
    {
        return offset_;
    }
 
    const std::map<std::string, const Descriptor*>& GetDescriptors() const
    {
        return descriptors_;
    }
    const RenderStage& GetSpecification() const
    {
        return *specification_;
    }
 
private:
    void RebuildDescriptors(const std::map<std::string, std::unique_ptr<RenderTarget2D>>& render_targets);
 
private:
    const RenderStage* specification_ = nullptr;
 
    std::unique_ptr<Renderpass> renderpass_;
    std::unique_ptr<ImageDepth> depth_stencil_;
    std::unique_ptr<Framebuffers> framebuffers_;
 
    std::map<std::string, const Descriptor*> descriptors_;
    std::map<uint32_t, const Descriptor*> descriptors_by_binding_;
 
    std::vector<VkClearValue> clear_values_;
    std::vector<uint32_t> subpass_attachment_count_;
    std::optional<Attachment> depth_attachment_;
    std::optional<Attachment> swapchain_attachment_;
    std::vector<bool> subpass_multisampled_;
 
    glm::uvec2 extent_{ 0, 0 };
    glm::ivec2 offset_{ 0, 0 };
    float aspect_ratio_ = 1.0f;
    bool out_of_date_ = false;
};
} // namespace pixelbullet