From fe6fe8ea522899354713dea7bca5ffd7e5a7a354 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Klemens=20Sch=C3=B6lhorn?= <klemens@schoelhorn.eu>
Date: Mon, 19 Feb 2018 23:37:10 +0100
Subject: [PATCH] Disable led display after 10 seconds to save energy

Going to sleep and waking on an interrupt generated by the button would
save even more energy, but it our current chip unfortunately has problems
waking up reliably.
---
 code/code.ino | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/code/code.ino b/code/code.ino
index 04943f8..15132dc 100644
--- a/code/code.ino
+++ b/code/code.ino
@@ -48,6 +48,7 @@ void setup() {
 }
 
 enum class State : uint8_t {
+    Sleep,
     Idle,
     Menu,
     Counting,
@@ -64,6 +65,12 @@ void loop() {
     auto event = button.event(debounce.event(buttonPressed));
 
     switch(state) {
+        case State::Sleep:
+            if(buttonPressed) {
+                digitalWrite(DISPLAY_ENABLE, 0);
+                state = State::Idle;
+                timeout = millis();
+            }
         case State::Idle:
             if(event == ButtonEvent::Press) {
                 state = State::Slowing;
@@ -72,6 +79,9 @@ void loop() {
                 timeout = millis();
             } else if(event == ButtonEvent::DoublePress) {
                 state = State::Menu;
+            } else if((uint16_t) millis() - timeout >= 10000) {
+                digitalWrite(DISPLAY_ENABLE, 1);
+                state = State::Sleep;
             }
             break;
         case State::Menu:
@@ -79,6 +89,7 @@ void loop() {
             display.setDash();
             display.write();
             state = State::Idle;
+            timeout = millis();
             break;
         case State::Counting:
             if(event == ButtonEvent::LongPress) {
@@ -95,6 +106,7 @@ void loop() {
         case State::Slowing:
             slowdown();
             state = State::Idle;
+            timeout = millis();
             break;
     }
 }