diff --git a/libavcodec/vaapi_encode.c b/libavcodec/vaapi_encode.c index 949051660c7a32b157b6d4cae41e9facb6ae1d6b..c55a78361cafb77562aa358ef51d9fe64ed6284e 100644 --- a/libavcodec/vaapi_encode.c +++ b/libavcodec/vaapi_encode.c @@ -293,6 +293,27 @@ static int vaapi_encode_issue(AVCodecContext *avctx, } } + if (ctx->codec->write_extra_header) { + for (i = 0;; i++) { + int type; + bit_len = 8 * sizeof(data); + err = ctx->codec->write_extra_header(avctx, pic, i, &type, + data, &bit_len); + if (err == AVERROR_EOF) + break; + if (err < 0) { + av_log(avctx, AV_LOG_ERROR, "Failed to write extra " + "header %d: %d.\n", i, err); + goto fail; + } + + err = vaapi_encode_make_packed_header(avctx, pic, type, + data, bit_len); + if (err < 0) + goto fail; + } + } + av_assert0(pic->nb_slices <= MAX_PICTURE_SLICES); for (i = 0; i < pic->nb_slices; i++) { slice = av_mallocz(sizeof(*slice)); diff --git a/libavcodec/vaapi_encode.h b/libavcodec/vaapi_encode.h index b67168b1ef8f0b06ed52f6cf9b6218ae59f86500..8f4f653812433aee1bc810bfce5f8c9f9e8583ab 100644 --- a/libavcodec/vaapi_encode.h +++ b/libavcodec/vaapi_encode.h @@ -215,6 +215,10 @@ typedef struct VAAPIEncodeType { VAAPIEncodePicture *pic, int index, int *type, char *data, size_t *data_len); + int (*write_extra_header)(AVCodecContext *avctx, + VAAPIEncodePicture *pic, + int index, int *type, + char *data, size_t *data_len); } VAAPIEncodeType;