pub fn ax_sleep_until(deadline: crate::time::AxTimeValue) {
#[cfg(feature = "multitask")]
axtask::sleep_until(deadline);
#[cfg(not(feature = "multitask"))]
axhal::time::busy_wait_until(deadline);
}
pub fn ax_yield_now() {
#[cfg(feature = "multitask")]
axtask::yield_now();
#[cfg(not(feature = "multitask"))]
if cfg!(feature = "irq") {
axhal::arch::wait_for_irqs();
} else {
core::hint::spin_loop();
}
}
pub fn ax_exit(_exit_code: i32) -> ! {
#[cfg(feature = "multitask")]
axtask::exit(_exit_code);
#[cfg(not(feature = "multitask"))]
axhal::misc::terminate();
}
cfg_task! {
use core::time::Duration;
pub struct AxTaskHandle {
inner: axtask::AxTaskRef,
id: u64,
}
impl AxTaskHandle {
pub fn id(&self) -> u64 {
self.id
}
}
pub struct AxWaitQueueHandle(axtask::WaitQueue);
impl AxWaitQueueHandle {
pub const fn new() -> Self {
Self(axtask::WaitQueue::new())
}
}
pub fn ax_current_task_id() -> u64 {
axtask::current().id().as_u64()
}
pub fn ax_spawn<F>(f: F, name: alloc::string::String, stack_size: usize) -> AxTaskHandle
where
F: FnOnce() + Send + 'static,
{
let inner = axtask::spawn_raw(f, name, stack_size);
AxTaskHandle {
id: inner.id().as_u64(),
inner,
}
}
pub fn ax_wait_for_exit(task: AxTaskHandle) -> Option<i32> {
axtask::join(&task.inner)
}
pub fn ax_set_current_priority(prio: isize) -> crate::AxResult {
if axtask::set_priority(prio) {
Ok(())
} else {
axerrno::ax_err!(
BadState,
"ax_set_current_priority: failed to set task priority"
)
}
}
pub fn ax_wait_queue_wait(
wq: &AxWaitQueueHandle,
until_condition: impl Fn() -> bool,
timeout: Option<Duration>,
) -> bool {
#[cfg(feature = "irq")]
if let Some(dur) = timeout {
return wq.0.wait_timeout_until(dur, until_condition);
}
if timeout.is_some() {
axlog::warn!("ax_wait_queue_wait: the `timeout` argument is ignored without the `irq` feature");
}
wq.0.wait_until(until_condition);
false
}
pub fn ax_wait_queue_wake(wq: &AxWaitQueueHandle, count: u32) {
if count == u32::MAX {
wq.0.notify_all(true);
} else {
for _ in 0..count {
wq.0.notify_one(true);
}
}
}
}