video_writer.c 2.06 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 *  Copyright (c) 2014 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include <stdlib.h>

#include "./ivfenc.h"
#include "./video_writer.h"
15
#include "aom/aom_encoder.h"
16

17 18
struct AvxVideoWriterStruct {
  AvxVideoInfo info;
19 20 21 22
  FILE *file;
  int frame_count;
};

23
static void write_header(FILE *file, const AvxVideoInfo *info,
24
                         int frame_count) {
25
  struct aom_codec_enc_cfg cfg;
26 27 28 29 30 31 32 33
  cfg.g_w = info->frame_width;
  cfg.g_h = info->frame_height;
  cfg.g_timebase.num = info->time_base.numerator;
  cfg.g_timebase.den = info->time_base.denominator;

  ivf_write_file_header(file, &cfg, info->codec_fourcc, frame_count);
}

34 35 36
AvxVideoWriter *aom_video_writer_open(const char *filename,
                                      AvxContainer container,
                                      const AvxVideoInfo *info) {
37
  if (container == kContainerIVF) {
38
    AvxVideoWriter *writer = NULL;
39
    FILE *const file = fopen(filename, "wb");
clang-format's avatar
clang-format committed
40
    if (!file) return NULL;
41 42

    writer = malloc(sizeof(*writer));
clang-format's avatar
clang-format committed
43
    if (!writer) return NULL;
44 45 46 47 48 49 50 51 52 53 54 55 56

    writer->frame_count = 0;
    writer->info = *info;
    writer->file = file;

    write_header(writer->file, info, 0);

    return writer;
  }

  return NULL;
}

57
void aom_video_writer_close(AvxVideoWriter *writer) {
58 59 60 61 62 63 64 65 66 67
  if (writer) {
    // Rewriting frame header with real frame count
    rewind(writer->file);
    write_header(writer->file, &writer->info, writer->frame_count);

    fclose(writer->file);
    free(writer);
  }
}

68
int aom_video_writer_write_frame(AvxVideoWriter *writer, const uint8_t *buffer,
clang-format's avatar
clang-format committed
69
                                 size_t size, int64_t pts) {
70
  ivf_write_frame_header(writer->file, pts, size);
clang-format's avatar
clang-format committed
71
  if (fwrite(buffer, 1, size, writer->file) != size) return 0;
72 73 74 75 76

  ++writer->frame_count;

  return 1;
}