From 800792a67dbd73a7b45be3cebf9aa68d638c41fa Mon Sep 17 00:00:00 2001
From: Philip Rebohle <philip.rebohle@tu-dortmund.de>
Date: Fri, 18 Oct 2024 03:25:49 +0200
Subject: [PATCH] [dxvk] Add function to safely acquire a resource

---
 src/dxvk/dxvk_sparse.h | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/dxvk/dxvk_sparse.h b/src/dxvk/dxvk_sparse.h
index ee026a7b5..237ced820 100644
--- a/src/dxvk/dxvk_sparse.h
+++ b/src/dxvk/dxvk_sparse.h
@@ -513,6 +513,28 @@ namespace dxvk {
       return m_useCount.load(std::memory_order_acquire) >= getIncrement(access);
     }
     
+    /**
+     * \brief Tries to acquire reference
+     *
+     * If the reference count is zero at the time this is called,
+     * the method will fail, otherwise the reference count will
+     * be incremented by one. This is useful to safely obtain a
+     * pointer from a look-up table that does not own references.
+     * \returns \c true on success
+     */
+    Rc<DxvkPagedResource> tryAcquire() {
+      uint64_t increment = getIncrement(DxvkAccess::None);
+      uint64_t refCount = m_useCount.load(std::memory_order_acquire);
+
+      do {
+        if (!refCount)
+          return nullptr;
+      } while (!m_useCount.compare_exchange_strong( refCount,
+        refCount + increment, std::memory_order_relaxed));
+
+      return Rc<DxvkPagedResource>::unsafeCreate(this);
+    }
+
     /**
      * \brief Queries sparse page table
      *