#include "RHITestFixture.h"
#include <cstdlib>
#include <iostream>
#include <windows.h>

#include "XCEngine/RHI/D3D12/D3D12Device.h"
#include "XCEngine/RHI/OpenGL/OpenGLDevice.h"

namespace XCEngine {
namespace RHI {

INSTANTIATE_TEST_SUITE_P(D3D12, RHITestFixture, ::testing::Values(RHIType::D3D12));
INSTANTIATE_TEST_SUITE_P(OpenGL, RHITestFixture, ::testing::Values(RHIType::OpenGL));

void RHITestFixture::SetUpTestSuite() {
}

void RHITestFixture::TearDownTestSuite() {
}

void RHITestFixture::SetUp() {
    mDevice = RHIFactory::CreateRHIDevice(GetParam());
    ASSERT_NE(mDevice, nullptr);

    WNDCLASSEXW wc = {};
    wc.cbSize = sizeof(WNDCLASSEXW);
    wc.lpfnWndProc = DefWindowProcW;
    wc.hInstance = GetModuleHandle(nullptr);
    wc.lpszClassName = L"RHIUnitTestClass";
    RegisterClassExW(&wc);

    mWindow = CreateWindowExW(0, L"RHIUnitTestClass", L"RHIUnitTest", WS_OVERLAPPEDWINDOW,
        CW_USEDEFAULT, CW_USEDEFAULT, 800, 600, nullptr, nullptr, GetModuleHandle(nullptr), nullptr);

    bool initResult = false;
    if (GetParam() == RHIType::D3D12) {
        RHIDeviceDesc desc = {};
        desc.enableDebugLayer = false;
        initResult = mDevice->Initialize(desc);
    } else if (GetParam() == RHIType::OpenGL) {
        auto* oglDevice = static_cast<OpenGLDevice*>(mDevice);
        initResult = oglDevice->InitializeWithExistingWindow(mWindow);
    }
    ASSERT_TRUE(initResult);
}

void RHITestFixture::WaitForGPU() {
    if (mDevice == nullptr) {
        return;
    }

    if (GetParam() == RHIType::D3D12) {
        auto* device = static_cast<D3D12Device*>(mDevice);
        
        CommandQueueDesc queueDesc = {};
        queueDesc.queueType = static_cast<uint32_t>(CommandQueueType::Direct);
        queueDesc.priority = 0;
        queueDesc.nodeMask = 0;
        queueDesc.flags = 0;
        auto* commandQueue = device->CreateCommandQueue(queueDesc);
        if (commandQueue) {
            FenceDesc fenceDesc = {};
            fenceDesc.initialValue = 0;
            fenceDesc.flags = 0;
            auto* fence = mDevice->CreateFence(fenceDesc);
            if (fence) {
                commandQueue->Signal(fence, 1);
                fence->Wait(1);
                
                for (int i = 0; i < 100; i++) {
                    if (fence->GetCompletedValue() >= 1) {
                        break;
                    }
                    Sleep(10);
                }
                
                fence->Shutdown();
                delete fence;
            }
            commandQueue->Shutdown();
            delete commandQueue;
        }
        
        Sleep(100);
    }
}

void RHITestFixture::TearDown() {
    WaitForGPU();

    if (mDevice != nullptr) {
        mDevice->Shutdown();
        delete mDevice;
        mDevice = nullptr;
    }
    if (mWindow) {
        DestroyWindow(mWindow);
        mWindow = nullptr;
    }
}

} // namespace RHI
} // namespace XCEngine