1 /* 2 * Copyright (C) 2011 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #ifndef RECOVERY_UI_H 18 #define RECOVERY_UI_H 19 20 #include <linux/input.h> 21 #include <pthread.h> 22 #include <time.h> 23 24 // Abstract class for controlling the user interface during recovery. 25 class RecoveryUI { 26 public: 27 RecoveryUI(); 28 ~RecoveryUI()29 virtual ~RecoveryUI() { } 30 31 // Initialize the object; called before anything else. 32 virtual void Init(); 33 // Show a stage indicator. Call immediately after Init(). 34 virtual void SetStage(int current, int max) = 0; 35 36 // After calling Init(), you can tell the UI what locale it is operating in. 37 virtual void SetLocale(const char* locale) = 0; 38 39 // Set the overall recovery state ("background image"). 40 enum Icon { NONE, INSTALLING_UPDATE, ERASING, NO_COMMAND, ERROR }; 41 virtual void SetBackground(Icon icon) = 0; 42 virtual void SetSystemUpdateText(bool security_update) = 0; 43 44 // --- progress indicator --- 45 enum ProgressType { EMPTY, INDETERMINATE, DETERMINATE }; 46 virtual void SetProgressType(ProgressType determinate) = 0; 47 48 // Show a progress bar and define the scope of the next operation: 49 // portion - fraction of the progress bar the next operation will use 50 // seconds - expected time interval (progress bar moves at this minimum rate) 51 virtual void ShowProgress(float portion, float seconds) = 0; 52 53 // Set progress bar position (0.0 - 1.0 within the scope defined 54 // by the last call to ShowProgress). 55 virtual void SetProgress(float fraction) = 0; 56 57 // --- text log --- 58 59 virtual void ShowText(bool visible) = 0; 60 61 virtual bool IsTextVisible() = 0; 62 63 virtual bool WasTextEverVisible() = 0; 64 65 // Write a message to the on-screen log (shown if the user has 66 // toggled on the text display). Print() will also dump the message 67 // to stdout / log file, while PrintOnScreenOnly() not. 68 virtual void Print(const char* fmt, ...) __printflike(2, 3) = 0; 69 virtual void PrintOnScreenOnly(const char* fmt, ...) __printflike(2, 3) = 0; 70 71 virtual void ShowFile(const char* filename) = 0; 72 73 // --- key handling --- 74 75 // Wait for a key and return it. May return -1 after timeout. 76 virtual int WaitKey(); 77 78 virtual bool IsKeyPressed(int key); 79 virtual bool IsLongPress(); 80 81 // Returns true if you have the volume up/down and power trio typical 82 // of phones and tablets, false otherwise. 83 virtual bool HasThreeButtons(); 84 85 // Erase any queued-up keys. 86 virtual void FlushKeys(); 87 88 // Called on each key press, even while operations are in progress. 89 // Return value indicates whether an immediate operation should be 90 // triggered (toggling the display, rebooting the device), or if 91 // the key should be enqueued for use by the main thread. 92 enum KeyAction { ENQUEUE, TOGGLE, REBOOT, IGNORE }; 93 virtual KeyAction CheckKey(int key, bool is_long_press); 94 95 // Called when a key is held down long enough to have been a 96 // long-press (but before the key is released). This means that 97 // if the key is eventually registered (released without any other 98 // keys being pressed in the meantime), CheckKey will be called with 99 // 'is_long_press' true. 100 virtual void KeyLongPress(int key); 101 102 // Normally in recovery there's a key sequence that triggers 103 // immediate reboot of the device, regardless of what recovery is 104 // doing (with the default CheckKey implementation, it's pressing 105 // the power button 7 times in row). Call this to enable or 106 // disable that feature. It is enabled by default. 107 virtual void SetEnableReboot(bool enabled); 108 109 // --- menu display --- 110 111 // Display some header text followed by a menu of items, which appears 112 // at the top of the screen (in place of any scrolling ui_print() 113 // output, if necessary). 114 virtual void StartMenu(const char* const * headers, const char* const * items, 115 int initial_selection) = 0; 116 117 // Set the menu highlight to the given index, wrapping if necessary. 118 // Returns the actual item selected. 119 virtual int SelectMenu(int sel) = 0; 120 121 // End menu mode, resetting the text overlay so that ui_print() 122 // statements will be displayed. 123 virtual void EndMenu() = 0; 124 125 protected: 126 void EnqueueKey(int key_code); 127 128 private: 129 // Key event input queue 130 pthread_mutex_t key_queue_mutex; 131 pthread_cond_t key_queue_cond; 132 int key_queue[256], key_queue_len; 133 char key_pressed[KEY_MAX + 1]; // under key_queue_mutex 134 int key_last_down; // under key_queue_mutex 135 bool key_long_press; // under key_queue_mutex 136 int key_down_count; // under key_queue_mutex 137 bool enable_reboot; // under key_queue_mutex 138 int rel_sum; 139 140 int consecutive_power_keys; 141 int last_key; 142 143 bool has_power_key; 144 bool has_up_key; 145 bool has_down_key; 146 147 struct key_timer_t { 148 RecoveryUI* ui; 149 int key_code; 150 int count; 151 }; 152 153 pthread_t input_thread_; 154 155 void OnKeyDetected(int key_code); 156 157 static int InputCallback(int fd, uint32_t epevents, void* data); 158 int OnInputEvent(int fd, uint32_t epevents); 159 void ProcessKey(int key_code, int updown); 160 161 bool IsUsbConnected(); 162 163 static void* time_key_helper(void* cookie); 164 void time_key(int key_code, int count); 165 }; 166 167 #endif // RECOVERY_UI_H 168