From 48bc180211f226b0406aae28a85d5e5ee77455a2 Mon Sep 17 00:00:00 2001
From: Keith Packard <keithp@keithp.com>
Date: Mon, 22 Jun 2020 17:42:03 -0700
Subject: [PATCH] altos: Add ao_delay_until to ao_notask

Used on micropeak

Signed-off-by: Keith Packard <keithp@keithp.com>
---
 src/kernel/ao_notask.c | 19 +++++++++++--------
 src/kernel/ao_notask.h |  3 +++
 2 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/kernel/ao_notask.c b/src/kernel/ao_notask.c
index e973c7ae..8c5743fd 100644
--- a/src/kernel/ao_notask.c
+++ b/src/kernel/ao_notask.c
@@ -40,18 +40,21 @@ ao_sleep(void *wchan)
 }
 
 #if HAS_AO_DELAY
+void
+ao_delay_until(AO_TICK_TYPE target)
+{
+	ao_arch_block_interrupts();
+	while ((int16_t) (target - ao_tick_count) > 0)
+		ao_sleep((void *) &ao_tick_count);
+	ao_arch_release_interrupts();
+}
+
 void
 ao_delay(AO_TICK_TYPE ticks)
 {
-	AO_TICK_TYPE	target;
-
-	if (!ticks)
-		ticks = 1;
-	target = ao_tick_count + ticks;
-	do {
-		ao_sleep(&ao_time);
-	} while ((int16_t) (target - ao_tick_count) > 0);
+	ao_delay_until(ao_time() + ticks);
 }
+
 #endif
 
 void
diff --git a/src/kernel/ao_notask.h b/src/kernel/ao_notask.h
index 6ad15a56..f698a4ee 100644
--- a/src/kernel/ao_notask.h
+++ b/src/kernel/ao_notask.h
@@ -19,6 +19,9 @@
 #ifndef _AO_NOTASK_H_
 #define _AO_NOTASK_H_
 
+void
+ao_delay_until(AO_TICK_TYPE target);
+
 uint8_t
 ao_sleep(void *wchan);
 
-- 
2.30.2