Commit 1616106f authored by Mark Thompson's avatar Mark Thompson

vaapi_encode_vp9: Move options and common structures into context

parent 58d3ac31
...@@ -33,21 +33,25 @@ ...@@ -33,21 +33,25 @@
typedef struct VAAPIEncodeVP9Context { typedef struct VAAPIEncodeVP9Context {
VAAPIEncodeContext common;
// User options.
int loop_filter_level;
int loop_filter_sharpness;
// Derived settings.
int q_idx_idr; int q_idx_idr;
int q_idx_p; int q_idx_p;
int q_idx_b; int q_idx_b;
// Stream state.
// Reference direction for B-like frames: // Reference direction for B-like frames:
// 0 - most recent P/IDR frame is last. // 0 - most recent P/IDR frame is last.
// 1 - most recent P frame is golden. // 1 - most recent P frame is golden.
int last_ref_dir; int last_ref_dir;
} VAAPIEncodeVP9Context; } VAAPIEncodeVP9Context;
typedef struct VAAPIEncodeVP9Options {
int loop_filter_level;
int loop_filter_sharpness;
} VAAPIEncodeVP9Options;
#define vseq_var(name) vseq->name, name #define vseq_var(name) vseq->name, name
#define vseq_field(name) vseq->seq_fields.bits.name, name #define vseq_field(name) vseq->seq_fields.bits.name, name
...@@ -82,10 +86,8 @@ static int vaapi_encode_vp9_init_sequence_params(AVCodecContext *avctx) ...@@ -82,10 +86,8 @@ static int vaapi_encode_vp9_init_sequence_params(AVCodecContext *avctx)
static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx, static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx,
VAAPIEncodePicture *pic) VAAPIEncodePicture *pic)
{ {
VAAPIEncodeContext *ctx = avctx->priv_data; VAAPIEncodeVP9Context *priv = avctx->priv_data;
VAEncPictureParameterBufferVP9 *vpic = pic->codec_picture_params; VAEncPictureParameterBufferVP9 *vpic = pic->codec_picture_params;
VAAPIEncodeVP9Context *priv = ctx->priv_data;
VAAPIEncodeVP9Options *opt = ctx->codec_options;
int i; int i;
vpic->reconstructed_frame = pic->recon_surface; vpic->reconstructed_frame = pic->recon_surface;
...@@ -169,8 +171,8 @@ static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx, ...@@ -169,8 +171,8 @@ static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx,
vpic->chroma_ac_qindex_delta = 0; vpic->chroma_ac_qindex_delta = 0;
vpic->chroma_dc_qindex_delta = 0; vpic->chroma_dc_qindex_delta = 0;
vpic->filter_level = opt->loop_filter_level; vpic->filter_level = priv->loop_filter_level;
vpic->sharpness_level = opt->loop_filter_sharpness; vpic->sharpness_level = priv->loop_filter_sharpness;
if (avctx->max_b_frames > 0 && pic->type == PICTURE_TYPE_P) if (avctx->max_b_frames > 0 && pic->type == PICTURE_TYPE_P)
priv->last_ref_dir = !priv->last_ref_dir; priv->last_ref_dir = !priv->last_ref_dir;
...@@ -180,8 +182,7 @@ static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx, ...@@ -180,8 +182,7 @@ static int vaapi_encode_vp9_init_picture_params(AVCodecContext *avctx,
static av_cold int vaapi_encode_vp9_configure(AVCodecContext *avctx) static av_cold int vaapi_encode_vp9_configure(AVCodecContext *avctx)
{ {
VAAPIEncodeContext *ctx = avctx->priv_data; VAAPIEncodeVP9Context *priv = avctx->priv_data;
VAAPIEncodeVP9Context *priv = ctx->priv_data;
priv->q_idx_p = av_clip(avctx->global_quality, 0, VP9_MAX_QUANT); priv->q_idx_p = av_clip(avctx->global_quality, 0, VP9_MAX_QUANT);
if (avctx->i_quant_factor > 0.0) if (avctx->i_quant_factor > 0.0)
...@@ -266,8 +267,7 @@ static av_cold int vaapi_encode_vp9_init(AVCodecContext *avctx) ...@@ -266,8 +267,7 @@ static av_cold int vaapi_encode_vp9_init(AVCodecContext *avctx)
return ff_vaapi_encode_init(avctx); return ff_vaapi_encode_init(avctx);
} }
#define OFFSET(x) (offsetof(VAAPIEncodeContext, codec_options_data) + \ #define OFFSET(x) offsetof(VAAPIEncodeVP9Context, x)
offsetof(VAAPIEncodeVP9Options, x))
#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM) #define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)
static const AVOption vaapi_encode_vp9_options[] = { static const AVOption vaapi_encode_vp9_options[] = {
{ "loop_filter_level", "Loop filter level", { "loop_filter_level", "Loop filter level",
...@@ -298,8 +298,7 @@ AVCodec ff_vp9_vaapi_encoder = { ...@@ -298,8 +298,7 @@ AVCodec ff_vp9_vaapi_encoder = {
.long_name = NULL_IF_CONFIG_SMALL("VP9 (VAAPI)"), .long_name = NULL_IF_CONFIG_SMALL("VP9 (VAAPI)"),
.type = AVMEDIA_TYPE_VIDEO, .type = AVMEDIA_TYPE_VIDEO,
.id = AV_CODEC_ID_VP9, .id = AV_CODEC_ID_VP9,
.priv_data_size = (sizeof(VAAPIEncodeContext) + .priv_data_size = sizeof(VAAPIEncodeVP9Context),
sizeof(VAAPIEncodeVP9Options)),
.init = &vaapi_encode_vp9_init, .init = &vaapi_encode_vp9_init,
.encode2 = &ff_vaapi_encode2, .encode2 = &ff_vaapi_encode2,
.close = &ff_vaapi_encode_close, .close = &ff_vaapi_encode_close,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment