REFACTOR: NAV-16 - Minor cleanup and tuning of generator parameters.

naviqore · May 26, 2024 · d5bcb17 · d5bcb17
1 parent ded19f1
commit d5bcb17
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/main/java/ch/naviqore/api/controller/DummyData.java b/src/main/java/ch/naviqore/api/controller/DummyData.java
@@ -364,9 +364,9 @@ public static List<EarliestArrival> getIsolines(String fromStopId, LocalDateTime
         HashMap<Stop, Leg> earliestArrivals = new HashMap<>();
 
         // find nearest stop and walk there by foot
-        List<DistanceToStop> closestStops = getNearestStops(from.getCoordinates().getLatitude(),
+        List<DistanceToStop> closestStops = getNearestStops(stops, from.getCoordinates().getLatitude(),
                 from.getCoordinates().getLongitude(), Integer.MAX_VALUE, 10);
-        Stop nearestStop = closestStops.get(0).getStop();
+        Stop nearestStop = closestStops.getFirst().getStop();
         spawnStops.add(nearestStop);
         stops.remove(nearestStop);
         earliestArrivals.put(nearestStop, buildFootpathLeg(from, nearestStop, departureTime));
@@ -388,8 +388,11 @@ public static List<EarliestArrival> getIsolines(String fromStopId, LocalDateTime
             } else {
                 spawnStop = spawnStops.removeFirst();
             }
-            int maxConnections = stops.size() / 3;
-            int numConnections = random.nextInt(maxConnections + 2);
+            int maxConnections = stops.size() / 4;
+            if (maxConnections < 2) {
+                maxConnections = 2;
+            }
+            int numConnections = random.nextInt(maxConnections);
 
             List<DistanceToStop> nearestStops = getNearestStops(stops, spawnStop.getCoordinates().getLatitude(),
                     spawnStop.getCoordinates().getLongitude(), Integer.MAX_VALUE, 5 + numConnections);