AuroraMiddleware/VulkanMemoryAllocator
1
0
Fork 0
mirror of https://github.com/GPUOpen-LibrariesAndSDKs/VulkanMemoryAllocator synced 2024-12-04 01:01:08 +00:00
Code Issues Projects Releases Wiki Activity

documentation

Browse Source
This commit is contained in:
Ilya Doroshenko 2024-08-27 10:39:54 +02:00
parent 9402a6bef9
commit 65afd9e98b
2 changed files with 13 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Doxyfile
View File

@ -2466,6 +2466,7 @@ PREDEFINED = VMA_CALL_PRE= \
VMA_MEMORY_PRIORITY=1 \ VMA_MEMORY_PRIORITY=1 \
VMA_KHR_MAINTENANCE4=1 \ VMA_KHR_MAINTENANCE4=1 \
VMA_KHR_MAINTENANCE5=1 \ VMA_KHR_MAINTENANCE5=1 \
VMA_EXTERNAL_MEMORY_WIN32=1 \
VMA_EXTERNAL_MEMORY=1 \ VMA_EXTERNAL_MEMORY=1 \
VMA_EXTENDS_VK_STRUCT= \ VMA_EXTENDS_VK_STRUCT= \
VMA_STATS_STRING_ENABLED=1 VMA_STATS_STRING_ENABLED=1

31
include/vk_mem_alloc.h
View File

