ref: 2f52ae2384846b1ccd2c18db75bbb0867c1c9f5e
dir: /video_writer.c/
/* * 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" #include "vpx/vpx_encoder.h" struct VpxVideoWriterStruct { VpxVideoInfo info; FILE *file; int frame_count; }; static void write_header(FILE *file, const VpxVideoInfo *info, int frame_count) { struct vpx_codec_enc_cfg cfg; 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); } VpxVideoWriter *vpx_video_writer_open(const char *filename, VpxContainer container, const VpxVideoInfo *info) { if (container == kContainerIVF) { VpxVideoWriter *writer = NULL; FILE *const file = fopen(filename, "wb"); if (!file) { fprintf(stderr, "%s can't be written to.\n", filename); return NULL; } writer = malloc(sizeof(*writer)); if (!writer) { fprintf(stderr, "Can't allocate VpxVideoWriter.\n"); return NULL; } writer->frame_count = 0; writer->info = *info; writer->file = file; write_header(writer->file, info, 0); return writer; } fprintf(stderr, "VpxVideoWriter supports only IVF.\n"); return NULL; } void vpx_video_writer_close(VpxVideoWriter *writer) { 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); } } int vpx_video_writer_write_frame(VpxVideoWriter *writer, const uint8_t *buffer, size_t size, int64_t pts) { ivf_write_frame_header(writer->file, pts, size); if (fwrite(buffer, 1, size, writer->file) != size) return 0; ++writer->frame_count; return 1; }