From 35bca4cc432613af3926da3bed217a6baa9cbd26 Mon Sep 17 00:00:00 2001
From: lucy96chen <47800040+lucy96chen@users.noreply.github.com>
Date: Fri, 1 Oct 2021 13:08:25 -0700
Subject: Add getNativeHandle() to ICommandQueue and ICommandBuffer (#1952)

* Added support for getting command buffer and command queue handles to ICommandBuffer and ICommandQueue; D3D12Device, VkDevice, and DebugDevice modifieid to implement this new functionality; immediate-renderer-base.cpp also modified to implement the new functions

* Removed excess boilerplate

* Changed readRef() to get() in D3D12 getNativeHandle() implementation for ICommandBuffer and ICommandQueue

* Added unit tests for new getNativeHandle() implementations, unfinished

* Queue test added; Minor cleanup changes

* getBufferHandleTestImpl() now closes the command buffer before returning

* Added getNativeHandle() implementations to CUDADevice

* Added comment clarifying that the Vulkan check is checking for a null handle, which is defined to be 0
---
 tools/gfx/vulkan/render-vk.cpp | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'tools/gfx/vulkan/render-vk.cpp')

diff --git a/tools/gfx/vulkan/render-vk.cpp b/tools/gfx/vulkan/render-vk.cpp
index 186c7eb40..7af7d086f 100644
--- a/tools/gfx/vulkan/render-vk.cpp
+++ b/tools/gfx/vulkan/render-vk.cpp
@@ -4492,6 +4492,13 @@ public:
             }
             vkAPI.vkEndCommandBuffer(m_commandBuffer);
         }
+
+        virtual SLANG_NO_THROW Result SLANG_MCALL
+            getNativeHandle(NativeHandle* outHandle) override
+        {
+            *outHandle = (uint64_t)m_commandBuffer;
+            return SLANG_OK;
+        }
     };
 
     class CommandQueueImpl
@@ -4551,6 +4558,13 @@ public:
             vkAPI.vkQueueWaitIdle(m_queue);
         }
 
+        virtual SLANG_NO_THROW Result SLANG_MCALL
+            getNativeHandle(NativeHandle* outHandle) override
+        {
+            *outHandle = (uint64_t)m_queue;
+            return SLANG_OK;
+        }
+
         virtual SLANG_NO_THROW const Desc& SLANG_MCALL getDesc() override
         {
             return m_desc;
-- 
cgit v1.2.3