1use 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 fn mark_topo_dirty(&self) {
24 if let Ok(mut g) = self.topo_graph.lock() {
25 g.mark_dirty();
26 }
27 }
28
29 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 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 self.groups[group_idx].lines_mut()[line_idx]
82 .serves_mut()
83 .push(eid);
84
85 self.groups[group_idx].push_stop(eid);
87
88 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 pub fn add_elevator(
110 &mut self,
111 params: &ElevatorParams,
112 line: EntityId,
113 starting_position: f64,
114 ) -> Result<EntityId, SimError> {
115 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 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 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 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 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 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 let elevator_ids: Vec<EntityId> = self.groups[group_idx].lines()[line_idx]
287 .elevators()
288 .to_vec();
289
290 for eid in &elevator_ids {
292 let _ = self.disable(*eid);
294 }
295
296 self.groups[group_idx].lines_mut().remove(line_idx);
298
299 self.groups[group_idx].rebuild_caches();
301
302 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 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 let _ = self.disable(elevator);
331
332 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 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 self.world.despawn(elevator);
367
368 self.mark_topo_dirty();
369 Ok(())
370 }
371
372 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 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 let _ = self.disable(stop);
400
401 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 if let Some(calls) = self.world.car_calls_mut(eid) {
421 calls.retain(|c| c.floor != stop);
422 }
423 }
424
425 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 if let Some(sorted) = self.world.resource_mut::<crate::world::SortedStops>() {
435 sorted.0.retain(|&(_, s)| s != stop);
436 }
437
438 self.stop_lookup.retain(|_, &mut eid| eid != stop);
440
441 self.events.emit(Event::StopRemoved {
442 stop,
443 tick: self.tick,
444 });
445
446 self.world.despawn(stop);
448
449 self.mark_topo_dirty();
450 Ok(())
451 }
452
453 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 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 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 if old_group_id == new_group {
501 return Ok(old_group_id);
502 }
503
504 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 let line_info = self.groups[old_group_idx].lines_mut().remove(line_idx);
519 self.groups[old_group_idx].rebuild_caches();
520
521 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 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 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 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 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 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; self.events.emit(Event::ElevatorReassigned {
621 elevator,
622 old_line,
623 new_line,
624 tick: self.tick,
625 });
626
627 Ok(())
628 }
629
630 pub fn add_stop_to_line(&mut self, stop: EntityId, line: EntityId) -> Result<(), SimError> {
637 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 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 self.groups[group_idx].rebuild_caches();
673
674 self.mark_topo_dirty();
675 Ok(())
676 }
677
678 #[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 #[must_use]
691 pub fn line_count(&self) -> usize {
692 self.groups.iter().map(|g| g.lines().len()).sum()
693 }
694
695 #[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 #[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 #[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 #[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 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 #[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 #[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 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 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 pub fn transfer_points(&self) -> Vec<EntityId> {
785 self.ensure_graph_built();
786 TopologyGraph::transfer_points(&self.groups)
787 }
788
789 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}