Add find_convex_hull using Graham scan (#2889)

This commit is contained in:
Adrià Arrufat 2023-11-13 10:10:47 +09:00 committed by GitHub
parent f7d99ae0dc
commit b20e97446b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 105 additions and 0 deletions

View File

@ -1308,6 +1308,72 @@ namespace dlib
return (s0>0&&s1>0&&s2>0&&s3>0) || (s0<0&&s1<0&&s2<0&&s3<0); return (s0>0&&s1>0&&s2>0&&s3>0) || (s0<0&&s1<0&&s2<0&&s3<0);
} }
// ----------------------------------------------------------------------------------------
namespace impl
{
enum class points_orientation
{
collinear,
clockwise,
couter_clockwise,
};
template <typename T>
std::enable_if_t<std::is_same<T, point>::value || std::is_same<T, dpoint>::value, points_orientation>
find_points_orientation(const T& a, const T& b, const T& c)
{
const auto v = a.x() * (b.y() - c.y()) + b.x() * (c.y() - a.y()) + c.x() * (a.y() - b.y());
if (v < 0) return points_orientation::clockwise;
if (v > 0) return points_orientation::couter_clockwise;
return points_orientation::collinear;
}
}
template <typename T>
std::vector<T> find_convex_hull(std::vector<T>& points)
{
static_assert(std::is_same<T, point>::value || std::is_same<T, dpoint>::value, "find_convex_hull() only works for 2D dlib::vector types.");
if (points.size() < 3)
return {};
// find the point with the lowest y coordinate, and the left-most in case of ties.
const auto p0 = *std::min_element(points.begin(), points.end(), [](const auto& a, const auto& b){
return std::make_pair(a.y(), a.x()) < std::make_pair(b.y(), b.x());
});
// sort the points by polar angle in clockwise order
std::sort(points.begin(), points.end(), [&p0](const auto& a, const auto& b){
switch (impl::find_points_orientation(p0, a, b))
{
case impl::points_orientation::clockwise:
return true;
case impl::points_orientation::couter_clockwise:
return false;
case impl::points_orientation::collinear:
return length_squared(p0-a) < length_squared(p0-b);
default:
DLIB_CASSERT(false, "this should be impossible");
}
});
std::vector<T> hull;
for (const auto& p : points)
{
while (hull.size() > 1 &&
impl::find_points_orientation(hull.at(hull.size() - 2), hull.back(), p) != impl::points_orientation::clockwise
)
{
hull.pop_back();
}
hull.push_back(p);
}
// If all the points were collinear, we'll have only two points in the hull, so we need to clear it.
if (hull.size() < 3)
hull.clear();
return hull;
}
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
template < template <

View File

@ -456,6 +456,22 @@ namespace dlib
false if not. false if not.
!*/ !*/
// ----------------------------------------------------------------------------------------
template <typename T>
std::vector<T> find_convex_hull(
std::vector<T>& points
);
/*!
requires
- T == dlib::point or dlib::dpoint
ensures
- If points.size() < 3: it returns an empty vector.
- Else: Finds the convex hull of points using the Graham scan algorithm. That is,
the smallest convex shape that contains all points. Moreover, in case all points
are collinear, that is, along the same line, it will also return an empty vector.
!*/
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
template < template <

View File

@ -382,6 +382,28 @@ namespace
} }
// ----------------------------------------------------------------------------------------
void test_find_convex_hull()
{
print_spinner();
std::vector<dpoint> points{
{ 0.0, 0.0 },
{ 1.0, 1.0 },
{ 2.0, 2.0 },
{ 3.0, 1.0 },
{ 4.0, 0.0 },
{ 2.0, 4.0 },
{ 1.0, 3.0 },
};
const auto hull = find_convex_hull(points);
DLIB_TEST(hull.size() == 4);
DLIB_TEST(hull[0] == dpoint(0, 0));
DLIB_TEST(hull[1] == dpoint(1, 3));
DLIB_TEST(hull[2] == dpoint(2, 4));
DLIB_TEST(hull[3] == dpoint(4, 0));
}
// ---------------------------------------------------------------------------------------- // ----------------------------------------------------------------------------------------
void test_border_enumerator() void test_border_enumerator()
@ -956,6 +978,7 @@ namespace
test_find_similarity_transform2<float>(); test_find_similarity_transform2<float>();
test_line(); test_line();
test_polygon(); test_polygon();
test_find_convex_hull();
} }
} a; } a;