blob: f5327c9284e4d65bafc3b3c0ca11f8e7b733007a [file] [log] [blame]
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -08001/*
Yaowu Xu9c01aa12016-09-01 14:32:49 -07002 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -08003 *
Yaowu Xu9c01aa12016-09-01 14:32:49 -07004 * 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.
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080010 */
Tom Finegandd3e2a52018-05-23 14:33:09 -070011#include "common/video_reader.h"
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080012
13#include <stdlib.h>
14#include <string.h>
15
Yaowu Xuc27fc142016-08-22 16:08:15 -070016#include "aom_ports/mem_ops.h"
Tom Finegandd3e2a52018-05-23 14:33:09 -070017#include "common/ivfdec.h"
Dmitry Kovalev4334c072014-02-26 16:32:09 -080018
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080019static const char *const kIVFSignature = "DKIF";
20
Yaowu Xuf883b422016-08-30 14:01:10 -070021struct AvxVideoReaderStruct {
22 AvxVideoInfo info;
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080023 FILE *file;
24 uint8_t *buffer;
25 size_t buffer_size;
26 size_t frame_size;
Yunqing Wangdb7f0c52018-06-01 16:39:59 -070027 aom_codec_pts_t pts;
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080028};
29
Yaowu Xuf883b422016-08-30 14:01:10 -070030AvxVideoReader *aom_video_reader_open(const char *filename) {
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080031 char header[32];
Yaowu Xuf883b422016-08-30 14:01:10 -070032 AvxVideoReader *reader = NULL;
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080033 FILE *const file = fopen(filename, "rb");
clang-format6c4d83e2016-08-08 19:03:30 -070034 if (!file) return NULL; // Can't open file
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080035
clang-format6c4d83e2016-08-08 19:03:30 -070036 if (fread(header, 1, 32, file) != 32) return NULL; // Can't read file header
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080037
38 if (memcmp(kIVFSignature, header, 4) != 0)
39 return NULL; // Wrong IVF signature
40
clang-format6c4d83e2016-08-08 19:03:30 -070041 if (mem_get_le16(header + 4) != 0) return NULL; // Wrong IVF version
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080042
Nathan E. Egge48f2fe12016-10-11 14:20:20 -040043 reader = (AvxVideoReader *)calloc(1, sizeof(*reader));
Yaowu Xuf883b422016-08-30 14:01:10 -070044 if (!reader) return NULL; // Can't allocate AvxVideoReader
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080045
46 reader->file = file;
47 reader->info.codec_fourcc = mem_get_le32(header + 8);
48 reader->info.frame_width = mem_get_le16(header + 12);
49 reader->info.frame_height = mem_get_le16(header + 14);
50 reader->info.time_base.numerator = mem_get_le32(header + 16);
51 reader->info.time_base.denominator = mem_get_le32(header + 20);
52
53 return reader;
54}
55
Yaowu Xuf883b422016-08-30 14:01:10 -070056void aom_video_reader_close(AvxVideoReader *reader) {
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080057 if (reader) {
58 fclose(reader->file);
59 free(reader->buffer);
60 free(reader);
61 }
62}
63
Yaowu Xuf883b422016-08-30 14:01:10 -070064int aom_video_reader_read_frame(AvxVideoReader *reader) {
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080065 return !ivf_read_frame(reader->file, &reader->buffer, &reader->frame_size,
Yunqing Wangdb7f0c52018-06-01 16:39:59 -070066 &reader->buffer_size, &reader->pts);
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080067}
68
Yaowu Xuf883b422016-08-30 14:01:10 -070069const uint8_t *aom_video_reader_get_frame(AvxVideoReader *reader,
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080070 size_t *size) {
clang-format6c4d83e2016-08-08 19:03:30 -070071 if (size) *size = reader->frame_size;
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080072
73 return reader->buffer;
74}
75
Yunqing Wangdb7f0c52018-06-01 16:39:59 -070076int64_t aom_video_reader_get_frame_pts(AvxVideoReader *reader) {
77 return (int64_t)reader->pts;
78}
79
80FILE *aom_video_reader_get_file(AvxVideoReader *reader) { return reader->file; }
81
Yaowu Xuf883b422016-08-30 14:01:10 -070082const AvxVideoInfo *aom_video_reader_get_info(AvxVideoReader *reader) {
Dmitry Kovalev37e6fd32014-02-05 18:34:46 -080083 return &reader->info;
84}