EncoderConfigAV1: Fix GetMaxBFrameCount

To support driver which does not support B Frame, GetMaxBFrameCount has been introduced in #103 but AV1 was not supported yet. Signed-off-by: Stéphane Cerveau <[email protected]>
nvpro-samples · Nov 27, 2024 · 83b5cd1 · 83b5cd1
1 parent eb97ba3
commit 83b5cd1
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,