Skip to content

geometry_distance.hpp

SECTIONGeometry INCLUDEnoya/geometry/geometry_distance.hpp

求平面点在线上的投影、对称点及点线段距离;零长线段按点处理,退化直线返回空结果。距离计算使用相对坐标,结果为 long double。

Complexity: Time: \(O(1)\) per projection, reflection, or distance query. Space: \(O(1)\).

跳到代码 · GitHub ↗

Implementation

当前头文件,省略 include guard;依赖见 #include。

/// @complexity Time: O(1) per projection, reflection, or distance query.
/// Space: O(1).

#include "noya/geometry/geometry_base.hpp"

#include <algorithm>
#include <cassert>
#include <cmath>
#include <optional>

namespace noya {

namespace geometry_distance_internal {
template <class T> point<long double> cast(const point<T> &p) {
  return {static_cast<long double>(p.x), static_cast<long double>(p.y)};
}
} // namespace geometry_distance_internal

/// @brief Project p onto the infinite line through a and b. A zero-length
/// defining pair returns nullopt. Coordinates are converted before arithmetic;
/// results use long double and are subject to floating-point precision.
template <class T>
std::optional<point<long double>>
line_projection(const point<T> &p, const point<T> &a, const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return std::nullopt;
  }
  return u + v * (dot(q, v) / len);
}

/// @brief Reflect p across the infinite line ab; nullopt for a == b.
template <class T>
std::optional<point<long double>>
line_reflection(const point<T> &p, const point<T> &a, const point<T> &b) {
  auto q = line_projection(p, a, b);
  if (!q) {
    return std::nullopt;
  }
  return *q * 2.0L - geometry_distance_internal::cast(p);
}

/// @brief Return the point-to-line distance; nullopt for a == b.
template <class T>
std::optional<long double> point_line_distance(const point<T> &p,
                                              const point<T> &a,
                                              const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = std::hypot(v.x, v.y);
  if (len == 0) {
    return std::nullopt;
  }
  return std::abs(cross(v, q)) / len;
}

/// @brief Return the nearest point on the closed segment [a, b]. A zero-length
/// segment is its single endpoint.
template <class T>
point<long double> closest_point_on_segment(const point<T> &p,
                                            const point<T> &a,
                                            const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return u;
  }
  long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
  return u + v * t;
}

/// @brief Return the distance from p to the closed segment [a, b].
template <class T>
long double point_segment_distance(const point<T> &p, const point<T> &a,
                                   const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = norm2(v);
  // Keep the residual relative to a; an absolute foot loses small distances.
  if (len != 0) {
    long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
    q = t == 1 ? cast(p) - cast(b) : q - v * t;
  }
  return std::hypot(q.x, q.y);
}

/// @brief Return the distance between closed segments, including point segments.
/// eps is an absolute tolerance for the intersection predicates (cross products
/// and coordinate comparisons), not a relative or angular tolerance.
template <class T>
long double segment_distance(const point<T> &a, const point<T> &b,
                             const point<T> &c, const point<T> &d,
                             long double eps = 0) {
  assert(eps >= 0);
  using geometry_distance_internal::cast;
  auto p = cast(a), q = cast(b), r = cast(c), s = cast(d);
  if (segments_intersect(p, q, r, s, eps)) {
    return 0;
  }
  return std::min({point_segment_distance(p, r, s),
                   point_segment_distance(q, r, s),
                   point_segment_distance(r, p, q),
                   point_segment_distance(s, p, q)});
}

} // namespace noya

 // NOYA_GEOMETRY_DISTANCE_HPP
#ifndef NOYA_GEOMETRY_DISTANCE_HPP
#define NOYA_GEOMETRY_DISTANCE_HPP 1

/// @complexity Time: O(1) per projection, reflection, or distance query.
/// Space: O(1).

#include "noya/geometry/geometry_base.hpp"

#include <algorithm>
#include <cassert>
#include <cmath>
#include <optional>

namespace noya {

namespace geometry_distance_internal {
template <class T> point<long double> cast(const point<T> &p) {
  return {static_cast<long double>(p.x), static_cast<long double>(p.y)};
}
} // namespace geometry_distance_internal

/// @brief Project p onto the infinite line through a and b. A zero-length
/// defining pair returns nullopt. Coordinates are converted before arithmetic;
/// results use long double and are subject to floating-point precision.
template <class T>
std::optional<point<long double>>
line_projection(const point<T> &p, const point<T> &a, const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return std::nullopt;
  }
  return u + v * (dot(q, v) / len);
}

