From: spartazhc <[email protected]> Add av_packet_clean to remove AVPackets whose stream_index is not in st_index list.
Generally s->internal->packet_buffer may have pkts from different stream, and stream_index will be used to discard pkt that is not needed. But in case of abr, several streams may pass the stream_index check. So we need a function to remove AVPackets not needed in pktl added by hls_read_header. Signed-off-by: spartazhc <[email protected]> --- libavformat/avformat.h | 9 +++++++ libavformat/internal.h | 15 +++++++++++ libavformat/utils.c | 59 ++++++++++++++++++++++++++++++++++++++++++ libavformat/version.h | 2 +- 4 files changed, 84 insertions(+), 1 deletion(-) diff --git a/libavformat/avformat.h b/libavformat/avformat.h index e91e7f1d33..90dee0d075 100644 --- a/libavformat/avformat.h +++ b/libavformat/avformat.h @@ -2474,6 +2474,15 @@ int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts, int */ int avformat_flush(AVFormatContext *s); +/** + * Remove the AVPackets do not needed in the packet list. + * The behaviour is undefined if the packet list is empty. + * + * @param s media file handle + * @param st_index the stream_index list which is needed + */ +int av_packet_clean(AVFormatContext *s, int *st_index); + /** * Start playing a network-based stream (e.g. RTSP stream) at the * current position. diff --git a/libavformat/internal.h b/libavformat/internal.h index 17a6ab07d3..ac943b1441 100644 --- a/libavformat/internal.h +++ b/libavformat/internal.h @@ -764,6 +764,21 @@ int ff_packet_list_put(AVPacketList **head, AVPacketList **tail, int ff_packet_list_get(AVPacketList **head, AVPacketList **tail, AVPacket *pkt); +/** + * Remove the AVPackets do not needed in the packet list. + * The behaviour is undefined if the packet list is empty. + * + * @note The pkt will be overwritten completely. The caller owns the + * packet and must unref it by itself. + * + * @param head List head element + * @param tail List tail element + * @param st_index the stream_index list which is needed + * @return 0 on success. Success is guaranteed + * if the packet list is not empty. + */ +int ff_packet_list_clean(AVPacketList **head, AVPacketList **tail, + int *st_index); /** * Wipe the list and unref all the packets in it. * diff --git a/libavformat/utils.c b/libavformat/utils.c index 45a4179552..26b5a08a19 100644 --- a/libavformat/utils.c +++ b/libavformat/utils.c @@ -1565,6 +1565,65 @@ int ff_packet_list_get(AVPacketList **pkt_buffer, return 0; } +/** + * return 1 if needed + */ +static int ff_check_st_index(int st, int *st_index) +{ + for (int i = 0; i < AVMEDIA_TYPE_NB; ++i) { + if (st_index[i] == st) + return 1; + } + return 0; +} + +int ff_packet_list_clean(AVPacketList **pkt_buffer, + AVPacketList **pkt_buffer_end, + int *st_index) +{ + AVPacketList *pktl, *pktn; + av_assert0(*pkt_buffer); + pktl = *pkt_buffer; + pktn = pktl->next; + + /* num >= 3 */ + while (pktn && pktn != *pkt_buffer_end) { + if (!ff_check_st_index(pktn->pkt.stream_index, st_index)) { + av_packet_unref(&pktn->pkt); + pktl->next = pktn->next; + av_freep(pktn); + pktn = pktl->next; + } else { + pktl = pktn; + pktn = pktn->next; + } + } + /* last one*/ + if (pktn && !ff_check_st_index(pktn->pkt.stream_index, st_index)) { + av_packet_unref(&pktn->pkt); + av_freep(pktn); + pktl->next = NULL; + *pkt_buffer_end = pktl; + } + /* first one*/ + pktl = *pkt_buffer; + if (!ff_check_st_index(pktl->pkt.stream_index, st_index)) { + av_packet_unref(&pktl->pkt); + *pkt_buffer = pktl->next; + if (!pktl->next) + *pkt_buffer_end = NULL; + av_freep(&pktl); + } + + return 0; +} + +int av_packet_clean(AVFormatContext *s, int *st_index) { + int ret = ff_packet_list_clean(&s->internal->packet_buffer, + &s->internal->packet_buffer_end, st_index); + return ret; +} + static int64_t ts_to_samples(AVStream *st, int64_t ts) { return av_rescale(ts, st->time_base.num * st->codecpar->sample_rate, st->time_base.den); diff --git a/libavformat/version.h b/libavformat/version.h index 3c1957b00c..75c03fde0a 100644 --- a/libavformat/version.h +++ b/libavformat/version.h @@ -32,7 +32,7 @@ // Major bumping may affect Ticket5467, 5421, 5451(compatibility with Chromium) // Also please add any ticket numbers that you believe might be affected here #define LIBAVFORMAT_VERSION_MAJOR 58 -#define LIBAVFORMAT_VERSION_MINOR 47 +#define LIBAVFORMAT_VERSION_MINOR 48 #define LIBAVFORMAT_VERSION_MICRO 100 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \ -- 2.27.0 _______________________________________________ ffmpeg-devel mailing list [email protected] https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email [email protected] with subject "unsubscribe".
