1/*
2 *  Copyright (c) 2014 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 <stdlib.h>
12
13#include "./ivfenc.h"
14#include "./video_writer.h"
15#include "vpx/vpx_encoder.h"
16
17struct VpxVideoWriterStruct {
18  VpxVideoInfo info;
19  FILE *file;
20  int frame_count;
21};
22
23static void write_header(FILE *file, const VpxVideoInfo *info,
24                         int frame_count) {
25  struct vpx_codec_enc_cfg cfg;
26  cfg.g_w = info->frame_width;
27  cfg.g_h = info->frame_height;
28  cfg.g_timebase.num = info->time_base.numerator;
29  cfg.g_timebase.den = info->time_base.denominator;
30
31  ivf_write_file_header(file, &cfg, info->codec_fourcc, frame_count);
32}
33
34VpxVideoWriter *vpx_video_writer_open(const char *filename,
35                                      VpxContainer container,
36                                      const VpxVideoInfo *info) {
37  if (container == kContainerIVF) {
38    VpxVideoWriter *writer = NULL;
39    FILE *const file = fopen(filename, "wb");
40    if (!file)
41      return NULL;
42
43    writer = malloc(sizeof(*writer));
44    if (!writer)
45      return NULL;
46
47    writer->frame_count = 0;
48    writer->info = *info;
49    writer->file = file;
50
51    write_header(writer->file, info, 0);
52
53    return writer;
54  }
55
56  return NULL;
57}
58
59void vpx_video_writer_close(VpxVideoWriter *writer) {
60  if (writer) {
61    // Rewriting frame header with real frame count
62    rewind(writer->file);
63    write_header(writer->file, &writer->info, writer->frame_count);
64
65    fclose(writer->file);
66    free(writer);
67  }
68}
69
70int vpx_video_writer_write_frame(VpxVideoWriter *writer,
71                                 const uint8_t *buffer, size_t size,
72                                 int64_t pts) {
73  ivf_write_frame_header(writer->file, pts, size);
74  if (fwrite(buffer, 1, size, writer->file) != size)
75    return 0;
76
77  ++writer->frame_count;
78
79  return 1;
80}
81