Link helper to get an End pointer based on position

2 files changed, 14 insertions, 0 deletions

diff --git a/game/network/link.cpp b/game/network/link.cpp
index b8ffee2..1a4bb89 100644
--- a/game/network/link.cpp
+++ b/game/network/link.cpp
@@ -7,6 +7,18 @@
 
 Link::Link(End endA, End endB, float len) : ends {{std::move(endA), std::move(endB)}}, length {len} { }
 
+const Link::End *
+Link::endAt(const GlobalPosition3D & queryPos) const
+{
+	if (ends.front().node->pos == queryPos) {
+		return &ends.front();
+	}
+	if (ends.back().node->pos == queryPos) {
+		return &ends.back();
+	}
+	return nullptr;
+}
+
 LinkCurve::LinkCurve(GlobalPosition3D centre, RelativeDistance radius, Arc arc) :
 	centreBase {centre}, radius {radius}, arc {std::move(arc)}
 {
diff --git a/game/network/link.h b/game/network/link.h
index 0b58558..7ef1387 100644
--- a/game/network/link.h
+++ b/game/network/link.h
@@ -48,6 +48,8 @@ public:
 	[[nodiscard]] virtual bool intersectRay(const Ray<GlobalPosition3D> &) const = 0;
 	[[nodiscard]] virtual std::vector<GlobalPosition3D> getBase(RelativeDistance width) const = 0;
 
+	[[nodiscard]] const End * endAt(const GlobalPosition3D &) const;
+
 	std::array<End, 2> ends;
 	float length;