wear_ui.cpp 11.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * Copyright (C) 2014 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

17 18
#include "wear_ui.h"

19 20 21 22 23 24 25 26 27 28 29
#include <errno.h>
#include <fcntl.h>
#include <stdarg.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/types.h>
#include <time.h>
#include <unistd.h>

30
#include <string>
31 32
#include <vector>

Tao Bao's avatar
Tao Bao committed
33 34 35 36 37
#include <android-base/properties.h>
#include <android-base/strings.h>
#include <android-base/stringprintf.h>
#include <minui/minui.h>

38 39 40 41 42 43 44 45 46 47
#include "common.h"
#include "device.h"

// There's only (at most) one of these objects, and global callbacks
// (for pthread_create, and the input event system) need to find it,
// so use a global variable.
static WearRecoveryUI* self = NULL;

// Return the current time as a double (including fractions of a second).
static double now() {
48 49 50
  struct timeval tv;
  gettimeofday(&tv, NULL);
  return tv.tv_sec + tv.tv_usec / 1000000.0;
51 52
}

53
WearRecoveryUI::WearRecoveryUI()
54 55 56
    : kProgressBarBaseline(RECOVERY_UI_PROGRESS_BAR_BASELINE),
      kMenuUnusableRows(RECOVERY_UI_MENU_UNUSABLE_ROWS) {
  // TODO: kMenuUnusableRows should be computed based on the lines in draw_screen_locked().
57 58 59

  // TODO: The following three variables are likely not needed. The first two are detected
  // automatically in ScreenRecoveryUI::LoadAnimation(), based on the actual files seen on device.
60 61
  intro_frames = 22;
  loop_frames = 60;
62 63

  touch_screen_allowed_ = true;
64

65
  for (size_t i = 0; i < 5; i++) backgroundIcon[i] = NULL;
66

67
  self = this;
68 69
}

70
int WearRecoveryUI::GetProgressBaseline() const {
71
  return kProgressBarBaseline;
72 73
}

74 75
// Draw background frame on the screen.  Does not flip pages.
// Should only be called with updateMutex locked.
76
// TODO merge drawing routines with screen_ui
77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
void WearRecoveryUI::draw_background_locked() {
  pagesIdentical = false;
  gr_color(0, 0, 0, 255);
  gr_fill(0, 0, gr_fb_width(), gr_fb_height());

  if (currentIcon != NONE) {
    GRSurface* surface;
    if (currentIcon == INSTALLING_UPDATE || currentIcon == ERASING) {
      if (!intro_done) {
        surface = introFrames[current_frame];
      } else {
        surface = loopFrames[current_frame];
      }
    } else {
      surface = backgroundIcon[currentIcon];
    }
93

94 95
    int width = gr_get_width(surface);
    int height = gr_get_height(surface);
96

97 98
    int x = (gr_fb_width() - width) / 2;
    int y = (gr_fb_height() - height) / 2;
99

100 101
    gr_blit(surface, 0, 0, width, height, x, y);
  }
102 103
}

104
static const char* SWIPE_HELP[] = {
105 106 107 108
  "Swipe up/down to move.",
  "Swipe left/right to select.",
  "",
  NULL
109 110
};

111
// TODO merge drawing routines with screen_ui
112 113 114 115 116 117 118 119 120
void WearRecoveryUI::draw_screen_locked() {
  char cur_selection_str[50];

  draw_background_locked();
  if (!show_text) {
    draw_foreground_locked();
  } else {
    SetColor(TEXT_FILL);
    gr_fill(0, 0, gr_fb_width(), gr_fb_height());
121

122 123
    int y = kMarginHeight;
    int x = kMarginWidth;
124 125 126 127
    if (show_menu) {
      std::string recovery_fingerprint =
          android::base::GetProperty("ro.bootimage.build.fingerprint", "");
      SetColor(HEADER);
128
      y += DrawTextLine(x + 4, y, "Android Recovery", true);
129
      for (auto& chunk : android::base::Split(recovery_fingerprint, ":")) {
130
        y += DrawTextLine(x + 4, y, chunk.c_str(), false);
131
      }
132

133
      // This is actually the help strings.
134
      y += DrawTextLines(x + 4, y, SWIPE_HELP);
135
      SetColor(HEADER);
136
      y += DrawTextLines(x + 4, y, menu_headers_);
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156

      // Show the current menu item number in relation to total number if
      // items don't fit on the screen.
      if (menu_items > menu_end - menu_start) {
        sprintf(cur_selection_str, "Current item: %d/%d", menu_sel + 1, menu_items);
        gr_text(gr_sys_font(), x + 4, y, cur_selection_str, 1);
        y += char_height_ + 4;
      }

      // Menu begins here
      SetColor(MENU);

      for (int i = menu_start; i < menu_end; ++i) {
        if (i == menu_sel) {
          // draw the highlight bar
          SetColor(MENU_SEL_BG);
          gr_fill(x, y - 2, gr_fb_width() - x, y + char_height_ + 2);
          // white text of selected item
          SetColor(MENU_SEL_FG);
          if (menu_[i][0]) {
157
            gr_text(gr_sys_font(), x + 4, y, menu_[i].c_str(), 1);
158 159 160
          }
          SetColor(MENU);
        } else if (menu_[i][0]) {
161
          gr_text(gr_sys_font(), x + 4, y, menu_[i].c_str(), 0);
162
        }
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178
        y += char_height_ + 4;
      }
      SetColor(MENU);
      y += 4;
      gr_fill(0, y, gr_fb_width(), y + 2);
      y += 4;
    }

    SetColor(LOG);

    // display from the bottom up, until we hit the top of the
    // screen, the bottom of the menu, or we've displayed the
    // entire text buffer.
    int ty;
    int row = (text_top_ + text_rows_ - 1) % text_rows_;
    size_t count = 0;
179
    for (int ty = gr_fb_height() - char_height_ - kMarginHeight; ty > y + 2 && count < text_rows_;
180 181 182 183
         ty -= char_height_, ++count) {
      gr_text(gr_sys_font(), x + 4, ty, text_[row], 0);
      --row;
      if (row < 0) row = text_rows_ - 1;
184
    }
185
  }
186 187
}

