374 lines
13 KiB
C++
374 lines
13 KiB
C++
/*
|
|
* Copyright (C) 2017-2018 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/context/context.inl"
|
|
#include "runtime/os_interface/windows/d3d_sharing_functions.h"
|
|
#include "runtime/sharings/sharing_factory.h"
|
|
|
|
using namespace OCLRT;
|
|
|
|
template class D3DSharingFunctions<D3DTypesHelper::D3D10>;
|
|
template class D3DSharingFunctions<D3DTypesHelper::D3D11>;
|
|
|
|
const uint32_t D3DSharingFunctions<D3DTypesHelper::D3D10>::sharingId = SharingType::D3D10_SHARING;
|
|
const uint32_t D3DSharingFunctions<D3DTypesHelper::D3D11>::sharingId = SharingType::D3D11_SHARING;
|
|
|
|
static const DXGI_FORMAT DXGIFormats[] = {
|
|
DXGI_FORMAT_R32G32B32A32_TYPELESS,
|
|
DXGI_FORMAT_R32G32B32A32_FLOAT,
|
|
DXGI_FORMAT_R32G32B32A32_UINT,
|
|
DXGI_FORMAT_R32G32B32A32_SINT,
|
|
DXGI_FORMAT_R32G32B32_TYPELESS,
|
|
DXGI_FORMAT_R32G32B32_FLOAT,
|
|
DXGI_FORMAT_R32G32B32_UINT,
|
|
DXGI_FORMAT_R32G32B32_SINT,
|
|
DXGI_FORMAT_R16G16B16A16_TYPELESS,
|
|
DXGI_FORMAT_R16G16B16A16_FLOAT,
|
|
DXGI_FORMAT_R16G16B16A16_UNORM,
|
|
DXGI_FORMAT_R16G16B16A16_UINT,
|
|
DXGI_FORMAT_R16G16B16A16_SNORM,
|
|
DXGI_FORMAT_R16G16B16A16_SINT,
|
|
DXGI_FORMAT_R32G32_TYPELESS,
|
|
DXGI_FORMAT_R32G32_FLOAT,
|
|
DXGI_FORMAT_R32G32_UINT,
|
|
DXGI_FORMAT_R32G32_SINT,
|
|
DXGI_FORMAT_R32G8X24_TYPELESS,
|
|
DXGI_FORMAT_D32_FLOAT_S8X24_UINT,
|
|
DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS,
|
|
DXGI_FORMAT_X32_TYPELESS_G8X24_UINT,
|
|
DXGI_FORMAT_R10G10B10A2_TYPELESS,
|
|
DXGI_FORMAT_R10G10B10A2_UNORM,
|
|
DXGI_FORMAT_R10G10B10A2_UINT,
|
|
DXGI_FORMAT_R11G11B10_FLOAT,
|
|
DXGI_FORMAT_R8G8B8A8_TYPELESS,
|
|
DXGI_FORMAT_R8G8B8A8_UNORM,
|
|
DXGI_FORMAT_R8G8B8A8_UNORM_SRGB,
|
|
DXGI_FORMAT_R8G8B8A8_UINT,
|
|
DXGI_FORMAT_R8G8B8A8_SNORM,
|
|
DXGI_FORMAT_R8G8B8A8_SINT,
|
|
DXGI_FORMAT_R16G16_TYPELESS,
|
|
DXGI_FORMAT_R16G16_FLOAT,
|
|
DXGI_FORMAT_R16G16_UNORM,
|
|
DXGI_FORMAT_R16G16_UINT,
|
|
DXGI_FORMAT_R16G16_SNORM,
|
|
DXGI_FORMAT_R16G16_SINT,
|
|
DXGI_FORMAT_R32_TYPELESS,
|
|
DXGI_FORMAT_D32_FLOAT,
|
|
DXGI_FORMAT_R32_FLOAT,
|
|
DXGI_FORMAT_R32_UINT,
|
|
DXGI_FORMAT_R32_SINT,
|
|
DXGI_FORMAT_R24G8_TYPELESS,
|
|
DXGI_FORMAT_D24_UNORM_S8_UINT,
|
|
DXGI_FORMAT_R24_UNORM_X8_TYPELESS,
|
|
DXGI_FORMAT_X24_TYPELESS_G8_UINT,
|
|
DXGI_FORMAT_R8G8_TYPELESS,
|
|
DXGI_FORMAT_R8G8_UNORM,
|
|
DXGI_FORMAT_R8G8_UINT,
|
|
DXGI_FORMAT_R8G8_SNORM,
|
|
DXGI_FORMAT_R8G8_SINT,
|
|
DXGI_FORMAT_R16_TYPELESS,
|
|
DXGI_FORMAT_R16_FLOAT,
|
|
DXGI_FORMAT_D16_UNORM,
|
|
DXGI_FORMAT_R16_UNORM,
|
|
DXGI_FORMAT_R16_UINT,
|
|
DXGI_FORMAT_R16_SNORM,
|
|
DXGI_FORMAT_R16_SINT,
|
|
DXGI_FORMAT_R8_TYPELESS,
|
|
DXGI_FORMAT_R8_UNORM,
|
|
DXGI_FORMAT_R8_UINT,
|
|
DXGI_FORMAT_R8_SNORM,
|
|
DXGI_FORMAT_R8_SINT,
|
|
DXGI_FORMAT_A8_UNORM,
|
|
DXGI_FORMAT_R1_UNORM,
|
|
DXGI_FORMAT_R9G9B9E5_SHAREDEXP,
|
|
DXGI_FORMAT_R8G8_B8G8_UNORM,
|
|
DXGI_FORMAT_G8R8_G8B8_UNORM,
|
|
DXGI_FORMAT_BC1_TYPELESS,
|
|
DXGI_FORMAT_BC1_UNORM,
|
|
DXGI_FORMAT_BC1_UNORM_SRGB,
|
|
DXGI_FORMAT_BC2_TYPELESS,
|
|
DXGI_FORMAT_BC2_UNORM,
|
|
DXGI_FORMAT_BC2_UNORM_SRGB,
|
|
DXGI_FORMAT_BC3_TYPELESS,
|
|
DXGI_FORMAT_BC3_UNORM,
|
|
DXGI_FORMAT_BC3_UNORM_SRGB,
|
|
DXGI_FORMAT_BC4_TYPELESS,
|
|
DXGI_FORMAT_BC4_UNORM,
|
|
DXGI_FORMAT_BC4_SNORM,
|
|
DXGI_FORMAT_BC5_TYPELESS,
|
|
DXGI_FORMAT_BC5_UNORM,
|
|
DXGI_FORMAT_BC5_SNORM,
|
|
DXGI_FORMAT_B5G6R5_UNORM,
|
|
DXGI_FORMAT_B5G5R5A1_UNORM,
|
|
DXGI_FORMAT_B8G8R8A8_UNORM,
|
|
DXGI_FORMAT_B8G8R8X8_UNORM,
|
|
DXGI_FORMAT_R10G10B10_XR_BIAS_A2_UNORM,
|
|
DXGI_FORMAT_B8G8R8A8_TYPELESS,
|
|
DXGI_FORMAT_B8G8R8A8_UNORM_SRGB,
|
|
DXGI_FORMAT_B8G8R8X8_TYPELESS,
|
|
DXGI_FORMAT_B8G8R8X8_UNORM_SRGB,
|
|
DXGI_FORMAT_BC6H_TYPELESS,
|
|
DXGI_FORMAT_BC6H_UF16,
|
|
DXGI_FORMAT_BC6H_SF16,
|
|
DXGI_FORMAT_BC7_TYPELESS,
|
|
DXGI_FORMAT_BC7_UNORM,
|
|
DXGI_FORMAT_BC7_UNORM_SRGB,
|
|
DXGI_FORMAT_AYUV,
|
|
DXGI_FORMAT_Y410,
|
|
DXGI_FORMAT_Y416,
|
|
DXGI_FORMAT_NV12,
|
|
DXGI_FORMAT_P010,
|
|
DXGI_FORMAT_P016,
|
|
DXGI_FORMAT_420_OPAQUE,
|
|
DXGI_FORMAT_YUY2,
|
|
DXGI_FORMAT_Y210,
|
|
DXGI_FORMAT_Y216,
|
|
DXGI_FORMAT_NV11,
|
|
DXGI_FORMAT_AI44,
|
|
DXGI_FORMAT_IA44,
|
|
DXGI_FORMAT_P8,
|
|
DXGI_FORMAT_A8P8,
|
|
DXGI_FORMAT_B4G4R4A4_UNORM,
|
|
DXGI_FORMAT_P208,
|
|
DXGI_FORMAT_V208,
|
|
DXGI_FORMAT_V408,
|
|
DXGI_FORMAT_FORCE_UINT};
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::createQuery(D3DQuery **query) {
|
|
D3DQueryDesc desc = {};
|
|
d3dDevice->CreateQuery(&desc, query);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::updateDevice(D3DResource *resource) {
|
|
resource->GetDevice(&d3dDevice);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::fillCreateBufferDesc(D3DBufferDesc &desc, unsigned int width) {
|
|
desc.ByteWidth = width;
|
|
desc.MiscFlags = D3DResourceFlags::MISC_SHARED;
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::fillCreateTexture2dDesc(D3DTexture2dDesc &desc, D3DTexture2dDesc *srcDesc, cl_uint subresource) {
|
|
desc.Width = srcDesc->Width;
|
|
desc.Height = srcDesc->Height;
|
|
desc.MipLevels = 1;
|
|
desc.ArraySize = 1;
|
|
desc.Format = srcDesc->Format;
|
|
desc.MiscFlags = D3DResourceFlags::MISC_SHARED;
|
|
desc.SampleDesc.Count = srcDesc->SampleDesc.Count;
|
|
desc.SampleDesc.Quality = srcDesc->SampleDesc.Quality;
|
|
|
|
for (uint32_t i = 0u; i < (subresource % srcDesc->MipLevels); i++) {
|
|
desc.Width /= 2;
|
|
desc.Height /= 2;
|
|
}
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::fillCreateTexture3dDesc(D3DTexture3dDesc &desc, D3DTexture3dDesc *srcDesc, cl_uint subresource) {
|
|
desc.Width = srcDesc->Width;
|
|
desc.Height = srcDesc->Height;
|
|
desc.Depth = srcDesc->Depth;
|
|
desc.MipLevels = 1;
|
|
desc.Format = srcDesc->Format;
|
|
desc.MiscFlags = D3DResourceFlags::MISC_SHARED;
|
|
|
|
for (uint32_t i = 0u; i < (subresource % srcDesc->MipLevels); i++) {
|
|
desc.Width /= 2;
|
|
desc.Height /= 2;
|
|
desc.Depth /= 2;
|
|
}
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::createBuffer(D3DBufferObj **buffer, unsigned int width) {
|
|
D3DBufferDesc stagingDesc = {};
|
|
fillCreateBufferDesc(stagingDesc, width);
|
|
d3dDevice->CreateBuffer(&stagingDesc, nullptr, buffer);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::createTexture2d(D3DTexture2d **texture, D3DTexture2dDesc *desc, cl_uint subresource) {
|
|
D3DTexture2dDesc stagingDesc = {};
|
|
fillCreateTexture2dDesc(stagingDesc, desc, subresource);
|
|
d3dDevice->CreateTexture2D(&stagingDesc, nullptr, texture);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::createTexture3d(D3DTexture3d **texture, D3DTexture3dDesc *desc, cl_uint subresource) {
|
|
D3DTexture3dDesc stagingDesc = {};
|
|
fillCreateTexture3dDesc(stagingDesc, desc, subresource);
|
|
d3dDevice->CreateTexture3D(&stagingDesc, nullptr, texture);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::checkFormatSupport(DXGI_FORMAT format, UINT *pFormat) {
|
|
d3dDevice->CheckFormatSupport(format, pFormat);
|
|
}
|
|
|
|
template <typename D3D>
|
|
std::vector<DXGI_FORMAT> &D3DSharingFunctions<D3D>::retrieveTextureFormats(cl_mem_object_type imageType) {
|
|
auto cached = textureFormatCache.find(imageType);
|
|
if (cached == textureFormatCache.end()) {
|
|
bool success;
|
|
std::tie(cached, success) = textureFormatCache.emplace(imageType, std::vector<DXGI_FORMAT>(0));
|
|
if (!success) {
|
|
return DXGINoFormats;
|
|
}
|
|
std::vector<DXGI_FORMAT> &cached_formats = cached->second;
|
|
cached_formats.reserve(arrayCount(DXGIFormats));
|
|
for (auto DXGIFormat : DXGIFormats) {
|
|
UINT format = 0;
|
|
checkFormatSupport(DXGIFormat, &format);
|
|
if (memObjectFormatSupport(imageType, format)) {
|
|
cached_formats.push_back(DXGIFormat);
|
|
}
|
|
}
|
|
cached_formats.shrink_to_fit();
|
|
}
|
|
return cached->second;
|
|
}
|
|
|
|
template <>
|
|
bool D3DSharingFunctions<D3DTypesHelper::D3D10>::memObjectFormatSupport(cl_mem_object_type objectType, UINT format) {
|
|
auto d3dformat = static_cast<D3D10_FORMAT_SUPPORT>(format);
|
|
return ((objectType & CL_MEM_OBJECT_BUFFER) && (d3dformat & D3D10_FORMAT_SUPPORT_BUFFER)) || ((objectType & CL_MEM_OBJECT_IMAGE2D) && (d3dformat & D3D10_FORMAT_SUPPORT_TEXTURE2D)) || ((objectType & CL_MEM_OBJECT_IMAGE3D) && (d3dformat & D3D10_FORMAT_SUPPORT_TEXTURE3D));
|
|
}
|
|
|
|
template <>
|
|
bool D3DSharingFunctions<D3DTypesHelper::D3D11>::memObjectFormatSupport(cl_mem_object_type objectType, UINT format) {
|
|
auto d3dformat = static_cast<D3D11_FORMAT_SUPPORT>(format);
|
|
return ((objectType & CL_MEM_OBJECT_BUFFER) && (d3dformat & D3D11_FORMAT_SUPPORT_BUFFER)) || ((objectType & CL_MEM_OBJECT_IMAGE2D) && (d3dformat & D3D11_FORMAT_SUPPORT_TEXTURE2D)) || ((objectType & CL_MEM_OBJECT_IMAGE3D) && (d3dformat & D3D11_FORMAT_SUPPORT_TEXTURE3D));
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getBufferDesc(D3DBufferDesc *bufferDesc, D3DBufferObj *buffer) {
|
|
buffer->GetDesc(bufferDesc);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getTexture2dDesc(D3DTexture2dDesc *textureDesc, D3DTexture2d *texture) {
|
|
texture->GetDesc(textureDesc);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getTexture3dDesc(D3DTexture3dDesc *textureDesc, D3DTexture3d *texture) {
|
|
texture->GetDesc(textureDesc);
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getSharedHandle(D3DResource *resource, void **handle) {
|
|
IDXGIResource *dxgiResource = nullptr;
|
|
resource->QueryInterface(__uuidof(IDXGIResource), (void **)&dxgiResource);
|
|
dxgiResource->GetSharedHandle(handle);
|
|
dxgiResource->Release();
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getSharedNTHandle(D3DResource *resource, void **handle) {
|
|
IDXGIResource *dxgiResource = nullptr;
|
|
IDXGIResource1 *dxgiResource1 = nullptr;
|
|
resource->QueryInterface(__uuidof(IDXGIResource), (void **)&dxgiResource);
|
|
dxgiResource->QueryInterface(__uuidof(IDXGIResource1), (void **)&dxgiResource1);
|
|
dxgiResource1->CreateSharedHandle(nullptr, DXGI_SHARED_RESOURCE_READ | DXGI_SHARED_RESOURCE_WRITE, nullptr, handle);
|
|
dxgiResource1->Release();
|
|
dxgiResource->Release();
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::addRef(D3DResource *resource) {
|
|
resource->AddRef();
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::release(IUnknown *resource) {
|
|
resource->Release();
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::lockRect(D3DTexture2d *resource, D3DLOCKED_RECT *lockedRect, uint32_t flags) {
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::unlockRect(D3DTexture2d *resource) {
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::updateSurface(D3DTexture2d *src, D3DTexture2d *dst) {
|
|
}
|
|
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getRenderTargetData(D3DTexture2d *renderTarget, D3DTexture2d *dstSurface) {
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D10>::copySubresourceRegion(D3DResource *dst, cl_uint dstSubresource,
|
|
D3DResource *src, cl_uint srcSubresource) {
|
|
d3dDevice->CopySubresourceRegion(dst, dstSubresource, 0, 0, 0, src, srcSubresource, nullptr);
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D11>::copySubresourceRegion(D3DResource *dst, cl_uint dstSubresource,
|
|
D3DResource *src, cl_uint srcSubresource) {
|
|
d3d11DeviceContext->CopySubresourceRegion(dst, dstSubresource, 0, 0, 0, src, srcSubresource, nullptr);
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D10>::flushAndWait(D3DQuery *query) {
|
|
query->End();
|
|
d3dDevice->Flush();
|
|
while (query->GetData(nullptr, 0, 0) != S_OK)
|
|
;
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D11>::flushAndWait(D3DQuery *query) {
|
|
d3d11DeviceContext->End(query);
|
|
d3d11DeviceContext->Flush();
|
|
while (d3d11DeviceContext->GetData(query, nullptr, 0, 0) != S_OK)
|
|
;
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D10>::getDeviceContext(D3DQuery *query) {
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D11>::getDeviceContext(D3DQuery *query) {
|
|
d3dDevice->GetImmediateContext(&d3d11DeviceContext);
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D10>::releaseDeviceContext(D3DQuery *query) {
|
|
}
|
|
|
|
template <>
|
|
void D3DSharingFunctions<D3DTypesHelper::D3D11>::releaseDeviceContext(D3DQuery *query) {
|
|
d3d11DeviceContext->Release();
|
|
d3d11DeviceContext = nullptr;
|
|
}
|
|
template <typename D3D>
|
|
void D3DSharingFunctions<D3D>::getDxgiDesc(DXGI_ADAPTER_DESC *dxgiDesc, IDXGIAdapter *adapter, D3DDevice *device) {
|
|
if (!adapter) {
|
|
IDXGIDevice *dxgiDevice = nullptr;
|
|
device->QueryInterface(__uuidof(IDXGIDevice), (void **)&dxgiDevice);
|
|
dxgiDevice->GetAdapter(&adapter);
|
|
dxgiDevice->Release();
|
|
} else {
|
|
adapter->AddRef();
|
|
}
|
|
adapter->GetDesc(dxgiDesc);
|
|
adapter->Release();
|
|
}
|
|
|
|
template D3DSharingFunctions<D3DTypesHelper::D3D10> *Context::getSharing<D3DSharingFunctions<D3DTypesHelper::D3D10>>();
|
|
template D3DSharingFunctions<D3DTypesHelper::D3D11> *Context::getSharing<D3DSharingFunctions<D3DTypesHelper::D3D11>>();
|