104 lines
2.9 KiB
C++
104 lines
2.9 KiB
C++
#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
|