188 189
// TODO merge drawing routines with screen_ui
void WearRecoveryUI::update_progress_locked() {
190 191
  draw_screen_locked();
  gr_flip();
192 193
}

Sen Jiang's avatar
Sen Jiang committed
194
bool WearRecoveryUI::InitTextParams() {
195 196 197
  if (!ScreenRecoveryUI::InitTextParams()) {
    return false;
  }
198

199
  text_cols_ = (gr_fb_width() - (kMarginWidth * 2)) / char_width_;
200

201 202
  if (text_rows_ > kMaxRows) text_rows_ = kMaxRows;
  if (text_cols_ > kMaxCols) text_cols_ = kMaxCols;
203

204
  visible_text_rows = (gr_fb_height() - (kMarginHeight * 2)) / char_height_;
205
  return true;
206
}
207

208 209 210 211
bool WearRecoveryUI::Init(const std::string& locale) {
  if (!ScreenRecoveryUI::Init(locale)) {
    return false;
  }
212

213 214
  LoadBitmap("icon_error", &backgroundIcon[ERROR]);
  backgroundIcon[NO_COMMAND] = backgroundIcon[ERROR];
215

216 217
  // This leaves backgroundIcon[INSTALLING_UPDATE] and backgroundIcon[ERASING]
  // as NULL which is fine since draw_background_locked() doesn't use them.
218

219
  return true;
220 221
}

222 223
void WearRecoveryUI::SetStage(int current, int max) {
}
224

225 226 227 228 229 230 231 232 233 234 235 236 237 238 239
void WearRecoveryUI::Print(const char* fmt, ...) {
  char buf[256];
  va_list ap;
  va_start(ap, fmt);
  vsnprintf(buf, 256, fmt, ap);
  va_end(ap);

  fputs(buf, stdout);

  // This can get called before ui_init(), so be careful.
  pthread_mutex_lock(&updateMutex);
  if (text_rows_ > 0 && text_cols_ > 0) {
    char* ptr;
    for (ptr = buf; *ptr != '\0'; ++ptr) {
      if (*ptr == '\n' || text_col_ >= text_cols_) {
240
        text_[text_row_][text_col_] = '\0';
241 242 243 244 245
        text_col_ = 0;
        text_row_ = (text_row_ + 1) % text_rows_;
        if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
      }
      if (*ptr != '\n') text_[text_row_][text_col_++] = *ptr;
246
    }
247 248 249 250
    text_[text_row_][text_col_] = '\0';
    update_screen_locked();
  }
  pthread_mutex_unlock(&updateMutex);
251 252
}

253
void WearRecoveryUI::StartMenu(const char* const* headers, const char* const* items,
254
                               int initial_selection) {
255 256 257
  pthread_mutex_lock(&updateMutex);
  if (text_rows_ > 0 && text_cols_ > 0) {
    menu_headers_ = headers;
258
    menu_.clear();
259 260 261 262 263
    // "i < text_rows_" is removed from the loop termination condition,
    // which is different from the one in ScreenRecoveryUI::StartMenu().
    // Because WearRecoveryUI supports scrollable menu, it's fine to have
    // more entries than text_rows_. The menu may be truncated otherwise.
    // Bug: 23752519
264 265
    for (size_t i = 0; items[i] != nullptr; i++) {
      menu_.emplace_back(std::string(items[i], strnlen(items[i], text_cols_ - 1)));
266
    }
267
    menu_items = static_cast<int>(menu_.size());
268 269 270
    show_menu = true;
    menu_sel = initial_selection;
    menu_start = 0;
271
    menu_end = visible_text_rows - 1 - kMenuUnusableRows;
272 273 274 275
    if (menu_items <= menu_end) menu_end = menu_items;
    update_screen_locked();
  }
  pthread_mutex_unlock(&updateMutex);
276 277 278
}

