Store coordinate in Node

onthegomap · Nov 22, 2024 · b7cb952 · b7cb952
1 parent 60a3c1c
commit b7cb952
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/planetiler-core/src/main/java/com/onthegomap/planetiler/util/LoopLineMerger.java b/planetiler-core/src/main/java/com/onthegomap/planetiler/util/LoopLineMerger.java
@@ -392,15 +392,15 @@ private void buildNodes(List<List<Coordinate>> edges) {
       Coordinate first = coordinateSequence.getFirst();
       Node firstNode = nodes.get(first);
       if (firstNode == null) {
-        firstNode = new Node();
+        firstNode = new Node(first);
         nodes.put(first, firstNode);
         output.add(firstNode);
       }
 
       Coordinate last = coordinateSequence.getLast();
       Node lastNode = nodes.get(last);
       if (lastNode == null) {
-        lastNode = new Node();
+        lastNode = new Node(last);
         nodes.put(last, lastNode);
         output.add(lastNode);
       }
@@ -456,6 +456,11 @@ private List<List<Coordinate>> nodeLines(List<LineString> input) {
   private class Node {
     final int id = nodes++;
     final List<Edge> edge = new ArrayList<>();
+    Coordinate coordinate;
+
+    public Node(Coordinate coordinate) {
+      this.coordinate = coordinate;
+    }
 
     void addEdge(Edge edge) {
       for (Edge other : this.edge) {
@@ -480,12 +485,7 @@ public String toString() {
     }
 
     public double distance(Node end) {
-      if (!getEdges().isEmpty() && !end.getEdges().isEmpty()) {
-        Coordinate a = getEdges().getFirst().coordinates.getFirst();
-        Coordinate b = end.getEdges().getFirst().coordinates.getFirst();
-        return a.distance(b);
-      }
-      return Double.POSITIVE_INFINITY;
+      return coordinate.distance(end.coordinate);
     }
   }