Skip to main content

elevator_core/sim/
topology.rs

1//! Dynamic topology mutation and queries.
2//!
3//! Add/remove/reassign lines, elevators, stops, and groups at runtime, plus
4//! read-only topology queries (reachability, shortest route, transfer
5//! points). Split out from `sim.rs` to keep each concern readable.
6
7use crate::components::Route;
8use crate::components::{Elevator, ElevatorPhase, Line, Position, Stop, Velocity};
9use crate::dispatch::{BuiltinStrategy, DispatchStrategy, ElevatorGroup, LineInfo};
10use crate::door::DoorState;
11use crate::entity::EntityId;
12use crate::error::SimError;
13use crate::events::Event;
14use crate::ids::GroupId;
15use crate::topology::TopologyGraph;
16
17use super::{ElevatorParams, LineParams, Simulation};
18
19impl Simulation {
20    // ── Dynamic topology ────────────────────────────────────────────
21
22    /// Mark the topology graph dirty so it is rebuilt on next query.
23    fn mark_topo_dirty(&self) {
24        if let Ok(mut g) = self.topo_graph.lock() {
25            g.mark_dirty();
26        }
27    }
28
29    /// Find the (`group_index`, `line_index`) for a line entity.
30    fn find_line(&self, line: EntityId) -> Result<(usize, usize), SimError> {
31        self.groups
32            .iter()
33            .enumerate()
34            .find_map(|(gi, g)| {
35                g.lines()
36                    .iter()
37                    .position(|li| li.entity() == line)
38                    .map(|li_idx| (gi, li_idx))
39            })
40            .ok_or(SimError::LineNotFound(line))
41    }
42
43    /// Add a new stop to a group at runtime. Returns its `EntityId`.
44    ///
45    /// Runtime-added stops have no `StopId` — they are identified purely
46    /// by `EntityId`. The `stop_lookup` (config `StopId` → `EntityId`)
47    /// is not updated.
48    ///
49    /// # Errors
50    ///
51    /// Returns [`SimError::LineNotFound`] if the line entity does not exist.
52    pub fn add_stop(
53        &mut self,
54        name: String,
55        position: f64,
56        line: EntityId,
57    ) -> Result<EntityId, SimError> {
58        if !position.is_finite() {
59            return Err(SimError::InvalidConfig {
60                field: "position",
61                reason: format!(
62                    "stop position must be finite (got {position}); NaN/±inf \
63                     corrupt SortedStops ordering and find_stop_at_position lookup"
64                ),
65            });
66        }
67
68        let group_id = self
69            .world
70            .line(line)
71            .map(|l| l.group)
72            .ok_or(SimError::LineNotFound(line))?;
73
74        let (group_idx, line_idx) = self.find_line(line)?;
75
76        let eid = self.world.spawn();
77        self.world.set_stop(eid, Stop { name, position });
78        self.world.set_position(eid, Position { value: position });
79
80        // Add to the line's serves list.
81        self.groups[group_idx].lines_mut()[line_idx]
82            .serves_mut()
83            .push(eid);
84
85        // Add to the group's flat cache.
86        self.groups[group_idx].push_stop(eid);
87
88        // Maintain sorted-stops index for O(log n) PassingFloor detection.
89        if let Some(sorted) = self.world.resource_mut::<crate::world::SortedStops>() {
90            let idx = sorted.0.partition_point(|&(p, _)| p < position);
91            sorted.0.insert(idx, (position, eid));
92        }
93
94        self.mark_topo_dirty();
95        self.events.emit(Event::StopAdded {
96            stop: eid,
97            line,
98            group: group_id,
99            tick: self.tick,
100        });
101        Ok(eid)
102    }
103
104    /// Add a new elevator to a line at runtime. Returns its `EntityId`.
105    ///
106    /// # Errors
107    ///
108    /// Returns [`SimError::LineNotFound`] if the line entity does not exist.
109    pub fn add_elevator(
110        &mut self,
111        params: &ElevatorParams,
112        line: EntityId,
113        starting_position: f64,
114    ) -> Result<EntityId, SimError> {
115        // Reject malformed params before they reach the world. Without this,
116        // zero/negative physics or zero door ticks crash later phases.
117        super::construction::validate_elevator_physics(
118            params.max_speed.value(),
119            params.acceleration.value(),
120            params.deceleration.value(),
121            params.weight_capacity.value(),
122            params.inspection_speed_factor,
123            params.door_transition_ticks,
124            params.door_open_ticks,
125            params.bypass_load_up_pct,
126            params.bypass_load_down_pct,
127        )?;
128        if !starting_position.is_finite() {
129            return Err(SimError::InvalidConfig {
130                field: "starting_position",
131                reason: format!(
132                    "must be finite (got {starting_position}); NaN/±inf corrupt \
133                     SortedStops ordering and find_stop_at_position lookup"
134                ),
135            });
136        }
137
138        let group_id = self
139            .world
140            .line(line)
141            .map(|l| l.group)
142            .ok_or(SimError::LineNotFound(line))?;
143
144        let (group_idx, line_idx) = self.find_line(line)?;
145
146        // Enforce max_cars limit.
147        if let Some(max) = self.world.line(line).and_then(Line::max_cars) {
148            let current_count = self.groups[group_idx].lines()[line_idx].elevators().len();
149            if current_count >= max {
150                return Err(SimError::InvalidConfig {
151                    field: "line.max_cars",
152                    reason: format!("line already has {current_count} cars (max {max})"),
153                });
154            }
155        }
156
157        let eid = self.world.spawn();
158        self.world.set_position(
159            eid,
160            Position {
161                value: starting_position,
162            },
163        );
164        self.world.set_velocity(eid, Velocity { value: 0.0 });
165        self.world.set_elevator(
166            eid,
167            Elevator {
168                phase: ElevatorPhase::Idle,
169                door: DoorState::Closed,
170                max_speed: params.max_speed,
171                acceleration: params.acceleration,
172                deceleration: params.deceleration,
173                weight_capacity: params.weight_capacity,
174                current_load: crate::components::Weight::ZERO,
175                riders: Vec::new(),
176                target_stop: None,
177                door_transition_ticks: params.door_transition_ticks,
178                door_open_ticks: params.door_open_ticks,
179                line,
180                repositioning: false,
181                restricted_stops: params.restricted_stops.clone(),
182                inspection_speed_factor: params.inspection_speed_factor,
183                going_up: true,
184                going_down: true,
185                move_count: 0,
186                door_command_queue: Vec::new(),
187                manual_target_velocity: None,
188                bypass_load_up_pct: params.bypass_load_up_pct,
189                bypass_load_down_pct: params.bypass_load_down_pct,
190            },
191        );
192        self.world
193            .set_destination_queue(eid, crate::components::DestinationQueue::new());
194        self.groups[group_idx].lines_mut()[line_idx]
195            .elevators_mut()
196            .push(eid);
197        self.groups[group_idx].push_elevator(eid);
198
199        // Tag the elevator with its line's "line:{name}" tag.
200        let line_name = self.world.line(line).map(|l| l.name.clone());
201        if let Some(name) = line_name
202            && let Some(tags) = self
203                .world
204                .resource_mut::<crate::tagged_metrics::MetricTags>()
205        {
206            tags.tag(eid, format!("line:{name}"));
207        }
208
209        self.mark_topo_dirty();
210        self.events.emit(Event::ElevatorAdded {
211            elevator: eid,
212            line,
213            group: group_id,
214            tick: self.tick,
215        });
216        Ok(eid)
217    }
218
219    // ── Line / group topology ───────────────────────────────────────
220
221    /// Add a new line to a group. Returns the line entity.
222    ///
223    /// # Errors
224    ///
225    /// Returns [`SimError::GroupNotFound`] if the specified group does not exist.
226    pub fn add_line(&mut self, params: &LineParams) -> Result<EntityId, SimError> {
227        let group_id = params.group;
228        let group = self
229            .groups
230            .iter_mut()
231            .find(|g| g.id() == group_id)
232            .ok_or(SimError::GroupNotFound(group_id))?;
233
234        let line_tag = format!("line:{}", params.name);
235
236        let eid = self.world.spawn();
237        self.world.set_line(
238            eid,
239            Line {
240                name: params.name.clone(),
241                group: group_id,
242                orientation: params.orientation,
243                position: params.position,
244                min_position: params.min_position,
245                max_position: params.max_position,
246                max_cars: params.max_cars,
247            },
248        );
249
250        group
251            .lines_mut()
252            .push(LineInfo::new(eid, Vec::new(), Vec::new()));
253
254        // Tag the line entity with "line:{name}" for per-line metrics.
255        if let Some(tags) = self
256            .world
257            .resource_mut::<crate::tagged_metrics::MetricTags>()
258        {
259            tags.tag(eid, line_tag);
260        }
261
262        self.mark_topo_dirty();
263        self.events.emit(Event::LineAdded {
264            line: eid,
265            group: group_id,
266            tick: self.tick,
267        });
268        Ok(eid)
269    }
270
271    /// Remove a line and all its elevators from the simulation.
272    ///
273    /// Elevators on the line are disabled (not despawned) so riders are
274    /// properly ejected to the nearest stop.
275    ///
276    /// # Errors
277    ///
278    /// Returns [`SimError::LineNotFound`] if the line entity is not found
279    /// in any group.
280    pub fn remove_line(&mut self, line: EntityId) -> Result<(), SimError> {
281        let (group_idx, line_idx) = self.find_line(line)?;
282
283        let group_id = self.groups[group_idx].id();
284
285        // Collect elevator entities to disable.
286        let elevator_ids: Vec<EntityId> = self.groups[group_idx].lines()[line_idx]
287            .elevators()
288            .to_vec();
289
290        // Disable each elevator (ejects riders properly).
291        for eid in &elevator_ids {
292            // Ignore errors from already-disabled elevators.
293            let _ = self.disable(*eid);
294        }
295
296        // Remove the LineInfo from the group.
297        self.groups[group_idx].lines_mut().remove(line_idx);
298
299        // Rebuild flat caches.
300        self.groups[group_idx].rebuild_caches();
301
302        // Remove Line component from world.
303        self.world.remove_line(line);
304
305        self.mark_topo_dirty();
306        self.events.emit(Event::LineRemoved {
307            line,
308            group: group_id,
309            tick: self.tick,
310        });
311        Ok(())
312    }
313
314    /// Remove an elevator from the simulation.
315    ///
316    /// The elevator is disabled first (ejecting any riders), then removed
317    /// from its line and despawned from the world.
318    ///
319    /// # Errors
320    ///
321    /// Returns [`SimError::EntityNotFound`] if the elevator does not exist.
322    pub fn remove_elevator(&mut self, elevator: EntityId) -> Result<(), SimError> {
323        let line = self
324            .world
325            .elevator(elevator)
326            .ok_or(SimError::EntityNotFound(elevator))?
327            .line();
328
329        // Disable first to eject riders and reset state.
330        let _ = self.disable(elevator);
331
332        // Find and remove from group/line topology. If `find_line` fails
333        // the elevator's `line` ref points at a removed/moved line — an
334        // inconsistent state, but we still want to despawn for cleanup.
335        //
336        // The `disable` call above already fired `notify_removed` on the
337        // group's dispatcher — the cache still includes the elevator at
338        // that point — so no additional notify is needed here. Custom
339        // `DispatchStrategy::notify_removed` impls that count invocations
340        // (e.g. tests with an `AtomicUsize`) can assume exactly one call
341        // per removal.
342        let resolved_group: Option<GroupId> = match self.find_line(line) {
343            Ok((group_idx, line_idx)) => {
344                self.groups[group_idx].lines_mut()[line_idx]
345                    .elevators_mut()
346                    .retain(|&e| e != elevator);
347                self.groups[group_idx].rebuild_caches();
348                Some(self.groups[group_idx].id())
349            }
350            Err(_) => None,
351        };
352
353        // Only emit ElevatorRemoved when we resolved the actual group.
354        // Pre-fix this fired with `GroupId(0)` as a sentinel, masquerading
355        // a dangling-line cleanup as a legitimate group-0 removal (#266).
356        if let Some(group_id) = resolved_group {
357            self.events.emit(Event::ElevatorRemoved {
358                elevator,
359                line,
360                group: group_id,
361                tick: self.tick,
362            });
363        }
364
365        // Despawn from world.
366        self.world.despawn(elevator);
367
368        self.mark_topo_dirty();
369        Ok(())
370    }
371
372    /// Remove a stop from the simulation.
373    ///
374    /// The stop is disabled first (invalidating routes that reference it),
375    /// then removed from all lines and despawned from the world.
376    ///
377    /// # Errors
378    ///
379    /// Returns [`SimError::EntityNotFound`] if the stop does not exist.
380    pub fn remove_stop(&mut self, stop: EntityId) -> Result<(), SimError> {
381        if self.world.stop(stop).is_none() {
382            return Err(SimError::EntityNotFound(stop));
383        }
384
385        // Warn if resident riders exist at the stop before we disable it
386        // (disabling will abandon them, clearing the residents index).
387        let residents: Vec<EntityId> = self
388            .rider_index
389            .residents_at(stop)
390            .iter()
391            .copied()
392            .collect();
393        if !residents.is_empty() {
394            self.events
395                .emit(Event::ResidentsAtRemovedStop { stop, residents });
396        }
397
398        // Disable first to invalidate routes referencing this stop.
399        let _ = self.disable(stop);
400
401        // Scrub references to the removed stop from every elevator so the
402        // post-despawn tick loop does not chase a dead EntityId through
403        // `target_stop`, the destination queue, or access-control checks.
404        let elevator_ids: Vec<EntityId> =
405            self.world.iter_elevators().map(|(eid, _, _)| eid).collect();
406        for eid in elevator_ids {
407            if let Some(car) = self.world.elevator_mut(eid) {
408                if car.target_stop == Some(stop) {
409                    car.target_stop = None;
410                }
411                car.restricted_stops.remove(&stop);
412            }
413            if let Some(q) = self.world.destination_queue_mut(eid) {
414                q.retain(|s| s != stop);
415            }
416            // Drop any car-call whose floor is the removed stop. Built-in
417            // strategies don't currently route on car_calls but the public
418            // `sim.car_calls(car)` accessor and custom strategies (via
419            // `car_calls_for`) would otherwise return dangling refs (#293).
420            if let Some(calls) = self.world.car_calls_mut(eid) {
421                calls.retain(|c| c.floor != stop);
422            }
423        }
424
425        // Remove from all lines and groups.
426        for group in &mut self.groups {
427            for line_info in group.lines_mut() {
428                line_info.serves_mut().retain(|&s| s != stop);
429            }
430            group.rebuild_caches();
431        }
432
433        // Remove from SortedStops resource.
434        if let Some(sorted) = self.world.resource_mut::<crate::world::SortedStops>() {
435            sorted.0.retain(|&(_, s)| s != stop);
436        }
437
438        // Remove from stop_lookup.
439        self.stop_lookup.retain(|_, &mut eid| eid != stop);
440
441        self.events.emit(Event::StopRemoved {
442            stop,
443            tick: self.tick,
444        });
445
446        // Despawn from world.
447        self.world.despawn(stop);
448
449        self.mark_topo_dirty();
450        Ok(())
451    }
452
453    /// Create a new dispatch group. Returns the group ID.
454    pub fn add_group(
455        &mut self,
456        name: impl Into<String>,
457        dispatch: impl DispatchStrategy + 'static,
458    ) -> GroupId {
459        let next_id = self
460            .groups
461            .iter()
462            .map(|g| g.id().0)
463            .max()
464            .map_or(0, |m| m + 1);
465        let group_id = GroupId(next_id);
466
467        self.groups
468            .push(ElevatorGroup::new(group_id, name.into(), Vec::new()));
469
470        self.dispatchers.insert(group_id, Box::new(dispatch));
471        self.strategy_ids.insert(group_id, BuiltinStrategy::Scan);
472        self.mark_topo_dirty();
473        group_id
474    }
475
476    /// Reassign a line to a different group. Returns the old `GroupId`.
477    ///
478    /// # Errors
479    ///
480    /// Returns [`SimError::LineNotFound`] if the line is not found in any group.
481    /// Returns [`SimError::GroupNotFound`] if `new_group` does not exist.
482    pub fn assign_line_to_group(
483        &mut self,
484        line: EntityId,
485        new_group: GroupId,
486    ) -> Result<GroupId, SimError> {
487        let (old_group_idx, line_idx) = self.find_line(line)?;
488
489        // Verify new group exists.
490        if !self.groups.iter().any(|g| g.id() == new_group) {
491            return Err(SimError::GroupNotFound(new_group));
492        }
493
494        let old_group_id = self.groups[old_group_idx].id();
495
496        // Same-group reassign is a no-op. Skip BEFORE the notify_removed
497        // calls or we'd needlessly clear each elevator's dispatcher state
498        // (direction tracking in SCAN/LOOK, etc.) on a redundant move.
499        // Matches the early-return pattern in `reassign_elevator_to_line`.
500        if old_group_id == new_group {
501            return Ok(old_group_id);
502        }
503
504        // Notify the old dispatcher that these elevators are leaving — its
505        // per-elevator state (e.g. ScanDispatch.direction keyed by EntityId)
506        // would otherwise leak indefinitely as lines move between groups.
507        // Mirrors the cleanup `reassign_elevator_to_line` already does. (#257)
508        let elevators_to_notify: Vec<EntityId> = self.groups[old_group_idx].lines()[line_idx]
509            .elevators()
510            .to_vec();
511        if let Some(dispatcher) = self.dispatchers.get_mut(&old_group_id) {
512            for eid in &elevators_to_notify {
513                dispatcher.notify_removed(*eid);
514            }
515        }
516
517        // Remove LineInfo from old group.
518        let line_info = self.groups[old_group_idx].lines_mut().remove(line_idx);
519        self.groups[old_group_idx].rebuild_caches();
520
521        // Re-lookup new_group_idx by ID — we didn't capture it before the
522        // mutation. (Removal of a `LineInfo` from a group's inner `lines`
523        // vec doesn't shift `self.groups` indices, so this is purely about
524        // not having stored the index earlier, not about index invalidation.)
525        let new_group_idx = self
526            .groups
527            .iter()
528            .position(|g| g.id() == new_group)
529            .ok_or(SimError::GroupNotFound(new_group))?;
530        self.groups[new_group_idx].lines_mut().push(line_info);
531        self.groups[new_group_idx].rebuild_caches();
532
533        // Update Line component's group field.
534        if let Some(line_comp) = self.world.line_mut(line) {
535            line_comp.group = new_group;
536        }
537
538        self.mark_topo_dirty();
539        self.events.emit(Event::LineReassigned {
540            line,
541            old_group: old_group_id,
542            new_group,
543            tick: self.tick,
544        });
545
546        Ok(old_group_id)
547    }
548
549    /// Reassign an elevator to a different line (swing-car pattern).
550    ///
551    /// The elevator is moved from its current line to the target line.
552    /// Both lines must be in the same group, or you must reassign the
553    /// line first via [`assign_line_to_group`](Self::assign_line_to_group).
554    ///
555    /// # Errors
556    ///
557    /// Returns [`SimError::EntityNotFound`] if the elevator does not exist.
558    /// Returns [`SimError::LineNotFound`] if the target line is not found in any group.
559    pub fn reassign_elevator_to_line(
560        &mut self,
561        elevator: EntityId,
562        new_line: EntityId,
563    ) -> Result<(), SimError> {
564        let old_line = self
565            .world
566            .elevator(elevator)
567            .ok_or(SimError::EntityNotFound(elevator))?
568            .line();
569
570        if old_line == new_line {
571            return Ok(());
572        }
573
574        // Validate both lines exist BEFORE mutating anything.
575        let (old_group_idx, old_line_idx) = self.find_line(old_line)?;
576        let (new_group_idx, new_line_idx) = self.find_line(new_line)?;
577
578        // Enforce max_cars on target line.
579        if let Some(max) = self.world.line(new_line).and_then(Line::max_cars) {
580            let current_count = self.groups[new_group_idx].lines()[new_line_idx]
581                .elevators()
582                .len();
583            if current_count >= max {
584                return Err(SimError::InvalidConfig {
585                    field: "line.max_cars",
586                    reason: format!("target line already has {current_count} cars (max {max})"),
587                });
588            }
589        }
590
591        let old_group_id = self.groups[old_group_idx].id();
592        let new_group_id = self.groups[new_group_idx].id();
593
594        self.groups[old_group_idx].lines_mut()[old_line_idx]
595            .elevators_mut()
596            .retain(|&e| e != elevator);
597        self.groups[new_group_idx].lines_mut()[new_line_idx]
598            .elevators_mut()
599            .push(elevator);
600
601        if let Some(car) = self.world.elevator_mut(elevator) {
602            car.line = new_line;
603        }
604
605        self.groups[old_group_idx].rebuild_caches();
606        if new_group_idx != old_group_idx {
607            self.groups[new_group_idx].rebuild_caches();
608
609            // Notify the old group's dispatcher so it clears per-elevator
610            // state (ScanDispatch/LookDispatch track direction by
611            // EntityId). Matches the symmetry with `remove_elevator`.
612            if let Some(old_dispatcher) = self.dispatchers.get_mut(&old_group_id) {
613                old_dispatcher.notify_removed(elevator);
614            }
615        }
616
617        self.mark_topo_dirty();
618
619        let _ = new_group_id; // reserved for symmetric notify_added once the trait gains one
620        self.events.emit(Event::ElevatorReassigned {
621            elevator,
622            old_line,
623            new_line,
624            tick: self.tick,
625        });
626
627        Ok(())
628    }
629
630    /// Add a stop to a line's served stops.
631    ///
632    /// # Errors
633    ///
634    /// Returns [`SimError::EntityNotFound`] if the stop does not exist.
635    /// Returns [`SimError::LineNotFound`] if the line is not found in any group.
636    pub fn add_stop_to_line(&mut self, stop: EntityId, line: EntityId) -> Result<(), SimError> {
637        // Verify stop exists.
638        if self.world.stop(stop).is_none() {
639            return Err(SimError::EntityNotFound(stop));
640        }
641
642        let (group_idx, line_idx) = self.find_line(line)?;
643
644        let li = &mut self.groups[group_idx].lines_mut()[line_idx];
645        if !li.serves().contains(&stop) {
646            li.serves_mut().push(stop);
647        }
648
649        self.groups[group_idx].push_stop(stop);
650
651        self.mark_topo_dirty();
652        Ok(())
653    }
654
655    /// Remove a stop from a line's served stops.
656    ///
657    /// # Errors
658    ///
659    /// Returns [`SimError::LineNotFound`] if the line is not found in any group.
660    pub fn remove_stop_from_line(
661        &mut self,
662        stop: EntityId,
663        line: EntityId,
664    ) -> Result<(), SimError> {
665        let (group_idx, line_idx) = self.find_line(line)?;
666
667        self.groups[group_idx].lines_mut()[line_idx]
668            .serves_mut()
669            .retain(|&s| s != stop);
670
671        // Rebuild group's stop_entities from all lines.
672        self.groups[group_idx].rebuild_caches();
673
674        self.mark_topo_dirty();
675        Ok(())
676    }
677
678    // ── Line / group queries ────────────────────────────────────────
679
680    /// Get all line entities across all groups.
681    #[must_use]
682    pub fn all_lines(&self) -> Vec<EntityId> {
683        self.groups
684            .iter()
685            .flat_map(|g| g.lines().iter().map(LineInfo::entity))
686            .collect()
687    }
688
689    /// Number of lines in the simulation.
690    #[must_use]
691    pub fn line_count(&self) -> usize {
692        self.groups.iter().map(|g| g.lines().len()).sum()
693    }
694
695    /// Get all line entities in a group.
696    #[must_use]
697    pub fn lines_in_group(&self, group: GroupId) -> Vec<EntityId> {
698        self.groups
699            .iter()
700            .find(|g| g.id() == group)
701            .map_or_else(Vec::new, |g| {
702                g.lines().iter().map(LineInfo::entity).collect()
703            })
704    }
705
706    /// Get elevator entities on a specific line.
707    #[must_use]
708    pub fn elevators_on_line(&self, line: EntityId) -> Vec<EntityId> {
709        self.groups
710            .iter()
711            .flat_map(ElevatorGroup::lines)
712            .find(|li| li.entity() == line)
713            .map_or_else(Vec::new, |li| li.elevators().to_vec())
714    }
715
716    /// Get stop entities served by a specific line.
717    #[must_use]
718    pub fn stops_served_by_line(&self, line: EntityId) -> Vec<EntityId> {
719        self.groups
720            .iter()
721            .flat_map(ElevatorGroup::lines)
722            .find(|li| li.entity() == line)
723            .map_or_else(Vec::new, |li| li.serves().to_vec())
724    }
725
726    /// Get the line entity for an elevator.
727    #[must_use]
728    pub fn line_for_elevator(&self, elevator: EntityId) -> Option<EntityId> {
729        self.groups
730            .iter()
731            .flat_map(ElevatorGroup::lines)
732            .find(|li| li.elevators().contains(&elevator))
733            .map(LineInfo::entity)
734    }
735
736    /// Iterate over elevators currently repositioning.
737    pub fn iter_repositioning_elevators(&self) -> impl Iterator<Item = EntityId> + '_ {
738        self.world
739            .iter_elevators()
740            .filter_map(|(id, _pos, car)| if car.repositioning() { Some(id) } else { None })
741    }
742
743    /// Get all line entities that serve a given stop.
744    #[must_use]
745    pub fn lines_serving_stop(&self, stop: EntityId) -> Vec<EntityId> {
746        self.groups
747            .iter()
748            .flat_map(ElevatorGroup::lines)
749            .filter(|li| li.serves().contains(&stop))
750            .map(LineInfo::entity)
751            .collect()
752    }
753
754    /// Get all group IDs that serve a given stop.
755    #[must_use]
756    pub fn groups_serving_stop(&self, stop: EntityId) -> Vec<GroupId> {
757        self.groups
758            .iter()
759            .filter(|g| g.stop_entities().contains(&stop))
760            .map(ElevatorGroup::id)
761            .collect()
762    }
763
764    // ── Topology queries ─────────────────────────────────────────────
765
766    /// Rebuild the topology graph if any mutation has invalidated it.
767    fn ensure_graph_built(&self) {
768        if let Ok(mut graph) = self.topo_graph.lock()
769            && graph.is_dirty()
770        {
771            graph.rebuild(&self.groups);
772        }
773    }
774
775    /// All stops reachable from a given stop through the line/group topology.
776    pub fn reachable_stops_from(&self, stop: EntityId) -> Vec<EntityId> {
777        self.ensure_graph_built();
778        self.topo_graph
779            .lock()
780            .map_or_else(|_| Vec::new(), |g| g.reachable_stops_from(stop))
781    }
782
783    /// Stops that serve as transfer points between groups.
784    pub fn transfer_points(&self) -> Vec<EntityId> {
785        self.ensure_graph_built();
786        TopologyGraph::transfer_points(&self.groups)
787    }
788
789    /// Find the shortest route between two stops, possibly spanning multiple groups.
790    pub fn shortest_route(&self, from: EntityId, to: EntityId) -> Option<Route> {
791        self.ensure_graph_built();
792        self.topo_graph
793            .lock()
794            .ok()
795            .and_then(|g| g.shortest_route(from, to))
796    }
797}