1
0
Fork 0
mirror of https://github.com/ultravideo/uvg266.git synced 2024-12-19 11:24:06 +00:00
uvg266/tests/satd_tests.c
Ari Koivula 09baa6bd5e Fix intra_sad and satd_tests
The tests were not being run, because the names have changed.
2015-09-18 15:26:37 +03:00

202 lines
5.7 KiB
C

/*****************************************************************************
* This file is part of Kvazaar HEVC encoder.
*
* Copyright (C) 2013-2015 Tampere University of Technology and others (see
* COPYING file).
*
* Kvazaar is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License version 2.1 as
* published by the Free Software Foundation.
*
* Kvazaar is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Kvazaar. If not, see <http://www.gnu.org/licenses/>.
****************************************************************************/
#include "greatest/greatest.h"
#include "test_strategies.h"
#include "src/image.h"
#include <math.h>
//////////////////////////////////////////////////////////////////////////
// MACROS
#define NUM_TESTS 3
#define LCU_MAX_LOG_W 6
#define LCU_MIN_LOG_W 2
//////////////////////////////////////////////////////////////////////////
// GLOBALS
static kvz_pixel * satd_bufs[NUM_TESTS][7][2];
static struct {
int log_width; // for selecting dim from satd_bufs
cost_pixel_nxn_func * tested_func;
} satd_test_env;
//////////////////////////////////////////////////////////////////////////
// SETUP, TEARDOWN AND HELPER FUNCTIONS
static void setup_tests()
{
for (int test = 0; test < NUM_TESTS; ++test) {
for (int w = 0; w <= LCU_MIN_LOG_W; ++w) {
satd_bufs[test][w][0] = NULL;
satd_bufs[test][w][1] = NULL;
}
for (int w = LCU_MIN_LOG_W; w <= LCU_MAX_LOG_W; ++w) {
unsigned size = 1 << (w * 2);
satd_bufs[test][w][0] = malloc(size * sizeof(kvz_pixel));
satd_bufs[test][w][1] = malloc(size * sizeof(kvz_pixel));
}
}
//Black and white buffers
int test = 0;
for (int w = LCU_MIN_LOG_W; w <= LCU_MAX_LOG_W; ++w) {
unsigned size = 1 << (w * 2);
FILL_ARRAY(satd_bufs[test][w][0], 0, size);
FILL_ARRAY(satd_bufs[test][w][1], 255, size);
}
//Checker patterns, buffer 1 is negative of buffer 2
test = 1;
for (int w = LCU_MIN_LOG_W; w <= LCU_MAX_LOG_W; ++w) {
unsigned size = 1 << (w * 2);
for (int i = 0; i < size; ++i){
satd_bufs[test][w][0][i] = 255 * ( ( ((i >> w)%2) + (i % 2) ) % 2);
satd_bufs[test][w][1][i] = (satd_bufs[test][w][0][i] + 1) % 2 ;
}
}
//Gradient test pattern
test = 2;
for (int w = LCU_MIN_LOG_W; w <= LCU_MAX_LOG_W; ++w) {
unsigned size = 1 << (w * 2);
for (int i = 0; i < size; ++i){
int column = (i % (1 << w) );
int row = (i / (1 << w) );
int r = sqrt(row * row + column * column);
satd_bufs[test][w][0][i] = 255 / (r + 1);
satd_bufs[test][w][1][i] = 255 - 255 / (r + 1);
}
}
}
static void satd_tear_down_tests()
{
for (int test = 0; test < NUM_TESTS; ++test) {
for (int log_width = 2; log_width <= 6; ++log_width) {
free(satd_bufs[test][log_width][0]);
free(satd_bufs[test][log_width][1]);
}
}
}
//////////////////////////////////////////////////////////////////////////
// TESTS
TEST satd_test_black_and_white(void)
{
const int satd_results[5] = {2040, 4080, 16320, 65280, 261120};
const int test = 0;
kvz_pixel * buf1 = satd_bufs[test][satd_test_env.log_width][0];
kvz_pixel * buf2 = satd_bufs[test][satd_test_env.log_width][1];
unsigned result1 = satd_test_env.tested_func(buf1, buf2);
unsigned result2 = satd_test_env.tested_func(buf2, buf1);
ASSERT_EQ(result1, result2);
ASSERT_EQ(result1, satd_results[satd_test_env.log_width - 2]);
PASS();
}
TEST satd_test_checkers(void)
{
const int satd_checkers_results[5] = { 2040, 4080, 16320, 65280, 261120 };
const int test = 1;
kvz_pixel * buf1 = satd_bufs[test][satd_test_env.log_width][0];
kvz_pixel * buf2 = satd_bufs[test][satd_test_env.log_width][1];
unsigned result1 = satd_test_env.tested_func(buf1, buf2);
unsigned result2 = satd_test_env.tested_func(buf2, buf1);
ASSERT_EQ(result1, result2);
ASSERT_EQ(result1, satd_checkers_results[satd_test_env.log_width - 2]);
PASS();
}
TEST satd_test_gradient(void)
{
const int satd_gradient_results[5] = {3140,9004,20481,67262,258672};
const int test = 2;
kvz_pixel * buf1 = satd_bufs[test][satd_test_env.log_width][0];
kvz_pixel * buf2 = satd_bufs[test][satd_test_env.log_width][1];
unsigned result1 = satd_test_env.tested_func(buf1, buf2);
unsigned result2 = satd_test_env.tested_func(buf2, buf1);
ASSERT_EQ(result1, result2);
ASSERT_EQ(result1, satd_gradient_results[satd_test_env.log_width - 2]);
PASS();
}
//////////////////////////////////////////////////////////////////////////
// TEST FIXTURES
SUITE(satd_tests)
{
setup_tests();
// Loop through all strategies picking out the intra sad ones and run
// selectec strategies though all tests.
for (unsigned i = 0; i < strategies.count; ++i) {
const char * type = strategies.strategies[i].type;
if (strcmp(type, "satd_4x4") == 0) {
satd_test_env.log_width = 2;
}
else if (strcmp(type, "satd_8x8") == 0) {
satd_test_env.log_width = 3;
}
else if (strcmp(type, "satd_16x16") == 0) {
satd_test_env.log_width = 4;
}
else if (strcmp(type, "satd_32x32") == 0) {
satd_test_env.log_width = 5;
}
else if (strcmp(type, "satd_64x64") == 0) {
satd_test_env.log_width = 6;
}
else {
continue;
}
satd_test_env.tested_func = strategies.strategies[i].fptr;
// Tests
RUN_TEST(satd_test_black_and_white);
RUN_TEST(satd_test_checkers);
RUN_TEST(satd_test_gradient);
}
satd_tear_down_tests();
}