OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebM project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef TEST_UTIL_H_ | 11 #ifndef TEST_UTIL_H_ |
12 #define TEST_UTIL_H_ | 12 #define TEST_UTIL_H_ |
13 | 13 |
14 #include <stdio.h> | 14 #include <stdio.h> |
15 #include <math.h> | 15 #include <math.h> |
16 #include "third_party/googletest/src/include/gtest/gtest.h" | 16 #include "third_party/googletest/src/include/gtest/gtest.h" |
17 #include "vpx/vpx_image.h" | 17 #include "vpx/vpx_image.h" |
18 | 18 |
19 // Macros | 19 // Macros |
20 #define PARAMS(...) ::testing::TestWithParam< std::tr1::tuple< __VA_ARGS__ > > | |
21 #define GET_PARAM(k) std::tr1::get< k >(GetParam()) | 20 #define GET_PARAM(k) std::tr1::get< k >(GetParam()) |
22 | 21 |
23 static double compute_psnr(const vpx_image_t *img1, | 22 static double compute_psnr(const vpx_image_t *img1, |
24 const vpx_image_t *img2) { | 23 const vpx_image_t *img2) { |
25 assert((img1->fmt == img2->fmt) && | 24 assert((img1->fmt == img2->fmt) && |
26 (img1->d_w == img2->d_w) && | 25 (img1->d_w == img2->d_w) && |
27 (img1->d_h == img2->d_h)); | 26 (img1->d_h == img2->d_h)); |
28 | 27 |
29 const unsigned int width_y = img1->d_w; | 28 const unsigned int width_y = img1->d_w; |
30 const unsigned int height_y = img1->d_h; | 29 const unsigned int height_y = img1->d_h; |
31 unsigned int i, j; | 30 unsigned int i, j; |
32 | 31 |
33 int64_t sqrerr = 0; | 32 int64_t sqrerr = 0; |
34 for (i = 0; i < height_y; ++i) | 33 for (i = 0; i < height_y; ++i) |
35 for (j = 0; j < width_y; ++j) { | 34 for (j = 0; j < width_y; ++j) { |
36 int64_t d = img1->planes[VPX_PLANE_Y][i * img1->stride[VPX_PLANE_Y] + j] - | 35 int64_t d = img1->planes[VPX_PLANE_Y][i * img1->stride[VPX_PLANE_Y] + j] - |
37 img2->planes[VPX_PLANE_Y][i * img2->stride[VPX_PLANE_Y] + j]; | 36 img2->planes[VPX_PLANE_Y][i * img2->stride[VPX_PLANE_Y] + j]; |
38 sqrerr += d * d; | 37 sqrerr += d * d; |
39 } | 38 } |
40 double mse = static_cast<double>(sqrerr) / (width_y * height_y); | 39 double mse = static_cast<double>(sqrerr) / (width_y * height_y); |
41 double psnr = 100.0; | 40 double psnr = 100.0; |
42 if (mse > 0.0) { | 41 if (mse > 0.0) { |
43 psnr = 10 * log10(255.0 * 255.0 / mse); | 42 psnr = 10 * log10(255.0 * 255.0 / mse); |
44 } | 43 } |
45 return psnr; | 44 return psnr; |
46 } | 45 } |
47 | 46 |
48 #endif // TEST_UTIL_H_ | 47 #endif // TEST_UTIL_H_ |
OLD | NEW |