40 #include "aom_scale/yv12config.h"
42 #include "common/tools_common.h"
43 #include "common/video_writer.h"
45 static const char *exec_name;
47 void 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 memcpy((uint8_t *)stats->
buf + stats->
sz, pkt_buf, pkt_size);
85 stats->
sz += pkt_size;
99 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to encode frame.");
106 if (!aom_video_writer_write_frame(writer, pkt->
data.
frame.buf,
109 die_codec(ctx,
"Failed to write compressed frame.");
110 printf(keyframe ?
"K" :
".");
120 if (FORCE_HIGHBITDEPTH_DECODING) {
123 aom_img_upshift(raw_shift, raw, input_shift);
124 *frame_to_encode = raw_shift;
126 *frame_to_encode = raw;
133 int lf_height,
int lf_blocksize,
int flags,
137 int image_size_bytes = img_size_bytes(raw);
138 int u_blocks, v_blocks;
144 die(
"Failed to initialize encoder");
146 die_codec(&codec,
"Failed to turn off auto altref");
148 die_codec(&codec,
"Failed to set frame parallel decoding");
151 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
152 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
154 printf(
"\n First pass: ");
156 for (bv = 0; bv < v_blocks; ++bv) {
157 for (bu = 0; bu < u_blocks; ++bu) {
158 const int block_u_min = bu * lf_blocksize;
159 const int block_v_min = bv * lf_blocksize;
160 int block_u_end = (bu + 1) * lf_blocksize;
161 int block_v_end = (bv + 1) * lf_blocksize;
162 int u_block_size, v_block_size;
163 int block_ref_u, block_ref_v;
165 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
166 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
167 u_block_size = block_u_end - block_u_min;
168 v_block_size = block_v_end - block_v_min;
169 block_ref_u = block_u_min + u_block_size / 2;
170 block_ref_v = block_v_min + v_block_size / 2;
172 printf(
"A%d, ", (block_ref_u + block_ref_v * lf_width));
173 fseek(infile, (block_ref_u + block_ref_v * lf_width) * image_size_bytes,
175 aom_img_read(raw, infile);
176 get_raw_image(&frame_to_encode, raw, raw_shift);
180 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
191 die_codec(&codec,
"Failed to set frame parallel decoding");
193 for (bv = 0; bv < v_blocks; ++bv) {
194 for (bu = 0; bu < u_blocks; ++bu) {
195 const int block_u_min = bu * lf_blocksize;
196 const int block_v_min = bv * lf_blocksize;
197 int block_u_end = (bu + 1) * lf_blocksize;
198 int block_v_end = (bv + 1) * lf_blocksize;
200 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
201 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
202 for (v = block_v_min; v < block_v_end; ++v) {
203 for (u = block_u_min; u < block_u_end; ++u) {
204 printf(
"C%d, ", (u + v * lf_width));
205 fseek(infile, (u + v * lf_width) * image_size_bytes, SEEK_SET);
206 aom_img_read(raw, infile);
207 get_raw_image(&frame_to_encode, raw, raw_shift);
210 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
223 while (get_frame_stats(&codec, NULL, frame_count, 1, 0, &stats)) {
228 printf(
"\nFirst pass complete. Processed %d frames.\n", frame_count);
233 static void pass1(
aom_image_t *raw, FILE *infile,
const char *outfile_name,
235 int lf_width,
int lf_height,
int lf_blocksize,
int flags,
237 AvxVideoInfo info = { get_fourcc_by_aom_encoder(encoder),
242 AvxVideoWriter *writer = NULL;
245 int image_size_bytes = img_size_bytes(raw);
247 int u_blocks, v_blocks;
249 aom_image_t reference_images[MAX_EXTERNAL_REFERENCES];
250 int reference_image_num = 0;
253 writer = aom_video_writer_open(outfile_name, kContainerIVF, &info);
254 if (!writer) die(
"Failed to open %s for writing", outfile_name);
257 die(
"Failed to initialize encoder");
259 die_codec(&codec,
"Failed to turn off auto altref");
261 die_codec(&codec,
"Failed to set frame parallel decoding");
263 die_codec(&codec,
"Failed to enable encoder ext_tile debug");
265 die_codec(&codec,
"Failed to set cpu-used");
274 die_codec(&codec,
"Failed to set SB size");
276 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
277 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
279 reference_image_num = u_blocks * v_blocks;
288 reference_image_num - 1))
289 die_codec(&codec,
"Failed to set max gf interval");
293 int border_in_pixels =
295 ? AOM_BORDER_IN_PIXELS
296 : AOM_ENC_NO_SCALE_BORDER;
297 for (i = 0; i < reference_image_num; i++) {
299 cfg->
g_h, 32, 8, border_in_pixels)) {
300 die(
"Failed to allocate image.");
304 printf(
"\n Second pass: ");
307 printf(
"Encoding Reference Images\n");
308 for (bv = 0; bv < v_blocks; ++bv) {
309 for (bu = 0; bu < u_blocks; ++bu) {
310 const int block_u_min = bu * lf_blocksize;
311 const int block_v_min = bv * lf_blocksize;
312 int block_u_end = (bu + 1) * lf_blocksize;
313 int block_v_end = (bv + 1) * lf_blocksize;
314 int u_block_size, v_block_size;
315 int block_ref_u, block_ref_v;
317 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
318 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
319 u_block_size = block_u_end - block_u_min;
320 v_block_size = block_v_end - block_v_min;
321 block_ref_u = block_u_min + u_block_size / 2;
322 block_ref_v = block_v_min + v_block_size / 2;
324 printf(
"A%d, ", (block_ref_u + block_ref_v * lf_width));
325 fseek(infile, (block_ref_u + block_ref_v * lf_width) * image_size_bytes,
327 aom_img_read(raw, infile);
329 get_raw_image(&frame_to_encode, raw, raw_shift);
333 printf(
"Encoding reference image %d of %d\n", bv * u_blocks + bu,
334 u_blocks * v_blocks);
335 encode_frame(&codec, frame_to_encode, frame_count, 1,
344 &reference_images[frame_count - 1]))
345 die_codec(&codec,
"Failed to copy decoder reference frame");
353 die_codec(&codec,
"Failed to configure encoder");
357 die_codec(&codec,
"Failed to set cq level");
359 die_codec(&codec,
"Failed to set frame parallel decoding");
361 die_codec(&codec,
"Failed to turn on single tile decoding");
365 die_codec(&codec,
"Failed to set tile width");
367 die_codec(&codec,
"Failed to set tile height");
369 for (bv = 0; bv < v_blocks; ++bv) {
370 for (bu = 0; bu < u_blocks; ++bu) {
371 const int block_u_min = bu * lf_blocksize;
372 const int block_v_min = bv * lf_blocksize;
373 int block_u_end = (bu + 1) * lf_blocksize;
374 int block_v_end = (bv + 1) * lf_blocksize;
376 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
377 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
378 for (v = block_v_min; v < block_v_end; ++v) {
379 for (u = block_u_min; u < block_u_end; ++u) {
383 ref.
img = reference_images[bv * u_blocks + bu];
385 die_codec(&codec,
"Failed to set reference frame");
387 printf(
"C%d, ", (u + v * lf_width));
388 fseek(infile, (u + v * lf_width) * image_size_bytes, SEEK_SET);
389 aom_img_read(raw, infile);
390 get_raw_image(&frame_to_encode, raw, raw_shift);
393 printf(
"Encoding image %d of %d\n",
394 frame_count - (u_blocks * v_blocks), lf_width * lf_height);
395 encode_frame(&codec, frame_to_encode, frame_count, 1,
409 while (encode_frame(&codec, NULL, -1, 1, 0, writer)) {
412 for (i = 0; i < reference_image_num; i++)
aom_img_free(&reference_images[i]);
418 aom_video_writer_set_fourcc(writer, LST_FOURCC);
419 aom_video_writer_close(writer);
421 printf(
"\nSecond pass complete. Processed %d frames.\n", frame_count);
424 int main(
int argc,
char **argv) {
428 int lf_width, lf_height;
442 const int bitrate = 200;
443 const char *
const width_arg = argv[1];
444 const char *
const height_arg = argv[2];
445 const char *
const infile_arg = argv[3];
446 const char *
const outfile_arg = argv[4];
447 const char *
const lf_width_arg = argv[5];
448 const char *
const lf_height_arg = argv[6];
449 const char *lf_blocksize_arg = argv[7];
452 if (argc < 8) die(
"Invalid number of arguments");
455 if (!encoder) die(
"Unsupported codec.");
457 w = (int)strtol(width_arg, NULL, 0);
458 h = (int)strtol(height_arg, NULL, 0);
459 lf_width = (int)strtol(lf_width_arg, NULL, 0);
460 lf_height = (int)strtol(lf_height_arg, NULL, 0);
461 lf_blocksize = (int)strtol(lf_blocksize_arg, NULL, 0);
462 lf_blocksize = lf_blocksize < lf_width ? lf_blocksize : lf_width;
463 lf_blocksize = lf_blocksize < lf_height ? lf_blocksize : lf_height;
465 if (w <= 0 || h <= 0 || (w % 2) != 0 || (h % 2) != 0)
466 die(
"Invalid frame size: %dx%d", w, h);
467 if (lf_width <= 0 || lf_height <= 0)
468 die(
"Invalid lf_width and/or lf_height: %dx%d", lf_width, lf_height);
469 if (lf_blocksize <= 0) die(
"Invalid lf_blocksize: %d", lf_blocksize);
472 die(
"Failed to allocate image.");
474 if (FORCE_HIGHBITDEPTH_DECODING) {
484 if (res) die_codec(&codec,
"Failed to get default codec config.");
500 if (!(infile = fopen(infile_arg,
"rb")))
501 die(
"Failed to open %s for reading", infile_arg);
505 stats = pass0(&raw, infile, encoder, &cfg, lf_width, lf_height, lf_blocksize,
512 pass1(&raw, infile, outfile_arg, encoder, &cfg, lf_width, lf_height,
513 lf_blocksize, flags, &raw_shift);
516 if (FORCE_HIGHBITDEPTH_DECODING)
aom_img_free(&raw_shift);