1ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org/* 2ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 3ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * 4ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * Use of this source code is governed by a BSD-style license 5ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * that can be found in the LICENSE file in the root of the source 6ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * tree. An additional intellectual property rights grant can be found 7ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * in the file PATENTS. All contributing project authors may 8ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org * be found in the AUTHORS file in the root of the source tree. 9ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org */ 10ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 11ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "third_party/googletest/src/include/gtest/gtest.h" 12ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "test/acm_random.h" 137c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org#include "test/clear_system_state.h" 14d348b8d765c019ee7250075d663a83db00c65c08tomfinegan@chromium.org#include "test/register_state_check.h" 15f9586bb54d74c97d07b09eb2512f8569c9c1c025fgalligan@chromium.org#include "./vpx_config.h" 16f9586bb54d74c97d07b09eb2512f8569c9c1c025fgalligan@chromium.org#include "./vp8_rtcd.h" 17ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "vp8/common/blockd.h" 18ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "vp8/encoder/block.h" 19ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#include "vpx_mem/vpx_mem.h" 20ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 21ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.orgtypedef void (*SubtractBlockFunc)(BLOCK *be, BLOCKD *bd, int pitch); 22ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 23ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgnamespace { 24ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 25ac4e313c19203132648a2a271703b6ee76fe4284johannkoenig@chromium.orgclass SubtractBlockTest : public ::testing::TestWithParam<SubtractBlockFunc> { 267c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org public: 277c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org virtual void TearDown() { 287c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org libvpx_test::ClearSystemState(); 297c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org } 307c1508f4449fe8184aa045737c20ba37e603dd03fgalligan@chromium.org}; 31ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 32ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgusing libvpx_test::ACMRandom; 33ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 34ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgTEST_P(SubtractBlockTest, SimpleSubtract) { 35ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org ACMRandom rnd(ACMRandom::DeterministicSeed()); 36ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org BLOCK be; 37ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org BLOCKD bd; 38ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // in libvpx, this stride is always 16 39ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org const int kDiffPredStride = 16; 40ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org const int kSrcStride[] = {32, 16, 8, 4, 0}; 41ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org const int kBlockWidth = 4; 42ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org const int kBlockHeight = 4; 43ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 44ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // Allocate... align to 16 for mmx/sse tests 45ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org uint8_t *source = reinterpret_cast<uint8_t*>( 46ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_memalign(16, kBlockHeight * kSrcStride[0] * sizeof(*source))); 47ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org be.src_diff = reinterpret_cast<int16_t*>( 48ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_memalign(16, kBlockHeight * kDiffPredStride * sizeof(*be.src_diff))); 49ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org bd.predictor = reinterpret_cast<unsigned char*>( 50ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_memalign(16, kBlockHeight * kDiffPredStride * sizeof(*bd.predictor))); 51ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 52f9586bb54d74c97d07b09eb2512f8569c9c1c025fgalligan@chromium.org for (int i = 0; kSrcStride[i] > 0; ++i) { 53ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // start at block0 54ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org be.src = 0; 55ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org be.base_src = &source; 56ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org be.src_stride = kSrcStride[i]; 57ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 58ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // set difference 59ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org int16_t *src_diff = be.src_diff; 60ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int r = 0; r < kBlockHeight; ++r) { 61ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int c = 0; c < kBlockWidth; ++c) { 6276e516e2154f353aa02c504bac88afb0f95fefa7johannkoenig@chromium.org src_diff[c] = static_cast<int16_t>(0xa5a5u); 63ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 64ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org src_diff += kDiffPredStride; 65ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 66ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 67ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // set destination 68ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org uint8_t *base_src = *be.base_src; 69ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int r = 0; r < kBlockHeight; ++r) { 70ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int c = 0; c < kBlockWidth; ++c) { 71ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org base_src[c] = rnd.Rand8(); 72ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 73ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org base_src += be.src_stride; 74ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 75ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 76ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org // set predictor 77ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org uint8_t *predictor = bd.predictor; 78ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int r = 0; r < kBlockHeight; ++r) { 79ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int c = 0; c < kBlockWidth; ++c) { 80ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org predictor[c] = rnd.Rand8(); 81ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 82ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org predictor += kDiffPredStride; 83ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 84ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 8595aa45d73048f952dcaad0037429cc6751b34f2fjohannkoenig@google.com ASM_REGISTER_STATE_CHECK(GetParam()(&be, &bd, kDiffPredStride)); 86ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 87ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org base_src = *be.base_src; 88ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org src_diff = be.src_diff; 89ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org predictor = bd.predictor; 90ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int r = 0; r < kBlockHeight; ++r) { 91ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org for (int c = 0; c < kBlockWidth; ++c) { 92ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org EXPECT_EQ(base_src[c], (src_diff[c] + predictor[c])) << "r = " << r 93ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org << ", c = " << c; 94ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 95ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org src_diff += kDiffPredStride; 96ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org predictor += kDiffPredStride; 97ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org base_src += be.src_stride; 98ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 99ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org } 100ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_free(be.src_diff); 101ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_free(source); 102ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org vpx_free(bd.predictor); 103ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org} 104ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 105ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgINSTANTIATE_TEST_CASE_P(C, SubtractBlockTest, 106ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org ::testing::Values(vp8_subtract_b_c)); 107ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 108ace65784417788374f0b19ce5a8abd06c9ccd007johannkoenig@chromium.org#if HAVE_NEON 109411971f94253c85e1866c281860d6344f6aa0c78fgalligan@chromium.orgINSTANTIATE_TEST_CASE_P(NEON, SubtractBlockTest, 110411971f94253c85e1866c281860d6344f6aa0c78fgalligan@chromium.org ::testing::Values(vp8_subtract_b_neon)); 111411971f94253c85e1866c281860d6344f6aa0c78fgalligan@chromium.org#endif 112411971f94253c85e1866c281860d6344f6aa0c78fgalligan@chromium.org 113ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#if HAVE_MMX 114ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgINSTANTIATE_TEST_CASE_P(MMX, SubtractBlockTest, 115ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org ::testing::Values(vp8_subtract_b_mmx)); 116ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#endif 117ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 118ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#if HAVE_SSE2 119ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.orgINSTANTIATE_TEST_CASE_P(SSE2, SubtractBlockTest, 120ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org ::testing::Values(vp8_subtract_b_sse2)); 121ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org#endif 122ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org 123ed759d81a39febed3a8a395386639d54307504aagrunell@chromium.org} // namespace 124