OMGRainbows

No longer will we get bad reviews on CWM recovery for its lack
of rainbows! Originally design by gweedo767, this patch brings
back the wrap count mechanism from v2.0.2.4 to toggle Rainbow
Mode for the user. In the original design, every character has
a different color, but that method requires modifying graphics.c
which will cause breakage in devices with custom graphics.c.
This Rainbow Mode has a different color for each line, instead.
The wrap_count has been set to 5 instead of the previous 3 so
the toggle won't occur as often, if on accident.

(port to CM12.1 by cyanogen)

Change-Id: Ie9f6f1b6ed94ee69d4d8d9fe27f9a5417cc3f212

improve Rainbow Mode

 * Require 5 consecutive menu "wraps" in the same direction to help
   prevent accidental toggling

 * Align colors into stripes (i.e. columns)
   a) less prone to induce dizziness in some people when they change
   b) better resembles a rainbow

 * Move the stripes to the right when the selection moves up and
   move the stripes to the left when the selection moves down

Change-Id: I3feae173b22f5703c554ca33e634881749ff54cf

1 files changed, 36 insertions, 4 deletions

diff --git a/screen_ui.cpp b/screen_ui.cpp
index f2fda2f..f23affa 100644
--- a/screen_ui.cpp
+++ b/screen_ui.cpp
@@ -76,7 +76,9 @@ ScreenRecoveryUI::ScreenRecoveryUI() :
     animation_fps(20),
     installing_frames(-1),
     stage(-1),
-    max_stage(-1) {
+    max_stage(-1),
+    rainbow(false),
+    wrap_count(0) {
 
     for (int i = 0; i < 5; i++) {
         backgroundIcon[i] = nullptr;
@@ -321,6 +323,12 @@ void* ScreenRecoveryUI::ProgressThreadStartRoutine(void* data) {
     return nullptr;
 }
 
+void ScreenRecoveryUI::OMGRainbows()
+{
+    rainbow = rainbow ? false : true;
+    set_rainbow_mode(rainbow);
+}
+
 void ScreenRecoveryUI::ProgressThreadLoop() {
     double interval = 1.0 / animation_fps;
     while (true) {
@@ -664,16 +672,40 @@ void ScreenRecoveryUI::StartMenu(const char* const * headers, const char* const
 }
 
 int ScreenRecoveryUI::SelectMenu(int sel) {
+    int wrapped = 0;
     pthread_mutex_lock(&updateMutex);
     if (show_menu) {
         int old_sel = menu_sel;
         menu_sel = sel;
 
         // Wrap at top and bottom.
-        if (menu_sel < 0) menu_sel = menu_items - 1;
-        if (menu_sel >= menu_items) menu_sel = 0;
-
+        if (rainbow) {
+            if (menu_sel > old_sel) {
+                move_rainbow(1);
+            } else if (menu_sel < old_sel) {
+                move_rainbow(-1);
+            }
+        }
+        if (menu_sel < 0) {
+            wrapped = -1;
+            menu_sel = menu_items - 1;
+        }
+        if (menu_sel >= menu_items) {
+            wrapped = 1;
+            menu_sel = 0;
+        }
         sel = menu_sel;
+        if (wrapped != 0) {
+            if (wrap_count / wrapped > 0) {
+                wrap_count += wrapped;
+            } else {
+                wrap_count = wrapped;
+            }
+            if (wrap_count / wrapped >= 5) {
+                wrap_count = 0;
+                OMGRainbows();
+            }
+        }
         if (menu_sel != old_sel) update_screen_locked();
     }
     pthread_mutex_unlock(&updateMutex);