blob: bcfaaa4bef8455bc20c41670c4ff579129c4b1bb [file] [log] [blame]
Doug Zongker211aebc2011-10-28 15:13:10 -07001/*
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
Elliott Hughes498cda62016-04-14 16:49:04 -070017#include <dirent.h>
Doug Zongker211aebc2011-10-28 15:13:10 -070018#include <errno.h>
19#include <fcntl.h>
20#include <linux/input.h>
21#include <pthread.h>
22#include <stdarg.h>
23#include <stdio.h>
24#include <stdlib.h>
25#include <string.h>
26#include <sys/stat.h>
27#include <sys/time.h>
28#include <sys/types.h>
29#include <time.h>
30#include <unistd.h>
31
Tao Bao736d59c2017-01-03 10:15:33 -080032#include <string>
Elliott Hughes95fc63e2015-04-10 19:12:01 -070033#include <vector>
34
Tianjie Xuc21edd42016-08-05 18:00:04 -070035#include <android-base/logging.h>
Elliott Hughescb220402016-09-23 15:30:55 -070036#include <android-base/properties.h>
Elliott Hughes4b166f02015-12-04 15:30:20 -080037#include <android-base/strings.h>
38#include <android-base/stringprintf.h>
Tao Baob6918c72015-05-19 17:02:16 -070039
Doug Zongker211aebc2011-10-28 15:13:10 -070040#include "common.h"
Doug Zongkerdaefc1d2011-10-31 09:34:15 -070041#include "device.h"
Doug Zongker32a0a472011-11-01 11:00:20 -070042#include "minui/minui.h"
43#include "screen_ui.h"
44#include "ui.h"
Doug Zongker211aebc2011-10-28 15:13:10 -070045
Doug Zongker211aebc2011-10-28 15:13:10 -070046// Return the current time as a double (including fractions of a second).
47static double now() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070048 struct timeval tv;
49 gettimeofday(&tv, nullptr);
50 return tv.tv_sec + tv.tv_usec / 1000000.0;
Doug Zongker211aebc2011-10-28 15:13:10 -070051}
52
Tao Bao736d59c2017-01-03 10:15:33 -080053ScreenRecoveryUI::ScreenRecoveryUI()
Tao Bao4521b702017-06-20 18:11:21 -070054 : kMarginWidth(RECOVERY_UI_MARGIN_WIDTH),
55 kMarginHeight(RECOVERY_UI_MARGIN_HEIGHT),
56 density_(static_cast<float>(android::base::GetIntProperty("ro.sf.lcd_density", 160)) / 160.f),
Tao Bao171b4c42017-06-19 23:10:44 -070057 currentIcon(NONE),
Tao Bao736d59c2017-01-03 10:15:33 -080058 progressBarType(EMPTY),
59 progressScopeStart(0),
60 progressScopeSize(0),
61 progress(0),
62 pagesIdentical(false),
63 text_cols_(0),
64 text_rows_(0),
65 text_(nullptr),
66 text_col_(0),
67 text_row_(0),
68 text_top_(0),
69 show_text(false),
70 show_text_ever(false),
71 menu_(nullptr),
72 show_menu(false),
73 menu_items(0),
74 menu_sel(0),
75 file_viewer_text_(nullptr),
76 intro_frames(0),
77 loop_frames(0),
78 current_frame(0),
79 intro_done(false),
80 animation_fps(30), // TODO: there's currently no way to infer this.
81 stage(-1),
82 max_stage(-1),
83 updateMutex(PTHREAD_MUTEX_INITIALIZER) {}
Doug Zongker211aebc2011-10-28 15:13:10 -070084
Tao Bao99b2d772017-06-23 22:47:03 -070085GRSurface* ScreenRecoveryUI::GetCurrentFrame() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070086 if (currentIcon == INSTALLING_UPDATE || currentIcon == ERASING) {
87 return intro_done ? loopFrames[current_frame] : introFrames[current_frame];
88 }
89 return error_icon;
Elliott Hughes498cda62016-04-14 16:49:04 -070090}
91
Tao Bao99b2d772017-06-23 22:47:03 -070092GRSurface* ScreenRecoveryUI::GetCurrentText() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -070093 switch (currentIcon) {
94 case ERASING:
95 return erasing_text;
96 case ERROR:
97 return error_text;
98 case INSTALLING_UPDATE:
99 return installing_text;
100 case NO_COMMAND:
101 return no_command_text;
102 case NONE:
103 abort();
104 }
Elliott Hughes498cda62016-04-14 16:49:04 -0700105}
106
Mikhail Lappob49767c2017-03-23 21:44:26 +0100107int ScreenRecoveryUI::PixelsFromDp(int dp) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700108 return dp * density_;
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700109}
110
111// Here's the intended layout:
112
Elliott Hughes6d089a92016-07-08 17:23:41 -0700113// | portrait large landscape large
114// ---------+-------------------------------------------------
115// gap | 220dp 366dp 142dp 284dp
116// icon | (200dp)
117// gap | 68dp 68dp 56dp 112dp
118// text | (14sp)
119// gap | 32dp 32dp 26dp 52dp
120// progress | (2dp)
121// gap | 194dp 340dp 131dp 262dp
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700122
123// Note that "baseline" is actually the *top* of each icon (because that's how our drawing
Elliott Hughesa3691042016-04-27 17:40:11 -0700124// routines work), so that's the more useful measurement for calling code.
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700125
Elliott Hughes6d089a92016-07-08 17:23:41 -0700126enum Layout { PORTRAIT = 0, PORTRAIT_LARGE = 1, LANDSCAPE = 2, LANDSCAPE_LARGE = 3, LAYOUT_MAX };
127enum Dimension { PROGRESS = 0, TEXT = 1, ICON = 2, DIMENSION_MAX };
128static constexpr int kLayouts[LAYOUT_MAX][DIMENSION_MAX] = {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700129 { 194, 32, 68, }, // PORTRAIT
130 { 340, 32, 68, }, // PORTRAIT_LARGE
131 { 131, 26, 56, }, // LANDSCAPE
132 { 262, 52, 112, }, // LANDSCAPE_LARGE
Elliott Hughes6d089a92016-07-08 17:23:41 -0700133};
134
Tao Bao99b2d772017-06-23 22:47:03 -0700135int ScreenRecoveryUI::GetAnimationBaseline() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700136 return GetTextBaseline() - PixelsFromDp(kLayouts[layout_][ICON]) - gr_get_height(loopFrames[0]);
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700137}
138
Tao Bao99b2d772017-06-23 22:47:03 -0700139int ScreenRecoveryUI::GetTextBaseline() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700140 return GetProgressBaseline() - PixelsFromDp(kLayouts[layout_][TEXT]) -
141 gr_get_height(installing_text);
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700142}
143
Tao Bao99b2d772017-06-23 22:47:03 -0700144int ScreenRecoveryUI::GetProgressBaseline() const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700145 return gr_fb_height() - PixelsFromDp(kLayouts[layout_][PROGRESS]) -
146 gr_get_height(progressBarFill);
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700147}
148
Doug Zongker211aebc2011-10-28 15:13:10 -0700149// Clear the screen and draw the currently selected background icon (if any).
150// Should only be called with updateMutex locked.
Elliott Hughes498cda62016-04-14 16:49:04 -0700151void ScreenRecoveryUI::draw_background_locked() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700152 pagesIdentical = false;
153 gr_color(0, 0, 0, 255);
154 gr_clear();
Doug Zongker211aebc2011-10-28 15:13:10 -0700155
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700156 if (currentIcon != NONE) {
157 if (max_stage != -1) {
158 int stage_height = gr_get_height(stageMarkerEmpty);
159 int stage_width = gr_get_width(stageMarkerEmpty);
160 int x = (gr_fb_width() - max_stage * gr_get_width(stageMarkerEmpty)) / 2;
161 int y = gr_fb_height() - stage_height;
162 for (int i = 0; i < max_stage; ++i) {
163 GRSurface* stage_surface = (i < stage) ? stageMarkerFill : stageMarkerEmpty;
164 gr_blit(stage_surface, 0, 0, stage_width, stage_height, x, y);
165 x += stage_width;
166 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700167 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700168
169 GRSurface* text_surface = GetCurrentText();
170 int text_x = (gr_fb_width() - gr_get_width(text_surface)) / 2;
171 int text_y = GetTextBaseline();
172 gr_color(255, 255, 255, 255);
173 gr_texticon(text_x, text_y, text_surface);
174 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700175}
176
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700177// Draws the animation and progress bar (if any) on the screen.
178// Does not flip pages.
Doug Zongker211aebc2011-10-28 15:13:10 -0700179// Should only be called with updateMutex locked.
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700180void ScreenRecoveryUI::draw_foreground_locked() {
Tao Bao736d59c2017-01-03 10:15:33 -0800181 if (currentIcon != NONE) {
182 GRSurface* frame = GetCurrentFrame();
183 int frame_width = gr_get_width(frame);
184 int frame_height = gr_get_height(frame);
185 int frame_x = (gr_fb_width() - frame_width) / 2;
186 int frame_y = GetAnimationBaseline();
187 gr_blit(frame, 0, 0, frame_width, frame_height, frame_x, frame_y);
188 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700189
Tao Bao736d59c2017-01-03 10:15:33 -0800190 if (progressBarType != EMPTY) {
191 int width = gr_get_width(progressBarEmpty);
192 int height = gr_get_height(progressBarEmpty);
Doug Zongker211aebc2011-10-28 15:13:10 -0700193
Tao Bao736d59c2017-01-03 10:15:33 -0800194 int progress_x = (gr_fb_width() - width) / 2;
195 int progress_y = GetProgressBaseline();
Doug Zongker211aebc2011-10-28 15:13:10 -0700196
Tao Bao736d59c2017-01-03 10:15:33 -0800197 // Erase behind the progress bar (in case this was a progress-only update)
198 gr_color(0, 0, 0, 255);
199 gr_fill(progress_x, progress_y, width, height);
Doug Zongker211aebc2011-10-28 15:13:10 -0700200
Tao Bao736d59c2017-01-03 10:15:33 -0800201 if (progressBarType == DETERMINATE) {
202 float p = progressScopeStart + progress * progressScopeSize;
203 int pos = static_cast<int>(p * width);
Doug Zongker211aebc2011-10-28 15:13:10 -0700204
Tao Bao736d59c2017-01-03 10:15:33 -0800205 if (rtl_locale_) {
206 // Fill the progress bar from right to left.
207 if (pos > 0) {
208 gr_blit(progressBarFill, width - pos, 0, pos, height, progress_x + width - pos,
209 progress_y);
Doug Zongker211aebc2011-10-28 15:13:10 -0700210 }
Tao Bao736d59c2017-01-03 10:15:33 -0800211 if (pos < width - 1) {
212 gr_blit(progressBarEmpty, 0, 0, width - pos, height, progress_x, progress_y);
213 }
214 } else {
215 // Fill the progress bar from left to right.
216 if (pos > 0) {
217 gr_blit(progressBarFill, 0, 0, pos, height, progress_x, progress_y);
218 }
219 if (pos < width - 1) {
220 gr_blit(progressBarEmpty, pos, 0, width - pos, height, progress_x + pos, progress_y);
221 }
222 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700223 }
Tao Bao736d59c2017-01-03 10:15:33 -0800224 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700225}
226
Tao Bao99b2d772017-06-23 22:47:03 -0700227void ScreenRecoveryUI::SetColor(UIElement e) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700228 switch (e) {
229 case INFO:
230 gr_color(249, 194, 0, 255);
231 break;
232 case HEADER:
233 gr_color(247, 0, 6, 255);
234 break;
235 case MENU:
236 case MENU_SEL_BG:
237 gr_color(0, 106, 157, 255);
238 break;
239 case MENU_SEL_BG_ACTIVE:
240 gr_color(0, 156, 100, 255);
241 break;
242 case MENU_SEL_FG:
243 gr_color(255, 255, 255, 255);
244 break;
245 case LOG:
246 gr_color(196, 196, 196, 255);
247 break;
248 case TEXT_FILL:
249 gr_color(0, 0, 0, 160);
250 break;
251 default:
252 gr_color(255, 255, 255, 255);
253 break;
254 }
Doug Zongkerc0441d12013-07-31 11:28:24 -0700255}
Doug Zongker211aebc2011-10-28 15:13:10 -0700256
Tao Bao99b2d772017-06-23 22:47:03 -0700257void ScreenRecoveryUI::DrawHorizontalRule(int* y) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700258 SetColor(MENU);
259 *y += 4;
260 gr_fill(0, *y, gr_fb_width(), *y + 2);
261 *y += 4;
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700262}
263
Luke Songe2bd8762017-06-12 16:08:33 -0700264void ScreenRecoveryUI::DrawHighlightBar(int x, int y, int width, int height) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700265 gr_fill(x, y, x + width, y + height);
Luke Songe2bd8762017-06-12 16:08:33 -0700266}
267
Mikhail Lappob49767c2017-03-23 21:44:26 +0100268void ScreenRecoveryUI::DrawTextLine(int x, int* y, const char* line, bool bold) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700269 gr_text(gr_sys_font(), x, *y, line, bold);
270 *y += char_height_ + 4;
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700271}
272
Mikhail Lappob49767c2017-03-23 21:44:26 +0100273void ScreenRecoveryUI::DrawTextLines(int x, int* y, const char* const* lines) const {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700274 for (size_t i = 0; lines != nullptr && lines[i] != nullptr; ++i) {
275 DrawTextLine(x, y, lines[i], false);
276 }
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700277}
278
279static const char* REGULAR_HELP[] = {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700280 "Use volume up/down and power.",
281 NULL
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700282};
283
284static const char* LONG_PRESS_HELP[] = {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700285 "Any button cycles highlight.",
286 "Long-press activates.",
287 NULL
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700288};
289
Tao Bao171b4c42017-06-19 23:10:44 -0700290// Redraws everything on the screen. Does not flip pages. Should only be called with updateMutex
291// locked.
Elliott Hughes8de52072015-04-08 20:06:50 -0700292void ScreenRecoveryUI::draw_screen_locked() {
Tao Bao171b4c42017-06-19 23:10:44 -0700293 if (!show_text) {
294 draw_background_locked();
295 draw_foreground_locked();
296 return;
297 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700298
Tao Bao171b4c42017-06-19 23:10:44 -0700299 gr_color(0, 0, 0, 255);
300 gr_clear();
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700301
Tao Bao171b4c42017-06-19 23:10:44 -0700302 static constexpr int TEXT_INDENT = 4;
Tao Bao4521b702017-06-20 18:11:21 -0700303 int x = TEXT_INDENT + kMarginWidth;
304 int y = kMarginHeight;
Tao Bao171b4c42017-06-19 23:10:44 -0700305 if (show_menu) {
306 std::string recovery_fingerprint =
307 android::base::GetProperty("ro.bootimage.build.fingerprint", "");
Elliott Hughes8fd86d72015-04-13 14:36:02 -0700308
Tao Bao171b4c42017-06-19 23:10:44 -0700309 SetColor(INFO);
310 DrawTextLine(x, &y, "Android Recovery", true);
311 for (const auto& chunk : android::base::Split(recovery_fingerprint, ":")) {
312 DrawTextLine(x, &y, chunk.c_str(), false);
Doug Zongker211aebc2011-10-28 15:13:10 -0700313 }
Tao Bao171b4c42017-06-19 23:10:44 -0700314 DrawTextLines(x, &y, HasThreeButtons() ? REGULAR_HELP : LONG_PRESS_HELP);
315
316 SetColor(HEADER);
317 DrawTextLines(x, &y, menu_headers_);
318
319 SetColor(MENU);
320 DrawHorizontalRule(&y);
321 y += 4;
322 for (int i = 0; i < menu_items; ++i) {
323 if (i == menu_sel) {
324 // Draw the highlight bar.
325 SetColor(IsLongPress() ? MENU_SEL_BG_ACTIVE : MENU_SEL_BG);
326 DrawHighlightBar(0, y - 2, gr_fb_width(), char_height_ + 4);
327 // Bold white text for the selected item.
328 SetColor(MENU_SEL_FG);
329 DrawTextLine(x, &y, menu_[i], true);
330 SetColor(MENU);
331 } else {
332 DrawTextLine(x, &y, menu_[i], false);
333 }
334 }
335 DrawHorizontalRule(&y);
336 }
337
338 // Display from the bottom up, until we hit the top of the screen, the bottom of the menu, or
339 // we've displayed the entire text buffer.
340 SetColor(LOG);
341 int row = (text_top_ + text_rows_ - 1) % text_rows_;
342 size_t count = 0;
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700343 for (int ty = gr_fb_height() - kMarginHeight - char_height_; ty >= y && count < text_rows_;
344 ty -= char_height_, ++count) {
Tao Bao171b4c42017-06-19 23:10:44 -0700345 int temp_y = ty;
346 DrawTextLine(x, &temp_y, text_[row], false);
347 --row;
348 if (row < 0) row = text_rows_ - 1;
349 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700350}
351
352// Redraw everything on the screen and flip the screen (make it visible).
353// Should only be called with updateMutex locked.
Elliott Hughes8de52072015-04-08 20:06:50 -0700354void ScreenRecoveryUI::update_screen_locked() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700355 draw_screen_locked();
356 gr_flip();
Doug Zongker211aebc2011-10-28 15:13:10 -0700357}
358
359// Updates only the progress bar, if possible, otherwise redraws the screen.
360// Should only be called with updateMutex locked.
Elliott Hughes8de52072015-04-08 20:06:50 -0700361void ScreenRecoveryUI::update_progress_locked() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700362 if (show_text || !pagesIdentical) {
363 draw_screen_locked(); // Must redraw the whole screen
364 pagesIdentical = true;
365 } else {
366 draw_foreground_locked(); // Draw only the progress bar and overlays
367 }
368 gr_flip();
Doug Zongker211aebc2011-10-28 15:13:10 -0700369}
370
371// Keeps the progress bar updated, even when the process is otherwise busy.
Elliott Hughes985022a2015-04-13 13:04:32 -0700372void* ScreenRecoveryUI::ProgressThreadStartRoutine(void* data) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700373 reinterpret_cast<ScreenRecoveryUI*>(data)->ProgressThreadLoop();
374 return nullptr;
Doug Zongker32a0a472011-11-01 11:00:20 -0700375}
376
Elliott Hughes985022a2015-04-13 13:04:32 -0700377void ScreenRecoveryUI::ProgressThreadLoop() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700378 double interval = 1.0 / animation_fps;
379 while (true) {
380 double start = now();
381 pthread_mutex_lock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700382
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700383 bool redraw = false;
Doug Zongker211aebc2011-10-28 15:13:10 -0700384
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700385 // update the installation animation, if active
386 // skip this if we have a text overlay (too expensive to update)
387 if ((currentIcon == INSTALLING_UPDATE || currentIcon == ERASING) && !show_text) {
388 if (!intro_done) {
389 if (current_frame == intro_frames - 1) {
390 intro_done = true;
391 current_frame = 0;
392 } else {
393 ++current_frame;
Doug Zongker211aebc2011-10-28 15:13:10 -0700394 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700395 } else {
396 current_frame = (current_frame + 1) % loop_frames;
397 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700398
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700399 redraw = true;
Doug Zongker211aebc2011-10-28 15:13:10 -0700400 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700401
402 // move the progress bar forward on timed intervals, if configured
403 int duration = progressScopeDuration;
404 if (progressBarType == DETERMINATE && duration > 0) {
405 double elapsed = now() - progressScopeTime;
406 float p = 1.0 * elapsed / duration;
407 if (p > 1.0) p = 1.0;
408 if (p > progress) {
409 progress = p;
410 redraw = true;
411 }
412 }
413
414 if (redraw) update_progress_locked();
415
416 pthread_mutex_unlock(&updateMutex);
417 double end = now();
418 // minimum of 20ms delay between frames
419 double delay = interval - (end - start);
420 if (delay < 0.02) delay = 0.02;
421 usleep(static_cast<useconds_t>(delay * 1000000));
422 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700423}
424
Elliott Hughes0a5cb0c2015-04-15 10:58:56 -0700425void ScreenRecoveryUI::LoadBitmap(const char* filename, GRSurface** surface) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700426 int result = res_create_display_surface(filename, surface);
427 if (result < 0) {
428 LOG(ERROR) << "couldn't load bitmap " << filename << " (error " << result << ")";
429 }
Doug Zongkereac881c2014-03-07 09:21:25 -0800430}
431
Elliott Hughes0a5cb0c2015-04-15 10:58:56 -0700432void ScreenRecoveryUI::LoadLocalizedBitmap(const char* filename, GRSurface** surface) {
Tao Bao736d59c2017-01-03 10:15:33 -0800433 int result = res_create_localized_alpha_surface(filename, locale_.c_str(), surface);
434 if (result < 0) {
435 LOG(ERROR) << "couldn't load bitmap " << filename << " (error " << result << ")";
436 }
Doug Zongker02ec6b82012-08-22 17:26:40 -0700437}
438
Elliott Hughesaa0d6af2015-04-08 12:42:50 -0700439static char** Alloc2d(size_t rows, size_t cols) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700440 char** result = new char*[rows];
441 for (size_t i = 0; i < rows; ++i) {
442 result[i] = new char[cols];
443 memset(result[i], 0, cols);
444 }
445 return result;
Elliott Hughesaa0d6af2015-04-08 12:42:50 -0700446}
447
Tianjie Xu35926c42016-04-28 18:06:26 -0700448// Choose the right background string to display during update.
449void ScreenRecoveryUI::SetSystemUpdateText(bool security_update) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700450 if (security_update) {
451 LoadLocalizedBitmap("installing_security_text", &installing_text);
452 } else {
453 LoadLocalizedBitmap("installing_text", &installing_text);
454 }
455 Redraw();
Tianjie Xu35926c42016-04-28 18:06:26 -0700456}
457
Sen Jiangd5304492016-12-09 16:20:49 -0800458bool ScreenRecoveryUI::InitTextParams() {
Tao Bao171b4c42017-06-19 23:10:44 -0700459 if (gr_init() < 0) {
460 return false;
461 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700462
Tao Bao171b4c42017-06-19 23:10:44 -0700463 gr_font_size(gr_sys_font(), &char_width_, &char_height_);
Tao Bao4521b702017-06-20 18:11:21 -0700464 text_rows_ = (gr_fb_height() - kMarginHeight * 2) / char_height_;
465 text_cols_ = (gr_fb_width() - kMarginWidth * 2) / char_width_;
Tao Bao171b4c42017-06-19 23:10:44 -0700466 return true;
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700467}
468
Tao Bao736d59c2017-01-03 10:15:33 -0800469bool ScreenRecoveryUI::Init(const std::string& locale) {
470 RecoveryUI::Init(locale);
471 if (!InitTextParams()) {
472 return false;
473 }
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700474
Tao Bao736d59c2017-01-03 10:15:33 -0800475 // Are we portrait or landscape?
476 layout_ = (gr_fb_width() > gr_fb_height()) ? LANDSCAPE : PORTRAIT;
477 // Are we the large variant of our base layout?
478 if (gr_fb_height() > PixelsFromDp(800)) ++layout_;
Elliott Hughesfaf36e02016-04-20 17:22:16 -0700479
Tao Bao736d59c2017-01-03 10:15:33 -0800480 text_ = Alloc2d(text_rows_, text_cols_ + 1);
481 file_viewer_text_ = Alloc2d(text_rows_, text_cols_ + 1);
482 menu_ = Alloc2d(text_rows_, text_cols_ + 1);
Doug Zongker55a36ac2013-03-04 15:49:02 -0800483
Tao Bao736d59c2017-01-03 10:15:33 -0800484 text_col_ = text_row_ = 0;
485 text_top_ = 1;
Doug Zongker211aebc2011-10-28 15:13:10 -0700486
Tao Bao736d59c2017-01-03 10:15:33 -0800487 LoadBitmap("icon_error", &error_icon);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700488
Tao Bao736d59c2017-01-03 10:15:33 -0800489 LoadBitmap("progress_empty", &progressBarEmpty);
490 LoadBitmap("progress_fill", &progressBarFill);
Elliott Hughes498cda62016-04-14 16:49:04 -0700491
Tao Bao736d59c2017-01-03 10:15:33 -0800492 LoadBitmap("stage_empty", &stageMarkerEmpty);
493 LoadBitmap("stage_fill", &stageMarkerFill);
Doug Zongker211aebc2011-10-28 15:13:10 -0700494
Tao Bao736d59c2017-01-03 10:15:33 -0800495 // Background text for "installing_update" could be "installing update"
496 // or "installing security update". It will be set after UI init according
497 // to commands in BCB.
498 installing_text = nullptr;
499 LoadLocalizedBitmap("erasing_text", &erasing_text);
500 LoadLocalizedBitmap("no_command_text", &no_command_text);
501 LoadLocalizedBitmap("error_text", &error_text);
Elliott Hughes498cda62016-04-14 16:49:04 -0700502
Tao Bao736d59c2017-01-03 10:15:33 -0800503 LoadAnimation();
Doug Zongker02ec6b82012-08-22 17:26:40 -0700504
Tao Bao736d59c2017-01-03 10:15:33 -0800505 pthread_create(&progress_thread_, nullptr, ProgressThreadStartRoutine, this);
Sen Jiangd5304492016-12-09 16:20:49 -0800506
Tao Bao736d59c2017-01-03 10:15:33 -0800507 return true;
Doug Zongker211aebc2011-10-28 15:13:10 -0700508}
509
Elliott Hughes498cda62016-04-14 16:49:04 -0700510void ScreenRecoveryUI::LoadAnimation() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700511 std::unique_ptr<DIR, decltype(&closedir)> dir(opendir("/res/images"), closedir);
512 dirent* de;
513 std::vector<std::string> intro_frame_names;
514 std::vector<std::string> loop_frame_names;
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700515
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700516 while ((de = readdir(dir.get())) != nullptr) {
517 int value, num_chars;
518 if (sscanf(de->d_name, "intro%d%n.png", &value, &num_chars) == 1) {
519 intro_frame_names.emplace_back(de->d_name, num_chars);
520 } else if (sscanf(de->d_name, "loop%d%n.png", &value, &num_chars) == 1) {
521 loop_frame_names.emplace_back(de->d_name, num_chars);
Elliott Hughes498cda62016-04-14 16:49:04 -0700522 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700523 }
Elliott Hughes498cda62016-04-14 16:49:04 -0700524
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700525 intro_frames = intro_frame_names.size();
526 loop_frames = loop_frame_names.size();
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700527
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700528 // It's okay to not have an intro.
529 if (intro_frames == 0) intro_done = true;
530 // But you must have an animation.
531 if (loop_frames == 0) abort();
Elliott Hughes498cda62016-04-14 16:49:04 -0700532
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700533 std::sort(intro_frame_names.begin(), intro_frame_names.end());
534 std::sort(loop_frame_names.begin(), loop_frame_names.end());
Damien Bargiacchi5e7cfb92016-08-24 18:28:43 -0700535
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700536 introFrames = new GRSurface*[intro_frames];
537 for (size_t i = 0; i < intro_frames; i++) {
538 LoadBitmap(intro_frame_names.at(i).c_str(), &introFrames[i]);
539 }
Elliott Hughes498cda62016-04-14 16:49:04 -0700540
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700541 loopFrames = new GRSurface*[loop_frames];
542 for (size_t i = 0; i < loop_frames; i++) {
543 LoadBitmap(loop_frame_names.at(i).c_str(), &loopFrames[i]);
544 }
Elliott Hughes498cda62016-04-14 16:49:04 -0700545}
546
Elliott Hughes8de52072015-04-08 20:06:50 -0700547void ScreenRecoveryUI::SetBackground(Icon icon) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700548 pthread_mutex_lock(&updateMutex);
Doug Zongker02ec6b82012-08-22 17:26:40 -0700549
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700550 currentIcon = icon;
551 update_screen_locked();
Doug Zongker52eeea4f2012-09-04 14:28:25 -0700552
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700553 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700554}
555
Elliott Hughes8de52072015-04-08 20:06:50 -0700556void ScreenRecoveryUI::SetProgressType(ProgressType type) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700557 pthread_mutex_lock(&updateMutex);
558 if (progressBarType != type) {
559 progressBarType = type;
560 }
561 progressScopeStart = 0;
562 progressScopeSize = 0;
563 progress = 0;
564 update_progress_locked();
565 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700566}
567
Elliott Hughes8de52072015-04-08 20:06:50 -0700568void ScreenRecoveryUI::ShowProgress(float portion, float seconds) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700569 pthread_mutex_lock(&updateMutex);
570 progressBarType = DETERMINATE;
571 progressScopeStart += progressScopeSize;
572 progressScopeSize = portion;
573 progressScopeTime = now();
574 progressScopeDuration = seconds;
575 progress = 0;
576 update_progress_locked();
577 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700578}
579
Elliott Hughes8de52072015-04-08 20:06:50 -0700580void ScreenRecoveryUI::SetProgress(float fraction) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700581 pthread_mutex_lock(&updateMutex);
582 if (fraction < 0.0) fraction = 0.0;
583 if (fraction > 1.0) fraction = 1.0;
584 if (progressBarType == DETERMINATE && fraction > progress) {
585 // Skip updates that aren't visibly different.
586 int width = gr_get_width(progressBarEmpty);
587 float scale = width * progressScopeSize;
588 if ((int)(progress * scale) != (int)(fraction * scale)) {
589 progress = fraction;
590 update_progress_locked();
Doug Zongker211aebc2011-10-28 15:13:10 -0700591 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700592 }
593 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700594}
595
Doug Zongkerc87bab12013-11-25 13:53:25 -0800596void ScreenRecoveryUI::SetStage(int current, int max) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700597 pthread_mutex_lock(&updateMutex);
598 stage = current;
599 max_stage = max;
600 pthread_mutex_unlock(&updateMutex);
Doug Zongkerc87bab12013-11-25 13:53:25 -0800601}
602
Tao Baob6918c72015-05-19 17:02:16 -0700603void ScreenRecoveryUI::PrintV(const char* fmt, bool copy_to_stdout, va_list ap) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700604 std::string str;
605 android::base::StringAppendV(&str, fmt, ap);
Doug Zongker211aebc2011-10-28 15:13:10 -0700606
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700607 if (copy_to_stdout) {
608 fputs(str.c_str(), stdout);
609 }
Doug Zongker211aebc2011-10-28 15:13:10 -0700610
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700611 pthread_mutex_lock(&updateMutex);
612 if (text_rows_ > 0 && text_cols_ > 0) {
613 for (const char* ptr = str.c_str(); *ptr != '\0'; ++ptr) {
614 if (*ptr == '\n' || text_col_ >= text_cols_) {
Elliott Hughesc0491632015-05-06 12:40:05 -0700615 text_[text_row_][text_col_] = '\0';
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700616 text_col_ = 0;
617 text_row_ = (text_row_ + 1) % text_rows_;
618 if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
619 }
620 if (*ptr != '\n') text_[text_row_][text_col_++] = *ptr;
Doug Zongker211aebc2011-10-28 15:13:10 -0700621 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700622 text_[text_row_][text_col_] = '\0';
623 update_screen_locked();
624 }
625 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700626}
627
Tao Baob6918c72015-05-19 17:02:16 -0700628void ScreenRecoveryUI::Print(const char* fmt, ...) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700629 va_list ap;
630 va_start(ap, fmt);
631 PrintV(fmt, true, ap);
632 va_end(ap);
Tao Baob6918c72015-05-19 17:02:16 -0700633}
634
635void ScreenRecoveryUI::PrintOnScreenOnly(const char *fmt, ...) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700636 va_list ap;
637 va_start(ap, fmt);
638 PrintV(fmt, false, ap);
639 va_end(ap);
Tao Baob6918c72015-05-19 17:02:16 -0700640}
641
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700642void ScreenRecoveryUI::PutChar(char ch) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700643 pthread_mutex_lock(&updateMutex);
644 if (ch != '\n') text_[text_row_][text_col_++] = ch;
645 if (ch == '\n' || text_col_ >= text_cols_) {
646 text_col_ = 0;
647 ++text_row_;
Elliott Hughesc0491632015-05-06 12:40:05 -0700648
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700649 if (text_row_ == text_top_) text_top_ = (text_top_ + 1) % text_rows_;
650 }
651 pthread_mutex_unlock(&updateMutex);
Elliott Hughes8de52072015-04-08 20:06:50 -0700652}
653
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700654void ScreenRecoveryUI::ClearText() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700655 pthread_mutex_lock(&updateMutex);
656 text_col_ = 0;
657 text_row_ = 0;
658 text_top_ = 1;
659 for (size_t i = 0; i < text_rows_; ++i) {
660 memset(text_[i], 0, text_cols_ + 1);
661 }
662 pthread_mutex_unlock(&updateMutex);
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700663}
664
665void ScreenRecoveryUI::ShowFile(FILE* fp) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700666 std::vector<off_t> offsets;
667 offsets.push_back(ftello(fp));
668 ClearText();
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700669
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700670 struct stat sb;
671 fstat(fileno(fp), &sb);
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700672
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700673 bool show_prompt = false;
674 while (true) {
675 if (show_prompt) {
676 PrintOnScreenOnly("--(%d%% of %d bytes)--",
677 static_cast<int>(100 * (double(ftello(fp)) / double(sb.st_size))),
678 static_cast<int>(sb.st_size));
679 Redraw();
680 while (show_prompt) {
681 show_prompt = false;
682 int key = WaitKey();
683 if (key == KEY_POWER || key == KEY_ENTER) {
684 return;
685 } else if (key == KEY_UP || key == KEY_VOLUMEUP) {
686 if (offsets.size() <= 1) {
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700687 show_prompt = true;
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700688 } else {
689 offsets.pop_back();
690 fseek(fp, offsets.back(), SEEK_SET);
691 }
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700692 } else {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700693 if (feof(fp)) {
694 return;
695 }
696 offsets.push_back(ftello(fp));
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700697 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700698 }
699 ClearText();
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700700 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700701
702 int ch = getc(fp);
703 if (ch == EOF) {
704 while (text_row_ < text_rows_ - 1) PutChar('\n');
705 show_prompt = true;
706 } else {
707 PutChar(ch);
708 if (text_col_ == 0 && text_row_ >= text_rows_ - 1) {
709 show_prompt = true;
710 }
711 }
712 }
Elliott Hughes95fc63e2015-04-10 19:12:01 -0700713}
714
Elliott Hughes8de52072015-04-08 20:06:50 -0700715void ScreenRecoveryUI::ShowFile(const char* filename) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700716 FILE* fp = fopen_path(filename, "re");
717 if (fp == nullptr) {
718 Print(" Unable to open %s: %s\n", filename, strerror(errno));
719 return;
720 }
Elliott Hughesc0491632015-05-06 12:40:05 -0700721
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700722 char** old_text = text_;
723 size_t old_text_col = text_col_;
724 size_t old_text_row = text_row_;
725 size_t old_text_top = text_top_;
Elliott Hughesc0491632015-05-06 12:40:05 -0700726
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700727 // Swap in the alternate screen and clear it.
728 text_ = file_viewer_text_;
729 ClearText();
Elliott Hughesc0491632015-05-06 12:40:05 -0700730
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700731 ShowFile(fp);
732 fclose(fp);
Elliott Hughesc0491632015-05-06 12:40:05 -0700733
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700734 text_ = old_text;
735 text_col_ = old_text_col;
736 text_row_ = old_text_row;
737 text_top_ = old_text_top;
Elliott Hughes8de52072015-04-08 20:06:50 -0700738}
739
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700740void ScreenRecoveryUI::StartMenu(const char* const* headers, const char* const* items,
Doug Zongker211aebc2011-10-28 15:13:10 -0700741 int initial_selection) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700742 pthread_mutex_lock(&updateMutex);
743 if (text_rows_ > 0 && text_cols_ > 0) {
744 menu_headers_ = headers;
745 size_t i = 0;
746 for (; i < text_rows_ && items[i] != nullptr; ++i) {
747 strncpy(menu_[i], items[i], text_cols_ - 1);
748 menu_[i][text_cols_ - 1] = '\0';
Doug Zongker211aebc2011-10-28 15:13:10 -0700749 }
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700750 menu_items = i;
751 show_menu = true;
752 menu_sel = initial_selection;
753 update_screen_locked();
754 }
755 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700756}
757
758int ScreenRecoveryUI::SelectMenu(int sel) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700759 pthread_mutex_lock(&updateMutex);
760 if (show_menu) {
761 int old_sel = menu_sel;
762 menu_sel = sel;
Elliott Hughesfc06f872015-03-23 13:45:31 -0700763
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700764 // Wrap at top and bottom.
765 if (menu_sel < 0) menu_sel = menu_items - 1;
766 if (menu_sel >= menu_items) menu_sel = 0;
Elliott Hughesfc06f872015-03-23 13:45:31 -0700767
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700768 sel = menu_sel;
769 if (menu_sel != old_sel) update_screen_locked();
770 }
771 pthread_mutex_unlock(&updateMutex);
772 return sel;
Doug Zongker211aebc2011-10-28 15:13:10 -0700773}
774
775void ScreenRecoveryUI::EndMenu() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700776 pthread_mutex_lock(&updateMutex);
777 if (show_menu && text_rows_ > 0 && text_cols_ > 0) {
778 show_menu = false;
779 update_screen_locked();
780 }
781 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700782}
783
Elliott Hughes8de52072015-04-08 20:06:50 -0700784bool ScreenRecoveryUI::IsTextVisible() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700785 pthread_mutex_lock(&updateMutex);
786 int visible = show_text;
787 pthread_mutex_unlock(&updateMutex);
788 return visible;
Doug Zongker211aebc2011-10-28 15:13:10 -0700789}
790
Elliott Hughes8de52072015-04-08 20:06:50 -0700791bool ScreenRecoveryUI::WasTextEverVisible() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700792 pthread_mutex_lock(&updateMutex);
793 int ever_visible = show_text_ever;
794 pthread_mutex_unlock(&updateMutex);
795 return ever_visible;
Doug Zongker211aebc2011-10-28 15:13:10 -0700796}
797
Elliott Hughes8de52072015-04-08 20:06:50 -0700798void ScreenRecoveryUI::ShowText(bool visible) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700799 pthread_mutex_lock(&updateMutex);
800 show_text = visible;
801 if (show_text) show_text_ever = true;
802 update_screen_locked();
803 pthread_mutex_unlock(&updateMutex);
Doug Zongker211aebc2011-10-28 15:13:10 -0700804}
Doug Zongkerc0441d12013-07-31 11:28:24 -0700805
Elliott Hughes8de52072015-04-08 20:06:50 -0700806void ScreenRecoveryUI::Redraw() {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700807 pthread_mutex_lock(&updateMutex);
808 update_screen_locked();
809 pthread_mutex_unlock(&updateMutex);
Doug Zongkerc0441d12013-07-31 11:28:24 -0700810}
Elliott Hughes642aaa72015-04-10 12:47:46 -0700811
812void ScreenRecoveryUI::KeyLongPress(int) {
Tao Bao5d2e3bd2017-06-23 22:23:50 -0700813 // Redraw so that if we're in the menu, the highlight
814 // will change color to indicate a successful long press.
815 Redraw();
Elliott Hughes642aaa72015-04-10 12:47:46 -0700816}