/// @brief Reflect p across the infinite line ab; nullopt for a == b.
template <class T>
std::optional<point<long double>>
line_reflection(const point<T> &p, const point<T> &a, const point<T> &b) {
  auto q = line_projection(p, a, b);
  if (!q) {
    return std::nullopt;
  }
  return *q * 2.0L - geometry_distance_internal::cast(p);
}

/// @brief Return the point-to-line distance; nullopt for a == b.
template <class T>
std::optional<long double> point_line_distance(const point<T> &p,
                                              const point<T> &a,
                                              const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = std::hypot(v.x, v.y);
  if (len == 0) {
    return std::nullopt;
  }
  return std::abs(cross(v, q)) / len;
}

/// @brief Return the nearest point on the closed segment [a, b]. A zero-length
/// segment is its single endpoint.
template <class T>
point<long double> closest_point_on_segment(const point<T> &p,
                                            const point<T> &a,
                                            const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return u;
  }
  long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
  return u + v * t;
}

/// @brief Return the distance from p to the closed segment [a, b].
template <class T>
long double point_segment_distance(const point<T> &p, const point<T> &a,
                                   const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = norm2(v);
  // Keep the residual relative to a; an absolute foot loses small distances.
  if (len != 0) {
    long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
    q = t == 1 ? cast(p) - cast(b) : q - v * t;
  }
  return std::hypot(q.x, q.y);
}

/// @brief Return the distance between closed segments, including point segments.
/// eps is an absolute tolerance for the intersection predicates (cross products
/// and coordinate comparisons), not a relative or angular tolerance.
template <class T>
long double segment_distance(const point<T> &a, const point<T> &b,
                             const point<T> &c, const point<T> &d,
                             long double eps = 0) {
  assert(eps >= 0);
  using geometry_distance_internal::cast;
  auto p = cast(a), q = cast(b), r = cast(c), s = cast(d);
  if (segments_intersect(p, q, r, s, eps)) {
    return 0;
  }
  return std::min({point_segment_distance(p, r, s),
                   point_segment_distance(q, r, s),
                   point_segment_distance(r, p, q),
                   point_segment_distance(s, p, q)});
}

} // namespace noya

#endif // NOYA_GEOMETRY_DISTANCE_HPP
/// @complexity Time: O(1) per projection, reflection, or distance query.
/// Space: O(1).

/// @complexity Time: O(1) per point/segment primitive, O(n) for polygon area or
/// point location, O(n log n) for convex hull. Space: O(1), except O(n) for hull.

#include <algorithm>
#include <cmath>
#include <optional>
#include <vector>

