1 /* 2 * Copyright (c) 2016, Alliance for Open Media. All rights reserved 3 * 4 * This source code is subject to the terms of the BSD 2 Clause License and 5 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License 6 * was not distributed with this source code in the LICENSE file, you can 7 * obtain it at www.aomedia.org/license/software. If the Alliance for Open 8 * Media Patent License 1.0 was not distributed with this source code in the 9 * PATENTS file, you can obtain it at www.aomedia.org/license/patent. 10 */ 11 12 #ifndef AOM_TEST_UTIL_H_ 13 #define AOM_TEST_UTIL_H_ 14 15 #include <stdio.h> 16 #include <math.h> 17 #include "third_party/googletest/src/googletest/include/gtest/gtest.h" 18 #include "aom/aom_integer.h" 19 #include "aom/aom_image.h" 20 #include "aom_ports/aom_timer.h" 21 22 // Macros 23 #define GET_PARAM(k) ::testing::get<k>(GetParam()) 24 compute_psnr(const aom_image_t * img1,const aom_image_t * img2)25inline double compute_psnr(const aom_image_t *img1, const aom_image_t *img2) { 26 assert((img1->fmt == img2->fmt) && (img1->d_w == img2->d_w) && 27 (img1->d_h == img2->d_h)); 28 29 const unsigned int width_y = img1->d_w; 30 const unsigned int height_y = img1->d_h; 31 unsigned int i, j; 32 33 int64_t sqrerr = 0; 34 for (i = 0; i < height_y; ++i) 35 for (j = 0; j < width_y; ++j) { 36 int64_t d = img1->planes[AOM_PLANE_Y][i * img1->stride[AOM_PLANE_Y] + j] - 37 img2->planes[AOM_PLANE_Y][i * img2->stride[AOM_PLANE_Y] + j]; 38 sqrerr += d * d; 39 } 40 double mse = static_cast<double>(sqrerr) / (width_y * height_y); 41 double psnr = 100.0; 42 if (mse > 0.0) { 43 psnr = 10 * log10(255.0 * 255.0 / mse); 44 } 45 return psnr; 46 } 47 get_time_mark(aom_usec_timer * t)48static INLINE double get_time_mark(aom_usec_timer *t) { 49 aom_usec_timer_mark(t); 50 return static_cast<double>(aom_usec_timer_elapsed(t)); 51 } 52 53 #endif // AOM_TEST_UTIL_H_ 54