AOMedia AV1 Codec
aomenc
1/*
2 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
3 *
4 * This source code is subject to the terms of the BSD 2 Clause License and
5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6 * was not distributed with this source code in the LICENSE file, you can
7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8 * Media Patent License 1.0 was not distributed with this source code in the
9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10 */
11
12#include "apps/aomenc.h"
13
14#include "config/aom_config.h"
15
16#include <assert.h>
17#include <limits.h>
18#include <math.h>
19#include <stdarg.h>
20#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
23
24#if CONFIG_AV1_DECODER
25#include "aom/aom_decoder.h"
26#include "aom/aomdx.h"
27#endif
28
29#include "aom/aom_encoder.h"
30#include "aom/aom_integer.h"
31#include "aom/aomcx.h"
32#include "aom_dsp/aom_dsp_common.h"
33#include "aom_ports/aom_timer.h"
34#include "aom_ports/mem_ops.h"
35#include "common/args.h"
36#include "common/ivfenc.h"
37#include "common/tools_common.h"
38#include "common/warnings.h"
39
40#if CONFIG_WEBM_IO
41#include "common/webmenc.h"
42#endif
43
44#include "common/y4minput.h"
45#include "examples/encoder_util.h"
46#include "stats/aomstats.h"
47#include "stats/rate_hist.h"
48
49#if CONFIG_LIBYUV
50#include <libyuv/scale.h>
51#endif
52
53/* Swallow warnings about unused results of fread/fwrite */
54static size_t wrap_fread(void *ptr, size_t size, size_t nmemb, FILE *stream) {
55 return fread(ptr, size, nmemb, stream);
56}
57#define fread wrap_fread
58
59static size_t wrap_fwrite(const void *ptr, size_t size, size_t nmemb,
60 FILE *stream) {
61 return fwrite(ptr, size, nmemb, stream);
62}
63#define fwrite wrap_fwrite
64
65static const char *exec_name;
66
67static AOM_TOOLS_FORMAT_PRINTF(3, 0) void warn_or_exit_on_errorv(
68 aom_codec_ctx_t *ctx, int fatal, const char *s, va_list ap) {
69 if (ctx->err) {
70 const char *detail = aom_codec_error_detail(ctx);
71
72 vfprintf(stderr, s, ap);
73 fprintf(stderr, ": %s\n", aom_codec_error(ctx));
74
75 if (detail) fprintf(stderr, " %s\n", detail);
76
77 if (fatal) exit(EXIT_FAILURE);
78 }
79}
80
81static AOM_TOOLS_FORMAT_PRINTF(2,
82 3) void ctx_exit_on_error(aom_codec_ctx_t *ctx,
83 const char *s, ...) {
84 va_list ap;
85
86 va_start(ap, s);
87 warn_or_exit_on_errorv(ctx, 1, s, ap);
88 va_end(ap);
89}
90
91static AOM_TOOLS_FORMAT_PRINTF(3, 4) void warn_or_exit_on_error(
92 aom_codec_ctx_t *ctx, int fatal, const char *s, ...) {
93 va_list ap;
94
95 va_start(ap, s);
96 warn_or_exit_on_errorv(ctx, fatal, s, ap);
97 va_end(ap);
98}
99
100static int read_frame(struct AvxInputContext *input_ctx, aom_image_t *img) {
101 FILE *f = input_ctx->file;
102 y4m_input *y4m = &input_ctx->y4m;
103 int shortread = 0;
104
105 if (input_ctx->file_type == FILE_TYPE_Y4M) {
106 if (y4m_input_fetch_frame(y4m, f, img) < 1) return 0;
107 } else {
108 shortread = read_yuv_frame(input_ctx, img);
109 }
110
111 return !shortread;
112}
113
114static int file_is_y4m(const char detect[4]) {
115 if (memcmp(detect, "YUV4", 4) == 0) {
116 return 1;
117 }
118 return 0;
119}
120
121static int fourcc_is_ivf(const char detect[4]) {
122 if (memcmp(detect, "DKIF", 4) == 0) {
123 return 1;
124 }
125 return 0;
126}
127
128static const int av1_arg_ctrl_map[] = { AOME_SET_CPUUSED,
129 AOME_SET_ENABLEAUTOALTREF,
130 AOME_SET_SHARPNESS,
131 AOME_SET_STATIC_THRESHOLD,
132 AV1E_SET_ROW_MT,
133 AV1E_SET_FP_MT,
134 AV1E_SET_TILE_COLUMNS,
135 AV1E_SET_TILE_ROWS,
136 AV1E_SET_ENABLE_TPL_MODEL,
137 AV1E_SET_ENABLE_KEYFRAME_FILTERING,
138 AOME_SET_ARNR_MAXFRAMES,
139 AOME_SET_ARNR_STRENGTH,
140 AOME_SET_TUNING,
141 AOME_SET_CQ_LEVEL,
142 AOME_SET_MAX_INTRA_BITRATE_PCT,
143 AV1E_SET_MAX_INTER_BITRATE_PCT,
144 AV1E_SET_GF_CBR_BOOST_PCT,
145 AV1E_SET_LOSSLESS,
146 AV1E_SET_ENABLE_CDEF,
147 AV1E_SET_ENABLE_RESTORATION,
148 AV1E_SET_ENABLE_RECT_PARTITIONS,
149 AV1E_SET_ENABLE_AB_PARTITIONS,
150 AV1E_SET_ENABLE_1TO4_PARTITIONS,
151 AV1E_SET_MIN_PARTITION_SIZE,
152 AV1E_SET_MAX_PARTITION_SIZE,
153 AV1E_SET_ENABLE_DUAL_FILTER,
154 AV1E_SET_ENABLE_CHROMA_DELTAQ,
155 AV1E_SET_ENABLE_INTRA_EDGE_FILTER,
156 AV1E_SET_ENABLE_ORDER_HINT,
157 AV1E_SET_ENABLE_TX64,
158 AV1E_SET_ENABLE_FLIP_IDTX,
159 AV1E_SET_ENABLE_RECT_TX,
160 AV1E_SET_ENABLE_DIST_WTD_COMP,
161 AV1E_SET_ENABLE_MASKED_COMP,
162 AV1E_SET_ENABLE_ONESIDED_COMP,
163 AV1E_SET_ENABLE_INTERINTRA_COMP,
164 AV1E_SET_ENABLE_SMOOTH_INTERINTRA,
165 AV1E_SET_ENABLE_DIFF_WTD_COMP,
166 AV1E_SET_ENABLE_INTERINTER_WEDGE,
167 AV1E_SET_ENABLE_INTERINTRA_WEDGE,
168 AV1E_SET_ENABLE_GLOBAL_MOTION,
169 AV1E_SET_ENABLE_WARPED_MOTION,
170 AV1E_SET_ENABLE_FILTER_INTRA,
171 AV1E_SET_ENABLE_SMOOTH_INTRA,
172 AV1E_SET_ENABLE_PAETH_INTRA,
173 AV1E_SET_ENABLE_CFL_INTRA,
174 AV1E_SET_ENABLE_DIAGONAL_INTRA,
175 AV1E_SET_FORCE_VIDEO_MODE,
176 AV1E_SET_ENABLE_OBMC,
177 AV1E_SET_ENABLE_OVERLAY,
178 AV1E_SET_ENABLE_PALETTE,
179 AV1E_SET_ENABLE_INTRABC,
180 AV1E_SET_ENABLE_ANGLE_DELTA,
181 AV1E_SET_DISABLE_TRELLIS_QUANT,
182 AV1E_SET_ENABLE_QM,
183 AV1E_SET_QM_MIN,
184 AV1E_SET_QM_MAX,
185 AV1E_SET_REDUCED_TX_TYPE_SET,
186 AV1E_SET_INTRA_DCT_ONLY,
187 AV1E_SET_INTER_DCT_ONLY,
188 AV1E_SET_INTRA_DEFAULT_TX_ONLY,
189 AV1E_SET_QUANT_B_ADAPT,
190 AV1E_SET_COEFF_COST_UPD_FREQ,
191 AV1E_SET_MODE_COST_UPD_FREQ,
192 AV1E_SET_MV_COST_UPD_FREQ,
193 AV1E_SET_FRAME_PARALLEL_DECODING,
194 AV1E_SET_ERROR_RESILIENT_MODE,
195 AV1E_SET_AQ_MODE,
196 AV1E_SET_DELTAQ_MODE,
197 AV1E_SET_DELTAQ_STRENGTH,
198 AV1E_SET_DELTALF_MODE,
199 AV1E_SET_FRAME_PERIODIC_BOOST,
200 AV1E_SET_NOISE_SENSITIVITY,
201 AV1E_SET_TUNE_CONTENT,
202 AV1E_SET_CDF_UPDATE_MODE,
203 AV1E_SET_COLOR_PRIMARIES,
204 AV1E_SET_TRANSFER_CHARACTERISTICS,
205 AV1E_SET_MATRIX_COEFFICIENTS,
206 AV1E_SET_CHROMA_SAMPLE_POSITION,
207 AV1E_SET_MIN_GF_INTERVAL,
208 AV1E_SET_MAX_GF_INTERVAL,
209 AV1E_SET_GF_MIN_PYRAMID_HEIGHT,
210 AV1E_SET_GF_MAX_PYRAMID_HEIGHT,
211 AV1E_SET_SUPERBLOCK_SIZE,
212 AV1E_SET_NUM_TG,
213 AV1E_SET_MTU,
214 AV1E_SET_TIMING_INFO_TYPE,
215 AV1E_SET_FILM_GRAIN_TEST_VECTOR,
216 AV1E_SET_FILM_GRAIN_TABLE,
217#if CONFIG_DENOISE
218 AV1E_SET_DENOISE_NOISE_LEVEL,
219 AV1E_SET_DENOISE_BLOCK_SIZE,
220 AV1E_SET_ENABLE_DNL_DENOISING,
221#endif // CONFIG_DENOISE
222 AV1E_SET_MAX_REFERENCE_FRAMES,
223 AV1E_SET_REDUCED_REFERENCE_SET,
224 AV1E_SET_ENABLE_REF_FRAME_MVS,
225 AV1E_SET_TARGET_SEQ_LEVEL_IDX,
226 AV1E_SET_TIER_MASK,
227 AV1E_SET_MIN_CR,
228 AV1E_SET_VBR_CORPUS_COMPLEXITY_LAP,
229 AV1E_SET_CHROMA_SUBSAMPLING_X,
230 AV1E_SET_CHROMA_SUBSAMPLING_Y,
231#if CONFIG_TUNE_VMAF
232 AV1E_SET_VMAF_MODEL_PATH,
233#endif
234 AV1E_SET_DV_COST_UPD_FREQ,
235 AV1E_SET_PARTITION_INFO_PATH,
236 AV1E_SET_ENABLE_DIRECTIONAL_INTRA,
237 AV1E_SET_ENABLE_TX_SIZE_SEARCH,
238 AV1E_SET_LOOPFILTER_CONTROL,
239 AV1E_SET_AUTO_INTRA_TOOLS_OFF,
240 0 };
241
242const arg_def_t *main_args[] = { &g_av1_codec_arg_defs.help,
243 &g_av1_codec_arg_defs.use_cfg,
244 &g_av1_codec_arg_defs.debugmode,
245 &g_av1_codec_arg_defs.outputfile,
246 &g_av1_codec_arg_defs.codecarg,
247 &g_av1_codec_arg_defs.passes,
248 &g_av1_codec_arg_defs.pass_arg,
249 &g_av1_codec_arg_defs.fpf_name,
250 &g_av1_codec_arg_defs.limit,
251 &g_av1_codec_arg_defs.skip,
252 &g_av1_codec_arg_defs.good_dl,
253 &g_av1_codec_arg_defs.rt_dl,
254 &g_av1_codec_arg_defs.ai_dl,
255 &g_av1_codec_arg_defs.quietarg,
256 &g_av1_codec_arg_defs.verbosearg,
257 &g_av1_codec_arg_defs.psnrarg,
258 &g_av1_codec_arg_defs.use_webm,
259 &g_av1_codec_arg_defs.use_ivf,
260 &g_av1_codec_arg_defs.use_obu,
261 &g_av1_codec_arg_defs.q_hist_n,
262 &g_av1_codec_arg_defs.rate_hist_n,
263 &g_av1_codec_arg_defs.disable_warnings,
264 &g_av1_codec_arg_defs.disable_warning_prompt,
265 &g_av1_codec_arg_defs.recontest,
266 NULL };
267
268const arg_def_t *global_args[] = {
269 &g_av1_codec_arg_defs.use_nv12,
270 &g_av1_codec_arg_defs.use_yv12,
271 &g_av1_codec_arg_defs.use_i420,
272 &g_av1_codec_arg_defs.use_i422,
273 &g_av1_codec_arg_defs.use_i444,
274 &g_av1_codec_arg_defs.usage,
275 &g_av1_codec_arg_defs.threads,
276 &g_av1_codec_arg_defs.profile,
277 &g_av1_codec_arg_defs.width,
278 &g_av1_codec_arg_defs.height,
279 &g_av1_codec_arg_defs.forced_max_frame_width,
280 &g_av1_codec_arg_defs.forced_max_frame_height,
281#if CONFIG_WEBM_IO
282 &g_av1_codec_arg_defs.stereo_mode,
283#endif
284 &g_av1_codec_arg_defs.timebase,
285 &g_av1_codec_arg_defs.framerate,
286 &g_av1_codec_arg_defs.global_error_resilient,
287 &g_av1_codec_arg_defs.bitdeptharg,
288 &g_av1_codec_arg_defs.inbitdeptharg,
289 &g_av1_codec_arg_defs.lag_in_frames,
290 &g_av1_codec_arg_defs.large_scale_tile,
291 &g_av1_codec_arg_defs.monochrome,
292 &g_av1_codec_arg_defs.full_still_picture_hdr,
293 &g_av1_codec_arg_defs.use_16bit_internal,
294 &g_av1_codec_arg_defs.save_as_annexb,
295 NULL
296};
297
298const arg_def_t *rc_args[] = { &g_av1_codec_arg_defs.dropframe_thresh,
299 &g_av1_codec_arg_defs.resize_mode,
300 &g_av1_codec_arg_defs.resize_denominator,
301 &g_av1_codec_arg_defs.resize_kf_denominator,
302 &g_av1_codec_arg_defs.superres_mode,
303 &g_av1_codec_arg_defs.superres_denominator,
304 &g_av1_codec_arg_defs.superres_kf_denominator,
305 &g_av1_codec_arg_defs.superres_qthresh,
306 &g_av1_codec_arg_defs.superres_kf_qthresh,
307 &g_av1_codec_arg_defs.end_usage,
308 &g_av1_codec_arg_defs.target_bitrate,
309 &g_av1_codec_arg_defs.min_quantizer,
310 &g_av1_codec_arg_defs.max_quantizer,
311 &g_av1_codec_arg_defs.undershoot_pct,
312 &g_av1_codec_arg_defs.overshoot_pct,
313 &g_av1_codec_arg_defs.buf_sz,
314 &g_av1_codec_arg_defs.buf_initial_sz,
315 &g_av1_codec_arg_defs.buf_optimal_sz,
316 &g_av1_codec_arg_defs.bias_pct,
317 &g_av1_codec_arg_defs.minsection_pct,
318 &g_av1_codec_arg_defs.maxsection_pct,
319 NULL };
320
321const arg_def_t *kf_args[] = { &g_av1_codec_arg_defs.fwd_kf_enabled,
322 &g_av1_codec_arg_defs.kf_min_dist,
323 &g_av1_codec_arg_defs.kf_max_dist,