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