[PATCH 4/4] vkd3d: Add a pending execution count to struct d3d12_command_signature.

[Conor McCarthy]

From: Conor McCarthy

Buffering of command list commands will depend upon command signatures
persisting until the buffer is flushed.
---
 libs/vkd3d/command.c       | 42 ++++++++++++++++++++++++++++----------
 libs/vkd3d/vkd3d_private.h | 15 ++++++++++++++
 2 files changed, 46 insertions(+), 11 deletions(-)

diff --git a/libs/vkd3d/command.c b/libs/vkd3d/command.c
index caf9964d..c8d1b5d4 100644
--- a/libs/vkd3d/command.c
+++ b/libs/vkd3d/command.c
@@ -1905,6 +1905,29 @@ HRESULT d3d12_command_allocator_create(struct d3d12_device *device,
     return S_OK;
 }

+static void d3d12_command_signature_destroy(struct d3d12_command_signature *signature)
+{
+    struct d3d12_device *device = signature->device;
+
+    vkd3d_private_store_destroy(&signature->private_store);
+
+    vkd3d_free((void *)signature->desc.pArgumentDescs);
+    vkd3d_free(signature);
+
+    d3d12_device_release(device);
+}
+
+static void d3d12_command_signature_add_pending(struct d3d12_command_signature *signature)
+{
+    vkd3d_atomic_increment(&signature->pending_count);
+}
+
+static void d3d12_command_signature_release_pending(struct d3d12_command_signature *signature)
+{
+    if (!vkd3d_atomic_decrement(&signature->pending_count) && !signature->refcount)
+        d3d12_command_signature_destroy(signature);
+}
+
 /* ID3D12CommandList */
 static inline struct d3d12_command_list *impl_from_ID3D12GraphicsCommandList2(ID3D12GraphicsCommandList2 *iface)
 {
@@ -5713,6 +5736,8 @@ static void STDMETHODCALLTYPE d3d12_command_list_ExecuteIndirect(ID3D12GraphicsC
         return;
     }

+    d3d12_command_signature_add_pending(sig_impl);
+
     signature_desc = &sig_impl->desc;
     for (i = 0; i < signature_desc->NumArgumentDescs; ++i)
     {
@@ -5775,6 +5800,7 @@ static void STDMETHODCALLTYPE d3d12_command_list_ExecuteIndirect(ID3D12GraphicsC
                 if (!d3d12_command_list_update_compute_state(list))
                 {
                     WARN("Failed to update compute state, ignoring dispatch.\n");
+                    d3d12_command_signature_release_pending(sig_impl);
                     return;
                 }

@@ -5787,6 +5813,8 @@ static void STDMETHODCALLTYPE d3d12_command_list_ExecuteIndirect(ID3D12GraphicsC
                 break;
         }
     }
+
+    d3d12_command_signature_release_pending(sig_impl);
 }

static void STDMETHODCALLTYPE d3d12_command_list_AtomicCopyBufferUINT(ID3D12GraphicsCommandList2 *iface,
@@ -7252,17 +7280,8 @@ static ULONG STDMETHODCALLTYPE d3d12_command_signature_Release(ID3D12CommandSign

TRACE("%p decreasing refcount to %u.\n", signature, refcount);

-    if (!refcount)
-    {
-        struct d3d12_device *device = signature->device;
-
-        vkd3d_private_store_destroy(&signature->private_store);
-
-        vkd3d_free((void *)signature->desc.pArgumentDescs);
-        vkd3d_free(signature);
-
-        d3d12_device_release(device);
-    }
+    if (!refcount && !signature->pending_count)
+        d3d12_command_signature_destroy(signature);

return refcount;
 }
@@ -7369,6 +7388,7 @@ HRESULT d3d12_command_signature_create(struct d3d12_device *device, const D3D12_

object->ID3D12CommandSignature_iface.lpVtbl = &d3d12_command_signature_vtbl;
     object->refcount = 1;
+    object->pending_count = 0;

object->desc = *desc;
     if (!(object->desc.pArgumentDescs = vkd3d_calloc(desc->NumArgumentDescs, sizeof(*desc->pArgumentDescs))))
diff --git a/libs/vkd3d/vkd3d_private.h b/libs/vkd3d/vkd3d_private.h
index c5259420..645cbb18 100644
--- a/libs/vkd3d/vkd3d_private.h
+++ b/libs/vkd3d/vkd3d_private.h
@@ -253,6 +253,11 @@ static inline void vkd3d_cond_destroy(struct vkd3d_cond *cond)
 {
 }

+static inline unsigned int vkd3d_atomic_increment(unsigned int volatile *x)
+{
+    return InterlockedIncrement((LONG volatile *)x);
+}
+
 static inline unsigned int vkd3d_atomic_decrement(unsigned int volatile *x)
 {
     return InterlockedDecrement((LONG volatile *)x);
@@ -387,6 +392,15 @@ static inline unsigned int vkd3d_atomic_decrement(unsigned int volatile *x)
 }
 # else
 #  error "vkd3d_atomic_decrement() not implemented for this platform"
+# endif  /* HAVE_SYNC_SUB_AND_FETCH */
+
+# if HAVE_SYNC_ADD_AND_FETCH
+static inline unsigned int vkd3d_atomic_increment(unsigned int volatile *x)
+{
+    return __sync_add_and_fetch(x, 1);
+}
+# else
+#  error "vkd3d_atomic_increment() not implemented for this platform"
 # endif  /* HAVE_SYNC_ADD_AND_FETCH */

# if HAVE_SYNC_BOOL_COMPARE_AND_SWAP
@@ -1575,6 +1589,7 @@ struct d3d12_command_signature
 {
     ID3D12CommandSignature ID3D12CommandSignature_iface;
     LONG refcount;
+    unsigned int pending_count;

D3D12_COMMAND_SIGNATURE_DESC desc;