1 /*
2 * Copyright (c) 2016, 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 #include "common/ivfenc.h"
13
14 #include "aom/aom_encoder.h"
15 #include "aom_ports/mem_ops.h"
16
ivf_write_file_header(FILE * outfile,const struct aom_codec_enc_cfg * cfg,unsigned int fourcc,int frame_cnt)17 void ivf_write_file_header(FILE *outfile, const struct aom_codec_enc_cfg *cfg,
18 unsigned int fourcc, int frame_cnt) {
19 char header[32];
20
21 header[0] = 'D';
22 header[1] = 'K';
23 header[2] = 'I';
24 header[3] = 'F';
25 mem_put_le16(header + 4, 0); // version
26 mem_put_le16(header + 6, 32); // header size
27 mem_put_le32(header + 8, fourcc); // fourcc
28 mem_put_le16(header + 12, cfg->g_w); // width
29 mem_put_le16(header + 14, cfg->g_h); // height
30 mem_put_le32(header + 16, cfg->g_timebase.den); // rate
31 mem_put_le32(header + 20, cfg->g_timebase.num); // scale
32 mem_put_le32(header + 24, frame_cnt); // length
33 mem_put_le32(header + 28, 0); // unused
34
35 fwrite(header, 1, 32, outfile);
36 }
37
ivf_write_frame_header(FILE * outfile,int64_t pts,size_t frame_size)38 void ivf_write_frame_header(FILE *outfile, int64_t pts, size_t frame_size) {
39 char header[12];
40
41 mem_put_le32(header, (int)frame_size);
42 mem_put_le32(header + 4, (int)(pts & 0xFFFFFFFF));
43 mem_put_le32(header + 8, (int)(pts >> 32));
44 fwrite(header, 1, 12, outfile);
45 }
46
ivf_write_frame_size(FILE * outfile,size_t frame_size)47 void ivf_write_frame_size(FILE *outfile, size_t frame_size) {
48 char header[4];
49
50 mem_put_le32(header, (int)frame_size);
51 fwrite(header, 1, 4, outfile);
52 }
53