1 /*
2 * Copyright (c) 2013 The WebM project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "./ivfenc.h"
12
13 #include "vpx/vpx_encoder.h"
14 #include "vpx_ports/mem_ops.h"
15
ivf_write_file_header_with_video_info(FILE * outfile,unsigned int fourcc,int frame_cnt,int frame_width,int frame_height,vpx_rational_t timebase)16 void ivf_write_file_header_with_video_info(FILE *outfile, unsigned int fourcc,
17 int frame_cnt, int frame_width,
18 int frame_height,
19 vpx_rational_t timebase) {
20 char header[32];
21
22 header[0] = 'D';
23 header[1] = 'K';
24 header[2] = 'I';
25 header[3] = 'F';
26 mem_put_le16(header + 4, 0); // version
27 mem_put_le16(header + 6, 32); // header size
28 mem_put_le32(header + 8, fourcc); // fourcc
29 mem_put_le16(header + 12, frame_width); // width
30 mem_put_le16(header + 14, frame_height); // height
31 mem_put_le32(header + 16, timebase.den); // rate
32 mem_put_le32(header + 20, timebase.num); // scale
33 mem_put_le32(header + 24, frame_cnt); // length
34 mem_put_le32(header + 28, 0); // unused
35
36 fwrite(header, 1, 32, outfile);
37 }
38
ivf_write_file_header(FILE * outfile,const struct vpx_codec_enc_cfg * cfg,unsigned int fourcc,int frame_cnt)39 void ivf_write_file_header(FILE *outfile, const struct vpx_codec_enc_cfg *cfg,
40 unsigned int fourcc, int frame_cnt) {
41 ivf_write_file_header_with_video_info(outfile, fourcc, frame_cnt, cfg->g_w,
42 cfg->g_h, cfg->g_timebase);
43 }
44
ivf_write_frame_header(FILE * outfile,int64_t pts,size_t frame_size)45 void ivf_write_frame_header(FILE *outfile, int64_t pts, size_t frame_size) {
46 char header[12];
47
48 mem_put_le32(header, (int)frame_size);
49 mem_put_le32(header + 4, (int)(pts & 0xFFFFFFFF));
50 mem_put_le32(header + 8, (int)(pts >> 32));
51 fwrite(header, 1, 12, outfile);
52 }
53
ivf_write_frame_size(FILE * outfile,size_t frame_size)54 void ivf_write_frame_size(FILE *outfile, size_t frame_size) {
55 char header[4];
56
57 mem_put_le32(header, (int)frame_size);
58 fwrite(header, 1, 4, outfile);
59 }
60