26 #define VPX_DISABLE_CTRL_TYPECHECKS 1
27 #define VPX_CODEC_DISABLE_COMPAT 1
28 #include <vpx/vpx_encoder.h>
29 #include <vpx/vp8cx.h>
70 [VP8E_UPD_ENTROPY] =
"VP8E_UPD_ENTROPY",
71 [VP8E_UPD_REFERENCE] =
"VP8E_UPD_REFERENCE",
72 [VP8E_USE_REFERENCE] =
"VP8E_USE_REFERENCE",
73 [VP8E_SET_ROI_MAP] =
"VP8E_SET_ROI_MAP",
74 [VP8E_SET_ACTIVEMAP] =
"VP8E_SET_ACTIVEMAP",
75 [VP8E_SET_SCALEMODE] =
"VP8E_SET_SCALEMODE",
76 [VP8E_SET_CPUUSED] =
"VP8E_SET_CPUUSED",
77 [VP8E_SET_ENABLEAUTOALTREF] =
"VP8E_SET_ENABLEAUTOALTREF",
78 [VP8E_SET_NOISE_SENSITIVITY] =
"VP8E_SET_NOISE_SENSITIVITY",
79 [VP8E_SET_SHARPNESS] =
"VP8E_SET_SHARPNESS",
80 [VP8E_SET_STATIC_THRESHOLD] =
"VP8E_SET_STATIC_THRESHOLD",
81 [VP8E_SET_TOKEN_PARTITIONS] =
"VP8E_SET_TOKEN_PARTITIONS",
82 [VP8E_GET_LAST_QUANTIZER] =
"VP8E_GET_LAST_QUANTIZER",
83 [VP8E_SET_ARNR_MAXFRAMES] =
"VP8E_SET_ARNR_MAXFRAMES",
84 [VP8E_SET_ARNR_STRENGTH] =
"VP8E_SET_ARNR_STRENGTH",
85 [VP8E_SET_ARNR_TYPE] =
"VP8E_SET_ARNR_TYPE",
91 const char *error = vpx_codec_error(&ctx->encoder);
92 const char *detail = vpx_codec_error_detail(&ctx->encoder);
100 const struct vpx_codec_enc_cfg *cfg)
105 av_log(avctx, level,
"vpx_codec_enc_cfg\n");
106 av_log(avctx, level,
"generic settings\n"
107 " %*s%u\n %*s%u\n %*s%u\n %*s%u\n %*s%u\n"
108 " %*s{%u/%u}\n %*s%u\n %*s%d\n %*s%u\n",
109 width,
"g_usage:", cfg->g_usage,
110 width,
"g_threads:", cfg->g_threads,
111 width,
"g_profile:", cfg->g_profile,
112 width,
"g_w:", cfg->g_w,
113 width,
"g_h:", cfg->g_h,
114 width,
"g_timebase:", cfg->g_timebase.num, cfg->g_timebase.den,
115 width,
"g_error_resilient:", cfg->g_error_resilient,
116 width,
"g_pass:", cfg->g_pass,
117 width,
"g_lag_in_frames:", cfg->g_lag_in_frames);
118 av_log(avctx, level,
"rate control settings\n"
119 " %*s%u\n %*s%u\n %*s%u\n %*s%u\n"
120 " %*s%d\n %*s%p(%zu)\n %*s%u\n",
121 width,
"rc_dropframe_thresh:", cfg->rc_dropframe_thresh,
122 width,
"rc_resize_allowed:", cfg->rc_resize_allowed,
123 width,
"rc_resize_up_thresh:", cfg->rc_resize_up_thresh,
124 width,
"rc_resize_down_thresh:", cfg->rc_resize_down_thresh,
125 width,
"rc_end_usage:", cfg->rc_end_usage,
126 width,
"rc_twopass_stats_in:", cfg->rc_twopass_stats_in.buf, cfg->rc_twopass_stats_in.sz,
127 width,
"rc_target_bitrate:", cfg->rc_target_bitrate);
128 av_log(avctx, level,
"quantizer settings\n"
130 width,
"rc_min_quantizer:", cfg->rc_min_quantizer,
131 width,
"rc_max_quantizer:", cfg->rc_max_quantizer);
132 av_log(avctx, level,
"bitrate tolerance\n"
134 width,
"rc_undershoot_pct:", cfg->rc_undershoot_pct,
135 width,
"rc_overshoot_pct:", cfg->rc_overshoot_pct);
136 av_log(avctx, level,
"decoder buffer model\n"
137 " %*s%u\n %*s%u\n %*s%u\n",
138 width,
"rc_buf_sz:", cfg->rc_buf_sz,
139 width,
"rc_buf_initial_sz:", cfg->rc_buf_initial_sz,
140 width,
"rc_buf_optimal_sz:", cfg->rc_buf_optimal_sz);
141 av_log(avctx, level,
"2 pass rate control settings\n"
142 " %*s%u\n %*s%u\n %*s%u\n",
143 width,
"rc_2pass_vbr_bias_pct:", cfg->rc_2pass_vbr_bias_pct,
144 width,
"rc_2pass_vbr_minsection_pct:", cfg->rc_2pass_vbr_minsection_pct,
145 width,
"rc_2pass_vbr_maxsection_pct:", cfg->rc_2pass_vbr_maxsection_pct);
146 av_log(avctx, level,
"keyframing settings\n"
147 " %*s%d\n %*s%u\n %*s%u\n",
148 width,
"kf_mode:", cfg->kf_mode,
149 width,
"kf_min_dist:", cfg->kf_min_dist,
150 width,
"kf_max_dist:", cfg->kf_max_dist);
151 av_log(avctx, level,
"\n");
182 enum vp8e_enc_control_id
id,
int val)
189 snprintf(buf,
sizeof(buf),
"%s:",
ctlidstr[
id]);
192 res = vpx_codec_control(&ctx->encoder,
id, val);
193 if (res != VPX_CODEC_OK) {
194 snprintf(buf,
sizeof(buf),
"Failed to set %s codec control",
199 return res == VPX_CODEC_OK ? 0 :
AVERROR(EINVAL);
206 vpx_codec_destroy(&ctx->encoder);
217 const struct vpx_codec_iface *iface = &vpx_codec_vp8_cx_algo;
218 struct vpx_codec_enc_cfg enccfg;
224 if ((res = vpx_codec_enc_config_default(iface, &enccfg, 0)) != VPX_CODEC_OK) {
226 vpx_codec_err_to_string(res));
231 enccfg.g_w = avctx->
width;
232 enccfg.g_h = avctx->
height;
237 if (ctx->lag_in_frames >= 0)
238 enccfg.g_lag_in_frames = ctx->lag_in_frames;
241 enccfg.g_pass = VPX_RC_FIRST_PASS;
243 enccfg.g_pass = VPX_RC_LAST_PASS;
245 enccfg.g_pass = VPX_RC_ONE_PASS;
249 enccfg.rc_end_usage = VPX_CBR;
253 enccfg.rc_min_quantizer = avctx->
qmin;
255 enccfg.rc_max_quantizer = avctx->
qmax;
260 enccfg.rc_2pass_vbr_minsection_pct =
263 enccfg.rc_2pass_vbr_maxsection_pct =
270 enccfg.rc_buf_initial_sz =
272 enccfg.rc_buf_optimal_sz = enccfg.rc_buf_sz * 5 / 6;
278 enccfg.kf_max_dist = avctx->
gop_size;
280 if (enccfg.g_pass == VPX_RC_FIRST_PASS)
281 enccfg.g_lag_in_frames = 0;
282 else if (enccfg.g_pass == VPX_RC_LAST_PASS) {
290 ctx->twopass_stats.sz = strlen(avctx->
stats_in) * 3 / 4;
291 ctx->twopass_stats.buf =
av_malloc(ctx->twopass_stats.sz);
292 if (!ctx->twopass_stats.buf) {
294 "Stat buffer alloc (%zu bytes) failed\n",
295 ctx->twopass_stats.sz);
299 ctx->twopass_stats.sz);
300 if (decode_size < 0) {
305 ctx->twopass_stats.sz = decode_size;
306 enccfg.rc_twopass_stats_in = ctx->twopass_stats;
313 enccfg.g_profile = avctx->
profile;
315 enccfg.g_error_resilient = ctx->error_resilient;
319 res = vpx_codec_enc_init(&ctx->encoder, iface, &enccfg, 0);
320 if (res != VPX_CODEC_OK) {
327 if (ctx->cpu_used != INT_MIN)
329 if (ctx->auto_alt_ref >= 0)
330 codecctl_int(avctx, VP8E_SET_ENABLEAUTOALTREF, ctx->auto_alt_ref);
331 if (ctx->arnr_max_frames >= 0)
332 codecctl_int(avctx, VP8E_SET_ARNR_MAXFRAMES, ctx->arnr_max_frames);
333 if (ctx->arnr_strength >= 0)
334 codecctl_int(avctx, VP8E_SET_ARNR_STRENGTH, ctx->arnr_strength);
335 if (ctx->arnr_type >= 0)
336 codecctl_int(avctx, VP8E_SET_ARNR_TYPE, ctx->arnr_type);
342 vpx_img_wrap(&ctx->rawimg, VPX_IMG_FMT_I420, avctx->
width, avctx->
height, 1,
355 const struct vpx_codec_cx_pkt *src)
357 dst->
pts = src->data.frame.pts;
358 dst->
duration = src->data.frame.duration;
359 dst->
flags = src->data.frame.flags;
360 dst->
sz = src->data.frame.sz;
361 dst->
buf = src->data.frame.buf;
373 uint8_t *buf,
int buf_size,
AVFrame *coded_frame)
375 if ((
int) cx_frame->
sz <= buf_size) {
376 buf_size = cx_frame->
sz;
377 memcpy(buf, cx_frame->
buf, buf_size);
378 coded_frame->
pts = cx_frame->
pts;
387 "Compressed frame larger than storage provided! (%zu/%d)\n",
388 cx_frame->
sz, buf_size);
406 const struct vpx_codec_cx_pkt *pkt;
407 const void *iter =
NULL;
410 if (ctx->coded_frame_list) {
413 size =
storeframe(avctx, cx_frame, buf, buf_size, coded_frame);
416 ctx->coded_frame_list = cx_frame->
next;
422 while ((pkt = vpx_codec_get_cx_data(&ctx->encoder, &iter))) {
424 case VPX_CODEC_CX_FRAME_PKT:
430 assert(!ctx->coded_frame_list);
432 size =
storeframe(avctx, &cx_frame, buf, buf_size, coded_frame);
441 "Frame queue element alloc failed\n");
447 if (!cx_frame->
buf) {
449 "Data buffer alloc (%zu bytes) failed\n",
453 memcpy(cx_frame->
buf, pkt->data.frame.buf, pkt->data.frame.sz);
457 case VPX_CODEC_STATS_PKT: {
458 struct vpx_fixed_buf *stats = &ctx->twopass_stats;
460 stats->sz + pkt->data.twopass_stats.sz);
465 memcpy((uint8_t*)stats->buf + stats->sz,
466 pkt->data.twopass_stats.buf, pkt->data.twopass_stats.sz);
467 stats->sz += pkt->data.twopass_stats.sz;
470 case VPX_CODEC_PSNR_PKT:
471 case VPX_CODEC_CUSTOM_PKT:
485 struct vpx_image *rawimg =
NULL;
486 int64_t timestamp = 0;
490 rawimg = &ctx->rawimg;
491 rawimg->planes[VPX_PLANE_Y] = frame->
data[0];
492 rawimg->planes[VPX_PLANE_U] = frame->
data[1];
493 rawimg->planes[VPX_PLANE_V] = frame->
data[2];
494 rawimg->stride[VPX_PLANE_Y] = frame->
linesize[0];
495 rawimg->stride[VPX_PLANE_U] = frame->
linesize[1];
496 rawimg->stride[VPX_PLANE_V] = frame->
linesize[2];
497 timestamp = frame->
pts;
500 res = vpx_codec_encode(&ctx->encoder, rawimg, timestamp,
502 if (res != VPX_CODEC_OK) {
518 ctx->twopass_stats.sz);
523 #define OFFSET(x) offsetof(VP8Context, x)
524 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
526 {
"cpu-used",
"Quality/Speed ratio modifier",
OFFSET(cpu_used),
AV_OPT_TYPE_INT, {INT_MIN}, INT_MIN, INT_MAX,
VE},
527 {
"auto-alt-ref",
"Enable use of alternate reference "
529 {
"lag-in-frames",
"Number of frames to look ahead for "
531 {
"arnr-maxframes",
"altref noise reduction max frame count",
OFFSET(arnr_max_frames),
AV_OPT_TYPE_INT, {-1}, -1, INT_MAX,
VE},
532 {
"arnr-strength",
"altref noise reduction filter strength",
OFFSET(arnr_strength),
AV_OPT_TYPE_INT, {-1}, -1, INT_MAX,
VE},
533 {
"arnr-type",
"altref noise reduction filter type",
OFFSET(arnr_type),
AV_OPT_TYPE_INT, {-1}, -1, INT_MAX,
VE,
"arnr_type"},
537 {
"deadline",
"Time to spend encoding, in microseconds.",
OFFSET(deadline),
AV_OPT_TYPE_INT, {VPX_DL_GOOD_QUALITY}, INT_MIN, INT_MAX,
VE,
"quality"},
541 {
"error-resilient",
"Error resilience configuration",
OFFSET(error_resilient),
AV_OPT_TYPE_FLAGS, {0}, INT_MIN, INT_MAX,
VE,
"er"},
542 #ifdef VPX_ERROR_RESILIENT_DEFAULT
543 {
"default",
"Improve resiliency against losses of whole frames", 0,
AV_OPT_TYPE_CONST, {VPX_ERROR_RESILIENT_DEFAULT}, 0, 0,
VE,
"er"},
544 {
"partitions",
"The frame partitions are independently decodable "
545 "by the bool decoder, meaning that partitions can be decoded even "
546 "though earlier partitions have been lost. Note that intra predicition"
547 " is still done over the partition boundary.", 0,
AV_OPT_TYPE_CONST, {VPX_ERROR_RESILIENT_PARTITIONS}, 0, 0,
VE,
"er"},
563 {
"keyint_min",
"-1" },
578 .priv_class = &
class,