--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/rust/hwphysics/src/time.rs Thu Jul 25 19:58:19 2019 +0300
@@ -0,0 +1,96 @@
+use crate::common::{GearDataProcessor, GearId};
+use fpnum::{fp, FPNum};
+use std::{
+ cmp::{Eq, Ord, Ordering, PartialEq, PartialOrd},
+ collections::BinaryHeap,
+};
+
+pub type EventId = u16;
+
+struct TimeEvent {
+ time: FPNum,
+ gear_id: GearId,
+ event_id: EventId,
+}
+
+impl PartialOrd for TimeEvent {
+ #[inline]
+ fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
+ self.time.partial_cmp(&other.time)
+ }
+}
+
+impl PartialEq for TimeEvent {
+ #[inline]
+ fn eq(&self, other: &Self) -> bool {
+ self.time.eq(&other.time)
+ }
+}
+
+impl Ord for TimeEvent {
+ #[inline]
+ fn cmp(&self, other: &Self) -> Ordering {
+ self.time.cmp(&other.time)
+ }
+}
+
+impl Eq for TimeEvent {}
+
+pub struct OccurredEvents {
+ events: Vec<(GearId, EventId)>,
+}
+
+impl OccurredEvents {
+ fn new() -> Self {
+ Self { events: vec![] }
+ }
+
+ fn clear(&mut self) {
+ self.events.clear()
+ }
+}
+
+pub struct TimeProcessor {
+ current_event_id: EventId,
+ current_time: FPNum,
+ events: BinaryHeap<TimeEvent>,
+ timeouts: OccurredEvents,
+}
+
+impl TimeProcessor {
+ pub fn new() -> Self {
+ Self {
+ current_event_id: 0,
+ current_time: fp!(0),
+ events: BinaryHeap::with_capacity(1024),
+ timeouts: OccurredEvents::new(),
+ }
+ }
+
+ pub fn register(&mut self, gear_id: GearId, timeout: FPNum) -> EventId {
+ let event_id = self.current_event_id;
+ self.current_event_id.wrapping_add(1);
+ let event = TimeEvent {
+ time: self.current_time + timeout,
+ gear_id,
+ event_id,
+ };
+ self.events.push(event);
+ event_id
+ }
+
+ pub fn process(&mut self, time_step: FPNum) -> &OccurredEvents {
+ self.timeouts.clear();
+ self.current_time += time_step;
+ while self
+ .events
+ .peek()
+ .filter(|e| e.time <= self.current_time)
+ .is_some()
+ {
+ let event = self.events.pop().unwrap();
+ self.timeouts.events.push((event.gear_id, event.event_id))
+ }
+ &self.timeouts
+ }
+}