blob: d9595ddf7628220789847e92f11f0e9384841d09 [file] [log] [blame]
#include <gtest/gtest.h>
#include <hardware/hardware.h>
#include "separate_rects.h"
using namespace separate_rects;
#define RectSet RectSet<TId, TNum>
#define Rect Rect<TNum>
#define IdSet IdSet<TId>
typedef uint64_t TId;
typedef float TNum;
struct SeparateRectTest : public testing::Test {
bool IsEquality(std::vector<RectSet> &out,
std::vector<RectSet> &expected_out) {
// Test for rects missing from out
for (size_t i = 0; i < expected_out.size(); i++) {
RectSet &ex_out = expected_out[i];
if (std::find(out.begin(), out.end(), ex_out) == out.end()) {
return false;
}
}
// Test for presence of unexpected rects in out
for (size_t i = 0; i < out.size(); i++) {
RectSet &actual_out = out[i];
if (std::find(expected_out.begin(), expected_out.end(), actual_out) ==
expected_out.end()) {
return false;
}
}
return true;
}
};
TEST_F(SeparateRectTest, test_separate_rect) {
std::vector<Rect> in;
std::vector<RectSet> out;
std::vector<RectSet> expected_out;
in.push_back({0, 0, 4, 5});
in.push_back({2, 0, 6, 6});
in.push_back({4, 0, 8, 5});
in.push_back({0, 7, 8, 9});
in.push_back({10, 0, 18, 5});
in.push_back({12, 0, 16, 5});
in.push_back({20, 11, 24, 17});
in.push_back({22, 13, 26, 21});
in.push_back({32, 33, 36, 37});
in.push_back({30, 31, 38, 39});
in.push_back({40, 43, 48, 45});
in.push_back({44, 41, 46, 47});
in.push_back({50, 51, 52, 53});
in.push_back({50, 51, 52, 53});
in.push_back({50, 51, 52, 53});
in.push_back({0, 0, 0, 10});
in.push_back({0, 0, 10, 0});
in.push_back({10, 0, 0, 10});
in.push_back({0, 10, 10, 0});
for (int i = 0; i < 100000; i++) {
out.clear();
separate_frects_64(in, &out);
}
expected_out.push_back(RectSet(IdSet(0), Rect(0, 0, 2, 5)));
expected_out.push_back(RectSet(IdSet(1), Rect(2, 5, 6, 6)));
expected_out.push_back(RectSet(IdSet(1) | 0, Rect(2, 0, 4, 5)));
expected_out.push_back(RectSet(IdSet(1) | 2, Rect(4, 0, 6, 5)));
expected_out.push_back(RectSet(IdSet(2), Rect(6, 0, 8, 5)));
expected_out.push_back(RectSet(IdSet(3), Rect(0, 7, 8, 9)));
expected_out.push_back(RectSet(IdSet(4), Rect(10, 0, 12, 5)));
expected_out.push_back(RectSet(IdSet(5) | 4, Rect(12, 0, 16, 5)));
expected_out.push_back(RectSet(IdSet(4), Rect(16, 0, 18, 5)));
expected_out.push_back(RectSet(IdSet(6), Rect(20, 11, 22, 17)));
expected_out.push_back(RectSet(IdSet(6) | 7, Rect(22, 13, 24, 17)));
expected_out.push_back(RectSet(IdSet(6), Rect(22, 11, 24, 13)));
expected_out.push_back(RectSet(IdSet(7), Rect(22, 17, 24, 21)));
expected_out.push_back(RectSet(IdSet(7), Rect(24, 13, 26, 21)));
expected_out.push_back(RectSet(IdSet(9), Rect(30, 31, 32, 39)));
expected_out.push_back(RectSet(IdSet(8) | 9, Rect(32, 33, 36, 37)));
expected_out.push_back(RectSet(IdSet(9), Rect(32, 37, 36, 39)));
expected_out.push_back(RectSet(IdSet(9), Rect(32, 31, 36, 33)));
expected_out.push_back(RectSet(IdSet(9), Rect(36, 31, 38, 39)));
expected_out.push_back(RectSet(IdSet(10), Rect(40, 43, 44, 45)));
expected_out.push_back(RectSet(IdSet(10) | 11, Rect(44, 43, 46, 45)));
expected_out.push_back(RectSet(IdSet(11), Rect(44, 41, 46, 43)));
expected_out.push_back(RectSet(IdSet(11), Rect(44, 45, 46, 47)));
expected_out.push_back(RectSet(IdSet(10), Rect(46, 43, 48, 45)));
expected_out.push_back(RectSet(IdSet(12) | 13 | 14, Rect(50, 51, 52, 53)));
ASSERT_TRUE(IsEquality(out, expected_out));
}