NMF_ImGui.h

#pragma once

#ifndef NMF_VERSION
#include "NMF.h"
#endif

#ifdef NMF_USE_IMGUI

#include <functional>
#include <map>
#include <mutex>
#include <stdint.h>
#include <string>

// TODO: imconfig.h ?
#include "lib/imgui/imgui.h"

#if defined(NMF_IMGUI_DX9)
struct IDirect3D9;
struct IDirect3DDevice9;
#elif defined(NMF_IMGUI_DX10)
#error "Not implemented yet!"
#elif defined(NMF_IMGUI_DX11)
struct ID3D11Device;
struct ID3D11DeviceContext;
#elif defined(NMF_IMGUI_DX12)
#error "Not implemented yet!"
#else
#error "Invalid DX version!"
#endif

namespace NMF
{
    enum class ImGuiDrawTarget : uint8_t
    {
        MainMenu,
        General
    };

    struct ImGuiConsoleEntry
    {
        ImVec4 Color;
        std::string Message;
    };

    using ConsoleSendFunction = std::function<void(const char*)>;

    struct ImGuiConsole
    {
        ImGuiConsole()
            : Id(nullptr)
            , Name(nullptr)
            , TextInput("")
            , LastEntryCount(0)
            , HistoryIndex(-1)
            , IsShown(false)
            , SendFunction()
            , Entries()
            , History()
            , EntriesLock()
        {
        }

        const char* Id;
        const char* Name;
        char TextInput[512];
        size_t LastEntryCount;
        int HistoryIndex;
        bool IsShown;
        ConsoleSendFunction SendFunction;
        std::vector<ImGuiConsoleEntry> Entries;
        std::vector<std::string> History;
        std::mutex EntriesLock;
    };

    class NMF_EXPORT ImGuiManager final
    {
    public:
        static void Setup(HINSTANCE hinstance, int nShowCmd);
        static void Teardown();
        static void Render();
        static WndProcHandleResult WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
        static void ShowMenu(bool isShown);
        static bool IsMenuShown();
        static void EndScene();

        static bool AddConsole(const char* id, const char* name, ConsoleSendFunction sendFunction);
        static bool RemoveConsole(const char* id);
        static bool AddMessage(const char* id, const std::string& message);
        static bool AddMessage(const char* id, const std::string& message, const ImVec4& color);

#ifdef NMF_EXTERN_SET_GAME_FOCUS
        static void SetCatchControl(bool isCatching);
        static bool IsCatchingControl();
#endif

#if defined(NMF_IMGUI_DX9)
        static void SetupD3D(HWND gameHWND, IDirect3D9* instance, IDirect3DDevice9* device);
#elif defined(NMF_IMGUI_DX10)
#error "Not implemented yet!"
#elif defined(NMF_IMGUI_DX11)
        static void SetupD3D(HWND gameHWND, ID3D11Device* device, ID3D11DeviceContext* context);
#elif defined(NMF_IMGUI_DX12)
#error "Not implemented yet!"
#else
#error "Invalid DX version!"
#endif

    private:
        static void RecreateContext();

        static void RenderConsole(ImGuiDrawTarget target, ImGuiConsole* console);
        static void HandleCommand(ImGuiConsole* console);

        static bool IsMenuOpen;
        static HWND GameHWND;
        static HINSTANCE GameHINSTANCE;
        static int GamenShowCmd;

#ifdef NMF_EXTERN_SET_GAME_FOCUS
        static bool IsControlCaught;
#endif

#pragma warning(disable: 4251)
        static std::map<std::string, ImGuiConsole*> Consoles;
#pragma warning(default: 4251)

#ifdef NMF_IMGUI_POP_OUT
        static LRESULT WINAPI ExternalWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);

        static bool RegisterExternalWindow();
        static bool CreateExternalWindow();
        static bool ShowExternalWindow();
        static bool HideExternalWindow();
        static bool CreateExternalDevice();

        static bool IsExternalWindow;
        static HWND ExternalHWND;
        static constexpr const TCHAR* ExternalWindowClassName = TEXT("NMF_class");
        static constexpr const TCHAR* ExternalWindowName = TEXT("NMF");
#endif

#ifdef NMF_USE_LOGGING
        static Logger Logger;
#endif

#ifdef NMF_IMGUI_DEMO
        static bool IsImGUIDemoOpen;
#endif

#if defined(NMF_IMGUI_DX9)
        static IDirect3DDevice9* GameDevice;
        static IDirect3D9* GameInstance;

#ifdef NMF_IMGUI_POP_OUT
        static IDirect3DDevice9* ExternalDevice;
#endif

#ifdef NMF_IMGUI_DX9_HOOK_VIRTUAL
        static Hook* DeviceEndSceneHook;
        static Hook* DeviceResetHook;

        static HRESULT __stdcall DX9EndScene(void* device);
        static HRESULT __stdcall DX9Reset(void* device, void* pParam);
#endif
#elif defined(NMF_IMGUI_DX11)
        static ID3D11Device* GameDevice;
        static ID3D11DeviceContext* GameDeviceContext;

#ifdef NMF_IMGUI_POP_OUT
        static ID3D11Device* ExternalDevice;
#endif
#else
        static void* ExternalDevice;
#endif
};

#ifdef NMF_USE_CONSOLE

#ifdef NMF_USE_EXT_LUA_CONSOLE
    // wire the ext lua into the console
#else
    // wire extern functions into the console
#endif
#endif
}

#endif