optimize choosing primary_ref_frame

nvpro-samples · Nov 21, 2024 · 283c843 · 283c843
1 parent 579000e
commit 283c843
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/vk_video_encoder/libs/VkVideoEncoder/VkEncoderDpbAV1.cpp b/vk_video_encoder/libs/VkVideoEncoder/VkEncoderDpbAV1.cpp
@@ -416,6 +416,8 @@ VkVideoEncoderAV1PrimaryRefType VkEncDpbAV1::GetPrimaryRefType(StdVideoAV1Refere
         return GLD_FRAME;
     } else if (refName == STD_VIDEO_AV1_REFERENCE_NAME_BWDREF_FRAME) {
         return BRF_FRAME;
+    } else if (m_maxRefFramesL1 > 0) {
+        return INT_ARF_FRAME;
     } else {
         return REGULAR_FRAME;
     }
@@ -431,6 +433,14 @@ int32_t VkEncDpbAV1::GetPrimaryRefBufId(VkVideoEncoderAV1PrimaryRefType primaryR
         refBufId = m_primaryRefBufIdMap[primaryRefType];
     }
 
+    if (refBufId == VkEncDpbAV1::INVALID_IDX) {
+        if (primaryRefType == INT_ARF_FRAME) {
+            refBufId = m_primaryRefBufIdMap[ARF_FRAME];
+        } else {
+            refBufId = m_primaryRefBufIdMap[(m_maxRefFramesL1 > 0) ? BRF_FRAME : REGULAR_FRAME];
+        }
+    }
+
     return refBufId;
 }