diff mbox series

[FFmpeg-devel,12/17] avformat/demux: propagate the internal decoder's bitrate properties

Message ID 20230904221116.21933-1-jamrial@gmail.com
State New
Headers show
Series [FFmpeg-devel,01/17] avcodec/avcodec: add side data to AVCodecContext | expand

Commit Message

James Almer Sept. 4, 2023, 10:11 p.m. UTC 
Muxers may access this information through cpb properties within the stream's
side data.

Signed-off-by: James Almer <jamrial@gmail.com>
---
 libavformat/demux.c | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
diff mbox series

Patch

diff --git a/libavformat/demux.c b/libavformat/demux.c
index 94128b00bc..606ccbd180 100644
--- a/libavformat/demux.c
+++ b/libavformat/demux.c
@@ -2976,6 +2976,24 @@  int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
             ret = add_coded_side_data(st, sti->avctx);
             if (ret < 0)
                 goto find_stream_info_err;
+
+            if (sti->avctx->rc_buffer_size > 0 || sti->avctx->rc_max_rate > 0 ||
+                sti->avctx->rc_min_rate) {
+                size_t cpb_size;
+                AVCPBProperties *props = av_cpb_properties_alloc(&cpb_size);
+                if (props) {
+                    if (sti->avctx->rc_buffer_size > 0)
+                        props->buffer_size = sti->avctx->rc_buffer_size;
+                    if (sti->avctx->rc_min_rate > 0)
+                        props->min_bitrate = sti->avctx->rc_min_rate;
+                    if (sti->avctx->rc_max_rate > 0)
+                        props->max_bitrate = sti->avctx->rc_max_rate;
+                    if (!av_packet_side_data_set_add(&st->codecpar->side_data,
+                                                     AV_PKT_DATA_CPB_PROPERTIES,
+                                                     (uint8_t *)props, cpb_size, 0))
+                        av_free(props);
+                }
+            }
         }
 
         sti->avctx_inited = 0;