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 11 #include <string> 12 13 #include "third_party/googletest/src/include/gtest/gtest.h" 14 15 #include "./vpx_config.h" 16 #include "./y4menc.h" 17 #include "test/md5_helper.h" 18 #include "test/util.h" 19 #include "test/y4m_video_source.h" 20 21 namespace { 22 23 using std::string; 24 25 static const unsigned int kWidth = 160; 26 static const unsigned int kHeight = 90; 27 static const unsigned int kFrames = 10; 28 29 struct Y4mTestParam { 30 const char *filename; 31 unsigned int bit_depth; 32 vpx_img_fmt format; 33 const char *md5raw; 34 }; 35 36 const Y4mTestParam kY4mTestVectors[] = { 37 { "park_joy_90p_8_420.y4m", 8, VPX_IMG_FMT_I420, 38 "e5406275b9fc6bb3436c31d4a05c1cab" }, 39 { "park_joy_90p_8_422.y4m", 8, VPX_IMG_FMT_I422, 40 "284a47a47133b12884ec3a14e959a0b6" }, 41 { "park_joy_90p_8_444.y4m", 8, VPX_IMG_FMT_I444, 42 "90517ff33843d85de712fd4fe60dbed0" }, 43 { "park_joy_90p_10_420.y4m", 10, VPX_IMG_FMT_I42016, 44 "63f21f9f717d8b8631bd2288ee87137b" }, 45 { "park_joy_90p_10_422.y4m", 10, VPX_IMG_FMT_I42216, 46 "48ab51fb540aed07f7ff5af130c9b605" }, 47 { "park_joy_90p_10_444.y4m", 10, VPX_IMG_FMT_I44416, 48 "067bfd75aa85ff9bae91fa3e0edd1e3e" }, 49 { "park_joy_90p_12_420.y4m", 12, VPX_IMG_FMT_I42016, 50 "9e6d8f6508c6e55625f6b697bc461cef" }, 51 { "park_joy_90p_12_422.y4m", 12, VPX_IMG_FMT_I42216, 52 "b239c6b301c0b835485be349ca83a7e3" }, 53 { "park_joy_90p_12_444.y4m", 12, VPX_IMG_FMT_I44416, 54 "5a6481a550821dab6d0192f5c63845e9" }, 55 }; 56 57 static void write_image_file(const vpx_image_t *img, FILE *file) { 58 int plane, y; 59 for (plane = 0; plane < 3; ++plane) { 60 const unsigned char *buf = img->planes[plane]; 61 const int stride = img->stride[plane]; 62 const int bytes_per_sample = (img->fmt & VPX_IMG_FMT_HIGHBITDEPTH) ? 2 : 1; 63 const int h = 64 (plane ? (img->d_h + img->y_chroma_shift) >> img->y_chroma_shift 65 : img->d_h); 66 const int w = 67 (plane ? (img->d_w + img->x_chroma_shift) >> img->x_chroma_shift 68 : img->d_w); 69 for (y = 0; y < h; ++y) { 70 fwrite(buf, bytes_per_sample, w, file); 71 buf += stride; 72 } 73 } 74 } 75 76 class Y4mVideoSourceTest : public ::testing::TestWithParam<Y4mTestParam>, 77 public ::libvpx_test::Y4mVideoSource { 78 protected: 79 Y4mVideoSourceTest() : Y4mVideoSource("", 0, 0) {} 80 81 virtual ~Y4mVideoSourceTest() { CloseSource(); } 82 83 virtual void Init(const std::string &file_name, int limit) { 84 file_name_ = file_name; 85 start_ = 0; 86 limit_ = limit; 87 frame_ = 0; 88 Begin(); 89 } 90 91 // Checks y4m header information 92 void HeaderChecks(unsigned int bit_depth, vpx_img_fmt_t fmt) { 93 ASSERT_TRUE(input_file_ != NULL); 94 ASSERT_EQ(y4m_.pic_w, (int)kWidth); 95 ASSERT_EQ(y4m_.pic_h, (int)kHeight); 96 ASSERT_EQ(img()->d_w, kWidth); 97 ASSERT_EQ(img()->d_h, kHeight); 98 ASSERT_EQ(y4m_.bit_depth, bit_depth); 99 ASSERT_EQ(y4m_.vpx_fmt, fmt); 100 if (fmt == VPX_IMG_FMT_I420 || fmt == VPX_IMG_FMT_I42016) { 101 ASSERT_EQ(y4m_.bps, (int)y4m_.bit_depth * 3 / 2); 102 ASSERT_EQ(img()->x_chroma_shift, 1U); 103 ASSERT_EQ(img()->y_chroma_shift, 1U); 104 } 105 if (fmt == VPX_IMG_FMT_I422 || fmt == VPX_IMG_FMT_I42216) { 106 ASSERT_EQ(y4m_.bps, (int)y4m_.bit_depth * 2); 107 ASSERT_EQ(img()->x_chroma_shift, 1U); 108 ASSERT_EQ(img()->y_chroma_shift, 0U); 109 } 110 if (fmt == VPX_IMG_FMT_I444 || fmt == VPX_IMG_FMT_I44416) { 111 ASSERT_EQ(y4m_.bps, (int)y4m_.bit_depth * 3); 112 ASSERT_EQ(img()->x_chroma_shift, 0U); 113 ASSERT_EQ(img()->y_chroma_shift, 0U); 114 } 115 } 116 117 // Checks MD5 of the raw frame data 118 void Md5Check(const string &expected_md5) { 119 ASSERT_TRUE(input_file_ != NULL); 120 libvpx_test::MD5 md5; 121 for (unsigned int i = start_; i < limit_; i++) { 122 md5.Add(img()); 123 Next(); 124 } 125 ASSERT_EQ(string(md5.Get()), expected_md5); 126 } 127 }; 128 129 TEST_P(Y4mVideoSourceTest, SourceTest) { 130 const Y4mTestParam t = GetParam(); 131 Init(t.filename, kFrames); 132 HeaderChecks(t.bit_depth, t.format); 133 Md5Check(t.md5raw); 134 } 135 136 INSTANTIATE_TEST_CASE_P(C, Y4mVideoSourceTest, 137 ::testing::ValuesIn(kY4mTestVectors)); 138 139 class Y4mVideoWriteTest : public Y4mVideoSourceTest { 140 protected: 141 Y4mVideoWriteTest() : tmpfile_(NULL) {} 142 143 virtual ~Y4mVideoWriteTest() { 144 delete tmpfile_; 145 input_file_ = NULL; 146 } 147 148 void ReplaceInputFile(FILE *input_file) { 149 CloseSource(); 150 frame_ = 0; 151 input_file_ = input_file; 152 rewind(input_file_); 153 ReadSourceToStart(); 154 } 155 156 // Writes out a y4m file and then reads it back 157 void WriteY4mAndReadBack() { 158 ASSERT_TRUE(input_file_ != NULL); 159 char buf[Y4M_BUFFER_SIZE] = { 0 }; 160 const struct VpxRational framerate = { y4m_.fps_n, y4m_.fps_d }; 161 tmpfile_ = new libvpx_test::TempOutFile; 162 ASSERT_TRUE(tmpfile_->file() != NULL); 163 y4m_write_file_header(buf, sizeof(buf), kWidth, kHeight, &framerate, 164 y4m_.vpx_fmt, y4m_.bit_depth); 165 fputs(buf, tmpfile_->file()); 166 for (unsigned int i = start_; i < limit_; i++) { 167 y4m_write_frame_header(buf, sizeof(buf)); 168 fputs(buf, tmpfile_->file()); 169 write_image_file(img(), tmpfile_->file()); 170 Next(); 171 } 172 ReplaceInputFile(tmpfile_->file()); 173 } 174 175 virtual void Init(const std::string &file_name, int limit) { 176 Y4mVideoSourceTest::Init(file_name, limit); 177 WriteY4mAndReadBack(); 178 } 179 libvpx_test::TempOutFile *tmpfile_; 180 }; 181 182 TEST_P(Y4mVideoWriteTest, WriteTest) { 183 const Y4mTestParam t = GetParam(); 184 Init(t.filename, kFrames); 185 HeaderChecks(t.bit_depth, t.format); 186 Md5Check(t.md5raw); 187 } 188 189 INSTANTIATE_TEST_CASE_P(C, Y4mVideoWriteTest, 190 ::testing::ValuesIn(kY4mTestVectors)); 191 } // namespace 192