summaryrefslogtreecommitdiffstats
path: root/third_party/aom/test/yuv_video_source.h
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/aom/test/yuv_video_source.h')
-rw-r--r--third_party/aom/test/yuv_video_source.h123
1 files changed, 0 insertions, 123 deletions
diff --git a/third_party/aom/test/yuv_video_source.h b/third_party/aom/test/yuv_video_source.h
deleted file mode 100644
index 774ecc008..000000000
--- a/third_party/aom/test/yuv_video_source.h
+++ /dev/null
@@ -1,123 +0,0 @@
-/*
- * Copyright (c) 2016, Alliance for Open Media. All rights reserved
- *
- * This source code is subject to the terms of the BSD 2 Clause License and
- * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
- * was not distributed with this source code in the LICENSE file, you can
- * obtain it at www.aomedia.org/license/software. If the Alliance for Open
- * Media Patent License 1.0 was not distributed with this source code in the
- * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
- */
-#ifndef AOM_TEST_YUV_VIDEO_SOURCE_H_
-#define AOM_TEST_YUV_VIDEO_SOURCE_H_
-
-#include <cstdio>
-#include <cstdlib>
-#include <string>
-
-#include "test/video_source.h"
-#include "aom/aom_image.h"
-
-namespace libaom_test {
-
-// This class extends VideoSource to allow parsing of raw YUV
-// formats of various color sampling and bit-depths so that we can
-// do actual file encodes.
-class YUVVideoSource : public VideoSource {
- public:
- YUVVideoSource(const std::string &file_name, aom_img_fmt format,
- unsigned int width, unsigned int height, int rate_numerator,
- int rate_denominator, unsigned int start, int limit)
- : file_name_(file_name), input_file_(NULL), img_(NULL), start_(start),
- limit_(limit), frame_(0), width_(0), height_(0),
- format_(AOM_IMG_FMT_NONE), framerate_numerator_(rate_numerator),
- framerate_denominator_(rate_denominator) {
- // This initializes format_, raw_size_, width_, height_ and allocates img.
- SetSize(width, height, format);
- }
-
- virtual ~YUVVideoSource() {
- aom_img_free(img_);
- if (input_file_) fclose(input_file_);
- }
-
- virtual void Begin() {
- if (input_file_) fclose(input_file_);
- input_file_ = OpenTestDataFile(file_name_);
- ASSERT_TRUE(input_file_ != NULL)
- << "Input file open failed. Filename: " << file_name_;
- if (start_)
- fseek(input_file_, static_cast<unsigned>(raw_size_) * start_, SEEK_SET);
-
- frame_ = start_;
- FillFrame();
- }
-
- virtual void Next() {
- ++frame_;
- FillFrame();
- }
-
- virtual aom_image_t *img() const { return (frame_ < limit_) ? img_ : NULL; }
-
- // Models a stream where Timebase = 1/FPS, so pts == frame.
- virtual aom_codec_pts_t pts() const { return frame_; }
-
- virtual unsigned long duration() const { return 1; }
-
- virtual aom_rational_t timebase() const {
- const aom_rational_t t = { framerate_denominator_, framerate_numerator_ };
- return t;
- }
-
- virtual unsigned int frame() const { return frame_; }
-
- virtual unsigned int limit() const { return limit_; }
-
- virtual void SetSize(unsigned int width, unsigned int height,
- aom_img_fmt format) {
- if (width != width_ || height != height_ || format != format_) {
- aom_img_free(img_);
- img_ = aom_img_alloc(NULL, format, width, height, 1);
- ASSERT_TRUE(img_ != NULL);
- width_ = width;
- height_ = height;
- format_ = format;
- switch (format) {
- case AOM_IMG_FMT_I420: raw_size_ = width * height * 3 / 2; break;
- case AOM_IMG_FMT_I422: raw_size_ = width * height * 2; break;
- case AOM_IMG_FMT_I444: raw_size_ = width * height * 3; break;
- case AOM_IMG_FMT_I42016: raw_size_ = width * height * 3; break;
- case AOM_IMG_FMT_I42216: raw_size_ = width * height * 4; break;
- case AOM_IMG_FMT_I44416: raw_size_ = width * height * 6; break;
- default: ASSERT_TRUE(0);
- }
- }
- }
-
- virtual void FillFrame() {
- ASSERT_TRUE(input_file_ != NULL);
- // Read a frame from input_file.
- if (fread(img_->img_data, raw_size_, 1, input_file_) == 0) {
- limit_ = frame_;
- }
- }
-
- protected:
- std::string file_name_;
- FILE *input_file_;
- aom_image_t *img_;
- size_t raw_size_;
- unsigned int start_;
- unsigned int limit_;
- unsigned int frame_;
- unsigned int width_;
- unsigned int height_;
- aom_img_fmt format_;
- int framerate_numerator_;
- int framerate_denominator_;
-};
-
-} // namespace libaom_test
-
-#endif // AOM_TEST_YUV_VIDEO_SOURCE_H_