mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2025-04-23 08:11:23 +01:00
119 lines
3.2 KiB
C++
Executable File
119 lines
3.2 KiB
C++
Executable File
/*
|
|
* Copyright 1993-2018 NVIDIA Corporation. All rights reserved.
|
|
*
|
|
* Please refer to the NVIDIA end user license agreement (EULA) associated
|
|
* with this source code for terms and conditions that govern your use of
|
|
* this software. Any use, reproduction, disclosure, or distribution of
|
|
* this software and related documentation outside the terms of the EULA
|
|
* is strictly prohibited.
|
|
*
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "DX12CudaSample.h"
|
|
#include <helper_string.h>
|
|
|
|
using namespace Microsoft::WRL;
|
|
|
|
DX12CudaSample::DX12CudaSample(UINT width, UINT height, std::string name) :
|
|
m_width(width),
|
|
m_height(height),
|
|
m_title(name),
|
|
m_useWarpDevice(false)
|
|
{
|
|
m_aspectRatio = static_cast<float>(width) / static_cast<float>(height);
|
|
}
|
|
|
|
DX12CudaSample::~DX12CudaSample()
|
|
{
|
|
}
|
|
|
|
std::wstring DX12CudaSample::string2wstring(const std::string& s)
|
|
{
|
|
int len;
|
|
int slength = (int)s.length() + 1;
|
|
len = MultiByteToWideChar(CP_ACP, 0, s.c_str(), slength, 0, 0);
|
|
wchar_t* buf = new wchar_t[len];
|
|
MultiByteToWideChar(CP_ACP, 0, s.c_str(), slength, buf, len);
|
|
std::wstring r(buf);
|
|
delete[] buf;
|
|
return r;
|
|
}
|
|
// Helper function for resolving the full path of assets.
|
|
std::wstring DX12CudaSample::GetAssetFullPath(const char* assetName)
|
|
{
|
|
LPTSTR lpBuffer = new char[4096];
|
|
GetCurrentDirectory(FILENAME_MAX, lpBuffer);
|
|
char *tmp = sdkFindFilePath((const char *)assetName, "simpleD3D12");
|
|
if (tmp == NULL)
|
|
{
|
|
throw std::exception("File not found");
|
|
}
|
|
for (int i = 0; i < strlen(tmp); i++)
|
|
{
|
|
if (tmp[i] == '/')
|
|
{
|
|
tmp[i] = '\\';
|
|
}
|
|
}
|
|
m_assetsPath = lpBuffer;
|
|
m_assetsPath = m_assetsPath + "\\" + tmp;
|
|
|
|
std::wstring stemp = string2wstring(m_assetsPath);
|
|
|
|
return stemp;
|
|
}
|
|
|
|
// Helper function for acquiring the first available hardware adapter that supports Direct3D 12.
|
|
// If no such adapter can be found, *ppAdapter will be set to nullptr.
|
|
_Use_decl_annotations_
|
|
void DX12CudaSample::GetHardwareAdapter(IDXGIFactory2* pFactory, IDXGIAdapter1** ppAdapter)
|
|
{
|
|
ComPtr<IDXGIAdapter1> adapter;
|
|
*ppAdapter = nullptr;
|
|
|
|
for (UINT adapterIndex = 0; DXGI_ERROR_NOT_FOUND != pFactory->EnumAdapters1(adapterIndex, &adapter); ++adapterIndex)
|
|
{
|
|
DXGI_ADAPTER_DESC1 desc;
|
|
adapter->GetDesc1(&desc);
|
|
|
|
if (desc.Flags & DXGI_ADAPTER_FLAG_SOFTWARE)
|
|
{
|
|
// Don't select the Basic Render Driver adapter.
|
|
// If you want a software adapter, pass in "/warp" on the command line.
|
|
continue;
|
|
}
|
|
|
|
// Check to see if the adapter supports Direct3D 12, but don't create the
|
|
// actual device yet.
|
|
if (SUCCEEDED(D3D12CreateDevice(adapter.Get(), D3D_FEATURE_LEVEL_11_0, _uuidof(ID3D12Device), nullptr)))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
*ppAdapter = adapter.Detach();
|
|
}
|
|
|
|
// Helper function for setting the window's title text.
|
|
void DX12CudaSample::SetCustomWindowText(const char* text)
|
|
{
|
|
std::string windowText = m_title + text;
|
|
SetWindowText(Win32Application::GetHwnd(), windowText.c_str());
|
|
}
|
|
|
|
// Helper function for parsing any supplied command line args.
|
|
_Use_decl_annotations_
|
|
void DX12CudaSample::ParseCommandLineArgs(WCHAR* argv[], int argc)
|
|
{
|
|
for (int i = 1; i < argc; ++i)
|
|
{
|
|
if (_wcsnicmp(argv[i], L"-warp", wcslen(argv[i])) == 0 ||
|
|
_wcsnicmp(argv[i], L"/warp", wcslen(argv[i])) == 0)
|
|
{
|
|
m_useWarpDevice = true;
|
|
m_title = m_title + " (WARP)";
|
|
}
|
|
}
|
|
}
|