26 #define AOM_DISABLE_CTRL_TYPECHECKS 1
27 #include <aom/aom_encoder.h>
28 #include <aom/aomcx.h>
138 [AOME_SET_CPUUSED] =
"AOME_SET_CPUUSED",
139 [AOME_SET_CQ_LEVEL] =
"AOME_SET_CQ_LEVEL",
140 [AOME_SET_ENABLEAUTOALTREF] =
"AOME_SET_ENABLEAUTOALTREF",
141 [AOME_SET_ARNR_MAXFRAMES] =
"AOME_SET_ARNR_MAXFRAMES",
142 [AOME_SET_ARNR_STRENGTH] =
"AOME_SET_ARNR_STRENGTH",
143 [AOME_SET_STATIC_THRESHOLD] =
"AOME_SET_STATIC_THRESHOLD",
144 [AV1E_SET_COLOR_RANGE] =
"AV1E_SET_COLOR_RANGE",
145 [AV1E_SET_COLOR_PRIMARIES] =
"AV1E_SET_COLOR_PRIMARIES",
146 [AV1E_SET_MATRIX_COEFFICIENTS] =
"AV1E_SET_MATRIX_COEFFICIENTS",
147 [AV1E_SET_TRANSFER_CHARACTERISTICS] =
"AV1E_SET_TRANSFER_CHARACTERISTICS",
148 [AV1E_SET_AQ_MODE] =
"AV1E_SET_AQ_MODE",
149 [AV1E_SET_FRAME_PARALLEL_DECODING] =
"AV1E_SET_FRAME_PARALLEL_DECODING",
150 [AV1E_SET_SUPERBLOCK_SIZE] =
"AV1E_SET_SUPERBLOCK_SIZE",
151 [AV1E_SET_TILE_COLUMNS] =
"AV1E_SET_TILE_COLUMNS",
152 [AV1E_SET_TILE_ROWS] =
"AV1E_SET_TILE_ROWS",
153 [AV1E_SET_ENABLE_RESTORATION] =
"AV1E_SET_ENABLE_RESTORATION",
154 #ifdef AOM_CTRL_AV1E_SET_ROW_MT
155 [AV1E_SET_ROW_MT] =
"AV1E_SET_ROW_MT",
157 #ifdef AOM_CTRL_AV1E_SET_DENOISE_NOISE_LEVEL
158 [AV1E_SET_DENOISE_NOISE_LEVEL] =
"AV1E_SET_DENOISE_NOISE_LEVEL",
160 #ifdef AOM_CTRL_AV1E_SET_DENOISE_BLOCK_SIZE
161 [AV1E_SET_DENOISE_BLOCK_SIZE] =
"AV1E_SET_DENOISE_BLOCK_SIZE",
163 #ifdef AOM_CTRL_AV1E_SET_MAX_REFERENCE_FRAMES
164 [AV1E_SET_MAX_REFERENCE_FRAMES] =
"AV1E_SET_MAX_REFERENCE_FRAMES",
166 #ifdef AOM_CTRL_AV1E_SET_ENABLE_GLOBAL_MOTION
167 [AV1E_SET_ENABLE_GLOBAL_MOTION] =
"AV1E_SET_ENABLE_GLOBAL_MOTION",
169 #ifdef AOM_CTRL_AV1E_SET_ENABLE_INTRABC
170 [AV1E_SET_ENABLE_INTRABC] =
"AV1E_SET_ENABLE_INTRABC",
172 [AV1E_SET_ENABLE_CDEF] =
"AV1E_SET_ENABLE_CDEF",
173 [AOME_SET_TUNING] =
"AOME_SET_TUNING",
174 #if AOM_ENCODER_ABI_VERSION >= 22
175 [AV1E_SET_ENABLE_1TO4_PARTITIONS] =
"AV1E_SET_ENABLE_1TO4_PARTITIONS",
176 [AV1E_SET_ENABLE_AB_PARTITIONS] =
"AV1E_SET_ENABLE_AB_PARTITIONS",
177 [AV1E_SET_ENABLE_RECT_PARTITIONS] =
"AV1E_SET_ENABLE_RECT_PARTITIONS",
178 [AV1E_SET_ENABLE_ANGLE_DELTA] =
"AV1E_SET_ENABLE_ANGLE_DELTA",
179 [AV1E_SET_ENABLE_CFL_INTRA] =
"AV1E_SET_ENABLE_CFL_INTRA",
180 [AV1E_SET_ENABLE_FILTER_INTRA] =
"AV1E_SET_ENABLE_FILTER_INTRA",
181 [AV1E_SET_ENABLE_INTRA_EDGE_FILTER] =
"AV1E_SET_ENABLE_INTRA_EDGE_FILTER",
182 [AV1E_SET_ENABLE_PAETH_INTRA] =
"AV1E_SET_ENABLE_PAETH_INTRA",
183 [AV1E_SET_ENABLE_SMOOTH_INTRA] =
"AV1E_SET_ENABLE_SMOOTH_INTRA",
184 [AV1E_SET_ENABLE_PALETTE] =
"AV1E_SET_ENABLE_PALETTE",
185 [AV1E_SET_ENABLE_FLIP_IDTX] =
"AV1E_SET_ENABLE_FLIP_IDTX",
186 [AV1E_SET_ENABLE_TX64] =
"AV1E_SET_ENABLE_TX64",
187 [AV1E_SET_INTRA_DCT_ONLY] =
"AV1E_SET_INTRA_DCT_ONLY",
188 [AV1E_SET_INTER_DCT_ONLY] =
"AV1E_SET_INTER_DCT_ONLY",
189 [AV1E_SET_INTRA_DEFAULT_TX_ONLY] =
"AV1E_SET_INTRA_DEFAULT_TX_ONLY",
190 [AV1E_SET_REDUCED_TX_TYPE_SET] =
"AV1E_SET_REDUCED_TX_TYPE_SET",
191 [AV1E_SET_ENABLE_DIFF_WTD_COMP] =
"AV1E_SET_ENABLE_DIFF_WTD_COMP",
192 [AV1E_SET_ENABLE_DIST_WTD_COMP] =
"AV1E_SET_ENABLE_DIST_WTD_COMP",
193 [AV1E_SET_ENABLE_DUAL_FILTER] =
"AV1E_SET_ENABLE_DUAL_FILTER",
194 [AV1E_SET_ENABLE_INTERINTER_WEDGE] =
"AV1E_SET_ENABLE_INTERINTER_WEDGE",
195 [AV1E_SET_ENABLE_INTERINTRA_WEDGE] =
"AV1E_SET_ENABLE_INTERINTRA_WEDGE",
196 [AV1E_SET_ENABLE_MASKED_COMP] =
"AV1E_SET_ENABLE_MASKED_COMP",
197 [AV1E_SET_ENABLE_INTERINTRA_COMP] =
"AV1E_SET_ENABLE_INTERINTRA_COMP",
198 [AV1E_SET_ENABLE_OBMC] =
"AV1E_SET_ENABLE_OBMC",
199 [AV1E_SET_ENABLE_ONESIDED_COMP] =
"AV1E_SET_ENABLE_ONESIDED_COMP",
200 [AV1E_SET_REDUCED_REFERENCE_SET] =
"AV1E_SET_REDUCED_REFERENCE_SET",
201 [AV1E_SET_ENABLE_SMOOTH_INTERINTRA] =
"AV1E_SET_ENABLE_SMOOTH_INTERINTRA",
202 [AV1E_SET_ENABLE_REF_FRAME_MVS] =
"AV1E_SET_ENABLE_REF_FRAME_MVS",
204 #ifdef AOM_CTRL_AV1E_GET_NUM_OPERATING_POINTS
205 [AV1E_GET_NUM_OPERATING_POINTS] =
"AV1E_GET_NUM_OPERATING_POINTS",
207 #ifdef AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX
208 [AV1E_GET_SEQ_LEVEL_IDX] =
"AV1E_GET_SEQ_LEVEL_IDX",
210 #ifdef AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX
211 [AV1E_GET_TARGET_SEQ_LEVEL_IDX] =
"AV1E_GET_TARGET_SEQ_LEVEL_IDX",
213 [AV1_GET_NEW_FRAME_IMAGE] =
"AV1_GET_NEW_FRAME_IMAGE",
219 const char *
error = aom_codec_error(&
ctx->encoder);
220 const char *detail = aom_codec_error_detail(&
ctx->encoder);
228 const struct aom_codec_enc_cfg *cfg,
235 " %*s%u\n %*s%u\n %*s%u\n %*s%u\n %*s%u\n"
237 " %*s{%u/%u}\n %*s%u\n %*s%d\n %*s%u\n",
238 width,
"g_usage:", cfg->g_usage,
239 width,
"g_threads:", cfg->g_threads,
240 width,
"g_profile:", cfg->g_profile,
241 width,
"g_w:", cfg->g_w,
242 width,
"g_h:", cfg->g_h,
243 width,
"g_bit_depth:", cfg->g_bit_depth,
244 width,
"g_input_bit_depth:", cfg->g_input_bit_depth,
245 width,
"g_timebase:", cfg->g_timebase.num, cfg->g_timebase.den,
246 width,
"g_error_resilient:", cfg->g_error_resilient,
247 width,
"g_pass:", cfg->g_pass,
248 width,
"g_lag_in_frames:", cfg->g_lag_in_frames);
251 width,
"rc_dropframe_thresh:", cfg->rc_dropframe_thresh,
252 width,
"rc_end_usage:", cfg->rc_end_usage,
253 width,
"rc_twopass_stats_in:", cfg->rc_twopass_stats_in.buf, cfg->rc_twopass_stats_in.sz,
254 width,
"rc_target_bitrate:", cfg->rc_target_bitrate);
257 width,
"rc_min_quantizer:", cfg->rc_min_quantizer,
258 width,
"rc_max_quantizer:", cfg->rc_max_quantizer);
261 width,
"rc_undershoot_pct:", cfg->rc_undershoot_pct,
262 width,
"rc_overshoot_pct:", cfg->rc_overshoot_pct);
264 " %*s%u\n %*s%u\n %*s%u\n",
265 width,
"rc_buf_sz:", cfg->rc_buf_sz,
266 width,
"rc_buf_initial_sz:", cfg->rc_buf_initial_sz,
267 width,
"rc_buf_optimal_sz:", cfg->rc_buf_optimal_sz);
268 av_log(avctx,
level,
"2 pass rate control settings\n"
269 " %*s%u\n %*s%u\n %*s%u\n",
270 width,
"rc_2pass_vbr_bias_pct:", cfg->rc_2pass_vbr_bias_pct,
271 width,
"rc_2pass_vbr_minsection_pct:", cfg->rc_2pass_vbr_minsection_pct,
272 width,
"rc_2pass_vbr_maxsection_pct:", cfg->rc_2pass_vbr_maxsection_pct);
274 " %*s%d\n %*s%u\n %*s%u\n",
275 width,
"kf_mode:", cfg->kf_mode,
276 width,
"kf_min_dist:", cfg->kf_min_dist,
277 width,
"kf_max_dist:", cfg->kf_max_dist);
280 width,
"tile_width_count:", cfg->tile_width_count,
281 width,
"tile_height_count:", cfg->tile_height_count);
314 aome_enc_control_id
id,
316 enum aome_enc_control_id
id,
328 res = aom_codec_control(&
ctx->encoder,
id,
val);
329 if (res != AOM_CODEC_OK) {
339 #if defined(AOM_CTRL_AV1E_GET_NUM_OPERATING_POINTS) && \
340 defined(AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX) && \
341 defined(AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX)
344 aome_enc_control_id
id,
346 enum aome_enc_control_id
id,
358 res = aom_codec_control(&
ctx->encoder,
id, ptr);
359 if (res != AOM_CODEC_OK) {
372 aome_enc_control_id
id,
374 enum aome_enc_control_id
id,
376 struct aom_image *
img)
384 res = aom_codec_control(&
ctx->encoder,
id,
img);
385 if (res != AOM_CODEC_OK) {
399 #if defined(AOM_CTRL_AV1E_GET_NUM_OPERATING_POINTS) && \
400 defined(AOM_CTRL_AV1E_GET_SEQ_LEVEL_IDX) && \
401 defined(AOM_CTRL_AV1E_GET_TARGET_SEQ_LEVEL_IDX)
403 int num_operating_points;
405 int target_levels[32];
407 if (!codecctl_intp(avctx, AV1E_GET_NUM_OPERATING_POINTS,
408 &num_operating_points) &&
409 !codecctl_intp(avctx, AV1E_GET_SEQ_LEVEL_IDX, levels) &&
410 !codecctl_intp(avctx, AV1E_GET_TARGET_SEQ_LEVEL_IDX,
412 for (
int i = 0;
i < num_operating_points;
i++) {
413 if (levels[
i] > target_levels[
i]) {
416 "Could not encode to target level %d.%d for "
417 "operating point %d. The output level is %d.%d.\n",
418 2 + (target_levels[
i] >> 2), target_levels[
i] & 3,
419 i, 2 + (levels[
i] >> 2), levels[
i] & 3);
420 }
else if (target_levels[
i] < 31) {
423 "Output level for operating point %d is %d.%d.\n",
424 i, 2 + (levels[
i] >> 2), levels[
i] & 3);
431 aom_codec_destroy(&
ctx->encoder);
440 struct aom_codec_enc_cfg *enccfg, aom_codec_flags_t *
flags,
441 aom_img_fmt_t *img_fmt)
445 enccfg->g_bit_depth = enccfg->g_input_bit_depth =
desc->comp[0].depth;
448 enccfg->monochrome = 1;
452 *img_fmt = AOM_IMG_FMT_I420;
456 *img_fmt = AOM_IMG_FMT_I422;
461 *img_fmt = AOM_IMG_FMT_I444;
465 enccfg->monochrome = 1;
469 if (codec_caps & AOM_CODEC_CAP_HIGHBITDEPTH) {
472 *img_fmt = AOM_IMG_FMT_I42016;
473 *
flags |= AOM_CODEC_USE_HIGHBITDEPTH;
479 if (codec_caps & AOM_CODEC_CAP_HIGHBITDEPTH) {
481 *img_fmt = AOM_IMG_FMT_I42216;
482 *
flags |= AOM_CODEC_USE_HIGHBITDEPTH;
490 if (codec_caps & AOM_CODEC_CAP_HIGHBITDEPTH) {
493 *img_fmt = AOM_IMG_FMT_I44416;
494 *
flags |= AOM_CODEC_USE_HIGHBITDEPTH;
507 aom_color_range_t aom_cr;
523 int sb_dim = (
dim + sb_size - 1) / sb_size;
524 int tile_dim = (sb_dim + (1 << tiles_log2) - 1) >> tiles_log2;
526 return (sb_dim + tile_dim - 1) / tile_dim;
530 struct aom_codec_enc_cfg *enccfg)
533 int sb_128x128_possible, sb_size, sb_width, sb_height;
534 int uniform_rows, uniform_cols;
535 int uniform_64x64_possible, uniform_128x128_possible;
536 int tile_size, rounding,
i;
538 if (
ctx->tile_cols_log2 >= 0)
539 ctx->tile_cols = 1 <<
ctx->tile_cols_log2;
540 if (
ctx->tile_rows_log2 >= 0)
541 ctx->tile_rows = 1 <<
ctx->tile_rows_log2;
543 if (
ctx->tile_cols == 0) {
546 if (
ctx->tile_cols > 1) {
548 "columns to fill width.\n",
ctx->tile_cols);
552 if (
ctx->tile_rows == 0) {
555 ctx->tile_cols - 1) /
ctx->tile_cols, 128);
559 if (
ctx->tile_rows > 1) {
561 "rows to fill area.\n",
ctx->tile_rows);
566 if ((avctx->
width + 63) / 64 <
ctx->tile_cols ||
567 (avctx->
height + 63) / 64 <
ctx->tile_rows) {
569 "large enough to fit specified tile arrangement.\n");
575 "not allow more than %dx%d tiles.\n",
581 "not allow tiles of width greater than %d.\n",
586 ctx->superblock_size = AOM_SUPERBLOCK_SIZE_DYNAMIC;
588 if (
ctx->tile_cols == 1 &&
ctx->tile_rows == 1) {
593 sb_128x128_possible =
594 (avctx->
width + 127) / 128 >=
ctx->tile_cols &&
595 (avctx->
height + 127) / 128 >=
ctx->tile_rows;
597 ctx->tile_cols_log2 =
ctx->tile_cols == 1 ? 0 :
599 ctx->tile_rows_log2 =
ctx->tile_rows == 1 ? 0 :
603 64,
ctx->tile_cols_log2);
605 64,
ctx->tile_rows_log2);
607 "-> %dx%d tiles.\n", uniform_cols, uniform_rows);
608 uniform_64x64_possible = uniform_cols ==
ctx->tile_cols &&
609 uniform_rows ==
ctx->tile_rows;
611 if (sb_128x128_possible) {
613 128,
ctx->tile_cols_log2);
615 128,
ctx->tile_rows_log2);
617 "-> %dx%d tiles.\n", uniform_cols, uniform_rows);
618 uniform_128x128_possible = uniform_cols ==
ctx->tile_cols &&
619 uniform_rows ==
ctx->tile_rows;
622 uniform_128x128_possible = 0;
625 ctx->uniform_tiles = 1;
626 if (uniform_64x64_possible && uniform_128x128_possible) {
628 "superblocks (tile_cols_log2 = %d, tile_rows_log2 = %d).\n",
629 ctx->tile_cols_log2,
ctx->tile_rows_log2);
632 if (uniform_64x64_possible && !sb_128x128_possible) {
634 "superblocks (tile_cols_log2 = %d, tile_rows_log2 = %d).\n",
635 ctx->tile_cols_log2,
ctx->tile_rows_log2);
636 ctx->superblock_size = AOM_SUPERBLOCK_SIZE_64X64;
639 if (uniform_128x128_possible) {
641 "superblocks (tile_cols_log2 = %d, tile_rows_log2 = %d).\n",
642 ctx->tile_cols_log2,
ctx->tile_rows_log2);
643 ctx->superblock_size = AOM_SUPERBLOCK_SIZE_128X128;
646 ctx->uniform_tiles = 0;
648 if (sb_128x128_possible) {
650 ctx->superblock_size = AOM_SUPERBLOCK_SIZE_128X128;
653 ctx->superblock_size = AOM_SUPERBLOCK_SIZE_64X64;
656 "superblocks (tile_cols = %d, tile_rows = %d).\n",
657 sb_size, sb_size,
ctx->tile_cols,
ctx->tile_rows);
659 enccfg->tile_width_count =
ctx->tile_cols;
660 enccfg->tile_height_count =
ctx->tile_rows;
662 sb_width = (avctx->
width + sb_size - 1) / sb_size;
663 sb_height = (avctx->
height + sb_size - 1) / sb_size;
665 tile_size = sb_width /
ctx->tile_cols;
666 rounding = sb_width %
ctx->tile_cols;
667 for (
i = 0;
i <
ctx->tile_cols;
i++) {
668 enccfg->tile_widths[
i] = tile_size +
670 i >
ctx->tile_cols - 1 - (rounding + 1) / 2);
673 tile_size = sb_height /
ctx->tile_rows;
674 rounding = sb_height %
ctx->tile_rows;
675 for (
i = 0;
i <
ctx->tile_rows;
i++) {
676 enccfg->tile_heights[
i] = tile_size +
678 i >
ctx->tile_rows - 1 - (rounding + 1) / 2);
685 const struct aom_codec_iface *iface)
689 struct aom_codec_enc_cfg enccfg = { 0 };
690 #ifdef AOM_FRAME_IS_INTRAONLY
691 aom_codec_flags_t
flags =
694 aom_codec_flags_t
flags = 0;
698 aom_img_fmt_t img_fmt;
699 aom_codec_caps_t codec_caps = aom_codec_get_caps(iface);
704 if ((res = aom_codec_enc_config_default(iface, &enccfg,
ctx->usage)) != AOM_CODEC_OK) {
706 aom_codec_err_to_string(res));
721 enccfg.g_w = avctx->
width;
722 enccfg.g_h = avctx->
height;
728 if (
ctx->lag_in_frames >= 0)
729 enccfg.g_lag_in_frames =
ctx->lag_in_frames;
732 enccfg.g_pass = AOM_RC_FIRST_PASS;
734 enccfg.g_pass = AOM_RC_LAST_PASS;
736 enccfg.g_pass = AOM_RC_ONE_PASS;
740 enccfg.rc_end_usage = AOM_CBR;
741 }
else if (
ctx->crf >= 0) {
742 enccfg.rc_end_usage = AOM_CQ;
744 enccfg.rc_end_usage = AOM_Q;
750 }
else if (enccfg.rc_end_usage != AOM_Q) {
751 enccfg.rc_end_usage = AOM_Q;
754 "Neither bitrate nor constrained quality specified, using default CRF of %d\n",
758 if (avctx->
qmin >= 0)
759 enccfg.rc_min_quantizer = avctx->
qmin;
760 if (avctx->
qmax >= 0) {
761 enccfg.rc_max_quantizer = avctx->
qmax;
762 }
else if (!
ctx->crf) {
763 enccfg.rc_max_quantizer = 0;
766 if (enccfg.rc_end_usage == AOM_CQ || enccfg.rc_end_usage == AOM_Q) {
767 if (
ctx->crf < enccfg.rc_min_quantizer ||
ctx->crf > enccfg.rc_max_quantizer) {
769 "CQ level %d must be between minimum and maximum quantizer value (%d-%d)\n",
770 ctx->crf, enccfg.rc_min_quantizer, enccfg.rc_max_quantizer);
775 enccfg.rc_dropframe_thresh =
ctx->drop_threshold;
779 if (
ctx->minsection_pct >= 0)
780 enccfg.rc_2pass_vbr_minsection_pct =
ctx->minsection_pct;
782 enccfg.rc_2pass_vbr_minsection_pct =
784 if (
ctx->maxsection_pct >= 0)
785 enccfg.rc_2pass_vbr_maxsection_pct =
ctx->maxsection_pct;
787 enccfg.rc_2pass_vbr_maxsection_pct =
794 enccfg.rc_buf_initial_sz =
796 enccfg.rc_buf_optimal_sz = enccfg.rc_buf_sz * 5 / 6;
798 if (
ctx->rc_undershoot_pct >= 0)
799 enccfg.rc_undershoot_pct =
ctx->rc_undershoot_pct;
800 if (
ctx->rc_overshoot_pct >= 0)
801 enccfg.rc_overshoot_pct =
ctx->rc_overshoot_pct;
807 enccfg.kf_max_dist = avctx->
gop_size;
809 if (enccfg.g_pass == AOM_RC_FIRST_PASS)
810 enccfg.g_lag_in_frames = 0;
811 else if (enccfg.g_pass == AOM_RC_LAST_PASS) {
812 int decode_size,
ret;
819 ctx->twopass_stats.sz = strlen(avctx->
stats_in) * 3 / 4;
824 ctx->twopass_stats.sz);
825 ctx->twopass_stats.sz = 0;
829 ctx->twopass_stats.sz);
830 if (decode_size < 0) {
835 ctx->twopass_stats.sz = decode_size;
836 enccfg.rc_twopass_stats_in =
ctx->twopass_stats;
843 enccfg.g_profile = avctx->
profile;
845 enccfg.g_error_resilient =
ctx->error_resilient;
851 if (
ctx->still_picture) {
857 enccfg.g_lag_in_frames = 0;
859 enccfg.kf_max_dist = 0;
860 enccfg.kf_mode = AOM_KF_DISABLED;
864 res = aom_codec_enc_init(&
ctx->encoder, iface, &enccfg,
flags);
865 if (res != AOM_CODEC_OK) {
875 if (
ctx->auto_alt_ref >= 0)
877 if (
ctx->arnr_max_frames >= 0)
879 if (
ctx->arnr_strength >= 0)
881 if (
ctx->enable_cdef >= 0)
883 if (
ctx->enable_restoration >= 0)
884 codecctl_int(avctx, AV1E_SET_ENABLE_RESTORATION,
ctx->enable_restoration);
885 #if AOM_ENCODER_ABI_VERSION >= 22
886 if (
ctx->enable_rect_partitions >= 0)
887 codecctl_int(avctx, AV1E_SET_ENABLE_RECT_PARTITIONS,
ctx->enable_rect_partitions);
888 if (
ctx->enable_1to4_partitions >= 0)
889 codecctl_int(avctx, AV1E_SET_ENABLE_1TO4_PARTITIONS,
ctx->enable_1to4_partitions);
890 if (
ctx->enable_ab_partitions >= 0)
891 codecctl_int(avctx, AV1E_SET_ENABLE_AB_PARTITIONS,
ctx->enable_ab_partitions);
892 if (
ctx->enable_angle_delta >= 0)
893 codecctl_int(avctx, AV1E_SET_ENABLE_ANGLE_DELTA,
ctx->enable_angle_delta);
894 if (
ctx->enable_cfl_intra >= 0)
896 if (
ctx->enable_filter_intra >= 0)
897 codecctl_int(avctx, AV1E_SET_ENABLE_FILTER_INTRA,
ctx->enable_filter_intra);
898 if (
ctx->enable_intra_edge_filter >= 0)
899 codecctl_int(avctx, AV1E_SET_ENABLE_INTRA_EDGE_FILTER,
ctx->enable_intra_edge_filter);
900 if (
ctx->enable_paeth_intra >= 0)
901 codecctl_int(avctx, AV1E_SET_ENABLE_PAETH_INTRA,
ctx->enable_paeth_intra);
902 if (
ctx->enable_smooth_intra >= 0)
903 codecctl_int(avctx, AV1E_SET_ENABLE_SMOOTH_INTRA,
ctx->enable_smooth_intra);
904 if (
ctx->enable_palette >= 0)
906 if (
ctx->enable_tx64 >= 0)
908 if (
ctx->enable_flip_idtx >= 0)
910 if (
ctx->use_intra_dct_only >= 0)
912 if (
ctx->use_inter_dct_only >= 0)
914 if (
ctx->use_intra_default_tx_only >= 0)
915 codecctl_int(avctx, AV1E_SET_INTRA_DEFAULT_TX_ONLY,
ctx->use_intra_default_tx_only);
916 if (
ctx->reduced_tx_type_set >= 0)
917 codecctl_int(avctx, AV1E_SET_REDUCED_TX_TYPE_SET,
ctx->reduced_tx_type_set);
918 if (
ctx->enable_ref_frame_mvs >= 0)
919 codecctl_int(avctx, AV1E_SET_ENABLE_REF_FRAME_MVS,
ctx->enable_ref_frame_mvs);
920 if (
ctx->enable_reduced_reference_set >= 0)
921 codecctl_int(avctx, AV1E_SET_REDUCED_REFERENCE_SET,
ctx->enable_reduced_reference_set);
922 if (
ctx->enable_diff_wtd_comp >= 0)
923 codecctl_int(avctx, AV1E_SET_ENABLE_DIFF_WTD_COMP,
ctx->enable_diff_wtd_comp);
924 if (
ctx->enable_dist_wtd_comp >= 0)
925 codecctl_int(avctx, AV1E_SET_ENABLE_DIST_WTD_COMP,
ctx->enable_dist_wtd_comp);
926 if (
ctx->enable_dual_filter >= 0)
927 codecctl_int(avctx, AV1E_SET_ENABLE_DUAL_FILTER,
ctx->enable_dual_filter);
928 if (
ctx->enable_interinter_wedge >= 0)
929 codecctl_int(avctx, AV1E_SET_ENABLE_INTERINTER_WEDGE,
ctx->enable_interinter_wedge);
930 if (
ctx->enable_masked_comp >= 0)
931 codecctl_int(avctx, AV1E_SET_ENABLE_MASKED_COMP,
ctx->enable_masked_comp);
932 if (
ctx->enable_interintra_comp >= 0)
933 codecctl_int(avctx, AV1E_SET_ENABLE_INTERINTRA_COMP,
ctx->enable_interintra_comp);
934 if (
ctx->enable_interintra_wedge >= 0)
935 codecctl_int(avctx, AV1E_SET_ENABLE_INTERINTRA_WEDGE,
ctx->enable_interintra_wedge);
936 if (
ctx->enable_obmc >= 0)
938 if (
ctx->enable_onesided_comp >= 0)
939 codecctl_int(avctx, AV1E_SET_ENABLE_ONESIDED_COMP,
ctx->enable_onesided_comp);
940 if (
ctx->enable_smooth_interintra >= 0)
941 codecctl_int(avctx, AV1E_SET_ENABLE_SMOOTH_INTERINTRA,
ctx->enable_smooth_interintra);
959 if (
ctx->aq_mode >= 0)
961 if (
ctx->frame_parallel >= 0)
962 codecctl_int(avctx, AV1E_SET_FRAME_PARALLEL_DECODING,
ctx->frame_parallel);
966 if (
ctx->uniform_tiles) {
971 #ifdef AOM_CTRL_AV1E_SET_DENOISE_NOISE_LEVEL
972 if (
ctx->denoise_noise_level >= 0)
973 codecctl_int(avctx, AV1E_SET_DENOISE_NOISE_LEVEL,
ctx->denoise_noise_level);
975 #ifdef AOM_CTRL_AV1E_SET_DENOISE_BLOCK_SIZE
976 if (
ctx->denoise_block_size >= 0)
977 codecctl_int(avctx, AV1E_SET_DENOISE_BLOCK_SIZE,
ctx->denoise_block_size);
979 #ifdef AOM_CTRL_AV1E_SET_ENABLE_GLOBAL_MOTION
980 if (
ctx->enable_global_motion >= 0)
981 codecctl_int(avctx, AV1E_SET_ENABLE_GLOBAL_MOTION,
ctx->enable_global_motion);
983 #ifdef AOM_CTRL_AV1E_SET_MAX_REFERENCE_FRAMES
984 if (avctx->
refs >= 3) {
988 #ifdef AOM_CTRL_AV1E_SET_ROW_MT
989 if (
ctx->row_mt >= 0)
992 #ifdef AOM_CTRL_AV1E_SET_ENABLE_INTRABC
993 if (
ctx->enable_intrabc >= 0)
997 #if AOM_ENCODER_ABI_VERSION >= 23
1002 int ret = aom_codec_set_option(&
ctx->encoder, en->
key, en->
value);
1003 if (
ret != AOM_CODEC_OK) {
1012 aom_img_wrap(&
ctx->rawimg, img_fmt, avctx->
width, avctx->
height, 1,
1015 if (codec_caps & AOM_CODEC_CAP_HIGHBITDEPTH)
1016 ctx->rawimg.bit_depth = enccfg.g_bit_depth;
1028 "not found. This is a bug, please report it.\n");
1044 if (enccfg.rc_end_usage == AOM_CBR ||
1045 enccfg.g_pass != AOM_RC_ONE_PASS) {
1057 const struct aom_codec_cx_pkt *
src)
1059 dst->
pts =
src->data.frame.pts;
1061 dst->
flags =
src->data.frame.flags;
1062 dst->
sz =
src->data.frame.sz;
1063 dst->
buf =
src->data.frame.buf;
1064 #ifdef AOM_FRAME_IS_INTRAONLY
1067 if (
ctx->have_sse) {
1070 memcpy(dst->
sse,
ctx->sse,
sizeof(dst->
sse));
1097 if (!!(cx_frame->
flags & AOM_FRAME_IS_KEY)) {
1099 #ifdef AOM_FRAME_IS_INTRAONLY
1101 }
else if (cx_frame->
flags & AOM_FRAME_IS_INTRAONLY) {
1108 cx_frame->
have_sse ? 3 : 0, pict_type);
1112 for (
i = 0;
i < 3; ++
i) {
1123 "failed to send input packet\n");
1130 "failed to receive output packet\n");
1148 const struct aom_codec_cx_pkt *
pkt;
1149 const void *iter =
NULL;
1152 if (
ctx->coded_frame_list) {
1158 ctx->coded_frame_list = cx_frame->
next;
1164 while ((
pkt = aom_codec_get_cx_data(&
ctx->encoder, &iter))) {
1165 switch (
pkt->kind) {
1166 case AOM_CODEC_CX_FRAME_PKT:
1183 "Frame queue element alloc failed\n");
1189 if (!cx_frame->
buf) {
1200 case AOM_CODEC_STATS_PKT:
1202 struct aom_fixed_buf *
stats = &
ctx->twopass_stats;
1206 pkt->
data.twopass_stats.sz)) < 0) {
1216 #ifdef AOM_FRAME_IS_INTRAONLY
1217 case AOM_CODEC_PSNR_PKT:
1228 case AOM_CODEC_CUSTOM_PKT:
1240 case AOM_IMG_FMT_I420:
1241 case AOM_IMG_FMT_I42016:
1242 if (
img->bit_depth == 8)
1244 else if (
img->bit_depth == 10)
1248 case AOM_IMG_FMT_I422:
1249 case AOM_IMG_FMT_I42216:
1250 if (
img->bit_depth == 8)
1252 else if (
img->bit_depth == 10)
1256 case AOM_IMG_FMT_I444:
1257 case AOM_IMG_FMT_I44416:
1258 if (
img->bit_depth == 8)
1260 else if (
img->bit_depth == 10)
1272 struct aom_image *rawimg =
NULL;
1273 int64_t timestamp = 0;
1274 int res, coded_size;
1275 aom_enc_frame_flags_t
flags = 0;
1278 rawimg = &
ctx->rawimg;
1279 rawimg->planes[AOM_PLANE_Y] =
frame->data[0];
1280 rawimg->planes[AOM_PLANE_U] =
frame->data[1];
1281 rawimg->planes[AOM_PLANE_V] =
frame->data[2];
1282 rawimg->stride[AOM_PLANE_Y] =
frame->linesize[0];
1283 rawimg->stride[AOM_PLANE_U] =
frame->linesize[1];
1284 rawimg->stride[AOM_PLANE_V] =
frame->linesize[2];
1285 timestamp =
frame->pts;
1286 switch (
frame->color_range) {
1288 rawimg->range = AOM_CR_STUDIO_RANGE;
1291 rawimg->range = AOM_CR_FULL_RANGE;
1296 flags |= AOM_EFLAG_FORCE_KF;
1299 res = aom_codec_encode(&
ctx->encoder, rawimg, timestamp,
1301 if (res != AOM_CODEC_OK) {
1317 ctx->twopass_stats.sz);
1320 *got_packet = !!coded_size;
1324 struct aom_image
img;
1335 "Unhandled reconstructed frame colorspace: %d\n",
1347 if ((
img.fmt & AOM_IMG_FMT_HIGHBITDEPTH) &&
img.bit_depth == 8)
1350 const uint8_t *
planes[4] = {
img.planes[0],
img.planes[1],
img.planes[2] };
1351 const int stride[4] = {
img.stride[0],
img.stride[1],
img.stride[2] };
1415 int supports_monochrome = aom_codec_version() >= 20001;
1416 aom_codec_caps_t codec_caps = aom_codec_get_caps(aom_codec_av1_cx());
1417 if (codec_caps & AOM_CODEC_CAP_HIGHBITDEPTH)
1424 if (aom_codec_version_major() < 2)
1430 return aom_init(avctx, aom_codec_av1_cx());
1433 #define OFFSET(x) offsetof(AOMContext, x)
1434 #define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
1437 {
"auto-alt-ref",
"Enable use of alternate reference "
1439 {
"lag-in-frames",
"Number of frames to look ahead at for "
1441 {
"arnr-max-frames",
"altref noise reduction max frame count",
OFFSET(arnr_max_frames),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
VE},
1442 {
"arnr-strength",
"altref noise reduction filter strength",
OFFSET(arnr_strength),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 6,
VE},
1443 {
"aq-mode",
"adaptive quantization mode",
OFFSET(aq_mode),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 4,
VE,
"aq_mode"},
1445 {
"variance",
"Variance based Aq", 0,
AV_OPT_TYPE_CONST, {.i64 = 1}, 0, 0,
VE,
"aq_mode"},
1446 {
"complexity",
"Complexity based Aq", 0,
AV_OPT_TYPE_CONST, {.i64 = 2}, 0, 0,
VE,
"aq_mode"},
1448 {
"error-resilience",
"Error resilience configuration",
OFFSET(error_resilient),
AV_OPT_TYPE_FLAGS, {.i64 = 0}, INT_MIN, INT_MAX,
VE,
"er"},
1449 {
"default",
"Improve resiliency against losses of whole frames", 0,
AV_OPT_TYPE_CONST, {.i64 = AOM_ERROR_RESILIENT_DEFAULT}, 0, 0,
VE,
"er"},
1450 {
"crf",
"Select the quality for constant quality mode", offsetof(AOMContext, crf),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 63,
VE },
1451 {
"static-thresh",
"A change threshold on blocks below which they will be skipped by the encoder",
OFFSET(static_thresh),
AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX,
VE },
1452 {
"drop-threshold",
"Frame drop threshold", offsetof(AOMContext, drop_threshold),
AV_OPT_TYPE_INT, {.i64 = 0 }, INT_MIN, INT_MAX,
VE },
1453 {
"denoise-noise-level",
"Amount of noise to be removed",
OFFSET(denoise_noise_level),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
VE},
1454 {
"denoise-block-size",
"Denoise block size ",
OFFSET(denoise_block_size),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX,
VE},
1455 {
"undershoot-pct",
"Datarate undershoot (min) target (%)",
OFFSET(rc_undershoot_pct),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 100,
VE},
1456 {
"overshoot-pct",
"Datarate overshoot (max) target (%)",
OFFSET(rc_overshoot_pct),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 1000,
VE},
1457 {
"minsection-pct",
"GOP min bitrate (% of target)",
OFFSET(minsection_pct),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 100,
VE},
1458 {
"maxsection-pct",
"GOP max bitrate (% of target)",
OFFSET(maxsection_pct),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 5000,
VE},
1459 {
"frame-parallel",
"Enable frame parallel decodability features",
OFFSET(frame_parallel),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1461 {
"tile-columns",
"Log2 of number of tile columns to use",
OFFSET(tile_cols_log2),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 6,
VE},
1462 {
"tile-rows",
"Log2 of number of tile rows to use",
OFFSET(tile_rows_log2),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, 6,
VE},
1465 {
"enable-global-motion",
"Enable global motion",
OFFSET(enable_global_motion),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1466 {
"enable-intrabc",
"Enable intra block copy prediction mode",
OFFSET(enable_intrabc),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1467 {
"enable-restoration",
"Enable Loop Restoration filtering",
OFFSET(enable_restoration),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1468 {
"usage",
"Quality and compression efficiency vs speed trade-off",
OFFSET(
usage),
AV_OPT_TYPE_INT, {.i64 = 0}, 0, INT_MAX,
VE,
"usage"},
1471 {
"allintra",
"All Intra encoding", 0,
AV_OPT_TYPE_CONST, {.i64 = 2 }, 0, 0,
VE,
"usage"},
1472 {
"tune",
"The metric that the encoder tunes for. Automatically chosen by the encoder by default",
OFFSET(tune),
AV_OPT_TYPE_INT, {.i64 = -1}, -1, AOM_TUNE_SSIM,
VE,
"tune"},
1476 {
"still-picture",
"Encode in single frame mode (typically used for still AVIF images).",
OFFSET(still_picture),
AV_OPT_TYPE_BOOL, {.i64 = 0}, -1, 1,
VE },
1477 {
"enable-rect-partitions",
"Enable rectangular partitions",
OFFSET(enable_rect_partitions),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1478 {
"enable-1to4-partitions",
"Enable 1:4/4:1 partitions",
OFFSET(enable_1to4_partitions),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1479 {
"enable-ab-partitions",
"Enable ab shape partitions",
OFFSET(enable_ab_partitions),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1480 {
"enable-angle-delta",
"Enable angle delta intra prediction",
OFFSET(enable_angle_delta),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1481 {
"enable-cfl-intra",
"Enable chroma predicted from luma intra prediction",
OFFSET(enable_cfl_intra),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1482 {
"enable-filter-intra",
"Enable filter intra predictor",
OFFSET(enable_filter_intra),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1483 {
"enable-intra-edge-filter",
"Enable intra edge filter",
OFFSET(enable_intra_edge_filter),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1484 {
"enable-smooth-intra",
"Enable smooth intra prediction mode",
OFFSET(enable_smooth_intra),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1485 {
"enable-paeth-intra",
"Enable paeth predictor in intra prediction",
OFFSET(enable_paeth_intra),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1486 {
"enable-palette",
"Enable palette prediction mode",
OFFSET(enable_palette),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1487 {
"enable-flip-idtx",
"Enable extended transform type",
OFFSET(enable_flip_idtx),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1489 {
"reduced-tx-type-set",
"Use reduced set of transform types",
OFFSET(reduced_tx_type_set),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1490 {
"use-intra-dct-only",
"Use DCT only for INTRA modes",
OFFSET(use_intra_dct_only),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1491 {
"use-inter-dct-only",
"Use DCT only for INTER modes",
OFFSET(use_inter_dct_only),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1492 {
"use-intra-default-tx-only",
"Use default-transform only for INTRA modes",
OFFSET(use_intra_default_tx_only),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1493 {
"enable-ref-frame-mvs",
"Enable temporal mv prediction",
OFFSET(enable_ref_frame_mvs),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1494 {
"enable-reduced-reference-set",
"Use reduced set of single and compound references",
OFFSET(enable_reduced_reference_set),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1496 {
"enable-dual-filter",
"Enable dual filter",
OFFSET(enable_dual_filter),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1497 {
"enable-diff-wtd-comp",
"Enable difference-weighted compound",
OFFSET(enable_diff_wtd_comp),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1498 {
"enable-dist-wtd-comp",
"Enable distance-weighted compound",
OFFSET(enable_dist_wtd_comp),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1499 {
"enable-onesided-comp",
"Enable one sided compound",
OFFSET(enable_onesided_comp),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1500 {
"enable-interinter-wedge",
"Enable interinter wedge compound",
OFFSET(enable_interinter_wedge),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1501 {
"enable-interintra-wedge",
"Enable interintra wedge compound",
OFFSET(enable_interintra_wedge),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1502 {
"enable-masked-comp",
"Enable masked compound",
OFFSET(enable_masked_comp),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1503 {
"enable-interintra-comp",
"Enable interintra compound",
OFFSET(enable_interintra_comp),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1504 {
"enable-smooth-interintra",
"Enable smooth interintra mode",
OFFSET(enable_smooth_interintra),
AV_OPT_TYPE_BOOL, {.i64 = -1}, -1, 1,
VE},
1505 #if AOM_ENCODER_ABI_VERSION >= 23
1506 {
"aom-params",
"Set libaom options using a :-separated list of key=value pairs",
OFFSET(aom_params),
AV_OPT_TYPE_DICT, { 0 }, 0, 0,
VE },
1516 {
"keyint_min",
"-1" },
1528 .
p.
name =
"libaom-av1",
1537 .p.wrapper_name =
"libaom",
1538 .priv_data_size =
sizeof(AOMContext),