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_Y4M_VIDEO_SOURCE_H_ 11 #define TEST_Y4M_VIDEO_SOURCE_H_ 12 #include <algorithm> 13 #include <string> 14 15 #include "test/video_source.h" 16 #include "./y4minput.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 Y4mVideoSource : public VideoSource { 23 public: Y4mVideoSource(const std::string & file_name,unsigned int start,int limit)24 Y4mVideoSource(const std::string &file_name, unsigned int start, int limit) 25 : file_name_(file_name), input_file_(NULL), img_(new vpx_image_t()), 26 start_(start), limit_(limit), frame_(0), framerate_numerator_(0), 27 framerate_denominator_(0), y4m_() {} 28 ~Y4mVideoSource()29 virtual ~Y4mVideoSource() { 30 vpx_img_free(img_.get()); 31 CloseSource(); 32 } 33 OpenSource()34 virtual void OpenSource() { 35 CloseSource(); 36 input_file_ = OpenTestDataFile(file_name_); 37 ASSERT_TRUE(input_file_ != NULL) << "Input file open failed. Filename: " 38 << file_name_; 39 } 40 ReadSourceToStart()41 virtual void ReadSourceToStart() { 42 ASSERT_TRUE(input_file_ != NULL); 43 ASSERT_FALSE(y4m_input_open(&y4m_, input_file_, NULL, 0, 0)); 44 framerate_numerator_ = y4m_.fps_n; 45 framerate_denominator_ = y4m_.fps_d; 46 frame_ = 0; 47 for (unsigned int i = 0; i < start_; i++) { 48 Next(); 49 } 50 FillFrame(); 51 } 52 Begin()53 virtual void Begin() { 54 OpenSource(); 55 ReadSourceToStart(); 56 } 57 Next()58 virtual void Next() { 59 ++frame_; 60 FillFrame(); 61 } 62 img()63 virtual vpx_image_t *img() const { 64 return (frame_ < limit_) ? img_.get() : NULL; 65 } 66 67 // Models a stream where Timebase = 1/FPS, so pts == frame. pts()68 virtual vpx_codec_pts_t pts() const { return frame_; } 69 duration()70 virtual unsigned long duration() const { return 1; } 71 timebase()72 virtual vpx_rational_t timebase() const { 73 const vpx_rational_t t = { framerate_denominator_, framerate_numerator_ }; 74 return t; 75 } 76 frame()77 virtual unsigned int frame() const { return frame_; } 78 limit()79 virtual unsigned int limit() const { return limit_; } 80 FillFrame()81 virtual void FillFrame() { 82 ASSERT_TRUE(input_file_ != NULL); 83 // Read a frame from input_file. 84 y4m_input_fetch_frame(&y4m_, input_file_, img_.get()); 85 } 86 87 // Swap buffers with another y4m source. This allows reading a new frame 88 // while keeping the old frame around. A whole Y4mSource is required and 89 // not just a vpx_image_t because of how the y4m reader manipulates 90 // vpx_image_t internals, SwapBuffers(Y4mVideoSource * other)91 void SwapBuffers(Y4mVideoSource *other) { 92 std::swap(other->y4m_.dst_buf, y4m_.dst_buf); 93 vpx_image_t *tmp; 94 tmp = other->img_.release(); 95 other->img_.reset(img_.release()); 96 img_.reset(tmp); 97 } 98 99 protected: CloseSource()100 void CloseSource() { 101 y4m_input_close(&y4m_); 102 y4m_ = y4m_input(); 103 if (input_file_ != NULL) { 104 fclose(input_file_); 105 input_file_ = NULL; 106 } 107 } 108 109 std::string file_name_; 110 FILE *input_file_; 111 testing::internal::scoped_ptr<vpx_image_t> img_; 112 unsigned int start_; 113 unsigned int limit_; 114 unsigned int frame_; 115 int framerate_numerator_; 116 int framerate_denominator_; 117 y4m_input y4m_; 118 }; 119 120 } // namespace libvpx_test 121 122 #endif // TEST_Y4M_VIDEO_SOURCE_H_ 123