mirror of https://github.com/m-labs/artiq
48 lines
1.1 KiB
Rust
48 lines
1.1 KiB
Rust
use board::clock;
|
|
|
|
#[derive(Debug, Clone, Copy)]
|
|
struct Watchdog {
|
|
active: bool,
|
|
threshold: u64
|
|
}
|
|
|
|
pub const MAX_WATCHDOGS: usize = 16;
|
|
|
|
#[derive(Debug)]
|
|
pub struct WatchdogSet {
|
|
watchdogs: [Watchdog; MAX_WATCHDOGS]
|
|
}
|
|
|
|
impl WatchdogSet {
|
|
pub fn new() -> WatchdogSet {
|
|
WatchdogSet {
|
|
watchdogs: [Watchdog { active: false, threshold: 0 }; MAX_WATCHDOGS]
|
|
}
|
|
}
|
|
|
|
pub fn set_ms(&mut self, interval: u64) -> Result<usize, ()> {
|
|
for (index, watchdog) in self.watchdogs.iter_mut().enumerate() {
|
|
if !watchdog.active {
|
|
watchdog.active = true;
|
|
watchdog.threshold = clock::get_ms() + interval;
|
|
return Ok(index)
|
|
}
|
|
}
|
|
|
|
Err(())
|
|
}
|
|
|
|
pub fn clear(&mut self, index: usize) {
|
|
if index < MAX_WATCHDOGS {
|
|
self.watchdogs[index].active = false
|
|
}
|
|
}
|
|
|
|
pub fn expired(&self) -> bool {
|
|
self.watchdogs.iter()
|
|
.filter(|wd| wd.active)
|
|
.min_by_key(|wd| wd.threshold)
|
|
.map_or(false, |wd| clock::get_ms() > wd.threshold)
|
|
}
|
|
}
|