util.h (1867B)
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 <math.h> 16 #include <stdio.h> 17 #include <string.h> 18 #include "gtest/gtest.h" 19 #include "aom/aom_image.h" 20 #include "aom_ports/aom_timer.h" 21 22 // Macros 23 #define GET_PARAM(k) std::get<k>(GetParam()) 24 25 inline int is_extension_y4m(const char *filename) { 26 const char *dot = strrchr(filename, '.'); 27 if (!dot || dot == filename) return 0; 28 29 return !strcmp(dot, ".y4m"); 30 } 31 32 inline double compute_psnr(const aom_image_t *img1, const aom_image_t *img2) { 33 assert((img1->fmt == img2->fmt) && (img1->d_w == img2->d_w) && 34 (img1->d_h == img2->d_h)); 35 36 const unsigned int width_y = img1->d_w; 37 const unsigned int height_y = img1->d_h; 38 unsigned int i, j; 39 40 int64_t sqrerr = 0; 41 for (i = 0; i < height_y; ++i) 42 for (j = 0; j < width_y; ++j) { 43 int64_t d = img1->planes[AOM_PLANE_Y][i * img1->stride[AOM_PLANE_Y] + j] - 44 img2->planes[AOM_PLANE_Y][i * img2->stride[AOM_PLANE_Y] + j]; 45 sqrerr += d * d; 46 } 47 double mse = static_cast<double>(sqrerr) / (width_y * height_y); 48 double psnr = 100.0; 49 if (mse > 0.0) { 50 psnr = 10 * log10(255.0 * 255.0 / mse); 51 } 52 return psnr; 53 } 54 55 static inline double get_time_mark(aom_usec_timer *t) { 56 aom_usec_timer_mark(t); 57 return static_cast<double>(aom_usec_timer_elapsed(t)); 58 } 59 60 #endif // AOM_TEST_UTIL_H_