EncoderConfigAV1: Fix GetMaxBFrameCount

To support driver which does not support B Frame, GetMaxBFrameCount has been introduced in nvpro-samples#103 but AV1 was not supported yet.
dabrain34 · Nov 27, 2024 · 1d4eb3e · 1d4eb3e
1 parent d6fe796
commit 1d4eb3e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vk_video_encoder/libs/VkVideoEncoder/VkEncoderConfigAV1.h b/vk_video_encoder/libs/VkVideoEncoder/VkEncoderConfigAV1.h
@@ -117,6 +117,8 @@ struct EncoderConfigAV1 : public EncoderConfig {
 
     virtual bool InitRateControl() override;
 
+    virtual uint8_t GetMaxBFrameCount() { return  static_cast<uint8_t>(av1EncodeCapabilities.maxBidirectionalCompoundReferenceCount); }
+
     bool GetRateControlParameters(VkVideoEncodeRateControlInfoKHR* rcInfo,
                                   VkVideoEncodeRateControlLayerInfoKHR* rcLayerInfo,
                                   VkVideoEncodeAV1RateControlInfoKHR* rcInfoAV1,