549e2b7539
See D11882 for a description of the iterators. Reviewed By: jbakker Differential Revision: https://developer.blender.org/D12001
142 lines
3.9 KiB
C++
142 lines
3.9 KiB
C++
/*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* Copyright 2021, Blender Foundation.
|
|
*/
|
|
|
|
#include "testing/testing.h"
|
|
|
|
#include "COM_BufferArea.h"
|
|
|
|
namespace blender::compositor::tests {
|
|
|
|
static rcti create_rect(int width, int height)
|
|
{
|
|
rcti rect;
|
|
BLI_rcti_init(&rect, 0, width, 0, height);
|
|
return rect;
|
|
}
|
|
|
|
static rcti create_rect(int width, int height, int offset)
|
|
{
|
|
rcti rect;
|
|
BLI_rcti_init(&rect, offset, offset + width, offset, offset + height);
|
|
return rect;
|
|
}
|
|
|
|
TEST(BufferArea, BufferConstructor)
|
|
{
|
|
const int width = 2;
|
|
const int height = 3;
|
|
BufferArea<float> area(nullptr, width, height, 4);
|
|
EXPECT_EQ(area.width(), width);
|
|
EXPECT_EQ(area.height(), height);
|
|
rcti rect = create_rect(width, height);
|
|
EXPECT_TRUE(BLI_rcti_compare(&area.get_rect(), &rect));
|
|
}
|
|
|
|
TEST(BufferArea, AreaConstructor)
|
|
{
|
|
const int buf_width = 5;
|
|
const int area_width = 1;
|
|
const int area_height = 3;
|
|
rcti area_rect = create_rect(area_width, area_height, 1);
|
|
BufferArea<float> area(nullptr, buf_width, area_rect, 4);
|
|
EXPECT_EQ(area.width(), area_width);
|
|
EXPECT_EQ(area.height(), area_height);
|
|
EXPECT_TRUE(BLI_rcti_compare(&area.get_rect(), &area_rect));
|
|
}
|
|
|
|
static void fill_buffer_with_indexes(float *buf, int buf_len)
|
|
{
|
|
for (int i = 0; i < buf_len; i++) {
|
|
buf[i] = i;
|
|
}
|
|
}
|
|
|
|
static void test_single_elem_iteration(float *buffer, BufferArea<float> area)
|
|
{
|
|
int elems_count = 0;
|
|
for (float *elem : area) {
|
|
EXPECT_EQ(elem, buffer);
|
|
elems_count++;
|
|
}
|
|
EXPECT_EQ(elems_count, 1);
|
|
}
|
|
|
|
static void test_full_buffer_iteration(
|
|
float *buf, int buf_width, int buf_len, int num_channels, BufferArea<float> area)
|
|
{
|
|
fill_buffer_with_indexes(buf, buf_len);
|
|
rcti rect = area.get_rect();
|
|
int x = rect.xmin;
|
|
int y = rect.ymin;
|
|
for (float *elem : area) {
|
|
for (int ch = 0; ch < num_channels; ch++) {
|
|
const int buf_index = y * buf_width * num_channels + x * num_channels + ch;
|
|
EXPECT_NEAR(elem[ch], buf_index, FLT_EPSILON);
|
|
}
|
|
x++;
|
|
if (x == rect.xmax) {
|
|
y++;
|
|
x = rect.xmin;
|
|
}
|
|
}
|
|
EXPECT_EQ(x, rect.xmin);
|
|
EXPECT_EQ(y, rect.ymax);
|
|
}
|
|
|
|
TEST(BufferArea, SingleElemBufferIteration)
|
|
{
|
|
const int buf_width = 4;
|
|
const int buf_height = 5;
|
|
const int area_width = 2;
|
|
const int area_height = 3;
|
|
const int num_channels = 4;
|
|
const int stride = 0;
|
|
float buf[num_channels];
|
|
{
|
|
BufferArea area(buf, buf_width, buf_height, stride);
|
|
test_single_elem_iteration(buf, area);
|
|
}
|
|
{
|
|
rcti area_rect = create_rect(area_width, area_height, 1);
|
|
BufferArea area(buf, buf_width, area_rect, stride);
|
|
test_single_elem_iteration(buf, area);
|
|
}
|
|
}
|
|
|
|
TEST(BufferArea, FullBufferIteration)
|
|
{
|
|
const int buf_width = 4;
|
|
const int area_width = 2;
|
|
const int area_height = 3;
|
|
const int buf_height = (area_height + 1);
|
|
const int num_channels = 4;
|
|
const int buf_len = buf_height * buf_width * num_channels;
|
|
float buf[buf_len];
|
|
{
|
|
BufferArea area(buf, buf_width, buf_height, num_channels);
|
|
test_full_buffer_iteration(buf, buf_width, buf_len, num_channels, area);
|
|
}
|
|
{
|
|
rcti area_rect = create_rect(area_width, area_height, 1);
|
|
BufferArea area(buf, buf_width, area_rect, num_channels);
|
|
test_full_buffer_iteration(buf, buf_width, buf_len, num_channels, area);
|
|
}
|
|
}
|
|
|
|
} // namespace blender::compositor::tests
|