56 #include "common/tools_common.h" 57 #include "common/video_writer.h" 59 static const char *exec_name;
61 void usage_exit(
void) {
63 "Usage: %s <codec> <width> <height> <infile> <outfile> " 64 "<limit(optional)>\n",
77 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to get frame stats.");
85 stats->
buf = realloc(stats->
buf, stats->
sz + pkt_size);
86 memcpy((uint8_t *)stats->
buf + stats->
sz, pkt_buf, pkt_size);
87 stats->
sz += pkt_size;
101 if (res !=
AOM_CODEC_OK) die_codec(ctx,
"Failed to encode frame.");
108 if (!aom_video_writer_write_frame(writer, pkt->
data.
frame.buf,
111 die_codec(ctx,
"Failed to write compressed frame.");
112 printf(keyframe ?
"K" :
".");
128 die(
"Failed to initialize encoder");
131 while (aom_img_read(raw, infile) && frame_count < limit) {
133 get_frame_stats(&codec, raw, frame_count, 1, 0, &stats);
137 while (get_frame_stats(&codec, NULL, frame_count, 1, 0, &stats)) {
140 printf(
"Pass 0 complete. Processed %d frames.\n", frame_count);
146 static void pass1(
aom_image_t *raw, FILE *infile,
const char *outfile_name,
149 AvxVideoInfo info = { get_fourcc_by_aom_encoder(encoder),
154 AvxVideoWriter *writer = NULL;
158 writer = aom_video_writer_open(outfile_name, kContainerIVF, &info);
159 if (!writer) die(
"Failed to open %s for writing", outfile_name);
162 die(
"Failed to initialize encoder");
165 die_codec(&codec,
"Failed to set cpu-used");
168 while (aom_img_read(raw, infile) && frame_count < limit) {
170 encode_frame(&codec, raw, frame_count, 1, 0, writer);
174 while (encode_frame(&codec, NULL, -1, 1, 0, writer)) {
181 aom_video_writer_close(writer);
183 printf(
"Pass 1 complete. Processed %d frames.\n", frame_count);
186 int main(
int argc,
char **argv) {
196 const int bitrate = 200;
197 const char *
const codec_arg = argv[1];
198 const char *
const width_arg = argv[2];
199 const char *
const height_arg = argv[3];
200 const char *
const infile_arg = argv[4];
201 const char *
const outfile_arg = argv[5];
205 if (argc < 6) die(
"Invalid number of arguments");
207 if (argc > 6) limit = (int)strtol(argv[6], NULL, 0);
209 if (limit == 0) limit = 100;
212 if (!encoder) die(
"Unsupported codec.");
214 w = (int)strtol(width_arg, NULL, 0);
215 h = (int)strtol(height_arg, NULL, 0);
217 if (w <= 0 || h <= 0 || (w % 2) != 0 || (h % 2) != 0)
218 die(
"Invalid frame size: %dx%d", w, h);
221 die(
"Failed to allocate image", w, h);
227 if (res) die_codec(&codec,
"Failed to get default codec config.");
235 if (!(infile = fopen(infile_arg,
"rb")))
236 die(
"Failed to open %s for reading", infile_arg);
240 stats = pass0(&raw, infile, encoder, &cfg, limit);
246 pass1(&raw, infile, outfile_arg, encoder, &cfg, limit);
void * buf
Definition: aom_encoder.h:77
Operation completed without error.
Definition: aom_codec.h:157
Definition: aom_encoder.h:161
unsigned int g_w
Width of the frame.
Definition: aom_encoder.h:406
unsigned int rc_target_bitrate
Target data rate.
Definition: aom_encoder.h:618
int64_t aom_codec_pts_t
Time Stamp Type.
Definition: aom_codec.h:235
Describes the encoder algorithm interface to applications.
#define aom_codec_enc_init(ctx, iface, cfg, flags)
Convenience macro for aom_codec_enc_init_ver()
Definition: aom_encoder.h:931
Encoder configuration structure.
Definition: aom_encoder.h:367
aom_fixed_buf_t rc_twopass_stats_in
Two-pass stats buffer.
Definition: aom_encoder.h:605
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.
Provides definitions for using AOM or AV1 encoder algorithm within the aom Codec Interface.
Codec context structure.
Definition: aom_codec.h:298
Image Descriptor.
Definition: aom_image.h:171
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.
Generic fixed size buffer structure.
Definition: aom_encoder.h:76
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.
const struct aom_codec_iface aom_codec_iface_t
Codec interface structure.
Definition: aom_codec.h:254
const char * aom_codec_iface_name(aom_codec_iface_t *iface)
Return the name for a given interface.
struct aom_rational g_timebase
Stream timebase units.
Definition: aom_encoder.h:464
aom_codec_err_t aom_codec_destroy(aom_codec_ctx_t *ctx)
Destroy a codec instance.
Definition: aom_encoder.h:99
enum aom_codec_cx_pkt_kind kind
Definition: aom_encoder.h:111
void aom_img_free(aom_image_t *img)
Close an image descriptor.
struct aom_codec_cx_pkt::@1::@2 frame
Definition: aom_encoder.h:98
Codec control function to set encoder internal speed settings, int parameter.
Definition: aomcx.h:213
const void * aom_codec_iter_t
Iterator.
Definition: aom_codec.h:288
#define AOM_FRAME_IS_KEY
Definition: aom_codec.h:271
aom_codec_err_t
Algorithm return codes.
Definition: aom_codec.h:155
aom_fixed_buf_t twopass_stats
Definition: aom_encoder.h:128
int den
Definition: aom_encoder.h:154
Encoder output packet.
Definition: aom_encoder.h:110
int num
Definition: aom_encoder.h:153
Definition: aom_image.h:45
long aom_enc_frame_flags_t
Encoded Frame Flags.
Definition: aom_encoder.h:357
union aom_codec_cx_pkt::@1 data
size_t sz
Definition: aom_encoder.h:78
Definition: aom_encoder.h:160
enum aom_enc_pass g_pass
Multi-pass Encoding Mode.
Definition: aom_encoder.h:479
aom_codec_err_t aom_codec_control(aom_codec_ctx_t *ctx, int ctrl_id,...)
Algorithm Control.
unsigned int g_h
Height of the frame.
Definition: aom_encoder.h:415