1233d2500723e5594f3e7c70896ffeeef32b9c950ywan/* 2233d2500723e5594f3e7c70896ffeeef32b9c950ywan * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3233d2500723e5594f3e7c70896ffeeef32b9c950ywan * 4233d2500723e5594f3e7c70896ffeeef32b9c950ywan * Use of this source code is governed by a BSD-style license 5233d2500723e5594f3e7c70896ffeeef32b9c950ywan * that can be found in the LICENSE file in the root of the source 6233d2500723e5594f3e7c70896ffeeef32b9c950ywan * tree. An additional intellectual property rights grant can be found 7233d2500723e5594f3e7c70896ffeeef32b9c950ywan * in the file PATENTS. All contributing project authors may 8233d2500723e5594f3e7c70896ffeeef32b9c950ywan * be found in the AUTHORS file in the root of the source tree. 9233d2500723e5594f3e7c70896ffeeef32b9c950ywan */ 10233d2500723e5594f3e7c70896ffeeef32b9c950ywan 11233d2500723e5594f3e7c70896ffeeef32b9c950ywan#ifndef TEST_UTIL_H_ 12233d2500723e5594f3e7c70896ffeeef32b9c950ywan#define TEST_UTIL_H_ 13233d2500723e5594f3e7c70896ffeeef32b9c950ywan 14233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <stdio.h> 15233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include <math.h> 16233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "third_party/googletest/src/include/gtest/gtest.h" 17233d2500723e5594f3e7c70896ffeeef32b9c950ywan#include "vpx/vpx_image.h" 18233d2500723e5594f3e7c70896ffeeef32b9c950ywan 19233d2500723e5594f3e7c70896ffeeef32b9c950ywan// Macros 20233d2500723e5594f3e7c70896ffeeef32b9c950ywan#define GET_PARAM(k) std::tr1::get< k >(GetParam()) 21233d2500723e5594f3e7c70896ffeeef32b9c950ywan 22233d2500723e5594f3e7c70896ffeeef32b9c950ywanstatic double compute_psnr(const vpx_image_t *img1, 23233d2500723e5594f3e7c70896ffeeef32b9c950ywan const vpx_image_t *img2) { 24233d2500723e5594f3e7c70896ffeeef32b9c950ywan assert((img1->fmt == img2->fmt) && 25233d2500723e5594f3e7c70896ffeeef32b9c950ywan (img1->d_w == img2->d_w) && 26233d2500723e5594f3e7c70896ffeeef32b9c950ywan (img1->d_h == img2->d_h)); 27233d2500723e5594f3e7c70896ffeeef32b9c950ywan 28233d2500723e5594f3e7c70896ffeeef32b9c950ywan const unsigned int width_y = img1->d_w; 29233d2500723e5594f3e7c70896ffeeef32b9c950ywan const unsigned int height_y = img1->d_h; 30233d2500723e5594f3e7c70896ffeeef32b9c950ywan unsigned int i, j; 31233d2500723e5594f3e7c70896ffeeef32b9c950ywan 32233d2500723e5594f3e7c70896ffeeef32b9c950ywan int64_t sqrerr = 0; 33233d2500723e5594f3e7c70896ffeeef32b9c950ywan for (i = 0; i < height_y; ++i) 34233d2500723e5594f3e7c70896ffeeef32b9c950ywan for (j = 0; j < width_y; ++j) { 35233d2500723e5594f3e7c70896ffeeef32b9c950ywan int64_t d = img1->planes[VPX_PLANE_Y][i * img1->stride[VPX_PLANE_Y] + j] - 36233d2500723e5594f3e7c70896ffeeef32b9c950ywan img2->planes[VPX_PLANE_Y][i * img2->stride[VPX_PLANE_Y] + j]; 37233d2500723e5594f3e7c70896ffeeef32b9c950ywan sqrerr += d * d; 38233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 39233d2500723e5594f3e7c70896ffeeef32b9c950ywan double mse = static_cast<double>(sqrerr) / (width_y * height_y); 40233d2500723e5594f3e7c70896ffeeef32b9c950ywan double psnr = 100.0; 41233d2500723e5594f3e7c70896ffeeef32b9c950ywan if (mse > 0.0) { 42233d2500723e5594f3e7c70896ffeeef32b9c950ywan psnr = 10 * log10(255.0 * 255.0 / mse); 43233d2500723e5594f3e7c70896ffeeef32b9c950ywan } 44233d2500723e5594f3e7c70896ffeeef32b9c950ywan return psnr; 45233d2500723e5594f3e7c70896ffeeef32b9c950ywan} 46233d2500723e5594f3e7c70896ffeeef32b9c950ywan 47233d2500723e5594f3e7c70896ffeeef32b9c950ywan#endif // TEST_UTIL_H_ 48