Refactor storing submissions into FrameSpan class, reduce AntiLag thread contention

1 files changed, 9 insertions, 12 deletions

diff --git a/src/strategies/anti_lag/queue_strategy.hh b/src/strategies/anti_lag/queue_strategy.hh
index 37c44a5..b1ae3e6 100644
--- a/src/strategies/anti_lag/queue_strategy.hh
+++ b/src/strategies/anti_lag/queue_strategy.hh
@@ -3,7 +3,7 @@
 
 #include "strategies/queue_strategy.hh"
 
-#include <deque>
+#include "frame_span.hh"
 #include <memory>
 #include <mutex>
 
@@ -12,25 +12,22 @@ namespace low_latency {
 class QueueContext;
 
 class AntiLagQueueStrategy final : public QueueStrategy {
-  private:
+  public:
     std::mutex mutex;
-    std::deque<std::unique_ptr<Submission>> pending_submissions;
+    std::unique_ptr<FrameSpan> frame_span; // Null represents no work.
 
   public:
     AntiLagQueueStrategy(QueueContext& queue);
     virtual ~AntiLagQueueStrategy();
 
   public:
-    virtual void notify_submit(const VkSubmitInfo& submit,
-                               std::unique_ptr<Submission> submission) override;
-    virtual void notify_submit(const VkSubmitInfo2& submit,
-                               std::unique_ptr<Submission> submission) override;
+    virtual void
+    notify_submit(const VkSubmitInfo& submit,
+                  std::shared_ptr<TimestampPool::Handle> handle) override;
+    virtual void
+    notify_submit(const VkSubmitInfo2& submit,
+                  std::shared_ptr<TimestampPool::Handle> handle) override;
     virtual void notify_present(const VkPresentInfoKHR& present) override;
-
-  public:
-    // Wait for all pending submissions to complete. Resets pending submissions
-    // once done.
-    void await_complete();
 };
 
 } // namespace low_latency