55 #include "common/tools_common.h"
56 #include "common/video_writer.h"
58 static const char *exec_name;
60 void usage_exit(
void) {
62 "Usage: %s <codec> <width> <height> <infile> <outfile> "
63 "<limit(optional)>\n",
76 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to get frame stats.");
84 stats->
buf = realloc(stats->
buf, stats->
sz + pkt_size);
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" :
".");
120 const AvxInterface *encoder,
127 die_codec(&codec,
"Failed to initialize encoder");
130 while (aom_img_read(raw, infile) && frame_count < limit) {
132 get_frame_stats(&codec, raw, frame_count, 1, 0, &stats);
136 while (get_frame_stats(&codec, NULL, frame_count, 1, 0, &stats)) {
139 printf(
"Pass 0 complete. Processed %d frames.\n", frame_count);
145 static void pass1(
aom_image_t *raw, FILE *infile,
const char *outfile_name,
148 AvxVideoInfo info = { encoder->fourcc,
153 AvxVideoWriter *writer = NULL;
157 writer = aom_video_writer_open(outfile_name, kContainerIVF, &info);
158 if (!writer) die(
"Failed to open %s for writing", outfile_name);
161 die_codec(&codec,
"Failed to initialize encoder");
164 while (aom_img_read(raw, infile) && frame_count < limit) {
166 encode_frame(&codec, raw, frame_count, 1, 0, writer);
170 while (encode_frame(&codec, NULL, -1, 1, 0, writer)) {
177 aom_video_writer_close(writer);
179 printf(
"Pass 1 complete. Processed %d frames.\n", frame_count);
182 int main(
int argc,
char **argv) {
191 const AvxInterface *encoder = NULL;
193 const int bitrate = 200;
194 const char *
const codec_arg = argv[1];
195 const char *
const width_arg = argv[2];
196 const char *
const height_arg = argv[3];
197 const char *
const infile_arg = argv[4];
198 const char *
const outfile_arg = argv[5];
202 if (argc < 6) die(
"Invalid number of arguments");
204 if (argc > 6) limit = (int)strtol(argv[6], NULL, 0);
206 if (limit == 0) limit = 100;
208 encoder = get_aom_encoder_by_name(codec_arg);
209 if (!encoder) die(
"Unsupported codec.");
211 w = (int)strtol(width_arg, NULL, 0);
212 h = (int)strtol(height_arg, NULL, 0);
214 if (w <= 0 || h <= 0 || (w % 2) != 0 || (h % 2) != 0)
215 die(
"Invalid frame size: %dx%d", w, h);
218 die(
"Failed to allocate image", w, h);
224 if (res) die_codec(&codec,
"Failed to get default codec config.");
232 if (!(infile = fopen(infile_arg,
"rb")))
233 die(
"Failed to open %s for reading", infile_arg);
237 stats = pass0(&raw, infile, encoder, &cfg, limit);
243 pass1(&raw, infile, outfile_arg, encoder, &cfg, limit);
@ AOM_IMG_FMT_I420
Definition: aom_image.h:45
Encoder configuration structure.
Definition: aom_encoder.h:228
Generic fixed size buffer structure.
Definition: aom_encoder.h:76
Codec context structure.
Definition: aom_codec.h:204
const typedef void * aom_codec_iter_t
Iterator.
Definition: aom_codec.h:194
@ AOM_CODEC_OK
Operation completed without error.
Definition: aom_codec.h:103
unsigned int rc_target_bitrate
Target data rate.
Definition: aom_encoder.h:481
@ AOM_RC_FIRST_PASS
Definition: aom_encoder.h:186
void aom_img_free(aom_image_t *img)
Close an image descriptor.
const char * aom_codec_iface_name(aom_codec_iface_t *iface)
Return the name for a given interface.
int den
Definition: aom_encoder.h:180
aom_codec_err_t aom_codec_destroy(aom_codec_ctx_t *ctx)
Destroy a codec instance.
@ AOM_RC_LAST_PASS
Definition: aom_encoder.h:187
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_fixed_buf_t twopass_stats
Definition: aom_encoder.h:156
enum aom_enc_pass g_pass
Multi-pass Encoding Mode.
Definition: aom_encoder.h:340
int64_t aom_codec_pts_t
Time Stamp Type.
Definition: aom_encoder.h:86
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition: aom_encoder.h:219
unsigned int g_w
Width of the frame.
Definition: aom_encoder.h:267
aom_fixed_buf_t rc_twopass_stats_in
Two-pass stats buffer.
Definition: aom_encoder.h:468
aom_codec_err_t
Algorithm return codes.
Definition: aom_codec.h:101
Describes the encoder algorithm interface to applications.
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.
union aom_codec_cx_pkt::@1 data
struct aom_codec_cx_pkt::@1::@2 frame
Encoder output packet.
Definition: aom_encoder.h:138
@ AOM_CODEC_STATS_PKT
Definition: aom_encoder.h:127
unsigned int g_h
Height of the frame.
Definition: aom_encoder.h:276
@ AOM_CODEC_CX_FRAME_PKT
Definition: aom_encoder.h:126
#define aom_codec_enc_init(ctx, iface, cfg, flags)
Convenience macro for aom_codec_enc_init_ver()
Definition: aom_encoder.h:764
Image Descriptor.
Definition: aom_image.h:141
size_t sz
Definition: aom_encoder.h:78
enum aom_codec_cx_pkt_kind kind
Definition: aom_encoder.h:139
struct aom_rational g_timebase
Stream timebase units.
Definition: aom_encoder.h:325
#define AOM_FRAME_IS_KEY
Definition: aom_encoder.h:96
void * buf
Definition: aom_encoder.h:77
aom_codec_err_t aom_codec_enc_config_default(aom_codec_iface_t *iface, aom_codec_enc_cfg_t *cfg, unsigned int reserved)
Get a default configuration.
int num
Definition: aom_encoder.h:179