On Fri, 2024-06-07 at 15:01 +0200, Anton Khirnov wrote: > Specifically, calling hwaccel end_frame, verifying frame checksum, > and printing the frame-was-decoded message. > --- > libavcodec/hevc/hevcdec.c | 187 +++++++++++++++++++----------------- > -- > libavcodec/hevc/hevcdec.h | 1 - > 2 files changed, 91 insertions(+), 97 deletions(-) > > diff --git a/libavcodec/hevc/hevcdec.c b/libavcodec/hevc/hevcdec.c > index 7263b80a24..a8c2172674 100644 > --- a/libavcodec/hevc/hevcdec.c > +++ b/libavcodec/hevc/hevcdec.c > @@ -2942,7 +2942,6 @@ static int hevc_frame_start(HEVCContext *s) > ff_hevc_clear_refs(s); > } > > - s->is_decoded = 0; > s->slice_idx = 0; > s->first_nal_type = s->nal_unit_type; > s->poc = s->sh.poc; > @@ -3038,6 +3037,75 @@ fail: > return ret; > } > > +static int verify_md5(HEVCContext *s, AVFrame *frame) > +{ > + const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame- > >format); > + char msg_buf[4 * (50 + 2 * 2 * 16 /* MD5-size */)]; > + int pixel_shift; > + int err = 0; > + int i, j; > + > + if (!desc) > + return AVERROR(EINVAL); > + > + pixel_shift = desc->comp[0].depth > 8; > + > + /* the checksums are LE, so we have to byteswap for >8bpp > formats > + * on BE arches */ > +#if HAVE_BIGENDIAN > + if (pixel_shift && !s->checksum_buf) { > + av_fast_malloc(&s->checksum_buf, &s->checksum_buf_size, > + FFMAX3(frame->linesize[0], frame- > >linesize[1], > + frame->linesize[2])); > + if (!s->checksum_buf) > + return AVERROR(ENOMEM); > + } > +#endif > + > + msg_buf[0] = '\0'; > + for (i = 0; frame->data[i]; i++) { > + int width = s->avctx->coded_width; > + int height = s->avctx->coded_height; > + int w = (i == 1 || i == 2) ? (width >> desc->log2_chroma_w) > : width; > + int h = (i == 1 || i == 2) ? (height >> desc->log2_chroma_h) > : height; > + uint8_t md5[16]; > + > + av_md5_init(s->md5_ctx); > + for (j = 0; j < h; j++) { > + const uint8_t *src = frame->data[i] + j * frame- > >linesize[i]; > +#if HAVE_BIGENDIAN > + if (pixel_shift) { > + s->bdsp.bswap16_buf((uint16_t *) s->checksum_buf, > + (const uint16_t *) src, w); > + src = s->checksum_buf; > + } > +#endif > + av_md5_update(s->md5_ctx, src, w << pixel_shift); > + } > + av_md5_final(s->md5_ctx, md5); > + > +#define MD5_PRI "%016" PRIx64 "%016" PRIx64 > +#define MD5_PRI_ARG(buf) AV_RB64(buf), AV_RB64((const uint8_t*)(buf) > + 8) > + > + if (!memcmp(md5, s->sei.picture_hash.md5[i], 16)) { > + av_strlcatf(msg_buf, sizeof(msg_buf), > + "plane %d - correct " MD5_PRI "; ", > + i, MD5_PRI_ARG(md5)); > + } else { > + av_strlcatf(msg_buf, sizeof(msg_buf), > + "mismatching checksum of plane %d - " MD5_PRI > " != " MD5_PRI "; ", > + i, MD5_PRI_ARG(md5), MD5_PRI_ARG(s- > >sei.picture_hash.md5[i])); > + err = AVERROR_INVALIDDATA; > + } > + } > + > + av_log(s->avctx, err < 0 ? AV_LOG_ERROR : AV_LOG_DEBUG, > + "Verifying checksum for frame with POC %d: %s\n", > + s->poc, msg_buf); > + > + return err; > + } > + > static int hevc_frame_end(HEVCContext *s) > { > HEVCFrame *out = s->cur_frame; > @@ -3062,6 +3130,28 @@ static int hevc_frame_end(HEVCContext *s) > av_assert1(ret >= 0); > } > > + if (s->avctx->hwaccel) { > + ret = FF_HW_SIMPLE_CALL(s->avctx, end_frame);
Hwaccel will not decode slice data to get ctb count, frame_end will be always missed for hwaccel. Maybe better to keep hwaccel frame_end in its original location. Thanks Fei > + if (ret < 0) { > + av_log(s->avctx, AV_LOG_ERROR, > + "hardware accelerator failed to decode > picture\n"); > + ff_hevc_unref_frame(s->cur_frame, ~0); > + return ret; > + } > + } else { > + if (s->avctx->err_recognition & AV_EF_CRCCHECK && > + s->sei.picture_hash.is_md5) { > + ret = verify_md5(s, s->cur_frame->f); > + if (ret < 0 && s->avctx->err_recognition & > AV_EF_EXPLODE) { > + ff_hevc_unref_frame(s->cur_frame, ~0); > + return ret; > + } > + } > + } > + s->sei.picture_hash.is_md5 = 0; > + > + av_log(s->avctx, AV_LOG_DEBUG, "Decoded frame with POC %d.\n", > s->poc); > + > return 0; > } > > @@ -3109,7 +3199,6 @@ static int decode_slice(HEVCContext *s, const > H2645NAL *nal, GetBitContext *gb) > ret = hevc_frame_end(s); > if (ret < 0) > return ret; > - s->is_decoded = 1; > } > > return 0; > @@ -3290,75 +3379,6 @@ fail: > return ret; > } > > -static int verify_md5(HEVCContext *s, AVFrame *frame) > -{ > - const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(frame- > >format); > - char msg_buf[4 * (50 + 2 * 2 * 16 /* MD5-size */)]; > - int pixel_shift; > - int err = 0; > - int i, j; > - > - if (!desc) > - return AVERROR(EINVAL); > - > - pixel_shift = desc->comp[0].depth > 8; > - > - /* the checksums are LE, so we have to byteswap for >8bpp > formats > - * on BE arches */ > -#if HAVE_BIGENDIAN > - if (pixel_shift && !s->checksum_buf) { > - av_fast_malloc(&s->checksum_buf, &s->checksum_buf_size, > - FFMAX3(frame->linesize[0], frame- > >linesize[1], > - frame->linesize[2])); > - if (!s->checksum_buf) > - return AVERROR(ENOMEM); > - } > -#endif > - > - msg_buf[0] = '\0'; > - for (i = 0; frame->data[i]; i++) { > - int width = s->avctx->coded_width; > - int height = s->avctx->coded_height; > - int w = (i == 1 || i == 2) ? (width >> desc->log2_chroma_w) > : width; > - int h = (i == 1 || i == 2) ? (height >> desc->log2_chroma_h) > : height; > - uint8_t md5[16]; > - > - av_md5_init(s->md5_ctx); > - for (j = 0; j < h; j++) { > - const uint8_t *src = frame->data[i] + j * frame- > >linesize[i]; > -#if HAVE_BIGENDIAN > - if (pixel_shift) { > - s->bdsp.bswap16_buf((uint16_t *) s->checksum_buf, > - (const uint16_t *) src, w); > - src = s->checksum_buf; > - } > -#endif > - av_md5_update(s->md5_ctx, src, w << pixel_shift); > - } > - av_md5_final(s->md5_ctx, md5); > - > -#define MD5_PRI "%016" PRIx64 "%016" PRIx64 > -#define MD5_PRI_ARG(buf) AV_RB64(buf), AV_RB64((const uint8_t*)(buf) > + 8) > - > - if (!memcmp(md5, s->sei.picture_hash.md5[i], 16)) { > - av_strlcatf(msg_buf, sizeof(msg_buf), > - "plane %d - correct " MD5_PRI "; ", > - i, MD5_PRI_ARG(md5)); > - } else { > - av_strlcatf(msg_buf, sizeof(msg_buf), > - "mismatching checksum of plane %d - " MD5_PRI > " != " MD5_PRI "; ", > - i, MD5_PRI_ARG(md5), MD5_PRI_ARG(s- > >sei.picture_hash.md5[i])); > - err = AVERROR_INVALIDDATA; > - } > - } > - > - av_log(s->avctx, err < 0 ? AV_LOG_ERROR : AV_LOG_DEBUG, > - "Verifying checksum for frame with POC %d: %s\n", > - s->poc, msg_buf); > - > - return err; > -} > - > static int hevc_decode_extradata(HEVCContext *s, uint8_t *buf, int > length, int first) > { > int ret, i; > @@ -3424,31 +3444,6 @@ static int hevc_decode_frame(AVCodecContext > *avctx, AVFrame *rframe, > if (ret < 0) > return ret; > > - if (avctx->hwaccel) { > - if (s->cur_frame && (ret = FF_HW_SIMPLE_CALL(avctx, > end_frame)) < 0) { > - av_log(avctx, AV_LOG_ERROR, > - "hardware accelerator failed to decode > picture\n"); > - ff_hevc_unref_frame(s->cur_frame, ~0); > - return ret; > - } > - } else { > - /* verify the SEI checksum */ > - if (avctx->err_recognition & AV_EF_CRCCHECK && s->cur_frame > && s->is_decoded && > - s->sei.picture_hash.is_md5) { > - ret = verify_md5(s, s->cur_frame->f); > - if (ret < 0 && avctx->err_recognition & AV_EF_EXPLODE) { > - ff_hevc_unref_frame(s->cur_frame, ~0); > - return ret; > - } > - } > - } > - s->sei.picture_hash.is_md5 = 0; > - > - if (s->is_decoded) { > - av_log(avctx, AV_LOG_DEBUG, "Decoded frame with POC %d.\n", > s->poc); > - s->is_decoded = 0; > - } > - > if (s->output_frame->buf[0]) { > av_frame_move_ref(rframe, s->output_frame); > *got_output = 1; > diff --git a/libavcodec/hevc/hevcdec.h b/libavcodec/hevc/hevcdec.h > index f0443b3ab9..da4d83e661 100644 > --- a/libavcodec/hevc/hevcdec.h > +++ b/libavcodec/hevc/hevcdec.h > @@ -488,7 +488,6 @@ typedef struct HEVCContext { > int bs_width; > int bs_height; > > - int is_decoded; > // NoRaslOutputFlag associated with the last IRAP frame > int no_rasl_output_flag; > _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".