namespace noya {

/// @brief Two-dimensional point with vector arithmetic and lexicographic order.
template <class T> struct point {
  T x{};
  T y{};

  point() = default;
  point(T x_, T y_) : x(x_), y(y_) {}

  point &operator+=(const point &rhs) {
    x += rhs.x;
    y += rhs.y;
    return *this;
  }
  point &operator-=(const point &rhs) {
    x -= rhs.x;
    y -= rhs.y;
    return *this;
  }
  point &operator*=(const T &scl) {
    x *= scl;
    y *= scl;
    return *this;
  }
  point &operator/=(const T &scl) {
    x /= scl;
    y /= scl;
    return *this;
  }

  friend point operator+(point l, const point &r) { return l += r; }
  friend point operator-(point l, const point &r) { return l -= r; }
  friend point operator*(point val, const T &scl) { return val *= scl; }
  friend point operator*(const T &scl, point val) { return val *= scl; }
  friend point operator/(point val, const T &scl) { return val /= scl; }
  friend bool operator==(const point &, const point &) = default;
  friend bool operator<(const point &l, const point &r) {
    return l.x < r.x || (l.x == r.x && l.y < r.y);
  }
};

/// @brief Circle represented by a center and a nonnegative radius.
template <class Real> struct circle {
  point<Real> o;
  Real r1{};
};

/// @brief Return the dot product of two vectors.
template <class T> T dot(const point<T> &a, const point<T> &b) {
  return a.x * b.x + a.y * b.y;
}

/// @brief Return the signed cross product of two vectors.
template <class T> T cross(const point<T> &a, const point<T> &b) {
  return a.x * b.y - a.y * b.x;
}

/// @brief Return cross(a - o1, b - o1).
template <class T>
T cross(const point<T> &o1, const point<T> &a, const point<T> &b) {
  return cross(a - o1, b - o1);
}

/// @brief Return the squared Euclidean norm.
template <class T> T norm2(const point<T> &val) { return dot(val, val); }

/// @brief Compare a value with zero using an optional absolute tolerance.
template <class T> int sign(const T &val, const T &eps = T{}) {
  return (val > eps) - (val < -eps);
}

/// @brief Return -1, 0, or 1 for a clockwise, collinear, or counter-clockwise
/// turn.
template <class T>
int orientation(const point<T> &a, const point<T> &b, const point<T> &c,
                const T &eps = T{}) {
  return sign(cross(a, b, c), eps);
}

/// @brief Test whether p lies on the closed segment [a, b].
template <class T>
bool on_segment(const point<T> &p, const point<T> &a, const point<T> &b,
                const T &eps = T{}) {
  if (orientation(a, b, p, eps) != 0) {
    return false;
  }
  return std::min(a.x, b.x) - eps <= p.x && p.x <= std::max(a.x, b.x) + eps &&
         std::min(a.y, b.y) - eps <= p.y && p.y <= std::max(a.y, b.y) + eps;
}

/// @brief Test whether the closed segments [a, b] and [c, d] intersect.
template <class T>
bool segments_intersect(const point<T> &a, const point<T> &b, const point<T> &c,
                        const point<T> &d, const T &eps = T{}) {
  int s1 = orientation(a, b, c, eps);
  int s2 = orientation(a, b, d, eps);
  int s3 = orientation(c, d, a, eps);
  int s4 = orientation(c, d, b, eps);
  if (s1 == 0 && on_segment(c, a, b, eps)) {
    return true;
  }
  if (s2 == 0 && on_segment(d, a, b, eps)) {
    return true;
  }
  if (s3 == 0 && on_segment(a, c, d, eps)) {
    return true;
  }
  if (s4 == 0 && on_segment(b, c, d, eps)) {
    return true;
  }
  return s1 * s2 < 0 && s3 * s4 < 0;
}

/// @brief Intersect the infinite lines through (a, b) and (c, d), returning
/// nullopt when they are parallel or coincident.
template <class T>
std::optional<point<long double>>
line_intersection(const point<T> &a, const point<T> &b, const point<T> &c,
                  const point<T> &d, long double eps = 0) {
  point<long double> lhs{static_cast<long double>(a.x),
                         static_cast<long double>(a.y)};
  point<long double> b1{static_cast<long double>(b.x),
                        static_cast<long double>(b.y)};
  point<long double> z{static_cast<long double>(c.x),
                       static_cast<long double>(c.y)};
  point<long double> d1{static_cast<long double>(d.x),
                        static_cast<long double>(d.y)};
  point<long double> da = b1 - lhs;
  point<long double> db = d1 - z;
  long double den = cross(da, db);
  if (std::abs(den) <= eps) {
    return std::nullopt;
  }
  long double rat = cross(z - lhs, db) / den;
  return lhs + da * rat;
}

/// @brief Return the convex hull in counter-clockwise order without repetition.
/// Starts at the lexicographically smallest point; kc retains boundary points.
/// Coordinate differences and determinant arithmetic must fit in T.
template <class T>
std::vector<point<T>> convex_hull(std::vector<point<T>> pt, bool kc = false) {
  std::sort(pt.begin(), pt.end());
  pt.erase(std::unique(pt.begin(), pt.end()), pt.end());
  if (pt.size() <= 1) {
    return pt;
  }
  if (kc) {
    bool col = true;
    for (int i = 2; i < int(pt.size()) && col; i++) {
      col = orientation(pt[0], pt[1], pt[i]) == 0;
    }
    if (col) return pt;
  }
  std::vector<point<T>> res;
  res.reserve(pt.size() + 1);
  for (const point<T> &p : pt) {
    while (res.size() >= 2) {
      int rot = orientation(res[res.size() - 2], res.back(), p);
      if (rot > 0 || (kc && rot == 0)) {
        break;
      }
      res.pop_back();
    }
    res.push_back(p);
  }
  auto k = res.size(); // Keep the lower chain while building the upper chain.
  for (int i = int(pt.size()) - 2; i >= 0; i--) {
    while (res.size() > k) {
      int rot = orientation(res[res.size() - 2], res.back(), pt[i]);
      if (rot > 0 || (kc && rot == 0)) {
        break;
      }
      res.pop_back();
    }
    res.push_back(pt[i]);
  }
  res.pop_back();
  return res;
}

/// @brief Return twice the signed area of a polygon.
/// Coordinate differences, products and the accumulated area must fit in T.
template <class T> T polygon_area2(const std::vector<point<T>> &pg) {
  T res{};
  if (pg.size() < 3) return res;
  auto prev = pg[1] - pg[0];
  for (int i = 2; i < int(pg.size()); i++) {
    auto cur = pg[i] - pg[0];
    res += cross(prev, cur);
    prev = cur;
  }
  return res;
}

/// @brief Classify a point relative to a polygon: -1 outside, 0 boundary, 1
/// inside.
template <class T>
int point_in_polygon(const point<T> &p, const std::vector<point<T>> &pg) {
  bool in = false;
  for (int i = 0; i < int(pg.size()); i++) {
    point<T> a = pg[i];
    point<T> b = pg[(i + 1) % pg.size()];
    if (on_segment(p, a, b)) {
      return 0;
    }
    if (a.y <= p.y && p.y < b.y && orientation(a, b, p) > 0) {
      in = !in;
    }
    if (b.y <= p.y && p.y < a.y && orientation(a, b, p) < 0) {
      in = !in;
    }
  }
  return in ? 1 : -1;
}

/// @brief Return the Euclidean distance between two points.
template <class T> long double distance(const point<T> &a, const point<T> &b) {
  return std::hypot(
      static_cast<long double>(a.x) - static_cast<long double>(b.x),
      static_cast<long double>(a.y) - static_cast<long double>(b.y));
}

} // namespace noya

