1 /* 2 * Copyright (c) 2012 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 #ifndef TEST_I420_VIDEO_SOURCE_H_ 11 #define TEST_I420_VIDEO_SOURCE_H_ 12 #include <cstdio> 13 #include <cstdlib> 14 #include <string> 15 16 #include "test/video_source.h" 17 18 namespace libvpx_test { 19 20 // This class extends VideoSource to allow parsing of raw yv12 21 // so that we can do actual file encodes. 22 class I420VideoSource : public VideoSource { 23 public: I420VideoSource(const std::string & file_name,unsigned int width,unsigned int height,int rate_numerator,int rate_denominator,unsigned int start,int limit)24 I420VideoSource(const std::string &file_name, 25 unsigned int width, unsigned int height, 26 int rate_numerator, int rate_denominator, 27 unsigned int start, int limit) 28 : file_name_(file_name), 29 input_file_(NULL), 30 img_(NULL), 31 start_(start), 32 limit_(limit), 33 frame_(0), 34 width_(0), 35 height_(0), 36 framerate_numerator_(rate_numerator), 37 framerate_denominator_(rate_denominator) { 38 // This initializes raw_sz_, width_, height_ and allocates an img. 39 SetSize(width, height); 40 } 41 ~I420VideoSource()42 virtual ~I420VideoSource() { 43 vpx_img_free(img_); 44 if (input_file_) 45 fclose(input_file_); 46 } 47 Begin()48 virtual void Begin() { 49 if (input_file_) 50 fclose(input_file_); 51 input_file_ = OpenTestDataFile(file_name_); 52 ASSERT_TRUE(input_file_ != NULL) << "Input file open failed. Filename: " 53 << file_name_; 54 if (start_) { 55 fseek(input_file_, static_cast<unsigned>(raw_sz_) * start_, SEEK_SET); 56 } 57 58 frame_ = start_; 59 FillFrame(); 60 } 61 Next()62 virtual void Next() { 63 ++frame_; 64 FillFrame(); 65 } 66 img()67 virtual vpx_image_t *img() const { return (frame_ < limit_) ? img_ : NULL; } 68 69 // Models a stream where Timebase = 1/FPS, so pts == frame. pts()70 virtual vpx_codec_pts_t pts() const { return frame_; } 71 duration()72 virtual unsigned long duration() const { return 1; } 73 timebase()74 virtual vpx_rational_t timebase() const { 75 const vpx_rational_t t = { framerate_denominator_, framerate_numerator_ }; 76 return t; 77 } 78 frame()79 virtual unsigned int frame() const { return frame_; } 80 limit()81 virtual unsigned int limit() const { return limit_; } 82 SetSize(unsigned int width,unsigned int height)83 void SetSize(unsigned int width, unsigned int height) { 84 if (width != width_ || height != height_) { 85 vpx_img_free(img_); 86 img_ = vpx_img_alloc(NULL, VPX_IMG_FMT_I420, width, height, 1); 87 ASSERT_TRUE(img_ != NULL); 88 width_ = width; 89 height_ = height; 90 raw_sz_ = width * height * 3 / 2; 91 } 92 } 93 FillFrame()94 virtual void FillFrame() { 95 ASSERT_TRUE(input_file_ != NULL); 96 // Read a frame from input_file. 97 if (fread(img_->img_data, raw_sz_, 1, input_file_) == 0) { 98 limit_ = frame_; 99 } 100 } 101 102 protected: 103 std::string file_name_; 104 FILE *input_file_; 105 vpx_image_t *img_; 106 size_t raw_sz_; 107 unsigned int start_; 108 unsigned int limit_; 109 unsigned int frame_; 110 unsigned int width_; 111 unsigned int height_; 112 int framerate_numerator_; 113 int framerate_denominator_; 114 }; 115 116 } // namespace libvpx_test 117 118 #endif // TEST_I420_VIDEO_SOURCE_H_ 119