feat(d3d11, dxmt): record potential hazard info in binding bitset

Since not all bound resource may introduce hazard, it would be a even sparser set.

Author: 3Shain

src/d3d11/d3d11_context_impl.cpp

@@ -2676,8 +2676,8 @@ template <typename ContextInternalState> class MTLD3D11DeviceContextImplBase : p
 
     auto &SRVs = state_.ShaderStages[Stage].SRVs;
 
-    // TODO: iterate _hazard_ set rather than _bound_ set (former is a subset)
-    for (const auto & [Slot, Bound] : SRVs) {
+    for (auto it = SRVs.hazard_begin(); it != SRVs.hazard_end(); it++) {
+      const auto & [Slot, Bound] = *it;
       if (CheckOverlap(Bound.SRV.ptr(), pViewOrBuffer)) {
         if (SRVs.unbind(Slot)) {
           EmitST([=](ArgumentEncodingContext& enc) {
@@ -2710,8 +2710,9 @@ template <typename ContextInternalState> class MTLD3D11DeviceContextImplBase : p
     }
 
     if (pViewOrBuffer->bindFlags() & D3D11_BIND_VERTEX_BUFFER) {
-      // TODO: iterate _hazard_ set rather than _bound_ set (former is a subset)
-      for (const auto &[Slot, Bound] : state_.InputAssembler.VertexBuffers) {
+      for (auto it = state_.InputAssembler.VertexBuffers.hazard_begin();
+           it != state_.InputAssembler.VertexBuffers.hazard_end(); it++) {
+        const auto &[Slot, Bound] = *it;
         if (CheckOverlap(Bound.Buffer.ptr(), pViewOrBuffer)) {
           if (state_.InputAssembler.VertexBuffers.unbind(Slot)) {
             EmitST([=](ArgumentEncodingContext &enc) { enc.bindVertexBuffer(Slot, 0, 0, {}); });
@@ -3492,7 +3493,7 @@ template <typename ContextInternalState> class MTLD3D11DeviceContextImplBase : p
       auto pView = static_cast<D3D11ShaderResourceView *>(ppShaderResourceViews[slot - StartSlot]);
       if (pView && ValidateSRVHazard<Stage>(pView)) {
         bool replaced = false;
-        auto &entry = ShaderStage.SRVs.bind(slot, {pView}, replaced);
+        auto &entry = ShaderStage.SRVs.bind(slot, {pView}, replaced, !pView->hazardsFree());
         if (!replaced)
           continue;
         entry.SRV = pView;
@@ -3641,7 +3642,7 @@ template <typename ContextInternalState> class MTLD3D11DeviceContextImplBase : p
       auto pVertexBuffer = reinterpret_cast<D3D11ResourceCommon *>(ppVertexBuffers[slot - StartSlot]);
       if (pVertexBuffer && (pVertexBuffer->bindFlags() & D3D11_BIND_VERTEX_BUFFER) && ValidateIAHazard(pVertexBuffer)) {
         bool replaced = false;
-        auto &entry = VertexBuffers.bind(slot, {pVertexBuffer}, replaced);
+        auto &entry = VertexBuffers.bind(slot, {pVertexBuffer}, replaced, !pVertexBuffer->hazardsFree());
         if (!replaced) {
           if (pStrides && pStrides[slot - StartSlot] != entry.Stride) {
             VertexBuffers.set_dirty(slot);

src/dxmt/dxmt_binding_set.hpp

@@ -14,6 +14,7 @@ Element is required to be move-assignable
 template <typename Element, size_t NumElements> class BindingSet {
   bit::bitset<NumElements> dirty;
   bit::bitset<NumElements> bound;
+  bit::bitset<NumElements> hazard;
   std::array<Element, NumElements> storage;
 
 public:
@@ -22,13 +23,15 @@ template <typename Element, size_t NumElements> class BindingSet {
   BindingSet(BindingSet &&move) : storage(std::move(move.storage)) {
     bound = move.bound;
     dirty = move.bound; // intended behavior
+    hazard = move.hazard;
   };
 
   BindingSet &
   operator=(BindingSet &&move) {
     storage = std::move(move.storage);
     bound = move.bound;
     dirty = move.bound; // intended behavior
+    hazard = move.hazard;
     return *this;
   }
 
@@ -121,13 +124,14 @@ template <typename Element, size_t NumElements> class BindingSet {
   (so no initialization overhead if no replacement)
   */
   inline Element &
-  bind(size_t slot, Element &&element, bool &replacement) {
+  bind(size_t slot, Element &&element, bool &replacement, bool hazard = true) {
     if (bound.get(slot)) {
       if (redunant_binding_trait<Element>::is_redunant(storage[slot], element)) {
         return storage[slot];
       }
     } else {
       bound.set(slot, true);
+      this->hazard.set(slot, hazard);
     }
     // new (storage.data() + slot) Element(std::forward<Element>(element));
     // std::construct_at(storage.data() + slot, std::forward<Element>(element));
@@ -149,6 +153,7 @@ template <typename Element, size_t NumElements> class BindingSet {
       // at bind()
       storage[slot] = {};
       bound.set(slot, false);
+      hazard.set(slot, false);
       dirty.set(slot, true);
       return true;
     }
@@ -223,5 +228,14 @@ template <typename Element, size_t NumElements> class BindingSet {
   end() const {
     return NumElements;
   }
+
+  bound_iterator
+  hazard_begin() const {
+    return bound_iterator(*this, hazard, 0);
+  }
+  uint64_t
+  hazard_end() const {
+    return NumElements;
+  }
 };
 } // namespace dxmt