File size: 2,317 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | // SPDX-License-Identifier: LGPL-2.1-or-later
#include <gtest/gtest.h>
#include <Mod/Mesh/App/Core/KDTree.h>
// NOLINTBEGIN(cppcoreguidelines-*,readability-*)
class KDTreeTest: public ::testing::Test
{
protected:
void SetUp() override
{
points.emplace_back(0.F, 0.F, 0.F);
points.emplace_back(0.F, 0.F, 1.F);
points.emplace_back(0.F, 1.F, 0.F);
points.emplace_back(0.F, 1.F, 1.F);
points.emplace_back(1.F, 0.F, 0.F);
points.emplace_back(1.F, 0.F, 1.F);
points.emplace_back(1.F, 1.F, 0.F);
points.emplace_back(1.F, 1.F, 1.F);
}
void TearDown() override
{}
const std::vector<Base::Vector3f>& GetPoints() const
{
return points;
}
private:
std::vector<Base::Vector3f> points;
};
TEST_F(KDTreeTest, TestKDTreeEmpty)
{
MeshCore::MeshKDTree tree;
EXPECT_EQ(tree.IsEmpty(), true);
}
TEST_F(KDTreeTest, TestKDTreeNearestEmpty)
{
MeshCore::MeshKDTree tree;
Base::Vector3f pnt;
Base::Vector3f nor;
float dist;
EXPECT_EQ(tree.FindNearest(pnt, nor, dist), MeshCore::POINT_INDEX_MAX);
}
TEST_F(KDTreeTest, TestKDTreeNearest)
{
MeshCore::MeshKDTree tree;
tree.AddPoints(GetPoints());
EXPECT_EQ(tree.IsEmpty(), false);
Base::Vector3f nor;
float dist;
EXPECT_EQ(tree.FindNearest(Base::Vector3f(0.9F, 0.1F, 0.1F), nor, dist), 4);
}
TEST_F(KDTreeTest, TestKDTreeNearestMaxDist)
{
MeshCore::MeshKDTree tree;
tree.AddPoints(GetPoints());
EXPECT_EQ(tree.IsEmpty(), false);
Base::Vector3f nor;
float dist;
EXPECT_EQ(
tree.FindNearest(Base::Vector3f(0.9F, 0.1F, 0.1F), 0.0F, nor, dist),
MeshCore::POINT_INDEX_MAX
);
}
TEST_F(KDTreeTest, TestKDTreeFindExact)
{
MeshCore::MeshKDTree tree;
tree.AddPoints(GetPoints());
EXPECT_EQ(tree.FindExact(Base::Vector3f(0.1F, 0, 0)), MeshCore::POINT_INDEX_MAX);
EXPECT_EQ(tree.FindExact(Base::Vector3f(0, 0, 0)), 0);
}
TEST_F(KDTreeTest, TestKDTreeFindRange)
{
MeshCore::MeshKDTree tree;
tree.AddPoints(GetPoints());
std::vector<MeshCore::PointIndex> index;
std::vector<MeshCore::PointIndex> result = {0, 4};
tree.FindInRange(Base::Vector3f(0.5F, 0, 0), 0.6F, index);
EXPECT_EQ(index, result);
}
// NOLINTEND(cppcoreguidelines-*,readability-*)
|