@ -242,7 +242,7 @@ extern "C" {
#endif #endif
#endif #endif
// Defined to 1 when VK_KHR_external_memory device extension is defined in Vulkan headers. // Defined to 1 when VK_KHR_external_memory_win32 device extension is defined in Vulkan headers.
#if !defined(VMA_EXTERNAL_MEMORY_WIN32) #if !defined(VMA_EXTERNAL_MEMORY_WIN32)
#if VK_KHR_external_memory_win32 #if VK_KHR_external_memory_win32
#define VMA_EXTERNAL_MEMORY_WIN32 1 #define VMA_EXTERNAL_MEMORY_WIN32 1
@ -1052,7 +1052,7 @@ typedef struct VmaVulkanFunctions
/// Fetch from "vkGetDeviceImageMemoryRequirements" on Vulkan >= 1.3, but you can also fetch it from "vkGetDeviceImageMemoryRequirementsKHR" if you enabled extension VK_KHR_maintenance4. /// Fetch from "vkGetDeviceImageMemoryRequirements" on Vulkan >= 1.3, but you can also fetch it from "vkGetDeviceImageMemoryRequirementsKHR" if you enabled extension VK_KHR_maintenance4.
PFN_vkGetDeviceImageMemoryRequirementsKHR VMA_NULLABLE vkGetDeviceImageMemoryRequirements; PFN_vkGetDeviceImageMemoryRequirementsKHR VMA_NULLABLE vkGetDeviceImageMemoryRequirements;
#endif #endif
#ifdef VK_USE_PLATFORM_WIN32_KHR #ifdef VMA_EXTERNAL_MEMORY_WIN32
PFN_vkGetMemoryWin32HandleKHR VMA_NULLABLE vkGetMemoryWin32HandleKHR; PFN_vkGetMemoryWin32HandleKHR VMA_NULLABLE vkGetMemoryWin32HandleKHR;
#else #else
void* VMA_NULLABLE vkGetMemoryWin32HandleKHR; void* VMA_NULLABLE vkGetMemoryWin32HandleKHR;
@ -6096,7 +6096,7 @@ class VmaWin32Handle
{ {
public: public:
VmaWin32Handle() noexcept : m_hHandle(VMA_NULL) { } VmaWin32Handle() noexcept : m_hHandle(VMA_NULL) { }
VmaWin32Handle(HANDLE hHandle) noexcept : m_hHandle(hHandle) { } explicit VmaWin32Handle(HANDLE hHandle) noexcept : m_hHandle(hHandle) { }
~VmaWin32Handle() noexcept { if (m_hHandle != VMA_NULL) { ::CloseHandle(m_hHandle); } } ~VmaWin32Handle() noexcept { if (m_hHandle != VMA_NULL) { ::CloseHandle(m_hHandle); } }
VMA_CLASS_NO_COPY_NO_MOVE(VmaWin32Handle) VMA_CLASS_NO_COPY_NO_MOVE(VmaWin32Handle)
@ -6105,32 +6105,26 @@ public:
VkResult GetHandle(VkDevice device, VkDeviceMemory memory, decltype(&vkGetMemoryWin32HandleKHR) pvkGetMemoryWin32HandleKHR, HANDLE hTargetProcess, bool useMutex, HANDLE* pHandle) noexcept VkResult GetHandle(VkDevice device, VkDeviceMemory memory, decltype(&vkGetMemoryWin32HandleKHR) pvkGetMemoryWin32HandleKHR, HANDLE hTargetProcess, bool useMutex, HANDLE* pHandle) noexcept
{ {
*pHandle = VMA_NULL; *pHandle = VMA_NULL;
// We only care about atomicity of handle retrieval, not about memory order.
HANDLE handle = m_hHandle.load(std::memory_order_relaxed);
// Try to get handle first. // Try to get handle first.
if (handle != VMA_NULL) if (m_hHandle != VMA_NULL)
{ {
*pHandle = Duplicate(hTargetProcess); *pHandle = Duplicate(hTargetProcess);
return VK_SUCCESS; return VK_SUCCESS;
} }
HANDLE hCreatedHandle = VMA_NULL;
VkResult res = VK_SUCCESS; VkResult res = VK_SUCCESS;
// If failed, try to create it. // If failed, try to create it.
{ {
VmaMutexLockWrite lock(m_Mutex, useMutex); VmaMutexLockWrite lock(m_Mutex, useMutex);
if (m_hHandle.load(std::memory_order_relaxed) == VMA_NULL) if (m_hHandle == VMA_NULL)
{ {
VkResult res = Create(device, memory, pvkGetMemoryWin32HandleKHR, &hCreatedHandle); res = Create(device, memory, pvkGetMemoryWin32HandleKHR, &m_hHandle);
m_hHandle.store(hCreatedHandle, std::memory_order_relaxed);
} }
} }
*pHandle = Duplicate(hTargetProcess); *pHandle = Duplicate(hTargetProcess);
return res; return res;
} }
operator bool() const noexcept { return m_hHandle != VMA_NULL; } operator bool() const noexcept { return m_hHandle != VMA_NULL; }
private: private:
@ -6150,20 +6144,19 @@ private:
} }
HANDLE Duplicate(HANDLE hTargetProcess = VMA_NULL) const noexcept HANDLE Duplicate(HANDLE hTargetProcess = VMA_NULL) const noexcept
{ {
HANDLE handle = m_hHandle.load(std::memory_order_relaxed); if (!m_hHandle)
if (!handle) return m_hHandle;
return handle;
HANDLE hCurrentProcess = ::GetCurrentProcess(); HANDLE hCurrentProcess = ::GetCurrentProcess();
HANDLE hDupHandle = VMA_NULL; HANDLE hDupHandle = VMA_NULL;
if (!::DuplicateHandle(hCurrentProcess, handle, hTargetProcess ? hTargetProcess : hCurrentProcess, &hDupHandle, 0, FALSE, DUPLICATE_SAME_ACCESS)) if (!::DuplicateHandle(hCurrentProcess, m_hHandle, hTargetProcess ? hTargetProcess : hCurrentProcess, &hDupHandle, 0, FALSE, DUPLICATE_SAME_ACCESS))
{ {
VMA_ASSERT(0 && "Failed to duplicate handle."); VMA_ASSERT(0 && "Failed to duplicate handle.");
} }
return hDupHandle; return hDupHandle;
} }
private: private:
std::atomic<HANDLE> m_hHandle; HANDLE m_hHandle;
VMA_RW_MUTEX m_Mutex; // Protects access m_Handle VMA_RW_MUTEX m_Mutex; // Protects access m_Handle
}; };
#else #else
@ -10804,7 +10797,7 @@ VkResult VmaDeviceMemoryBlock::BindImageMemory(
VkResult VmaDeviceMemoryBlock::CreateWin32Handle(const VmaAllocator hAllocator, decltype(&vkGetMemoryWin32HandleKHR) pvkGetMemoryWin32HandleKHR, HANDLE hTargetProcess, HANDLE* pHandle) noexcept VkResult VmaDeviceMemoryBlock::CreateWin32Handle(const VmaAllocator hAllocator, decltype(&vkGetMemoryWin32HandleKHR) pvkGetMemoryWin32HandleKHR, HANDLE hTargetProcess, HANDLE* pHandle) noexcept
{ {
VMA_ASSERT(pHandle); VMA_ASSERT(pHandle);
return m_Handle.GetHandle(hAllocator->m_hDevice, m_hMemory, &vkGetMemoryWin32HandleKHR, hTargetProcess, hAllocator->m_UseMutex, pHandle); return m_Handle.GetHandle(hAllocator->m_hDevice, m_hMemory, pvkGetMemoryWin32HandleKHR, hTargetProcess, hAllocator->m_UseMutex, pHandle);
} }
#endif // VMA_EXTERNAL_MEMORY_WIN32 #endif // VMA_EXTERNAL_MEMORY_WIN32
#endif // _VMA_DEVICE_MEMORY_BLOCK_FUNCTIONS #endif // _VMA_DEVICE_MEMORY_BLOCK_FUNCTIONS