WorldUiRenderer.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <unordered_map>
#include <vector>
#include <vk_mem_alloc.h>
#include <vulkan/vulkan.h>
 
struct ImGuiContext;
 
namespace Engine::Components
{
    class WidgetComponent;
}  // namespace Engine::Components
 
namespace Engine::Core
{
    class ApplicationContext;
    class AssetManager;
    class SceneManager;
}  // namespace Engine::Core
 
namespace Engine::Rendering
{
    class RenderProcess;
 
    class WorldUiRenderer
    {
    public:
        WorldUiRenderer() = default;
        ~WorldUiRenderer();
 
        WorldUiRenderer( const WorldUiRenderer & ) = delete;
        WorldUiRenderer & operator=( const WorldUiRenderer & ) = delete;
 
        void init( Core::ApplicationContext * context, uint32_t framesInFlight );
        void cleanup( Core::AssetManager * assetManager );
 
        [[nodiscard]] uint32_t ensureWidgetTexture(
            Components::WidgetComponent & widget,
            Core::AssetManager &          assetManager
        );
 
        void renderWidgets(
            VkCommandBuffer          commandBuffer,
            uint32_t                 frameIndex,
            RenderProcess &          renderProcess,
            Core::SceneManager *     sceneManager,
            Core::AssetManager *     assetManager
        );
 
    private:
        struct TargetFrame
        {
            VkImage       image         = VK_NULL_HANDLE;
            VmaAllocation allocation    = VK_NULL_HANDLE;
            VkImageView   imageView     = VK_NULL_HANDLE;
            VkImageLayout currentLayout = VK_IMAGE_LAYOUT_UNDEFINED;
        };
 
        struct Target
        {
            Components::WidgetComponent * widget = nullptr;
            uint32_t descriptorIndex = UINT32_MAX;
            uint32_t width = 0;
            uint32_t height = 0;
            std::vector<TargetFrame> frames;
        };
 
        Target & ensureTarget( Components::WidgetComponent & widget, Core::AssetManager & assetManager );
        void pruneInactiveTargets(
            const std::vector<Components::WidgetComponent *> & activeWidgets,
            Core::AssetManager &                                assetManager
        );
        void createTargetImages( Target & target ) const;
        void destroyTarget( Target & target, Core::AssetManager * assetManager ) const;
        void transitionTargetsToColorAttachment( VkCommandBuffer commandBuffer, uint32_t frameIndex );
        void transitionTargetsToShaderRead( VkCommandBuffer commandBuffer, uint32_t frameIndex );
        void renderTarget( VkCommandBuffer commandBuffer, Target & target, uint32_t frameIndex ) const;
 
        Core::ApplicationContext * context_ = nullptr;
        ImGuiContext *             imguiContext_ = nullptr;
        VkDescriptorPool           descriptorPool_ = VK_NULL_HANDLE;
        VkSampler                  sampler_ = VK_NULL_HANDLE;
        VkFormat                   format_ = VK_FORMAT_R8G8B8A8_UNORM;
        uint32_t                   framesInFlight_ = 0;
        bool                       initialized_ = false;
 
        std::unordered_map<Components::WidgetComponent *, Target> targets_;
    };
}  // namespace Engine::Rendering