1 /*
2 * Copyright (c) 2017, 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 // Lightfield Encoder
13 // ==================
14 //
15 // This is an example of a simple lightfield encoder. It builds upon the
16 // twopass_encoder.c example. It takes an input file in YV12 format,
17 // treating it as a planar lightfield instead of a video. The img_width
18 // and img_height arguments are the dimensions of the lightfield images,
19 // while the lf_width and lf_height arguments are the number of
20 // lightfield images in each dimension. The lf_blocksize determines the
21 // number of reference images used for MCP. For example, 5 means that there
22 // is a reference image for every 5x5 lightfield image block. All images
23 // within a block will use the center image in that block as the reference
24 // image for MCP.
25 // Run "make test" to download lightfield test data: vase10x10.yuv.
26 // Run lightfield encoder to encode whole lightfield:
27 // examples/lightfield_encoder 1024 1024 vase10x10.yuv vase10x10.ivf 10 10 5
28
29 // Note: In bitstream.c and encoder.c, define EXT_TILE_DEBUG as 1 will print
30 // out the uncompressed header and the frame contexts, which can be used to
31 // test the bit exactness of the headers and the frame contexts for large scale
32 // tile coded frames.
33
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <string.h>
37
38 #include "aom/aom_encoder.h"
39 #include "aom/aomcx.h"
40 #include "aom_scale/yv12config.h"
41 #include "av1/common/enums.h"
42 #include "common/tools_common.h"
43 #include "common/video_writer.h"
44
45 static const char *exec_name;
46
usage_exit(void)47 void usage_exit(void) {
48 fprintf(stderr,
49 "Usage: %s <img_width> <img_height> <infile> <outfile> "
50 "<lf_width> <lf_height> <lf_blocksize>\n",
51 exec_name);
52 exit(EXIT_FAILURE);
53 }
54
aom_img_size_bytes(aom_image_t * img)55 static int aom_img_size_bytes(aom_image_t *img) {
56 int image_size_bytes = 0;
57 int plane;
58 for (plane = 0; plane < 3; ++plane) {
59 const int w = aom_img_plane_width(img, plane) *
60 ((img->fmt & AOM_IMG_FMT_HIGHBITDEPTH) ? 2 : 1);
61 const int h = aom_img_plane_height(img, plane);
62 image_size_bytes += w * h;
63 }
64 return image_size_bytes;
65 }
66
get_frame_stats(aom_codec_ctx_t * ctx,const aom_image_t * img,aom_codec_pts_t pts,unsigned int duration,aom_enc_frame_flags_t flags,aom_fixed_buf_t * stats)67 static int get_frame_stats(aom_codec_ctx_t *ctx, const aom_image_t *img,
68 aom_codec_pts_t pts, unsigned int duration,
69 aom_enc_frame_flags_t flags,
70 aom_fixed_buf_t *stats) {
71 int got_pkts = 0;
72 aom_codec_iter_t iter = NULL;
73 const aom_codec_cx_pkt_t *pkt = NULL;
74 const aom_codec_err_t res = aom_codec_encode(ctx, img, pts, duration, flags);
75 if (res != AOM_CODEC_OK) die_codec(ctx, "Failed to get frame stats.");
76
77 while ((pkt = aom_codec_get_cx_data(ctx, &iter)) != NULL) {
78 got_pkts = 1;
79
80 if (pkt->kind == AOM_CODEC_STATS_PKT) {
81 const uint8_t *const pkt_buf = pkt->data.twopass_stats.buf;
82 const size_t pkt_size = pkt->data.twopass_stats.sz;
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;
86 }
87 }
88
89 return got_pkts;
90 }
91
encode_frame(aom_codec_ctx_t * ctx,const aom_image_t * img,aom_codec_pts_t pts,unsigned int duration,aom_enc_frame_flags_t flags,AvxVideoWriter * writer)92 static int encode_frame(aom_codec_ctx_t *ctx, const aom_image_t *img,
93 aom_codec_pts_t pts, unsigned int duration,
94 aom_enc_frame_flags_t flags, AvxVideoWriter *writer) {
95 int got_pkts = 0;
96 aom_codec_iter_t iter = NULL;
97 const aom_codec_cx_pkt_t *pkt = NULL;
98 const aom_codec_err_t res = aom_codec_encode(ctx, img, pts, duration, flags);
99 if (res != AOM_CODEC_OK) die_codec(ctx, "Failed to encode frame.");
100
101 while ((pkt = aom_codec_get_cx_data(ctx, &iter)) != NULL) {
102 got_pkts = 1;
103 if (pkt->kind == AOM_CODEC_CX_FRAME_PKT) {
104 const int keyframe = (pkt->data.frame.flags & AOM_FRAME_IS_KEY) != 0;
105
106 if (!aom_video_writer_write_frame(writer, pkt->data.frame.buf,
107 pkt->data.frame.sz,
108 pkt->data.frame.pts))
109 die_codec(ctx, "Failed to write compressed frame.");
110 printf(keyframe ? "K" : ".");
111 fflush(stdout);
112 }
113 }
114
115 return got_pkts;
116 }
117
get_raw_image(aom_image_t ** frame_to_encode,aom_image_t * raw,aom_image_t * raw_shift)118 static void get_raw_image(aom_image_t **frame_to_encode, aom_image_t *raw,
119 aom_image_t *raw_shift) {
120 if (!CONFIG_LOWBITDEPTH) {
121 // Need to allocate larger buffer to use hbd internal.
122 int input_shift = 0;
123 aom_img_upshift(raw_shift, raw, input_shift);
124 *frame_to_encode = raw_shift;
125 } else {
126 *frame_to_encode = raw;
127 }
128 }
129
pass0(aom_image_t * raw,FILE * infile,const AvxInterface * encoder,const aom_codec_enc_cfg_t * cfg,int lf_width,int lf_height,int lf_blocksize,int flags,aom_image_t * raw_shift)130 static aom_fixed_buf_t pass0(aom_image_t *raw, FILE *infile,
131 const AvxInterface *encoder,
132 const aom_codec_enc_cfg_t *cfg, int lf_width,
133 int lf_height, int lf_blocksize, int flags,
134 aom_image_t *raw_shift) {
135 aom_codec_ctx_t codec;
136 int frame_count = 0;
137 int image_size_bytes = aom_img_size_bytes(raw);
138 int u_blocks, v_blocks;
139 int bu, bv;
140 aom_fixed_buf_t stats = { NULL, 0 };
141 aom_image_t *frame_to_encode;
142
143 if (aom_codec_enc_init(&codec, encoder->codec_interface(), cfg, flags))
144 die_codec(&codec, "Failed to initialize encoder");
145 if (aom_codec_control(&codec, AOME_SET_ENABLEAUTOALTREF, 0))
146 die_codec(&codec, "Failed to turn off auto altref");
147 if (aom_codec_control(&codec, AV1E_SET_FRAME_PARALLEL_DECODING, 0))
148 die_codec(&codec, "Failed to set frame parallel decoding");
149
150 // How many reference images we need to encode.
151 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
152 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
153
154 printf("\n First pass: ");
155
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;
164
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;
171
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,
174 SEEK_SET);
175 aom_img_read(raw, infile);
176 get_raw_image(&frame_to_encode, raw, raw_shift);
177
178 // Reference frames can be encoded encoded without tiles.
179 ++frame_count;
180 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
181 AOM_EFLAG_NO_REF_LAST2 | AOM_EFLAG_NO_REF_LAST3 |
182 AOM_EFLAG_NO_REF_GF | AOM_EFLAG_NO_REF_ARF |
183 AOM_EFLAG_NO_REF_BWD | AOM_EFLAG_NO_REF_ARF2 |
184 AOM_EFLAG_NO_UPD_LAST | AOM_EFLAG_NO_UPD_GF |
185 AOM_EFLAG_NO_UPD_ARF,
186 &stats);
187 }
188 }
189
190 if (aom_codec_control(&codec, AV1E_SET_FRAME_PARALLEL_DECODING, 1))
191 die_codec(&codec, "Failed to set frame parallel decoding");
192
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;
199 int u, v;
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);
208
209 ++frame_count;
210 get_frame_stats(&codec, frame_to_encode, frame_count, 1,
211 AOM_EFLAG_NO_REF_LAST2 | AOM_EFLAG_NO_REF_LAST3 |
212 AOM_EFLAG_NO_REF_GF | AOM_EFLAG_NO_REF_ARF |
213 AOM_EFLAG_NO_REF_BWD | AOM_EFLAG_NO_REF_ARF2 |
214 AOM_EFLAG_NO_UPD_LAST | AOM_EFLAG_NO_UPD_GF |
215 AOM_EFLAG_NO_UPD_ARF | AOM_EFLAG_NO_UPD_ENTROPY,
216 &stats);
217 }
218 }
219 }
220 }
221 // Flush encoder.
222 // No ARF, this should not be needed.
223 while (get_frame_stats(&codec, NULL, frame_count, 1, 0, &stats)) {
224 }
225
226 if (aom_codec_destroy(&codec)) die_codec(&codec, "Failed to destroy codec.");
227
228 printf("\nFirst pass complete. Processed %d frames.\n", frame_count);
229
230 return stats;
231 }
232
pass1(aom_image_t * raw,FILE * infile,const char * outfile_name,const AvxInterface * encoder,aom_codec_enc_cfg_t * cfg,int lf_width,int lf_height,int lf_blocksize,int flags,aom_image_t * raw_shift)233 static void pass1(aom_image_t *raw, FILE *infile, const char *outfile_name,
234 const AvxInterface *encoder, aom_codec_enc_cfg_t *cfg,
235 int lf_width, int lf_height, int lf_blocksize, int flags,
236 aom_image_t *raw_shift) {
237 AvxVideoInfo info = { encoder->fourcc,
238 cfg->g_w,
239 cfg->g_h,
240 { cfg->g_timebase.num, cfg->g_timebase.den },
241 0 };
242 AvxVideoWriter *writer = NULL;
243 aom_codec_ctx_t codec;
244 int frame_count = 0;
245 int image_size_bytes = aom_img_size_bytes(raw);
246 int bu, bv;
247 int u_blocks, v_blocks;
248 aom_image_t *frame_to_encode;
249 aom_image_t reference_images[MAX_EXTERNAL_REFERENCES];
250 int reference_image_num = 0;
251 int i;
252
253 writer = aom_video_writer_open(outfile_name, kContainerIVF, &info);
254 if (!writer) die("Failed to open %s for writing", outfile_name);
255
256 if (aom_codec_enc_init(&codec, encoder->codec_interface(), cfg, flags))
257 die_codec(&codec, "Failed to initialize encoder");
258 if (aom_codec_control(&codec, AOME_SET_ENABLEAUTOALTREF, 0))
259 die_codec(&codec, "Failed to turn off auto altref");
260 if (aom_codec_control(&codec, AV1E_SET_FRAME_PARALLEL_DECODING, 0))
261 die_codec(&codec, "Failed to set frame parallel decoding");
262 // Note: The superblock is a sequence parameter and has to be the same for 1
263 // sequence. In lightfield application, must choose the superblock size(either
264 // 64x64 or 128x128) before the encoding starts. Otherwise, the default is
265 // AOM_SUPERBLOCK_SIZE_DYNAMIC, and the superblock size will be set to 64x64
266 // internally.
267 if (aom_codec_control(&codec, AV1E_SET_SUPERBLOCK_SIZE,
268 AOM_SUPERBLOCK_SIZE_64X64))
269 die_codec(&codec, "Failed to set SB size");
270
271 u_blocks = (lf_width + lf_blocksize - 1) / lf_blocksize;
272 v_blocks = (lf_height + lf_blocksize - 1) / lf_blocksize;
273
274 reference_image_num = u_blocks * v_blocks;
275 aom_img_fmt_t ref_fmt = AOM_IMG_FMT_I420;
276 if (!CONFIG_LOWBITDEPTH) ref_fmt |= AOM_IMG_FMT_HIGHBITDEPTH;
277 // Allocate memory with the border so that it can be used as a reference.
278 for (i = 0; i < reference_image_num; i++) {
279 if (!aom_img_alloc_with_border(&reference_images[i], ref_fmt, cfg->g_w,
280 cfg->g_h, 32, 8, AOM_BORDER_IN_PIXELS)) {
281 die("Failed to allocate image.");
282 }
283 }
284
285 printf("\n Second pass: ");
286
287 // Encode reference images first.
288 printf("Encoding Reference Images\n");
289 for (bv = 0; bv < v_blocks; ++bv) {
290 for (bu = 0; bu < u_blocks; ++bu) {
291 const int block_u_min = bu * lf_blocksize;
292 const int block_v_min = bv * lf_blocksize;
293 int block_u_end = (bu + 1) * lf_blocksize;
294 int block_v_end = (bv + 1) * lf_blocksize;
295 int u_block_size, v_block_size;
296 int block_ref_u, block_ref_v;
297
298 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
299 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
300 u_block_size = block_u_end - block_u_min;
301 v_block_size = block_v_end - block_v_min;
302 block_ref_u = block_u_min + u_block_size / 2;
303 block_ref_v = block_v_min + v_block_size / 2;
304
305 printf("A%d, ", (block_ref_u + block_ref_v * lf_width));
306 fseek(infile, (block_ref_u + block_ref_v * lf_width) * image_size_bytes,
307 SEEK_SET);
308 aom_img_read(raw, infile);
309
310 get_raw_image(&frame_to_encode, raw, raw_shift);
311
312 // Reference frames may be encoded without tiles.
313 ++frame_count;
314 printf("Encoding reference image %d of %d\n", bv * u_blocks + bu,
315 u_blocks * v_blocks);
316 encode_frame(&codec, frame_to_encode, frame_count, 1,
317 AOM_EFLAG_NO_REF_LAST2 | AOM_EFLAG_NO_REF_LAST3 |
318 AOM_EFLAG_NO_REF_GF | AOM_EFLAG_NO_REF_ARF |
319 AOM_EFLAG_NO_REF_BWD | AOM_EFLAG_NO_REF_ARF2 |
320 AOM_EFLAG_NO_UPD_LAST | AOM_EFLAG_NO_UPD_GF |
321 AOM_EFLAG_NO_UPD_ARF | AOM_EFLAG_NO_UPD_ENTROPY,
322 writer);
323
324 if (aom_codec_control(&codec, AV1_COPY_NEW_FRAME_IMAGE,
325 &reference_images[frame_count - 1]))
326 die_codec(&codec, "Failed to copy decoder reference frame");
327 }
328 }
329
330 cfg->large_scale_tile = 1;
331 // Fixed q encoding for camera frames.
332 cfg->rc_end_usage = AOM_Q;
333 if (aom_codec_enc_config_set(&codec, cfg))
334 die_codec(&codec, "Failed to configure encoder");
335
336 // The fixed q value used in encoding.
337 if (aom_codec_control(&codec, AOME_SET_CQ_LEVEL, 36))
338 die_codec(&codec, "Failed to set cq level");
339 if (aom_codec_control(&codec, AV1E_SET_FRAME_PARALLEL_DECODING, 1))
340 die_codec(&codec, "Failed to set frame parallel decoding");
341 if (aom_codec_control(&codec, AV1E_SET_SINGLE_TILE_DECODING, 1))
342 die_codec(&codec, "Failed to turn on single tile decoding");
343 // Set tile_columns and tile_rows to MAX values, which guarantees the tile
344 // size of 64 x 64 pixels(i.e. 1 SB) for <= 4k resolution.
345 if (aom_codec_control(&codec, AV1E_SET_TILE_COLUMNS, 6))
346 die_codec(&codec, "Failed to set tile width");
347 if (aom_codec_control(&codec, AV1E_SET_TILE_ROWS, 6))
348 die_codec(&codec, "Failed to set tile height");
349
350 for (bv = 0; bv < v_blocks; ++bv) {
351 for (bu = 0; bu < u_blocks; ++bu) {
352 const int block_u_min = bu * lf_blocksize;
353 const int block_v_min = bv * lf_blocksize;
354 int block_u_end = (bu + 1) * lf_blocksize;
355 int block_v_end = (bv + 1) * lf_blocksize;
356 int u, v;
357 block_u_end = block_u_end < lf_width ? block_u_end : lf_width;
358 block_v_end = block_v_end < lf_height ? block_v_end : lf_height;
359 for (v = block_v_min; v < block_v_end; ++v) {
360 for (u = block_u_min; u < block_u_end; ++u) {
361 av1_ref_frame_t ref;
362 ref.idx = 0;
363 ref.use_external_ref = 1;
364 ref.img = reference_images[bv * u_blocks + bu];
365 if (aom_codec_control(&codec, AV1_SET_REFERENCE, &ref))
366 die_codec(&codec, "Failed to set reference frame");
367
368 printf("C%d, ", (u + v * lf_width));
369 fseek(infile, (u + v * lf_width) * image_size_bytes, SEEK_SET);
370 aom_img_read(raw, infile);
371 get_raw_image(&frame_to_encode, raw, raw_shift);
372
373 ++frame_count;
374 printf("Encoding image %d of %d\n",
375 frame_count - (u_blocks * v_blocks), lf_width * lf_height);
376 encode_frame(&codec, frame_to_encode, frame_count, 1,
377 AOM_EFLAG_NO_REF_LAST2 | AOM_EFLAG_NO_REF_LAST3 |
378 AOM_EFLAG_NO_REF_GF | AOM_EFLAG_NO_REF_ARF |
379 AOM_EFLAG_NO_REF_BWD | AOM_EFLAG_NO_REF_ARF2 |
380 AOM_EFLAG_NO_UPD_LAST | AOM_EFLAG_NO_UPD_GF |
381 AOM_EFLAG_NO_UPD_ARF | AOM_EFLAG_NO_UPD_ENTROPY,
382 writer);
383 }
384 }
385 }
386 }
387
388 // Flush encoder.
389 // No ARF, this should not be needed.
390 while (encode_frame(&codec, NULL, -1, 1, 0, writer)) {
391 }
392
393 for (i = 0; i < reference_image_num; i++) aom_img_free(&reference_images[i]);
394
395 if (aom_codec_destroy(&codec)) die_codec(&codec, "Failed to destroy codec.");
396 aom_video_writer_close(writer);
397
398 printf("\nSecond pass complete. Processed %d frames.\n", frame_count);
399 }
400
main(int argc,char ** argv)401 int main(int argc, char **argv) {
402 FILE *infile = NULL;
403 int w, h;
404 // The number of lightfield images in the u and v dimensions.
405 int lf_width, lf_height;
406 // Defines how many images refer to the same reference image for MCP.
407 // lf_blocksize X lf_blocksize images will all use the reference image
408 // in the middle of the block of images.
409 int lf_blocksize;
410 aom_codec_ctx_t codec;
411 aom_codec_enc_cfg_t cfg;
412 aom_image_t raw;
413 aom_image_t raw_shift;
414 aom_codec_err_t res;
415 aom_fixed_buf_t stats;
416 int flags = 0;
417
418 const AvxInterface *encoder = NULL;
419 const int fps = 30;
420 const int bitrate = 200; // kbit/s
421 const char *const width_arg = argv[1];
422 const char *const height_arg = argv[2];
423 const char *const infile_arg = argv[3];
424 const char *const outfile_arg = argv[4];
425 const char *const lf_width_arg = argv[5];
426 const char *const lf_height_arg = argv[6];
427 const char *lf_blocksize_arg = argv[7];
428 exec_name = argv[0];
429
430 if (argc < 8) die("Invalid number of arguments");
431
432 encoder = get_aom_encoder_by_name("av1");
433 if (!encoder) die("Unsupported codec.");
434
435 w = (int)strtol(width_arg, NULL, 0);
436 h = (int)strtol(height_arg, NULL, 0);
437 lf_width = (int)strtol(lf_width_arg, NULL, 0);
438 lf_height = (int)strtol(lf_height_arg, NULL, 0);
439 lf_blocksize = (int)strtol(lf_blocksize_arg, NULL, 0);
440 lf_blocksize = lf_blocksize < lf_width ? lf_blocksize : lf_width;
441 lf_blocksize = lf_blocksize < lf_height ? lf_blocksize : lf_height;
442
443 if (w <= 0 || h <= 0 || (w % 2) != 0 || (h % 2) != 0)
444 die("Invalid frame size: %dx%d", w, h);
445 if (lf_width <= 0 || lf_height <= 0)
446 die("Invalid lf_width and/or lf_height: %dx%d", lf_width, lf_height);
447 if (lf_blocksize <= 0) die("Invalid lf_blocksize: %d", lf_blocksize);
448
449 if (!aom_img_alloc(&raw, AOM_IMG_FMT_I420, w, h, 32)) {
450 die("Failed to allocate image.");
451 }
452 if (!CONFIG_LOWBITDEPTH) {
453 // Need to allocate larger buffer to use hbd internal.
454 aom_img_alloc(&raw_shift, AOM_IMG_FMT_I420 | AOM_IMG_FMT_HIGHBITDEPTH, w, h,
455 32);
456 }
457
458 printf("Using %s\n", aom_codec_iface_name(encoder->codec_interface()));
459
460 // Configuration
461 res = aom_codec_enc_config_default(encoder->codec_interface(), &cfg, 0);
462 if (res) die_codec(&codec, "Failed to get default codec config.");
463
464 cfg.g_w = w;
465 cfg.g_h = h;
466 cfg.g_timebase.num = 1;
467 cfg.g_timebase.den = fps;
468 cfg.rc_target_bitrate = bitrate;
469 cfg.g_error_resilient = 0; // This is required.
470 cfg.g_lag_in_frames = 0; // need to set this since default is 19.
471 cfg.kf_mode = AOM_KF_DISABLED;
472 cfg.large_scale_tile = 0; // Only set it to 1 for camera frame encoding.
473 cfg.g_bit_depth = AOM_BITS_8;
474 flags |= (cfg.g_bit_depth > AOM_BITS_8 || !CONFIG_LOWBITDEPTH)
475 ? AOM_CODEC_USE_HIGHBITDEPTH
476 : 0;
477
478 if (!(infile = fopen(infile_arg, "rb")))
479 die("Failed to open %s for reading", infile_arg);
480
481 // Pass 0
482 cfg.g_pass = AOM_RC_FIRST_PASS;
483 stats = pass0(&raw, infile, encoder, &cfg, lf_width, lf_height, lf_blocksize,
484 flags, &raw_shift);
485
486 // Pass 1
487 rewind(infile);
488 cfg.g_pass = AOM_RC_LAST_PASS;
489 cfg.rc_twopass_stats_in = stats;
490 pass1(&raw, infile, outfile_arg, encoder, &cfg, lf_width, lf_height,
491 lf_blocksize, flags, &raw_shift);
492 free(stats.buf);
493
494 if (!CONFIG_LOWBITDEPTH) aom_img_free(&raw_shift);
495 aom_img_free(&raw);
496 fclose(infile);
497
498 return EXIT_SUCCESS;
499 }
500