[FFmpeg-devel] [PATCH] avformat/flvenc: write proper timestamps when muxing hevc

[Steven Liu]

Timo Rothenpieler <timo at rothenpieler.org> 于2023年7月25日周二 08:01写道：
>
> ---
>  libavformat/flvenc.c | 8 +++++++-
>  1 file changed, 7 insertions(+), 1 deletion(-)
>
> diff --git a/libavformat/flvenc.c b/libavformat/flvenc.c
> index 335d900415..41636ba1b8 100644
> --- a/libavformat/flvenc.c
> +++ b/libavformat/flvenc.c
> @@ -863,6 +863,9 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt)
>      else
>          flags_size = 1;
>
> +    if (par->codec_id == AV_CODEC_ID_HEVC && pkt->pts != pkt->dts)
> +        flags_size += 3;
> +
>      if (par->codec_id == AV_CODEC_ID_AAC || par->codec_id == AV_CODEC_ID_H264
>              || par->codec_id == AV_CODEC_ID_MPEG4 || par->codec_id == AV_CODEC_ID_HEVC
>              || par->codec_id == AV_CODEC_ID_AV1 || par->codec_id == AV_CODEC_ID_VP9) {
> @@ -998,8 +1001,11 @@ static int flv_write_packet(AVFormatContext *s, AVPacket *pkt)
>      } else {
>          av_assert1(flags>=0);
>          if (par->codec_id == AV_CODEC_ID_HEVC) {
> -            avio_w8(pb, FLV_IS_EX_HEADER | PacketTypeCodedFramesX | frametype); // ExVideoTagHeader mode with PacketTypeCodedFramesX
> +            int pkttype = (pkt->pts != pkt->dts) ? PacketTypeCodedFrames : PacketTypeCodedFramesX;
> +            avio_w8(pb, FLV_IS_EX_HEADER | pkttype | frametype); // ExVideoTagHeader mode with PacketTypeCodedFramesX
>              avio_write(pb, "hvc1", 4);
> +            if (pkttype == PacketTypeCodedFrames)
> +                avio_wb24(pb, pkt->pts - pkt->dts);
>          } else if (par->codec_id == AV_CODEC_ID_AV1 || par->codec_id == AV_CODEC_ID_VP9) {
>              avio_w8(pb, FLV_IS_EX_HEADER | PacketTypeCodedFrames | frametype);
>              avio_write(pb, par->codec_id == AV_CODEC_ID_AV1 ? "av01" : "vp09", 4);
> --
> 2.34.1
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request at ffmpeg.org with subject "unsubscribe".

LGTM