core: check offset when checking constraints in infra explorer for st…

…art location Signed-off-by: Erashin <[email protected]>
OpenRailAssociation · Oct 11, 2024 · f8a85d8 · f8a85d8
1 parent f219046
commit f8a85d8
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/core/src/main/kotlin/fr/sncf/osrd/stdcm/infra_exploration/InfraExplorer.kt b/core/src/main/kotlin/fr/sncf/osrd/stdcm/infra_exploration/InfraExplorer.kt
@@ -289,7 +289,14 @@ private class InfraExplorerImpl(
             } else {
                 // If a block cannot be explored, give up
                 val isRouteBlocked =
-                    constraints.any { constraint -> constraint.apply(block).isNotEmpty() }
+                    constraints.any { constraint ->
+                        constraint.apply(block).any {
+                            if (firstLocation != null)
+                                firstLocation.offset.distance >= it.start.distance &&
+                                    firstLocation.offset.distance < it.end.distance
+                            else true
+                        }
+                    }
                 if (isRouteBlocked) return false
                 val endLocationsOnBlock = stops.last().filter { it.edge == block }
                 val endPath = endLocationsOnBlock.isNotEmpty()