core: paths: add tooling for zone ranges

Signed-off-by: Eloi Charpentier <[email protected]>

Author: eckter

core/kt-osrd-path/src/main/kotlin/fr/sncf/osrd/path/implementations/TrainPathImpl.kt

@@ -34,6 +34,11 @@ data class TrainPathNoBacktrack(
 
     private val cachedEnvelopeSimPath by lazy { computeEnvelopeSimPath() }
 
+    private val cachedZonePaths by lazy {
+        assert(routes!!.isNotEmpty())
+        mapSubObjects(routes, rawInfra::getRoutePath, rawInfra::getZonePathLength)
+    }
+
     init {
         // The sanity checks here are quite exhaustive and might be expensive to compute.
         // Once the path types are stable, we can remove some of the tests.
@@ -86,6 +91,8 @@ data class TrainPathNoBacktrack(
 
     override fun getChunks(): List<DirChunkRange> = chunks
 
+    override fun getZonePaths(): List<ZonePathRange> = cachedZonePaths
+
     override val length: Double
         get() = pathProperties.getLength().meters
 

core/kt-osrd-path/src/main/kotlin/fr/sncf/osrd/path/interfaces/PathProperties.kt

@@ -28,7 +28,8 @@ sealed interface BlockPath
  * A marker type for Length and Offset. In TravelledPath, start refers to the real start of the head
  * of the train.
  */
-sealed interface TravelledPath
+// TODO path migration: remove TravelledPath entirely
+typealias TravelledPath = TrainPath
 
 @Suppress("INAPPLICABLE_JVM_NAME")
 interface PathProperties {

core/kt-osrd-path/src/main/kotlin/fr/sncf/osrd/path/interfaces/TrainPath.kt

@@ -48,13 +48,19 @@ interface TrainPath : PhysicsPath, PathProperties {
     fun getRoutes(): List<RouteRange>
 
     fun getChunks(): List<DirChunkRange>
+
+    fun getZonePaths(): List<ZonePathRange>
     // To be expanded as needed with other linear objects
 }
 
 fun concat(vararg paths: TrainPath): TrainPath {
     TODO("Required for actual backtracks, not necessary earlier than that")
 }
 
+fun TrainPath.getZones(rawInfra: RawInfra): List<ZoneRange> {
+    return getZonePaths().map { it.mapValue(rawInfra.getZonePathZone(it.value)) }
+}
+
 // Extension functions that help with backward compatibility.
 // These should only exist during the migration to enable more local changes,
 // to allow partial migration while still having a working core.

core/src/test/kotlin/fr/sncf/osrd/utils/PathUtils.kt

@@ -24,6 +24,7 @@ fun pathFromTracks(
     start: Distance,
     end: Distance,
     electricalProfileMapping: ElectricalProfileMapping? = null,
+    routeNames: List<String>? = null,
 ): TrainPath {
     val chunkList = mutableDirStaticIdxArrayListOf<TrackChunk>()
     trackIds
@@ -36,6 +37,7 @@ fun pathFromTracks(
         blockInfra,
         chunkPath,
         electricalProfileMapping = electricalProfileMapping,
+        routeNames = routeNames,
     )
 }
 
@@ -46,6 +48,7 @@ fun pathFromTracks(
     start: Distance,
     end: Distance,
     electricalProfileMapping: ElectricalProfileMapping? = null,
+    routeNames: List<String>? = null,
 ): TrainPath {
     return pathFromTracks(
         infra.rawInfra,
@@ -55,6 +58,7 @@ fun pathFromTracks(
         start,
         end,
         electricalProfileMapping = electricalProfileMapping,
+        routeNames = routeNames,
     )
 }