avcodec/h2645_sei: Factor updating H.2645 SEIs out

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2022-06-27 18:49:01 +02:00
parent 33239ebd07
commit de5891d371
6 changed files with 44 additions and 41 deletions

View File

@ -389,3 +389,32 @@ int ff_h2645_sei_message_decode(H2645SEI *h, enum SEIType type,
return FF_H2645_SEI_MESSAGE_UNHANDLED;
}
}
int ff_h2645_sei_ctx_replace(H2645SEI *dst, const H2645SEI *src)
{
int ret = av_buffer_replace(&dst->a53_caption.buf_ref,
src->a53_caption.buf_ref);
if (ret < 0)
return ret;
for (unsigned i = 0; i < dst->unregistered.nb_buf_ref; i++)
av_buffer_unref(&dst->unregistered.buf_ref[i]);
dst->unregistered.nb_buf_ref = 0;
if (src->unregistered.nb_buf_ref) {
ret = av_reallocp_array(&dst->unregistered.buf_ref,
src->unregistered.nb_buf_ref,
sizeof(*dst->unregistered.buf_ref));
if (ret < 0)
return ret;
for (unsigned i = 0; i < src->unregistered.nb_buf_ref; i++) {
dst->unregistered.buf_ref[i] = av_buffer_ref(src->unregistered.buf_ref[i]);
if (!dst->unregistered.buf_ref[i])
return AVERROR(ENOMEM);
dst->unregistered.nb_buf_ref++;
}
}
return 0;
}

View File

@ -130,4 +130,6 @@ int ff_h2645_sei_message_decode(H2645SEI *h, enum SEIType type,
enum AVCodecID codec_id, GetBitContext *gb,
GetByteContext *gbyte, void *logctx);
int ff_h2645_sei_ctx_replace(H2645SEI *dst, const H2645SEI *src);
#endif /* AVCODEC_H2645_SEI_H */

View File

@ -142,6 +142,12 @@ struct H264ParamSets;
int ff_h264_sei_decode(H264SEIContext *h, GetBitContext *gb,
const struct H264ParamSets *ps, void *logctx);
static inline int ff_h264_sei_ctx_replace(H264SEIContext *dst,
const H264SEIContext *src)
{
return ff_h2645_sei_ctx_replace(&dst->common, &src->common);
}
/**
* Reset SEI values at the beginning of the frame.
*/

View File

@ -434,29 +434,10 @@ int ff_h264_update_thread_context(AVCodecContext *dst,
h->frame_recovered = h1->frame_recovered;
ret = av_buffer_replace(&h->sei.common.a53_caption.buf_ref,
h1->sei.common.a53_caption.buf_ref);
ret = ff_h264_sei_ctx_replace(&h->sei, &h1->sei);
if (ret < 0)
return ret;
for (unsigned i = 0; i < h->sei.common.unregistered.nb_buf_ref; i++)
av_buffer_unref(&h->sei.common.unregistered.buf_ref[i]);
h->sei.common.unregistered.nb_buf_ref = 0;
if (h1->sei.common.unregistered.nb_buf_ref) {
ret = av_reallocp_array(&h->sei.common.unregistered.buf_ref,
h1->sei.common.unregistered.nb_buf_ref,
sizeof(*h->sei.common.unregistered.buf_ref));
if (ret < 0)
return ret;
for (unsigned i = 0; i < h1->sei.common.unregistered.nb_buf_ref; i++) {
h->sei.common.unregistered.buf_ref[i] = av_buffer_ref(h1->sei.common.unregistered.buf_ref[i]);
if (!h->sei.common.unregistered.buf_ref[i])
return AVERROR(ENOMEM);
h->sei.common.unregistered.nb_buf_ref++;
}
}
h->sei.common.unregistered.x264_build = h1->sei.common.unregistered.x264_build;
if (!h->cur_pic_ptr)

View File

@ -107,6 +107,11 @@ struct HEVCParamSets;
int ff_hevc_decode_nal_sei(GetBitContext *gb, void *logctx, HEVCSEI *s,
const struct HEVCParamSets *ps, enum HEVCNALUnitType type);
static inline int ff_hevc_sei_ctx_replace(HEVCSEI *dst, const HEVCSEI *src)
{
return ff_h2645_sei_ctx_replace(&dst->common, &src->common);
}
/**
* Reset SEI values that are stored on the Context.
* e.g. Caption data that was extracted during NAL

View File

@ -3729,30 +3729,10 @@ static int hevc_update_thread_context(AVCodecContext *dst,
s->max_ra = INT_MAX;
}
ret = av_buffer_replace(&s->sei.common.a53_caption.buf_ref,
s0->sei.common.a53_caption.buf_ref);
ret = ff_h2645_sei_ctx_replace(&s->sei.common, &s0->sei.common);
if (ret < 0)
return ret;
for (unsigned i = 0; i < s->sei.common.unregistered.nb_buf_ref; i++)
av_buffer_unref(&s->sei.common.unregistered.buf_ref[i]);
s->sei.common.unregistered.nb_buf_ref = 0;
if (s0->sei.common.unregistered.nb_buf_ref) {
ret = av_reallocp_array(&s->sei.common.unregistered.buf_ref,
s0->sei.common.unregistered.nb_buf_ref,
sizeof(*s->sei.common.unregistered.buf_ref));
if (ret < 0)
return ret;
for (unsigned i = 0; i < s0->sei.common.unregistered.nb_buf_ref; i++) {
s->sei.common.unregistered.buf_ref[i] = av_buffer_ref(s0->sei.common.unregistered.buf_ref[i]);
if (!s->sei.common.unregistered.buf_ref[i])
return AVERROR(ENOMEM);
s->sei.common.unregistered.nb_buf_ref++;
}
}
ret = av_buffer_replace(&s->sei.common.dynamic_hdr_plus.info,
s0->sei.common.dynamic_hdr_plus.info);
if (ret < 0)