vaapi_encode: Add support for writing arbitrary additional packed headers

This commit is contained in:
Mark Thompson 2016-04-09 16:48:27 +01:00
parent 081961f819
commit 19d7667a81
2 changed files with 25 additions and 0 deletions

View File

@ -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));

View File

@ -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;