imgui_layer.h

#pragma once
 
#include "pixelbullet/application/layer.h"
 
#include "pixelbullet/application/events/application_event.h"
#include "pixelbullet/application/events/key_event.h"
#include "pixelbullet/application/events/mouse_event.h"
 
#include <imgui.h>
#include <volk.h>
 
#include <cstdint>
#include <map>
#include <string>
#include <string_view>
#include <vector>
 
namespace pixelbullet
{
class Graphics;
class Window;
 
class ImGuiLayer : public Layer, public UiFrameLayer
{
public:
    ImGuiLayer(Window& window, Graphics& graphics);
    ~ImGuiLayer() = default;
 
    virtual void OnAttach() override;
    virtual void OnDetach() override;
    virtual EventResult OnEvent(Event& e) override;
 
    bool BeginUiFrame() override;
    void CancelUiFrame() override;
 
    bool Begin();
    void CancelFrame();
    void End(VkCommandBuffer commandBuffer);
    ImTextureID GetRenderTargetTexture(std::string_view name);
 
    void BlockEvents(bool block)
    {
        block_events_ = block;
    }
 
    void SetTheme();
 
    uint32_t GetActiveWidgetID() const;
 
private:
    bool EnsureVulkanInitialized();
    void CreateDescriptorPool(VkDevice device);
    void DestroyRenderTargetTextures();
 
    struct CachedRenderTargetTexture
    {
        std::vector<VkDescriptorSet> descriptor_sets;
        std::vector<VkImageView> image_views;
        std::vector<VkSampler> samplers;
    };
 
private:
    bool block_events_ = true;
    Window& window_;
    Graphics& graphics_;
    VkDescriptorPool descriptor_pool_ = VK_NULL_HANDLE;
    VkRenderPass render_pass_ = VK_NULL_HANDLE;
    uint32_t image_count_ = 0;
    bool vulkan_initialized_ = false;
    bool frame_started_ = false;
    std::map<std::string, CachedRenderTargetTexture> render_target_textures_;
};
 
} // namespace pixelbullet