#23 Keep length of path == 1 if both path ends are the same CP.

OpenBW · Mar 4, 2018 · c4678ec · c4678ec
1 parent c71393a
commit c4678ec
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/BWAPI4J/src/main/java/bwem/Graph.java b/BWAPI4J/src/main/java/bwem/Graph.java
@@ -776,10 +776,12 @@ private void setDistance(final ChokePoint cpA, final ChokePoint cpB, final int v
     private void setPath(final ChokePoint cpA, final ChokePoint cpB, final CPPath pathAB) {
         pathsBetweenChokePoints.get(cpA.getIndex().intValue()).set(cpB.getIndex().intValue(), pathAB);
 
-        pathsBetweenChokePoints.get(cpB.getIndex().intValue()).get(cpA.getIndex().intValue()).clear();
-        for (int i = pathAB.size() - 1; i >= 0; --i) {
-            final ChokePoint cp = pathAB.get(i);
-            pathsBetweenChokePoints.get(cpB.getIndex().intValue()).get(cpA.getIndex().intValue()).add(cp);
+        if (cpA != cpB) {
+            pathsBetweenChokePoints.get(cpB.getIndex().intValue()).get(cpA.getIndex().intValue()).clear();
+            for (int i = pathAB.size() - 1; i >= 0; --i) {
+                final ChokePoint cp = pathAB.get(i);
+                pathsBetweenChokePoints.get(cpB.getIndex().intValue()).get(cpA.getIndex().intValue()).add(cp);
+            }
         }
     }