From f7dbcbe882f9adb6598a7ab3229b60d1a2e836e3 Mon Sep 17 00:00:00 2001
From: Philip Rebohle <philip.rebohle@tu-dortmund.de>
Date: Thu, 10 Jan 2019 16:47:37 +0100
Subject: [PATCH] [dxvk] Introduce command type with extra data

The additional data is stored as a struct which can be
modified after submitting the command to the CS chunk.
---
 src/dxvk/dxvk_cs.h | 64 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)

diff --git a/src/dxvk/dxvk_cs.h b/src/dxvk/dxvk_cs.h
index bbc54e17..d7516a49 100644
--- a/src/dxvk/dxvk_cs.h
+++ b/src/dxvk/dxvk_cs.h
@@ -80,6 +80,42 @@ namespace dxvk {
     T m_command;
     
   };
+
+
+  /**
+   * \brief Typed command with metadata
+   * 
+   * Stores a function object and an arbitrary
+   * data structure which can be modified after
+   * submitting the command to a cs chunk.
+   */
+  template<typename T, typename M>
+  class alignas(16) DxvkCsDataCmd : public DxvkCsCmd {
+
+  public:
+
+    template<typename... Args>
+    DxvkCsDataCmd(T&& cmd, Args&&... args)
+    : m_command (std::move(cmd)),
+      m_data    (std::forward<Args>(args)...) { }
+    
+    DxvkCsDataCmd             (DxvkCsDataCmd&&) = delete;
+    DxvkCsDataCmd& operator = (DxvkCsDataCmd&&) = delete;
+
+    void exec(DxvkContext* ctx) const {
+      m_command(ctx, &m_data);
+    }
+
+    M* data() {
+      return &m_data;
+    }
+
+  private:
+
+    T m_command;
+    M m_data;
+
+  };
   
   
   /**
@@ -147,6 +183,34 @@ namespace dxvk {
       m_commandOffset += sizeof(FuncType);
       return true;
     }
+
+    /**
+     * \brief Adds a command with data to the chunk 
+     * 
+     * \param [in] command The command to add
+     * \param [in] args Constructor args for the data object
+     * \returns Pointer to the data object, or \c nullptr
+     */
+    template<typename M, typename T, typename... Args>
+    M* pushCmd(T& command, Args&&... args) {
+      using FuncType = DxvkCsDataCmd<T, M>;
+      
+      if (m_commandOffset + sizeof(FuncType) > MaxBlockSize)
+        return nullptr;
+      
+      FuncType* func = new (m_data + m_commandOffset)
+        FuncType(std::move(command), std::forward<Args>(args)...);
+      
+      if (m_tail != nullptr)
+        m_tail->setNext(func);
+      else
+        m_head = func;
+      m_tail = func;
+
+      m_commandCount  += 1;
+      m_commandOffset += sizeof(FuncType);
+      return func->data();
+    }
     
     /**
      * \brief Initializes chunk for recording