int WearRecoveryUI::SelectMenu(int sel) {
279 280 281 282 283 284 285 286 287 288 289 290 291
  int old_sel;
  pthread_mutex_lock(&updateMutex);
  if (show_menu) {
    old_sel = menu_sel;
    menu_sel = sel;
    if (menu_sel < 0) menu_sel = 0;
    if (menu_sel >= menu_items) menu_sel = menu_items - 1;
    if (menu_sel < menu_start) {
      menu_start--;
      menu_end--;
    } else if (menu_sel >= menu_end && menu_sel < menu_items) {
      menu_end++;
      menu_start++;
292
    }
293 294 295 296 297
    sel = menu_sel;
    if (menu_sel != old_sel) update_screen_locked();
  }
  pthread_mutex_unlock(&updateMutex);
  return sel;
298 299 300
}

void WearRecoveryUI::ShowFile(FILE* fp) {
301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321
  std::vector<off_t> offsets;
  offsets.push_back(ftello(fp));
  ClearText();

  struct stat sb;
  fstat(fileno(fp), &sb);

  bool show_prompt = false;
  while (true) {
    if (show_prompt) {
      Print("--(%d%% of %d bytes)--",
            static_cast<int>(100 * (double(ftello(fp)) / double(sb.st_size))),
            static_cast<int>(sb.st_size));
      Redraw();
      while (show_prompt) {
        show_prompt = false;
        int key = WaitKey();
        if (key == KEY_POWER || key == KEY_ENTER) {
          return;
        } else if (key == KEY_UP || key == KEY_VOLUMEUP) {
          if (offsets.size() <= 1) {
322
            show_prompt = true;
323 324 325 326
          } else {
            offsets.pop_back();
            fseek(fp, offsets.back(), SEEK_SET);
          }
327
        } else {
328 329 330 331
          if (feof(fp)) {
            return;
          }
          offsets.push_back(ftello(fp));
332
        }
333 334 335 336 337 338 339 340 341 342 343 344 345 346
      }
      ClearText();
    }

    int ch = getc(fp);
    if (ch == EOF) {
      text_row_ = text_top_ = text_rows_ - 2;
      show_prompt = true;
    } else {
      PutChar(ch);
      if (text_col_ == 0 && text_row_ >= text_rows_ - 2) {
        text_top_ = text_row_;
        show_prompt = true;
      }
347
    }
348
  }
349 350 351
}

void WearRecoveryUI::PutChar(char ch) {
352 353 354 355 356 357 358
  pthread_mutex_lock(&updateMutex);
  if (ch != '\n') text_[text_row_][text_col_++] = ch;
  if (ch == '\n' || text_col_ >= text_cols_) {
    text_col_ = 0;
    ++text_row_;
  }
  pthread_mutex_unlock(&updateMutex);
359 360 361
}

void WearRecoveryUI::ShowFile(const char* filename) {
362 363 364 365 366 367 368
  FILE* fp = fopen_path(filename, "re");
  if (fp == nullptr) {
    Print("  Unable to open %s: %s\n", filename, strerror(errno));
    return;
  }
  ShowFile(fp);
  fclose(fp);
369 370
}

371
void WearRecoveryUI::PrintOnScreenOnly(const char *fmt, ...) {
372 373 374 375
  va_list ap;
  va_start(ap, fmt);
  PrintV(fmt, false, ap);
  va_end(ap);
376 377 378
}

void WearRecoveryUI::PrintV(const char* fmt, bool copy_to_stdout, va_list ap) {
379 380
  std::string str;
  android::base::StringAppendV(&str, fmt, ap);
381

382 383 384
  if (copy_to_stdout) {
    fputs(str.c_str(), stdout);
  }
385

386 387 388 389
  pthread_mutex_lock(&updateMutex);
  if (text_rows_ > 0 && text_cols_ > 0) {
    for (const char* ptr = str.c_str(); *ptr != '\0'; ++ptr) {
      if (*ptr == '\n' || text_col_ >= text_cols_) {
390
        text_[text_row_][text_col_] = '\0';
391 392 393 394 395
        text_col_ = 0;
        text_row_ = (text_row_ + 1) % text_rows_;
        if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
      }
      if (*ptr != '\n') text_[text_row_][text_col_++] = *ptr;
396
    }
397 398 399 400
    text_[text_row_][text_col_] = '\0';
    update_screen_locked();
  }
  pthread_mutex_unlock(&updateMutex);
401
}