ref: 4caea79ffa54719a5a10320d22b3dc6a7f84918d
dir: /test/buffer.h/
/* * Copyright (c) 2016 The WebM project authors. All Rights Reserved. * * Use of this source code is governed by a BSD-style license * that can be found in the LICENSE file in the root of the source * tree. An additional intellectual property rights grant can be found * in the file PATENTS. All contributing project authors may * be found in the AUTHORS file in the root of the source tree. */ #ifndef VPX_TEST_BUFFER_H_ #define VPX_TEST_BUFFER_H_ #include <stdio.h> #include <limits> #include "third_party/googletest/src/include/gtest/gtest.h" #include "test/acm_random.h" #include "vpx/vpx_integer.h" #include "vpx_mem/vpx_mem.h" namespace libvpx_test { template <typename T> class Buffer { public: Buffer(int width, int height, int top_padding, int left_padding, int right_padding, int bottom_padding) : width_(width), height_(height), top_padding_(top_padding), left_padding_(left_padding), right_padding_(right_padding), bottom_padding_(bottom_padding), alignment_(0), padding_value_(0), stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(NULL) {} Buffer(int width, int height, int top_padding, int left_padding, int right_padding, int bottom_padding, unsigned int alignment) : width_(width), height_(height), top_padding_(top_padding), left_padding_(left_padding), right_padding_(right_padding), bottom_padding_(bottom_padding), alignment_(alignment), padding_value_(0), stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(NULL) {} Buffer(int width, int height, int padding) : width_(width), height_(height), top_padding_(padding), left_padding_(padding), right_padding_(padding), bottom_padding_(padding), alignment_(0), padding_value_(0), stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(NULL) {} Buffer(int width, int height, int padding, unsigned int alignment) : width_(width), height_(height), top_padding_(padding), left_padding_(padding), right_padding_(padding), bottom_padding_(padding), alignment_(alignment), padding_value_(0), stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(NULL) {} ~Buffer() { if (alignment_) { vpx_free(raw_buffer_); } else { delete[] raw_buffer_; } } T *TopLeftPixel() const; int stride() const { return stride_; } // Set the buffer (excluding padding) to 'value'. void Set(const T value); // Set the buffer (excluding padding) to the output of ACMRandom function // 'rand_func'. void Set(ACMRandom *rand_class, T (ACMRandom::*rand_func)()); // Set the buffer (excluding padding) to the output of ACMRandom function // 'RandRange' with range 'low' to 'high' which typically must be within // testing::internal::Random::kMaxRange (1u << 31). However, because we want // to allow negative low (and high) values, it is restricted to INT32_MAX // here. void Set(ACMRandom *rand_class, const T low, const T high); // Copy the contents of Buffer 'a' (excluding padding). void CopyFrom(const Buffer<T> &a); void DumpBuffer() const; // Highlight the differences between two buffers if they are the same size. void PrintDifference(const Buffer<T> &a) const; bool HasPadding() const; // Sets all the values in the buffer to 'padding_value'. void SetPadding(const T padding_value); // Checks if all the values (excluding padding) are equal to 'value' if the // Buffers are the same size. bool CheckValues(const T value) const; // Check that padding matches the expected value or there is no padding. bool CheckPadding() const; // Compare the non-padding portion of two buffers if they are the same size. bool CheckValues(const Buffer<T> &a) const; bool Init() { if (raw_buffer_ != NULL) return false; EXPECT_GT(width_, 0); EXPECT_GT(height_, 0); EXPECT_GE(top_padding_, 0); EXPECT_GE(left_padding_, 0); EXPECT_GE(right_padding_, 0); EXPECT_GE(bottom_padding_, 0); stride_ = left_padding_ + width_ + right_padding_; num_elements_ = stride_ * (top_padding_ + height_ + bottom_padding_); raw_size_ = num_elements_ * sizeof(T); if (alignment_) { EXPECT_GE(alignment_, sizeof(T)); // Ensure alignment of the first value will be preserved. EXPECT_EQ((left_padding_ * sizeof(T)) % alignment_, 0u); // Ensure alignment of the subsequent rows will be preserved when there is // a stride. if (stride_ != width_) { EXPECT_EQ((stride_ * sizeof(T)) % alignment_, 0u); } raw_buffer_ = reinterpret_cast<T *>(vpx_memalign(alignment_, raw_size_)); } else { raw_buffer_ = new (std::nothrow) T[num_elements_]; } EXPECT_TRUE(raw_buffer_ != NULL); SetPadding(std::numeric_limits<T>::max()); return !::testing::Test::HasFailure(); } private: bool BufferSizesMatch(const Buffer<T> &a) const; const int width_; const int height_; const int top_padding_; const int left_padding_; const int right_padding_; const int bottom_padding_; const unsigned int alignment_; T padding_value_; int stride_; int raw_size_; int num_elements_; T *raw_buffer_; }; template <typename T> T *Buffer<T>::TopLeftPixel() const { if (!raw_buffer_) return NULL; return raw_buffer_ + (top_padding_ * stride_) + left_padding_; } template <typename T> void Buffer<T>::Set(const T value) { if (!raw_buffer_) return; T *src = TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { src[width] = value; } src += stride_; } } template <typename T> void Buffer<T>::Set(ACMRandom *rand_class, T (ACMRandom::*rand_func)()) { if (!raw_buffer_) return; T *src = TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { src[width] = (*rand_class.*rand_func)(); } src += stride_; } } template <typename T> void Buffer<T>::Set(ACMRandom *rand_class, const T low, const T high) { if (!raw_buffer_) return; EXPECT_LE(low, high); EXPECT_LE(static_cast<int64_t>(high) - low, std::numeric_limits<int32_t>::max()); T *src = TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { // 'low' will be promoted to unsigned given the return type of RandRange. // Store the value as an int to avoid unsigned overflow warnings when // 'low' is negative. const int32_t value = static_cast<int32_t>((*rand_class).RandRange(high - low)); src[width] = static_cast<T>(value + low); } src += stride_; } } template <typename T> void Buffer<T>::CopyFrom(const Buffer<T> &a) { if (!raw_buffer_) return; if (!BufferSizesMatch(a)) return; T *a_src = a.TopLeftPixel(); T *b_src = this->TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { b_src[width] = a_src[width]; } a_src += a.stride(); b_src += this->stride(); } } template <typename T> void Buffer<T>::DumpBuffer() const { if (!raw_buffer_) return; for (int height = 0; height < height_ + top_padding_ + bottom_padding_; ++height) { for (int width = 0; width < stride_; ++width) { printf("%4d", raw_buffer_[height + width * stride_]); } printf("\n"); } } template <typename T> bool Buffer<T>::HasPadding() const { if (!raw_buffer_) return false; return top_padding_ || left_padding_ || right_padding_ || bottom_padding_; } template <typename T> void Buffer<T>::PrintDifference(const Buffer<T> &a) const { if (!raw_buffer_) return; if (!BufferSizesMatch(a)) return; T *a_src = a.TopLeftPixel(); T *b_src = TopLeftPixel(); printf("This buffer:\n"); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { if (a_src[width] != b_src[width]) { printf("*%3d", b_src[width]); } else { printf("%4d", b_src[width]); } } printf("\n"); a_src += a.stride(); b_src += this->stride(); } a_src = a.TopLeftPixel(); b_src = TopLeftPixel(); printf("Reference buffer:\n"); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { if (a_src[width] != b_src[width]) { printf("*%3d", a_src[width]); } else { printf("%4d", a_src[width]); } } printf("\n"); a_src += a.stride(); b_src += this->stride(); } } template <typename T> void Buffer<T>::SetPadding(const T padding_value) { if (!raw_buffer_) return; padding_value_ = padding_value; T *src = raw_buffer_; for (int i = 0; i < num_elements_; ++i) { src[i] = padding_value; } } template <typename T> bool Buffer<T>::CheckValues(const T value) const { if (!raw_buffer_) return false; T *src = TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { if (value != src[width]) { return false; } } src += stride_; } return true; } template <typename T> bool Buffer<T>::CheckPadding() const { if (!raw_buffer_) return false; if (!HasPadding()) return true; // Top padding. T const *top = raw_buffer_; for (int i = 0; i < stride_ * top_padding_; ++i) { if (padding_value_ != top[i]) { return false; } } // Left padding. T const *left = TopLeftPixel() - left_padding_; for (int height = 0; height < height_; ++height) { for (int width = 0; width < left_padding_; ++width) { if (padding_value_ != left[width]) { return false; } } left += stride_; } // Right padding. T const *right = TopLeftPixel() + width_; for (int height = 0; height < height_; ++height) { for (int width = 0; width < right_padding_; ++width) { if (padding_value_ != right[width]) { return false; } } right += stride_; } // Bottom padding T const *bottom = raw_buffer_ + (top_padding_ + height_) * stride_; for (int i = 0; i < stride_ * bottom_padding_; ++i) { if (padding_value_ != bottom[i]) { return false; } } return true; } template <typename T> bool Buffer<T>::CheckValues(const Buffer<T> &a) const { if (!raw_buffer_) return false; if (!BufferSizesMatch(a)) return false; T *a_src = a.TopLeftPixel(); T *b_src = this->TopLeftPixel(); for (int height = 0; height < height_; ++height) { for (int width = 0; width < width_; ++width) { if (a_src[width] != b_src[width]) { return false; } } a_src += a.stride(); b_src += this->stride(); } return true; } template <typename T> bool Buffer<T>::BufferSizesMatch(const Buffer<T> &a) const { if (!raw_buffer_) return false; if (a.width_ != this->width_ || a.height_ != this->height_) { printf( "Reference buffer of size %dx%d does not match this buffer which is " "size %dx%d\n", a.width_, a.height_, this->width_, this->height_); return false; } return true; } } // namespace libvpx_test #endif // VPX_TEST_BUFFER_H_