40#include "aom_scale/yv12config.h"
42#include "common/tools_common.h"
43#include "common/video_writer.h"
45static const char *exec_name;
47void usage_exit(
void) {
49 "Usage: %s <img_width> <img_height> <infile> <outfile> "
50 "<lf_width> <lf_height> <lf_blocksize>\n",
56 int image_size_bytes = 0;
58 for (plane = 0; plane < 3; ++plane) {
62 image_size_bytes += w * h;
64 return image_size_bytes;
75 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to get frame stats.");
83 stats->
buf = realloc(stats->
buf, stats->
sz + pkt_size);
84 if (!stats->
buf) die(
"Failed to allocate frame stats buffer.");
85 memcpy((uint8_t *)stats->
buf + stats->
sz, pkt_buf, pkt_size);
86 stats->
sz += pkt_size;
100 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to encode frame.");
107 if (!aom_video_writer_write_frame(writer, pkt->
data.
frame.
buf,
110 die_codec(ctx,
"Failed to write compressed frame.");
111 printf(keyframe ?
"K" :
".");
121 if (FORCE_HIGHBITDEPTH_DECODING) {
124 aom_img_upshift(raw_shift, raw, input_shift);
125 *frame_to_encode = raw_shift;
127 *frame_to_encode = raw;
134 int lf_height,
int lf_blocksize,
int flags,
138 int image_size_bytes = img_size_bytes(raw);
139 int u_blocks, v_blocks;
145 die(
"Failed to initialize encoder");
147 die_codec(&codec,
"Failed to turn off auto altref");
149 die_codec(&codec,
"Failed to set frame parallel decoding");
152 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
153 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
155 printf(
"\n First pass: ");
157 for (bv = 0; bv < v_blocks; ++bv) {
158 for (bu = 0; bu < u_blocks; ++bu) {
159 const int block_u_min = bu * lf_blocksize;
160 const int block_v_min = bv * lf_blocksize;
161 int block_u_end = (bu + 1) * lf_blocksize;
162 int block_v_end = (bv + 1) * lf_blocksize;
163 int u_block_size, v_block_size;
164 int block_ref_u, block_ref_v;
166 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
167 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
168 u_block_size = block_u_end - block_u_min;
169 v_block_size = block_v_end - block_v_min;
170 block_ref_u = block_u_min + u_block_size / 2;
171 block_ref_v = block_v_min + v_block_size / 2;
173 printf(
"A%d, ", (block_ref_u + block_ref_v * lf_width));
174 fseek(infile, (block_ref_u + block_ref_v * lf_width) * image_size_bytes,
176 aom_img_read(raw, infile);
177 get_raw_image(&frame_to_encode, raw, raw_shift);
181 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
192 die_codec(&codec,
"Failed to set frame parallel decoding");
194 for (bv = 0; bv < v_blocks; ++bv) {
195 for (bu = 0; bu < u_blocks; ++bu) {
196 const int block_u_min = bu * lf_blocksize;
197 const int block_v_min = bv * lf_blocksize;
198 int block_u_end = (bu + 1) * lf_blocksize;
199 int block_v_end = (bv + 1) * lf_blocksize;
201 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
202 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
203 for (v = block_v_min; v < block_v_end; ++v) {
204 for (u = block_u_min; u < block_u_end; ++u) {
205 printf(
"C%d, ", (u + v * lf_width));
206 fseek(infile, (u + v * lf_width) * image_size_bytes, SEEK_SET);
207 aom_img_read(raw, infile);
208 get_raw_image(&frame_to_encode, raw, raw_shift);
211 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
224 while (get_frame_stats(&codec, NULL, frame_count, 1, 0, &stats)) {
229 printf(
"\nFirst pass complete. Processed %d frames.\n", frame_count);
234static void pass1(
aom_image_t *raw, FILE *infile,
const char *outfile_name,
236 int lf_width,
int lf_height,
int lf_blocksize,
int flags,
238 AvxVideoInfo info = { get_fourcc_by_aom_encoder(encoder),
243 AvxVideoWriter *writer = NULL;
246 int image_size_bytes = img_size_bytes(raw);
248 int u_blocks, v_blocks;
250 aom_image_t reference_images[MAX_EXTERNAL_REFERENCES];
251 int reference_image_num = 0;
254 writer = aom_video_writer_open(outfile_name, kContainerIVF, &info);
255 if (!writer) die(
"Failed to open %s for writing", outfile_name);
258 die(
"Failed to initialize encoder");
260 die_codec(&codec,
"Failed to turn off auto altref");
262 die_codec(&codec,
"Failed to set frame parallel decoding");
264 die_codec(&codec,
"Failed to enable encoder ext_tile debug");
266 die_codec(&codec,
"Failed to set cpu-used");
275 die_codec(&codec,
"Failed to set SB size");
277 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
278 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
280 reference_image_num = u_blocks * v_blocks;
289 reference_image_num - 1))
290 die_codec(&codec,
"Failed to set max gf interval");
294 int border_in_pixels =
296 ? AOM_BORDER_IN_PIXELS
297 : AOM_ENC_NO_SCALE_BORDER;
298 for (i = 0; i < reference_image_num; i++) {
300 cfg->
g_h, 32, 8, border_in_pixels)) {
301 die(
"Failed to allocate image.");
305 printf(
"\n Second pass: ");
308 printf(
"Encoding Reference Images\n");
309 for (bv = 0; bv < v_blocks; ++bv) {
310 for (bu = 0; bu < u_blocks; ++bu) {
311 const int block_u_min = bu * lf_blocksize;
312 const int block_v_min = bv * lf_blocksize;
313 int block_u_end = (bu + 1) * lf_blocksize;
314 int block_v_end = (bv + 1) * lf_blocksize;
315 int u_block_size, v_block_size;
316 int block_ref_u, block_ref_v;
318 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
319 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
320 u_block_size = block_u_end - block_u_min;
321 v_block_size = block_v_end - block_v_min;
322 block_ref_u = block_u_min + u_block_size / 2;
323 block_ref_v = block_v_min + v_block_size / 2;
325 printf(
"A%d, ", (block_ref_u + block_ref_v * lf_width));
326 fseek(infile, (block_ref_u + block_ref_v * lf_width) * image_size_bytes,
328 aom_img_read(raw, infile);
330 get_raw_image(&frame_to_encode, raw, raw_shift);
334 printf(
"Encoding reference image %d of %d\n", bv * u_blocks + bu,
335 u_blocks * v_blocks);
336 encode_frame(&codec, frame_to_encode, frame_count, 1,
345 &reference_images[frame_count - 1]))
346 die_codec(&codec,
"Failed to copy decoder reference frame");
354 die_codec(&codec,
"Failed to configure encoder");
358 die_codec(&codec,
"Failed to set cq level");
360 die_codec(&codec,
"Failed to set frame parallel decoding");
362 die_codec(&codec,
"Failed to turn on single tile decoding");
366 die_codec(&codec,
"Failed to set tile width");
368 die_codec(&codec,
"Failed to set tile height");
370 for (bv = 0; bv < v_blocks; ++bv) {
371 for (bu = 0; bu < u_blocks; ++bu) {
372 const int block_u_min = bu * lf_blocksize;
373 const int block_v_min = bv * lf_blocksize;
374 int block_u_end = (bu + 1) * lf_blocksize;
375 int block_v_end = (bv + 1) * lf_blocksize;
377 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
378 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
379 for (v = block_v_min; v < block_v_end; ++v) {
380 for (u = block_u_min; u < block_u_end; ++u) {
384 ref.
img = reference_images[bv * u_blocks + bu];
386 die_codec(&codec,
"Failed to set reference frame");
388 printf(
"C%d, ", (u + v * lf_width));
389 fseek(infile, (u + v * lf_width) * image_size_bytes, SEEK_SET);
390 aom_img_read(raw, infile);
391 get_raw_image(&frame_to_encode, raw, raw_shift);
394 printf(
"Encoding image %d of %d\n",
395 frame_count - (u_blocks * v_blocks), lf_width * lf_height);
396 encode_frame(&codec, frame_to_encode, frame_count, 1,
410 while (encode_frame(&codec, NULL, -1, 1, 0, writer)) {
413 for (i = 0; i < reference_image_num; i++)
aom_img_free(&reference_images[i]);
419 aom_video_writer_set_fourcc(writer, LST_FOURCC);
420 aom_video_writer_close(writer);
422 printf(
"\nSecond pass complete. Processed %d frames.\n", frame_count);
425int main(
int argc,
char **argv) {
429 int lf_width, lf_height;
443 const int bitrate = 200;
444 const char *
const width_arg = argv[1];
445 const char *
const height_arg = argv[2];
446 const char *
const infile_arg = argv[3];
447 const char *
const outfile_arg = argv[4];
448 const char *
const lf_width_arg = argv[5];
449 const char *
const lf_height_arg = argv[6];
450 const char *lf_blocksize_arg = argv[7];
453 if (argc < 8) die(
"Invalid number of arguments");
456 if (!encoder) die(
"Unsupported codec.");
458 w = (int)strtol(width_arg, NULL, 0);
459 h = (int)strtol(height_arg, NULL, 0);
460 lf_width = (int)strtol(lf_width_arg, NULL, 0);
461 lf_height = (int)strtol(lf_height_arg, NULL, 0);
462 lf_blocksize = (int)strtol(lf_blocksize_arg, NULL, 0);
463 lf_blocksize = lf_blocksize < lf_width ? lf_blocksize : lf_width;
464 lf_blocksize = lf_blocksize < lf_height ? lf_blocksize : lf_height;
466 if (w <= 0 || h <= 0 || (w % 2) != 0 || (h % 2) != 0)
467 die(
"Invalid frame size: %dx%d", w, h);
468 if (lf_width <= 0 || lf_height <= 0)
469 die(
"Invalid lf_width and/or lf_height: %dx%d", lf_width, lf_height);
470 if (lf_blocksize <= 0) die(
"Invalid lf_blocksize: %d", lf_blocksize);
473 die(
"Failed to allocate image.");
475 if (FORCE_HIGHBITDEPTH_DECODING) {
485 if (res) die_codec(&codec,
"Failed to get default codec config.");
501 if (!(infile = fopen(infile_arg,
"rb")))
502 die(
"Failed to open %s for reading", infile_arg);
506 stats = pass0(&raw, infile, encoder, &cfg, lf_width, lf_height, lf_blocksize,
513 pass1(&raw, infile, outfile_arg, encoder, &cfg, lf_width, lf_height,
514 lf_blocksize, flags, &raw_shift);
517 if (FORCE_HIGHBITDEPTH_DECODING)
aom_img_free(&raw_shift);
Describes the encoder algorithm interface to applications.
aom_image_t * aom_img_alloc_with_border(aom_image_t *img, aom_img_fmt_t fmt, unsigned int d_w, unsigned int d_h, unsigned int align, unsigned int size_align, unsigned int border)
Open a descriptor, allocating storage for the underlying image with a border.
#define AOM_IMG_FMT_HIGHBITDEPTH
Definition: aom_image.h:38
aom_image_t * aom_img_alloc(aom_image_t *img, aom_img_fmt_t fmt, unsigned int d_w, unsigned int d_h, unsigned int align)
Open a descriptor, allocating storage for the underlying image.
@ AOM_IMG_FMT_I420
Definition: aom_image.h:45
enum aom_img_fmt aom_img_fmt_t
List of supported image formats.
int aom_img_plane_height(const aom_image_t *img, int plane)
Get the height of a plane.
int aom_img_plane_width(const aom_image_t *img, int plane)
Get the width of a plane.
void aom_img_free(aom_image_t *img)
Close an image descriptor.
Provides definitions for using AOM or AV1 encoder algorithm within the aom Codec Interface.
#define AOM_EFLAG_NO_UPD_ARF
Don't update the alternate reference frame.
Definition: aomcx.h:135
#define AOM_EFLAG_NO_REF_LAST2
Don't reference the last2 frame.
Definition: aomcx.h:78
#define AOM_EFLAG_NO_REF_BWD
Don't reference the bwd reference frame.
Definition: aomcx.h:107
#define AOM_EFLAG_NO_UPD_LAST
Don't update the last frame.
Definition: aomcx.h:121
#define AOM_EFLAG_NO_REF_ARF
Don't reference the alternate reference frame.
Definition: aomcx.h:100
#define AOM_EFLAG_NO_REF_LAST3
Don't reference the last3 frame.
Definition: aomcx.h:85
#define AOM_EFLAG_NO_UPD_GF
Don't update the golden frame.
Definition: aomcx.h:128
#define AOM_EFLAG_NO_REF_GF
Don't reference the golden frame.
Definition: aomcx.h:92
#define AOM_EFLAG_NO_UPD_ENTROPY
Disable entropy update.
Definition: aomcx.h:141
#define AOM_EFLAG_NO_REF_ARF2
Don't reference the alt2 reference frame.
Definition: aomcx.h:114
@ AV1E_SET_MAX_GF_INTERVAL
Codec control function to set minimum interval between GF/ARF frames, unsigned int parameter.
Definition: aomcx.h:593
@ AV1E_SET_FRAME_PARALLEL_DECODING
Codec control function to enable frame parallel decoding feature, unsigned int parameter.
Definition: aomcx.h:430
@ AV1E_SET_TILE_ROWS
Codec control function to set number of tile rows, unsigned int parameter.
Definition: aomcx.h:397
@ AV1E_SET_SUPERBLOCK_SIZE
Codec control function to set intended superblock size, unsigned int parameter.
Definition: aomcx.h:646
@ AOME_SET_ENABLEAUTOALTREF
Codec control function to enable automatic set and use alf frames, unsigned int parameter.
Definition: aomcx.h:227
@ AV1E_SET_TILE_COLUMNS
Codec control function to set number of tile columns. unsigned int parameter.
Definition: aomcx.h:379
@ AOME_SET_CPUUSED
Codec control function to set encoder internal speed settings, int parameter.
Definition: aomcx.h:219
@ AV1E_SET_SINGLE_TILE_DECODING
Codec control function to set the single tile decoding mode, unsigned int parameter.
Definition: aomcx.h:1144
@ AOME_SET_CQ_LEVEL
Codec control function to set constrained / constant quality level, unsigned int parameter.
Definition: aomcx.h:291
@ AV1E_ENABLE_EXT_TILE_DEBUG
Codec control function to enable EXT_TILE_DEBUG in AV1 encoder, unsigned int parameter.
Definition: aomcx.h:1297
@ AV1_SET_REFERENCE
Codec control function to write a frame into a reference buffer.
Definition: aom.h:57
@ AV1_COPY_NEW_FRAME_IMAGE
Codec control function to copy the new frame to an external buffer.
Definition: aom.h:76
const char * aom_codec_iface_name(aom_codec_iface_t *iface)
Return the name for a given interface.
aom_codec_err_t aom_codec_control(aom_codec_ctx_t *ctx, int ctrl_id,...)
Algorithm Control.
const struct aom_codec_iface aom_codec_iface_t
Codec interface structure.
Definition: aom_codec.h:254
int64_t aom_codec_pts_t
Time Stamp Type.
Definition: aom_codec.h:235
aom_codec_err_t aom_codec_destroy(aom_codec_ctx_t *ctx)
Destroy a codec instance.
aom_codec_err_t
Algorithm return codes.
Definition: aom_codec.h:155
const void * aom_codec_iter_t
Iterator.
Definition: aom_codec.h:288
#define AOM_FRAME_IS_KEY
Definition: aom_codec.h:271
@ AOM_BITS_8
Definition: aom_codec.h:319
@ AOM_CODEC_OK
Operation completed without error.
Definition: aom_codec.h:157
@ AOM_SUPERBLOCK_SIZE_64X64
Definition: aom_codec.h:331
const aom_codec_cx_pkt_t * aom_codec_get_cx_data(aom_codec_ctx_t *ctx, aom_codec_iter_t *iter)
Encoded data iterator.
aom_codec_err_t aom_codec_encode(aom_codec_ctx_t *ctx, const aom_image_t *img, aom_codec_pts_t pts, unsigned long duration, aom_enc_frame_flags_t flags)
Encode a frame.
#define aom_codec_enc_init(ctx, iface, cfg, flags)
Convenience macro for aom_codec_enc_init_ver()
Definition: aom_encoder.h:934
aom_codec_err_t aom_codec_enc_config_default(aom_codec_iface_t *iface, aom_codec_enc_cfg_t *cfg, unsigned int usage)
Get the default configuration for a usage.
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition: aom_encoder.h:376
#define AOM_CODEC_USE_HIGHBITDEPTH
Make the encoder output one partition at a time.
Definition: aom_encoder.h:81
aom_codec_err_t aom_codec_enc_config_set(aom_codec_ctx_t *ctx, const aom_codec_enc_cfg_t *cfg)
Set or change configuration.
@ AOM_Q
Definition: aom_encoder.h:188
@ AOM_RC_LAST_PASS
Definition: aom_encoder.h:180
@ AOM_RC_FIRST_PASS
Definition: aom_encoder.h:177
@ AOM_KF_DISABLED
Definition: aom_encoder.h:202
@ AOM_CODEC_CX_FRAME_PKT
Definition: aom_encoder.h:109
@ AOM_CODEC_STATS_PKT
Definition: aom_encoder.h:110
Codec context structure.
Definition: aom_codec.h:298
union aom_codec_ctx::@0 config
Encoder output packet.
Definition: aom_encoder.h:121
size_t sz
Definition: aom_encoder.h:126
enum aom_codec_cx_pkt_kind kind
Definition: aom_encoder.h:122
aom_fixed_buf_t twopass_stats
Definition: aom_encoder.h:139
union aom_codec_cx_pkt::@1 data
aom_codec_pts_t pts
time stamp to show frame (in timebase units)
Definition: aom_encoder.h:128
struct aom_codec_cx_pkt::@1::@2 frame
aom_codec_frame_flags_t flags
Definition: aom_encoder.h:131
void * buf
Definition: aom_encoder.h:125
Encoder configuration structure.
Definition: aom_encoder.h:386
struct aom_rational g_timebase
Stream timebase units.
Definition: aom_encoder.h:483
unsigned int g_h
Height of the frame.
Definition: aom_encoder.h:434
aom_superres_mode rc_superres_mode
Frame super-resolution scaling mode.
Definition: aom_encoder.h:567
enum aom_kf_mode kf_mode
Keyframe placement mode.
Definition: aom_encoder.h:761
enum aom_rc_mode rc_end_usage
Rate control algorithm to use.
Definition: aom_encoder.h:617
unsigned int g_lag_in_frames
Allow lagged encoding.
Definition: aom_encoder.h:512
aom_bit_depth_t g_bit_depth
Bit-depth of the codec.
Definition: aom_encoder.h:461
unsigned int g_w
Width of the frame.
Definition: aom_encoder.h:425
aom_codec_er_flags_t g_error_resilient
Enable error resilient modes.
Definition: aom_encoder.h:491
enum aom_enc_pass g_pass
Multi-pass Encoding Mode.
Definition: aom_encoder.h:498
unsigned int rc_target_bitrate
Target data rate.
Definition: aom_encoder.h:637
unsigned int rc_resize_mode
Mode for spatial resampling, if supported by the codec.
Definition: aom_encoder.h:543
aom_fixed_buf_t rc_twopass_stats_in
Two-pass stats buffer.
Definition: aom_encoder.h:624
unsigned int large_scale_tile
Tile coding mode.
Definition: aom_encoder.h:809
Generic fixed size buffer structure.
Definition: aom_encoder.h:87
size_t sz
Definition: aom_encoder.h:89
void * buf
Definition: aom_encoder.h:88
Image Descriptor.
Definition: aom_image.h:180
aom_img_fmt_t fmt
Definition: aom_image.h:181
int num
Definition: aom_encoder.h:164
int den
Definition: aom_encoder.h:165
AV1 specific reference frame data struct.
Definition: aom.h:89
int use_external_ref
Definition: aom.h:91
aom_image_t img
Definition: aom.h:92
int idx
Definition: aom.h:90