[dxvk] Implement waiting for specific present requests

src/d3d11/d3d11_swapchain.cpp

@@ -414,7 +414,7 @@ namespace dxvk {
       if (cHud != nullptr && !cFrameId)
         cHud->update();
 
-      m_device->presentImage(m_presenter, cPresentMode, &m_presentStatus);
+      m_device->presentImage(m_presenter, cPresentMode, 0, &m_presentStatus);
     });
 
     pContext->FlushCsChunk();

src/d3d9/d3d9_swapchain.cpp

@@ -833,7 +833,7 @@ namespace dxvk {
       if (cHud != nullptr && !cFrameId)
         cHud->update();
 
-      m_device->presentImage(m_presenter, cPresentMode, &m_presentStatus);
+      m_device->presentImage(m_presenter, cPresentMode, 0, &m_presentStatus);
     });
 
     m_parent->FlushCsChunk();

src/dxvk/dxvk_device.cpp

@@ -254,12 +254,14 @@ namespace dxvk {
   void DxvkDevice::presentImage(
     const Rc<Presenter>&            presenter,
           VkPresentModeKHR          presentMode,
+          uint64_t                  frameId,
           DxvkSubmitStatus*         status) {
     status->result = VK_NOT_READY;
 
     DxvkPresentInfo presentInfo = { };
     presentInfo.presenter = presenter;
     presentInfo.presentMode = presentMode;
+    presentInfo.frameId = frameId;
     m_submissionQueue.present(presentInfo, status);
     
     std::lock_guard<sync::Spinlock> statLock(m_statLock);
@@ -310,10 +312,13 @@ namespace dxvk {
   
   
   void DxvkDevice::waitForIdle() {
-    this->lockSubmission();
+    m_submissionQueue.waitForIdle();
+    m_submissionQueue.lockDeviceQueue();
+
     if (m_vkd->vkDeviceWaitIdle(m_vkd->device()) != VK_SUCCESS)
       Logger::err("DxvkDevice: waitForIdle: Operation failed");
-    this->unlockSubmission();
+
+    m_submissionQueue.unlockDeviceQueue();
   }
   
   

src/dxvk/dxvk_device.h

@@ -455,11 +455,13 @@ namespace dxvk {
      * can be retrieved with \ref waitForSubmission.
      * \param [in] presenter The presenter
      * \param [in] presenteMode Present mode
+     * \param [in] frameId Optional frame ID
      * \param [out] status Present status
      */
     void presentImage(
       const Rc<Presenter>&            presenter,
             VkPresentModeKHR          presentMode,
+            uint64_t                  frameId,
             DxvkSubmitStatus*         status);
     
     /**

src/dxvk/dxvk_queue.cpp

@@ -74,6 +74,19 @@ namespace dxvk {
   }
 
 
+  void DxvkSubmissionQueue::waitForIdle() {
+    std::unique_lock<dxvk::mutex> lock(m_mutex);
+
+    m_submitCond.wait(lock, [this] {
+      return m_submitQueue.empty();
+    });
+
+    m_finishCond.wait(lock, [this] {
+      return m_finishQueue.empty();
+    });
+  }
+
+
   void DxvkSubmissionQueue::lockDeviceQueue() {
     m_mutexQueue.lock();
 
@@ -107,8 +120,6 @@ namespace dxvk {
       lock.unlock();
 
       // Submit command buffer to device
-      VkResult status = VK_NOT_READY;
-
       if (m_lastError != VK_ERROR_DEVICE_LOST) {
         std::lock_guard<dxvk::mutex> lock(m_mutexQueue);
 
@@ -116,30 +127,34 @@ namespace dxvk {
           m_callback(true);
 
         if (entry.submit.cmdList != nullptr)
-          status = entry.submit.cmdList->submit();
+          entry.result = entry.submit.cmdList->submit();
         else if (entry.present.presenter != nullptr)
-          status = entry.present.presenter->presentImage(entry.present.presentMode, 0);
+          entry.result = entry.present.presenter->presentImage(entry.present.presentMode, entry.present.frameId);
 
         if (m_callback)
           m_callback(false);
       } else {
         // Don't submit anything after device loss
         // so that drivers get a chance to recover
-        status = VK_ERROR_DEVICE_LOST;
+        entry.result = VK_ERROR_DEVICE_LOST;
       }
 
       if (entry.status)
-        entry.status->result = status;
+        entry.status->result = entry.result;
       
       // On success, pass it on to the queue thread
       lock = std::unique_lock<dxvk::mutex>(m_mutex);
 
-      if (status == VK_SUCCESS) {
+      bool doForward = (entry.result == VK_SUCCESS) ||
-        if (entry.submit.cmdList != nullptr)
+        (entry.present.presenter != nullptr && entry.result != VK_ERROR_DEVICE_LOST);
+
+      if (doForward) {
         m_finishQueue.push(std::move(entry));
-      } else if (status == VK_ERROR_DEVICE_LOST || entry.submit.cmdList != nullptr) {
+      } else {
-        Logger::err(str::format("DxvkSubmissionQueue: Command submission failed: ", status));
+        Logger::err(str::format("DxvkSubmissionQueue: Command submission failed: ", entry.result));
-        m_lastError = status;
+        m_lastError = entry.result;
+
+        if (m_lastError != VK_ERROR_DEVICE_LOST)
           m_device->waitForIdle();
       }
 
@@ -172,6 +187,7 @@ namespace dxvk {
       DxvkSubmitEntry entry = std::move(m_finishQueue.front());
       lock.unlock();
       
+      if (entry.submit.cmdList != nullptr) {
         VkResult status = m_lastError.load();
         
         if (status != VK_ERROR_DEVICE_LOST)
@@ -179,12 +195,22 @@ namespace dxvk {
         
         if (status != VK_SUCCESS) {
           m_lastError = status;
+
+          if (status != VK_ERROR_DEVICE_LOST)
             m_device->waitForIdle();
         }
+      } else if (entry.present.presenter != nullptr) {
+        // Signal the frame and then immediately destroy the reference.
+        // This is necessary since the front-end may want to explicitly
+        // destroy the presenter object. 
+        entry.present.presenter->signalFrame(entry.result, entry.present.frameId);
+        entry.present.presenter = nullptr;
+      }
 
       // Release resources and signal events, then immediately wake
       // up any thread that's currently waiting on a resource in
       // order to reduce delays as much as possible.
+      if (entry.submit.cmdList != nullptr)
         entry.submit.cmdList->notifyObjects();
 
       lock.lock();
@@ -195,9 +221,11 @@ namespace dxvk {
       lock.unlock();
 
       // Free the command list and associated objects now
+      if (entry.submit.cmdList != nullptr) {
         entry.submit.cmdList->reset();
         m_device->recycleCommandList(entry.submit.cmdList);
       }
     }
+  }
   
 }

src/dxvk/dxvk_queue.h

@@ -44,6 +44,7 @@ namespace dxvk {
   struct DxvkPresentInfo {
     Rc<Presenter>       presenter;
     VkPresentModeKHR    presentMode;
+    uint64_t            frameId;
   };
 
 
@@ -51,6 +52,7 @@ namespace dxvk {
    * \brief Submission queue entry
    */
   struct DxvkSubmitEntry {
+    VkResult            result;
     DxvkSubmitStatus*   status;
     DxvkSubmitInfo      submit;
     DxvkPresentInfo     present;
@@ -160,6 +162,11 @@ namespace dxvk {
       m_finishCond.wait(lock, pred);
     }
 
+    /**
+     * \brief Waits for all submissions to complete
+     */
+    void waitForIdle();
+
     /**
      * \brief Locks device queue
      *