vk_graphics_context.h

Go to the documentation of this file.

#pragma once
 
#include <gui_core/gui_id.h>
#include <gui_core/graphics/gui_graphics_context.h>
#include <gui_core/graphics/gui_renderer.h>
#include <gui_core/graphics/vulkan/vk_device_context.h>
#include <gui_core/graphics/vulkan/vk_types.h>
 
namespace tachyon
{
    class IWindowBackend;
    class GradientManager;
    class StyleManager;
    class ScrollManagement;
    class IWindowContext;
    class StyleManager;
    class GradientManager;
    class ImageSet;
}
 
namespace tachyon::vulkan
{
 
    struct FrameData;
    class VulkanImageSet;
    class VulkanWindowContext;

    class VulkanGraphicsContext : public tachyon::GraphicsContext
    {
    public:
        VulkanImageSet* vulkanImageSet = nullptr;
        tachyon::Buffer* visualBoxStylesMesh = nullptr;
        tachyon::Buffer* colorStylesMesh = nullptr;
        std::vector<tachyon::ImageIndex> gradientPageImageIndices;
        VulkanDeviceContext deviceContext;
 
    public:
        VulkanGraphicsContext();
 
        tachyon::GraphicsAPI GetAPI() const override { return tachyon::GraphicsAPI::Vulkan; }
 
        // ---- Setup, before GUIApplication::Initialize (see VulkanDeviceContext) ----
        void SetExternalDevice(const VulkanDeviceContext::ExternalDevice& external) { deviceContext.SetExternalDevice(external); }
        void SetSurfaceFactory(VulkanDeviceContext::SurfaceFactory factory) { deviceContext.SetSurfaceFactory(std::move(factory)); }

        void SetShaderDirectory(std::string_view directory) { deviceContext.shaderDirectory = std::string(directory); }
        void SetAllocationHooks(const tachyon::AllocationHooks& hooks) override { deviceContext.SetAllocationHooks(hooks); }
        Result<void> Initialize(tachyon::IWindowBackend* windowBackend) override;
        void Cleanup() override;
        bool DeviceLost() const override { return tachyon::vulkan::IsDeviceLost(); }
        tachyon::Result<void> RecreateDevice(tachyon::IWindowBackend* windowBackend) override;
        void SimulateDeviceLoss() { tachyon::vulkan::SetDeviceLost(true); }
        void BeginFrame() override;
        void EndFrame() override {}
 
    public:
        tachyon::Image* CreateImage(tachyon::uint2 size, tachyon::ImageFormat format, tachyon::Sampler sampler, bool mipmapped = false);
        tachyon::Image* CreateImage(void* data, tachyon::uint2 size, tachyon::ImageFormat format, tachyon::Sampler sampler, bool mipmapped = false) override;
        void DestroyImage(tachyon::Image* img);
        void DestroyImageAtEnd(tachyon::Image* img, tachyon::DeviceScope scope,
                               const std::source_location& loc = std::source_location::current()) override;

        tachyon::Buffer* UploadVisualColorStyles(std::span<const tachyon::VisualColorStyle> colorStyles);
        void DestroyVisualColorStyles(tachyon::Buffer* buffer);
        void DestroyVisualColorStylesAtEnd(tachyon::Buffer* buffer, tachyon::DeviceScope scope,
                                           const std::source_location& loc = std::source_location::current());

        tachyon::Buffer* UploadVisualBoxStyles(std::span<const tachyon::VisualBoxStyle> visualBoxStyles);
        void DestroyVisualBoxStyles(tachyon::Buffer* buffer);
        void DestroyVisualBoxStylesAtEnd(tachyon::Buffer* buffer, tachyon::DeviceScope scope,
                                         const std::source_location& loc = std::source_location::current());
 
    public:
        tachyon::IWindowContext* CreateWindowContext(void* nativeHandle, tachyon::OSWindowHandle osHandle, 
                                                 tachyon::ID id, tachyon::int2 size, tachyon::WindowFlags windowFlags,
                                                 tachyon::OnGuiFunc onGui) override;
        void DestroyWindowContext(tachyon::IWindowContext* context) override;
 
        tachyon::TextAtlasManager* CreateTextAtlasManager() override;
 
        tachyon::ImageSet* GetImageSet() const override;
        tachyon::ImageIndex RegisterImage(tachyon::Image* image) override;
        void UnregisterImage(tachyon::ImageIndex index) override;
 
        // ---- Host-rendered images ----

        tachyon::ImageIndex RegisterExternalImage(VkImageView view, VkSampler sampler);
        void UpdateExternalImage(tachyon::ImageIndex index, VkImageView view, VkSampler sampler);

        void SetWindowImageTarget(tachyon::ID window, VkImage image, VkExtent2D extent);

        void AddFrameWait(VkSemaphore semaphore, uint64_t value, VkPipelineStageFlags2 stageMask)
        {
            deviceContext.frameWaits.Add(semaphore, value, stageMask);
        }

        bool StartResizing(tachyon::ID id) { return guiRenderer->StartResizing(id); }
        void StopResizing() { guiRenderer->StopResizing(); }
 
    private:
        struct ImageTarget { VkImage image; VkExtent2D extent; };
        std::unordered_map<tachyon::ID, ImageTarget>                  m_imageTargets;
        std::unordered_map<tachyon::ID, tachyon::vulkan::VulkanWindowContext*> m_windowContexts;
 
    public:
        void Reset() { guiRenderer->Reset(); }

        tachyon::Buffer* PrepareVisualBoxStylesBuffer(tachyon::StyleManager& styleManager);
        tachyon::Buffer* PrepareVisualColorStylesBuffer(tachyon::StyleManager& styleManager, tachyon::GradientManager& gradientManager);

        tachyon::GPUDrawPushConstants pushConstants{};

        void PrepareFrame(tachyon::StyleManager& styleManager, tachyon::GradientManager& gradientManager, float elapsedTimeSecs) override;
        void RePrepareFrame();
        bool RenderWindowFrame(VkCommandBuffer cmd, FrameData& currentFrame, tachyon::ID windowID, tachyon::uint2 extent);
 
        float GetLastFrameBuildSeconds() const override { return m_frameBuildSeconds; }
        void  SetFrameBuildSeconds(float seconds) { m_frameBuildSeconds = seconds; }
 
    private:
        tachyon::StyleManager*    m_lastStyleManager{ nullptr };
        tachyon::GradientManager* m_lastGradientManager{ nullptr };
        float                 m_lastElapsedTimeSecs{ 0.0f };
        float                 m_frameBuildSeconds{ 0.0f };
    };

}