From b3e9d59041ac38706288874cd2e5572a8d6e0cf7 Mon Sep 17 00:00:00 2001
From: Lars-Dominik Braun <lars@6xq.net>
Date: Tue, 20 Jan 2015 15:40:52 +0100
Subject: timer: Fix one-shot mode

---
 timer.c | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/timer.c b/timer.c
index e4dcd4e..0ca3586 100644
--- a/timer.c
+++ b/timer.c
@@ -25,6 +25,9 @@ ISR(TIMER1_COMPA_vect) {
 		OCR1A = lastcount;
 	} else if (hits >= maxhits) {
 		enableWakeup (WAKE_TIMER);
+		if (oneshot) {
+			timerStop ();
+		}
 	}
 }
 
@@ -37,9 +40,7 @@ uint32_t timerHit () {
 			disableWakeup (WAKE_TIMER);
 
 			ret = time;
-			if (oneshot) {
-				timerStop ();
-			} else {
+			if (!oneshot) {
 				/* reset timer, start again */
 				hits = 0;
 				time = 0;
-- 
cgit v1.2.3