diff mbox series

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

Message ID 20230905174336.3585-2-jamrial@gmail.com
State Accepted
Commit 8e972b13a97604d570ef26e8ac46767191d74f6f
Headers show
Series [FFmpeg-devel,1/4] Revert "avcodec/mpeg12dec: Do not alter avctx->rc_buffer_size" | expand

Commit Message

James Almer Sept. 5, 2023, 5:43 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 | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
diff mbox series

Patch

diff --git a/libavformat/demux.c b/libavformat/demux.c
index b218f64574..fcd5daf699 100644
--- a/libavformat/demux.c
+++ b/libavformat/demux.c
@@ -2974,6 +2974,23 @@  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_stream_add_side_data(st, AV_PKT_DATA_CPB_PROPERTIES,
+                                                (uint8_t *)props, cpb_size))
+                        av_free(props);
+                }
+            }
         }
 
         sti->avctx_inited = 0;