From 411c10ec317464a79c353dc5c18b805cf4bdf7d0 Mon Sep 17 00:00:00 2001
From: Kieran Cawthray <kieranc@gmail.com>
Date: Wed, 20 Oct 2021 18:07:08 +0200
Subject: [PATCH] Update GetNext/GetPrevious

---
 src/displayapp/screens/PineTimeStyle.cpp | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/src/displayapp/screens/PineTimeStyle.cpp b/src/displayapp/screens/PineTimeStyle.cpp
index fa9aec1c..a2b94b5e 100644
--- a/src/displayapp/screens/PineTimeStyle.cpp
+++ b/src/displayapp/screens/PineTimeStyle.cpp
@@ -561,22 +561,13 @@ void PineTimeStyle::UpdateSelected(lv_obj_t* object, lv_event_t event) {
 Pinetime::Controllers::Settings::Colors PineTimeStyle::GetNext(Pinetime::Controllers::Settings::Colors color) {
   auto colorAsInt = static_cast<uint8_t>(color);
   Pinetime::Controllers::Settings::Colors nextColor;
-  if (colorAsInt < 16) {
-    nextColor = static_cast<Controllers::Settings::Colors>(colorAsInt + 1);
-  } else {
-    nextColor = static_cast<Controllers::Settings::Colors>(0);
-  }
+  nextColor = static_cast<Controllers::Settings::Colors>((colorAsInt + 1) % 17);
   return nextColor;
 }
 
 Pinetime::Controllers::Settings::Colors PineTimeStyle::GetPrevious(Pinetime::Controllers::Settings::Colors color) {
   auto colorAsInt = static_cast<uint8_t>(color);
   Pinetime::Controllers::Settings::Colors prevColor;
-
-  if (colorAsInt > 0) {
-    prevColor = static_cast<Controllers::Settings::Colors>(colorAsInt - 1);
-  } else {
-    prevColor = static_cast<Controllers::Settings::Colors>(16);
-  }
+  prevColor = static_cast<Controllers::Settings::Colors>((colorAsInt - 1) % 17);
   return prevColor;
 }
\ No newline at end of file