added pool type traits,

added special constructor for pool types
KhronosGroup · Jul 23, 2024 · 85c2ae4 · 85c2ae4
1 parent 5f79a12
commit 85c2ae4
Show file tree

Hide file tree

Showing 4 changed files with 74 additions and 4 deletions.
diff --git a/snippets/SharedDestroy.hpp b/snippets/SharedDestroy.hpp
@@ -116,6 +116,8 @@ class PoolFreeShared
 public:
   using DestructorType = typename SharedHandleTraits<HandleType>::DestructorType;
 
+  using PoolTypeExport = PoolType;
+
   template <class Dispatcher>
   using ReturnType = decltype( std::declval<DestructorType>().free( PoolType(), 0u, nullptr, Dispatcher() ) );
 
@@ -135,7 +137,7 @@ class PoolFreeShared
 public:
   void destroy( DestructorType parent, HandleType handle ) const VULKAN_HPP_NOEXCEPT
   {
-    if( m_destroy && m_dispatch && m_pool )
+    if ( m_destroy && m_dispatch && m_pool )
       ( parent.*m_destroy )( m_pool.get(), 1u, &handle, *m_dispatch );
   }
 

diff --git a/snippets/SharedHandle.hpp b/snippets/SharedHandle.hpp
@@ -36,6 +36,28 @@ struct HasDestructor : std::integral_constant<bool, !std::is_same<DestructorType
 {
 };
 
+template <typename HandleType, typename = void>
+struct HasPoolType : std::false_type
+{
+};
+
+template <typename HandleType>
+struct HasPoolType<HandleType, decltype( (void)typename SharedHandleTraits<HandleType>::deleter::PoolTypeExport() )> : std::true_type
+{
+};
+
+template <typename HandleType, typename Enable = void>
+struct GetPoolType
+{
+  using type = NoDestructor;
+};
+
+template <typename HandleType>
+struct GetPoolType<HandleType, typename std::enable_if<HasPoolType<HandleType>::value>::type>
+{
+  using type = typename SharedHandleTraits<HandleType>::deleter::PoolTypeExport;
+};
+
 //=====================================================================================================================
 
 template <typename HandleType>
@@ -241,12 +263,23 @@ class SharedHandle : public SharedHandleBase<HandleType, SharedHeader<Destructor
 public:
   SharedHandle() = default;
 
-  template <typename T = HandleType, typename = typename std::enable_if<HasDestructor<T>::value>::type>
+  template <typename T = HandleType, typename = typename std::enable_if<HasDestructor<T>::value && !HasPoolType<T>::value>::type>
   explicit SharedHandle( HandleType handle, SharedHandle<DestructorTypeOf<HandleType>> parent, DeleterType deleter = DeleterType() ) VULKAN_HPP_NOEXCEPT
     : BaseType( handle, std::move( parent ), std::move( deleter ) )
   {
   }
 
+  template <typename Dispatcher = VULKAN_HPP_DEFAULT_DISPATCHER_TYPE,
+            typename T          = HandleType,
+            typename            = typename std::enable_if<HasDestructor<T>::value && HasPoolType<T>::value>::type>
+  explicit SharedHandle( HandleType                                           handle,
+                         SharedHandle<DestructorTypeOf<HandleType>>           parent,
+                         SharedHandle<typename GetPoolType<HandleType>::type> pool,
+                         const Dispatcher & dispatch                          VULKAN_HPP_DEFAULT_DISPATCHER_ASSIGNMENT ) VULKAN_HPP_NOEXCEPT
+    : BaseType( handle, std::move( parent ), DeleterType{ std::move( pool ), dispatch } )
+  {
+  }
+
   template <typename T = HandleType, typename = typename std::enable_if<!HasDestructor<T>::value>::type>
   explicit SharedHandle( HandleType handle, DeleterType deleter = DeleterType() ) VULKAN_HPP_NOEXCEPT : BaseType( handle, std::move( deleter ) )
   {

diff --git a/vulkan/vulkan.hpp b/vulkan/vulkan.hpp
@@ -16851,7 +16851,7 @@ namespace VULKAN_HPP_NAMESPACE
           m_library = dlopen( "libvulkan.1.dylib", RTLD_NOW | RTLD_LOCAL );
         }
 #  elif defined( _WIN32 )
-          m_library = ::LoadLibraryA( "vulkan-1.dll" );
+        m_library = ::LoadLibraryA( "vulkan-1.dll" );
 #  else
 #    error unsupported platform
 #  endif

diff --git a/vulkan/vulkan_shared.hpp b/vulkan/vulkan_shared.hpp
@@ -52,6 +52,28 @@ namespace VULKAN_HPP_NAMESPACE
   {
   };
 
+  template <typename HandleType, typename = void>
+  struct HasPoolType : std::false_type
+  {
+  };
+
+  template <typename HandleType>
+  struct HasPoolType<HandleType, decltype( (void)typename SharedHandleTraits<HandleType>::deleter::PoolTypeExport() )> : std::true_type
+  {
+  };
+
+  template <typename HandleType, typename Enable = void>
+  struct GetPoolType
+  {
+    using type = NoDestructor;
+  };
+
+  template <typename HandleType>
+  struct GetPoolType<HandleType, typename std::enable_if<HasPoolType<HandleType>::value>::type>
+  {
+    using type = typename SharedHandleTraits<HandleType>::deleter::PoolTypeExport;
+  };
+
   //=====================================================================================================================
 
   template <typename HandleType>
@@ -257,12 +279,23 @@ namespace VULKAN_HPP_NAMESPACE
   public:
     SharedHandle() = default;
 
-    template <typename T = HandleType, typename = typename std::enable_if<HasDestructor<T>::value>::type>
+    template <typename T = HandleType, typename = typename std::enable_if<HasDestructor<T>::value && !HasPoolType<T>::value>::type>
     explicit SharedHandle( HandleType handle, SharedHandle<DestructorTypeOf<HandleType>> parent, DeleterType deleter = DeleterType() ) VULKAN_HPP_NOEXCEPT
       : BaseType( handle, std::move( parent ), std::move( deleter ) )
     {
     }
 
+    template <typename Dispatcher = VULKAN_HPP_DEFAULT_DISPATCHER_TYPE,
+              typename T          = HandleType,
+              typename            = typename std::enable_if<HasDestructor<T>::value && HasPoolType<T>::value>::type>
+    explicit SharedHandle( HandleType                                           handle,
+                           SharedHandle<DestructorTypeOf<HandleType>>           parent,
+                           SharedHandle<typename GetPoolType<HandleType>::type> pool,
+                           const Dispatcher & dispatch                          VULKAN_HPP_DEFAULT_DISPATCHER_ASSIGNMENT ) VULKAN_HPP_NOEXCEPT
+      : BaseType( handle, std::move( parent ), DeleterType{ std::move( pool ), dispatch } )
+    {
+    }
+
     template <typename T = HandleType, typename = typename std::enable_if<!HasDestructor<T>::value>::type>
     explicit SharedHandle( HandleType handle, DeleterType deleter = DeleterType() ) VULKAN_HPP_NOEXCEPT : BaseType( handle, std::move( deleter ) )
     {
@@ -390,6 +423,8 @@ namespace VULKAN_HPP_NAMESPACE
   public:
     using DestructorType = typename SharedHandleTraits<HandleType>::DestructorType;
 
+    using PoolTypeExport = PoolType;
+
     template <class Dispatcher>
     using ReturnType = decltype( std::declval<DestructorType>().free( PoolType(), 0u, nullptr, Dispatcher() ) );