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 GET_PARAM(k) std::tr1::get< k >(GetParam()) | 20 #define GET_PARAM(k) std::tr1::get< k >(GetParam()) |
21 | 21 |
22 static double compute_psnr(const vpx_image_t *img1, | 22 inline double compute_psnr(const vpx_image_t *img1, const vpx_image_t *img2) { |
23 const vpx_image_t *img2) { | |
24 assert((img1->fmt == img2->fmt) && | 23 assert((img1->fmt == img2->fmt) && |
25 (img1->d_w == img2->d_w) && | 24 (img1->d_w == img2->d_w) && |
26 (img1->d_h == img2->d_h)); | 25 (img1->d_h == img2->d_h)); |
27 | 26 |
28 const unsigned int width_y = img1->d_w; | 27 const unsigned int width_y = img1->d_w; |
29 const unsigned int height_y = img1->d_h; | 28 const unsigned int height_y = img1->d_h; |
30 unsigned int i, j; | 29 unsigned int i, j; |
31 | 30 |
32 int64_t sqrerr = 0; | 31 int64_t sqrerr = 0; |
33 for (i = 0; i < height_y; ++i) | 32 for (i = 0; i < height_y; ++i) |
34 for (j = 0; j < width_y; ++j) { | 33 for (j = 0; j < width_y; ++j) { |
35 int64_t d = img1->planes[VPX_PLANE_Y][i * img1->stride[VPX_PLANE_Y] + j] - | 34 int64_t d = img1->planes[VPX_PLANE_Y][i * img1->stride[VPX_PLANE_Y] + j] - |
36 img2->planes[VPX_PLANE_Y][i * img2->stride[VPX_PLANE_Y] + j]; | 35 img2->planes[VPX_PLANE_Y][i * img2->stride[VPX_PLANE_Y] + j]; |
37 sqrerr += d * d; | 36 sqrerr += d * d; |
38 } | 37 } |
39 double mse = static_cast<double>(sqrerr) / (width_y * height_y); | 38 double mse = static_cast<double>(sqrerr) / (width_y * height_y); |
40 double psnr = 100.0; | 39 double psnr = 100.0; |
41 if (mse > 0.0) { | 40 if (mse > 0.0) { |
42 psnr = 10 * log10(255.0 * 255.0 / mse); | 41 psnr = 10 * log10(255.0 * 255.0 / mse); |
43 } | 42 } |
44 return psnr; | 43 return psnr; |
45 } | 44 } |
46 | 45 |
47 #endif // TEST_UTIL_H_ | 46 #endif // TEST_UTIL_H_ |
OLD | NEW |