window.cpp

#include <stdafx.h>

#include "window.h"

#include <CET.h>

using namespace std::chrono_literals;

static Window* s_pWindow = nullptr;

static BOOL CALLBACK EnumWindowsProcCP77(HWND ahWnd, LPARAM alParam)
{
    DWORD lpdwProcessId;
    GetWindowThreadProcessId(ahWnd, &lpdwProcessId);
    if (lpdwProcessId == GetCurrentProcessId())
    {
        wchar_t name[512] = {0};
        RealGetWindowClassW(ahWnd,name,511);
        if (wcscmp (L"W2ViewportClass", name) == 0)
        {
            *reinterpret_cast<HWND*>(alParam) = ahWnd;
            return FALSE;
        }
    }
    return TRUE;
}

LRESULT APIENTRY Window::WndProc(HWND ahWnd, UINT auMsg, WPARAM awParam, LPARAM alParam)
{
    if (s_pWindow)
    {
        if (auMsg == WM_WINDOWPOSCHANGED)
        {
            const auto* wp = reinterpret_cast<WINDOWPOS*>(alParam);

            if ((wp->flags & SWP_NOMOVE) == 0)
                s_pWindow->m_wndPos = {wp->x, wp->y};
            if ((wp->flags & SWP_NOSIZE) == 0)
                s_pWindow->m_wndSize = {wp->cx, wp->cy};

            RECT cr;
            GetClientRect(ahWnd, &cr);
            s_pWindow->m_clientPos = {cr.left, cr.top};
            s_pWindow->m_clientSize = {cr.right - cr.left, cr.bottom - cr.top};
        }

        {
            if (s_pWindow->m_pBindings->OnWndProc(ahWnd, auMsg, awParam, alParam))
                return 0; // VKBindings wants this input ignored!
        }

        {
            if (s_pWindow->m_pD3D12->OnWndProc(ahWnd, auMsg, awParam, alParam))
                return 0; // D3D12 wants this input ignored!
        }

        return CallWindowProc(s_pWindow->m_wndProc, ahWnd, auMsg, awParam, alParam);
    }

    return 0;
}

Window::Window(VKBindings* apBindings, D3D12* apD3D12)
    : m_pBindings(apBindings)
    , m_pD3D12(apD3D12)
{
    s_pWindow = this;

    std::thread t(
        [this]
        {
            while (m_hWnd == nullptr)
            {
                if (EnumWindows(EnumWindowsProcCP77, reinterpret_cast<LPARAM>(&m_hWnd)))
                    std::this_thread::sleep_for(50ms);
                else
                {
                    m_wndProc = reinterpret_cast<WNDPROC>(SetWindowLongPtr(m_hWnd, GWLP_WNDPROC, reinterpret_cast<LONG_PTR>(WndProc)));
                    Log::Info("Window::Initialize() - window hook complete.");
                }
            }

            m_initialized = true;
        });

    t.detach();
}

Window::~Window()
{
    s_pWindow = nullptr;

    if (m_hWnd != nullptr)
        SetWindowLongPtr(m_hWnd, GWLP_WNDPROC, reinterpret_cast<LONG_PTR>(m_wndProc));
}