Home | History | Annotate | Download | only in recovery
      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 
     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