#include <algorithm>
#include <cassert>
#include <cmath>
#include <optional>

namespace noya {

namespace geometry_distance_internal {
template <class T> point<long double> cast(const point<T> &p) {
  return {static_cast<long double>(p.x), static_cast<long double>(p.y)};
}
} // namespace geometry_distance_internal

/// @brief Project p onto the infinite line through a and b. A zero-length
/// defining pair returns nullopt. Coordinates are converted before arithmetic;
/// results use long double and are subject to floating-point precision.
template <class T>
std::optional<point<long double>>
line_projection(const point<T> &p, const point<T> &a, const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return std::nullopt;
  }
  return u + v * (dot(q, v) / len);
}

/// @brief Reflect p across the infinite line ab; nullopt for a == b.
template <class T>
std::optional<point<long double>>
line_reflection(const point<T> &p, const point<T> &a, const point<T> &b) {
  auto q = line_projection(p, a, b);
  if (!q) {
    return std::nullopt;
  }
  return *q * 2.0L - geometry_distance_internal::cast(p);
}

/// @brief Return the point-to-line distance; nullopt for a == b.
template <class T>
std::optional<long double> point_line_distance(const point<T> &p,
                                              const point<T> &a,
                                              const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = std::hypot(v.x, v.y);
  if (len == 0) {
    return std::nullopt;
  }
  return std::abs(cross(v, q)) / len;
}

/// @brief Return the nearest point on the closed segment [a, b]. A zero-length
/// segment is its single endpoint.
template <class T>
point<long double> closest_point_on_segment(const point<T> &p,
                                            const point<T> &a,
                                            const point<T> &b) {
  using geometry_distance_internal::cast;
  auto u = cast(a), v = cast(b) - u, q = cast(p) - u;
  long double len = norm2(v);
  if (len == 0) {
    return u;
  }
  long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
  return u + v * t;
}

/// @brief Return the distance from p to the closed segment [a, b].
template <class T>
long double point_segment_distance(const point<T> &p, const point<T> &a,
                                   const point<T> &b) {
  using geometry_distance_internal::cast;
  auto v = cast(b) - cast(a), q = cast(p) - cast(a);
  long double len = norm2(v);
  // Keep the residual relative to a; an absolute foot loses small distances.
  if (len != 0) {
    long double t = std::clamp(dot(q, v) / len, 0.0L, 1.0L);
    q = t == 1 ? cast(p) - cast(b) : q - v * t;
  }
  return std::hypot(q.x, q.y);
}

/// @brief Return the distance between closed segments, including point segments.
/// eps is an absolute tolerance for the intersection predicates (cross products
/// and coordinate comparisons), not a relative or angular tolerance.
template <class T>
long double segment_distance(const point<T> &a, const point<T> &b,
                             const point<T> &c, const point<T> &d,
                             long double eps = 0) {
  assert(eps >= 0);
  using geometry_distance_internal::cast;
  auto p = cast(a), q = cast(b), r = cast(c), s = cast(d);
  if (segments_intersect(p, q, r, s, eps)) {
    return 0;
  }
  return std::min({point_segment_distance(p, r, s),
                   point_segment_distance(q, r, s),
                   point_segment_distance(r, p, q),
                   point_segment_distance(s, p, q)});
}

} // namespace noya

 // NOYA_GEOMETRY_DISTANCE_HPP