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

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