You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
769 lines
23 KiB
769 lines
23 KiB
use crate::{
|
|
strutil::TString,
|
|
trezorhal::{
|
|
buffers::{
|
|
BufferBlurring, BufferBlurringTotals, BufferJpeg, BufferLine16bpp, BufferLine4bpp,
|
|
BufferText,
|
|
},
|
|
display,
|
|
dma2d::{dma2d_setup_4bpp_over_16bpp, dma2d_start_blend, dma2d_wait_for_transfer},
|
|
uzlib::UzlibContext,
|
|
},
|
|
ui::{
|
|
component::text::TextStyle,
|
|
constant::{screen, HEIGHT, WIDTH},
|
|
display::{
|
|
position_buffer, rect_fill_rounded_buffer, set_window,
|
|
tjpgd::{BufferInput, BufferOutput, JDEC},
|
|
Color, Icon,
|
|
},
|
|
geometry::{Offset, Point, Rect},
|
|
model_tt::theme,
|
|
util::icon_text_center,
|
|
},
|
|
};
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub struct HomescreenText<'a> {
|
|
pub text: TString<'a>,
|
|
pub style: TextStyle,
|
|
pub offset: Offset,
|
|
pub icon: Option<Icon>,
|
|
}
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub struct HomescreenNotification {
|
|
pub text: TString<'static>,
|
|
pub icon: Icon,
|
|
pub color: Color,
|
|
}
|
|
|
|
#[derive(Clone, Copy)]
|
|
struct HomescreenTextInfo {
|
|
pub text_area: Rect,
|
|
pub text_width: i16,
|
|
pub text_color: Color,
|
|
pub icon_area: Option<Rect>,
|
|
}
|
|
|
|
pub const HOMESCREEN_IMAGE_WIDTH: i16 = WIDTH;
|
|
pub const HOMESCREEN_IMAGE_HEIGHT: i16 = HEIGHT;
|
|
pub const HOMESCREEN_TOIF_SIZE: i16 = 144;
|
|
pub const HOMESCREEN_TOIF_Y_OFFSET: i16 = 27;
|
|
pub const HOMESCREEN_TOIF_X_OFFSET: usize =
|
|
((WIDTH.saturating_sub(HOMESCREEN_TOIF_SIZE)) / 2) as usize;
|
|
|
|
const HOMESCREEN_MAX_ICON_SIZE: i16 = 20;
|
|
const NOTIFICATION_HEIGHT: i16 = 36;
|
|
const NOTIFICATION_BORDER: i16 = 6;
|
|
const TEXT_ICON_SPACE: i16 = 2;
|
|
|
|
const HOMESCREEN_DIM_HEIGHT: i16 = 35;
|
|
const HOMESCREEN_DIM_START: i16 = HOMESCREEN_IMAGE_HEIGHT - 42;
|
|
const HOMESCREEN_DIM: f32 = 0.65;
|
|
const HOMESCREEN_DIM_BORDER: i16 = theme::BUTTON_SPACING;
|
|
|
|
const LOCKSCREEN_DIM: f32 = 0.55;
|
|
const LOCKSCREEN_DIM_BG: f32 = 0.0;
|
|
const LOCKSCREEN_DIM_ALL: bool = true;
|
|
|
|
const BLUR_SIZE: usize = 9;
|
|
const BLUR_DIV: u32 =
|
|
((65536_f32 * (1_f32 - LOCKSCREEN_DIM_BG)) as u32) / ((BLUR_SIZE * BLUR_SIZE) as u32);
|
|
const DECOMP_LINES: usize = BLUR_SIZE + 1;
|
|
const BLUR_RADIUS: i16 = (BLUR_SIZE / 2) as i16;
|
|
|
|
const COLORS: usize = 3;
|
|
const RED_IDX: usize = 0;
|
|
const GREEN_IDX: usize = 1;
|
|
const BLUE_IDX: usize = 2;
|
|
|
|
pub trait HomescreenDecompressor {
|
|
fn get_height(&self) -> i16;
|
|
fn decompress(&mut self);
|
|
fn get_data(&mut self) -> &mut BufferJpeg;
|
|
}
|
|
|
|
pub struct HomescreenJpeg<'i> {
|
|
pub output: BufferOutput,
|
|
pub input: BufferInput<'i>,
|
|
pub jdec: Option<JDEC<'i>>,
|
|
}
|
|
|
|
impl<'i> HomescreenJpeg<'i> {
|
|
pub fn new(mut input: BufferInput<'i>, pool: &'i mut [u8]) -> Self {
|
|
Self {
|
|
output: BufferOutput::new(WIDTH, 16),
|
|
jdec: JDEC::new(&mut input, pool).ok(),
|
|
input,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'i> HomescreenDecompressor for HomescreenJpeg<'i> {
|
|
fn get_height(&self) -> i16 {
|
|
if let Some(dec) = self.jdec.as_ref() {
|
|
return dec.mcu_height();
|
|
}
|
|
1
|
|
}
|
|
|
|
fn decompress(&mut self) {
|
|
self.jdec
|
|
.as_mut()
|
|
.map(|dec| dec.decomp(&mut self.input, &mut self.output));
|
|
}
|
|
|
|
fn get_data(&mut self) -> &mut BufferJpeg {
|
|
self.output.buffer()
|
|
}
|
|
}
|
|
|
|
pub struct HomescreenToif<'i> {
|
|
pub output: BufferOutput,
|
|
pub decomp_context: UzlibContext<'i>,
|
|
line: i16,
|
|
}
|
|
|
|
impl<'i> HomescreenToif<'i> {
|
|
pub fn new(context: UzlibContext<'i>) -> Self {
|
|
Self {
|
|
output: BufferOutput::new(WIDTH, 16),
|
|
decomp_context: context,
|
|
line: 0,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'i> HomescreenDecompressor for HomescreenToif<'i> {
|
|
fn get_height(&self) -> i16 {
|
|
1
|
|
}
|
|
|
|
fn decompress(&mut self) {
|
|
// SAFETY: Aligning to u8 slice is safe, because the original slice is aligned
|
|
// to 16 bits, therefore there are also no residuals (prefix/suffix).
|
|
// The data in the slices are integers, so these are valid for both u16
|
|
// and u8.
|
|
if self.line >= HOMESCREEN_TOIF_Y_OFFSET
|
|
&& self.line < HOMESCREEN_TOIF_Y_OFFSET + HOMESCREEN_TOIF_SIZE
|
|
{
|
|
let (_, workbuf, _) = unsafe { self.output.buffer().buffer.align_to_mut::<u8>() };
|
|
let result = self.decomp_context.uncompress(
|
|
&mut workbuf[2 * HOMESCREEN_TOIF_X_OFFSET
|
|
..2 * HOMESCREEN_TOIF_X_OFFSET + 2 * HOMESCREEN_TOIF_SIZE as usize],
|
|
);
|
|
|
|
if result.is_err() {
|
|
self.output.buffer().buffer.fill(0);
|
|
} else {
|
|
for i in 0..HOMESCREEN_TOIF_SIZE as usize {
|
|
workbuf.swap(
|
|
2 * HOMESCREEN_TOIF_X_OFFSET + 2 * i,
|
|
2 * HOMESCREEN_TOIF_X_OFFSET + 2 * i + 1,
|
|
);
|
|
}
|
|
}
|
|
} else {
|
|
self.output.buffer().buffer.fill(0);
|
|
}
|
|
self.line += 1;
|
|
}
|
|
|
|
fn get_data(&mut self) -> &mut BufferJpeg {
|
|
self.output.buffer()
|
|
}
|
|
}
|
|
|
|
fn homescreen_get_fg_text(
|
|
y_tmp: i16,
|
|
text_info: HomescreenTextInfo,
|
|
text_buffer: &BufferText,
|
|
fg_buffer: &mut BufferLine4bpp,
|
|
) -> bool {
|
|
if y_tmp >= text_info.text_area.y0 && y_tmp < text_info.text_area.y1 {
|
|
let y_pos = y_tmp - text_info.text_area.y0;
|
|
position_buffer(
|
|
&mut fg_buffer.buffer,
|
|
&text_buffer.buffer[(y_pos * WIDTH / 2) as usize..((y_pos + 1) * WIDTH / 2) as usize],
|
|
4,
|
|
text_info.text_area.x0,
|
|
text_info.text_width,
|
|
);
|
|
}
|
|
|
|
y_tmp == (text_info.text_area.y1 - 1)
|
|
}
|
|
|
|
fn homescreen_get_fg_icon(
|
|
y_tmp: i16,
|
|
text_info: HomescreenTextInfo,
|
|
icon_data: &[u8],
|
|
fg_buffer: &mut BufferLine4bpp,
|
|
) {
|
|
if let Some(icon_area) = text_info.icon_area {
|
|
let icon_size = icon_area.size();
|
|
if y_tmp >= icon_area.y0 && y_tmp < icon_area.y1 {
|
|
let y_pos = y_tmp - icon_area.y0;
|
|
position_buffer(
|
|
&mut fg_buffer.buffer,
|
|
&icon_data
|
|
[(y_pos * icon_size.x / 2) as usize..((y_pos + 1) * icon_size.x / 2) as usize],
|
|
4,
|
|
icon_area.x0,
|
|
icon_size.x,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn homescreen_position_text(
|
|
text: &HomescreenText,
|
|
buffer: &mut BufferText,
|
|
icon_buffer: &mut [u8],
|
|
) -> HomescreenTextInfo {
|
|
let text_width = text
|
|
.text
|
|
.map(|t| display::text_width(t, text.style.text_font.into()));
|
|
let font_max_height = display::text_max_height(text.style.text_font.into());
|
|
let font_baseline = display::text_baseline(text.style.text_font.into());
|
|
let text_width_clamped = text_width.clamp(0, screen().width());
|
|
|
|
let icon_size = if let Some(icon) = text.icon {
|
|
let size = icon.toif.size();
|
|
assert!(size.x <= HOMESCREEN_MAX_ICON_SIZE);
|
|
assert!(size.y <= HOMESCREEN_MAX_ICON_SIZE);
|
|
icon.toif.uncompress(icon_buffer);
|
|
size
|
|
} else {
|
|
Offset::zero()
|
|
};
|
|
|
|
let text_top = screen().y0 + text.offset.y - font_max_height + font_baseline;
|
|
let text_bottom = screen().y0 + text.offset.y + font_baseline;
|
|
|
|
let total_width = text_width_clamped + icon_size.x + TEXT_ICON_SPACE;
|
|
let icon_left = screen().center().x + text.offset.x - total_width / 2;
|
|
let text_left = icon_left + icon_size.x + TEXT_ICON_SPACE;
|
|
let text_right = screen().center().x + text.offset.x + total_width / 2;
|
|
|
|
let text_area = Rect::new(
|
|
Point::new(text_left, text_top),
|
|
Point::new(text_right, text_bottom),
|
|
);
|
|
|
|
let icon_area = if text.icon.is_some() {
|
|
Some(Rect::from_top_left_and_size(
|
|
Point::new(icon_left, text_bottom - icon_size.y - font_baseline),
|
|
icon_size,
|
|
))
|
|
} else {
|
|
None
|
|
};
|
|
|
|
text.text
|
|
.map(|t| display::text_into_buffer(t, text.style.text_font.into(), buffer, 0));
|
|
|
|
HomescreenTextInfo {
|
|
text_area,
|
|
text_width,
|
|
text_color: text.style.text_color,
|
|
icon_area,
|
|
}
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn homescreen_dim_area(x: i16, y: i16) -> bool {
|
|
y >= HOMESCREEN_DIM_START
|
|
&& (y > HOMESCREEN_DIM_START + 1
|
|
&& y < (HOMESCREEN_DIM_START + HOMESCREEN_DIM_HEIGHT - 1)
|
|
&& x > HOMESCREEN_DIM_BORDER
|
|
&& x < WIDTH - HOMESCREEN_DIM_BORDER)
|
|
|| (y > HOMESCREEN_DIM_START
|
|
&& y < (HOMESCREEN_DIM_START + HOMESCREEN_DIM_HEIGHT)
|
|
&& x > HOMESCREEN_DIM_BORDER + 1
|
|
&& x < WIDTH - (HOMESCREEN_DIM_BORDER + 1))
|
|
|| ((HOMESCREEN_DIM_START..=(HOMESCREEN_DIM_START + HOMESCREEN_DIM_HEIGHT)).contains(&y)
|
|
&& x > HOMESCREEN_DIM_BORDER + 2
|
|
&& x < WIDTH - (HOMESCREEN_DIM_BORDER + 2))
|
|
}
|
|
|
|
fn homescreen_line_blurred(
|
|
icon_data: &[u8],
|
|
text_buffer: &mut BufferText,
|
|
fg_buffer: &mut BufferLine4bpp,
|
|
img_buffer: &mut BufferLine16bpp,
|
|
text_info: HomescreenTextInfo,
|
|
blurring: &BlurringContext,
|
|
y: i16,
|
|
) -> bool {
|
|
fg_buffer.buffer.fill(0);
|
|
for x in 0..HOMESCREEN_IMAGE_WIDTH {
|
|
let c = if LOCKSCREEN_DIM_ALL {
|
|
let x = x as usize;
|
|
|
|
let coef = (65536_f32 * LOCKSCREEN_DIM) as u32;
|
|
|
|
let r = (blurring.totals.buffer[RED_IDX][x] as u32 * BLUR_DIV) >> 16;
|
|
let g = (blurring.totals.buffer[GREEN_IDX][x] as u32 * BLUR_DIV) >> 16;
|
|
let b = (blurring.totals.buffer[BLUE_IDX][x] as u32 * BLUR_DIV) >> 16;
|
|
|
|
let r = (((coef * r) >> 8) & 0xF800) as u16;
|
|
let g = (((coef * g) >> 13) & 0x07E0) as u16;
|
|
let b = (((coef * b) >> 19) & 0x001F) as u16;
|
|
|
|
r | g | b
|
|
} else {
|
|
let x = x as usize;
|
|
|
|
let r = (((blurring.totals.buffer[RED_IDX][x] as u32 * BLUR_DIV) >> 8) & 0xF800) as u16;
|
|
let g =
|
|
(((blurring.totals.buffer[GREEN_IDX][x] as u32 * BLUR_DIV) >> 13) & 0x07E0) as u16;
|
|
let b =
|
|
(((blurring.totals.buffer[BLUE_IDX][x] as u32 * BLUR_DIV) >> 19) & 0x001F) as u16;
|
|
r | g | b
|
|
};
|
|
|
|
let j = (2 * x) as usize;
|
|
img_buffer.buffer[j + 1] = (c >> 8) as u8;
|
|
img_buffer.buffer[j] = (c & 0xFF) as u8;
|
|
}
|
|
|
|
let done = homescreen_get_fg_text(y, text_info, text_buffer, fg_buffer);
|
|
homescreen_get_fg_icon(y, text_info, icon_data, fg_buffer);
|
|
|
|
dma2d_wait_for_transfer();
|
|
dma2d_setup_4bpp_over_16bpp(text_info.text_color.into());
|
|
unsafe {
|
|
dma2d_start_blend(&fg_buffer.buffer, &img_buffer.buffer, WIDTH);
|
|
}
|
|
|
|
done
|
|
}
|
|
|
|
#[allow(clippy::too_many_arguments)]
|
|
fn homescreen_line(
|
|
icon_data: &[u8],
|
|
text_buffer: &mut BufferText,
|
|
text_info: HomescreenTextInfo,
|
|
data_buffer: &mut BufferJpeg,
|
|
fg_buffer: &mut BufferLine4bpp,
|
|
img_buffer: &mut BufferLine16bpp,
|
|
mcu_height: i16,
|
|
y: i16,
|
|
) -> bool {
|
|
let image_data = get_data(data_buffer, y, mcu_height);
|
|
fg_buffer.buffer.fill(0);
|
|
|
|
for x in 0..HOMESCREEN_IMAGE_WIDTH {
|
|
let d = image_data[x as usize];
|
|
|
|
let c = if homescreen_dim_area(x, y) {
|
|
let coef = (65536_f32 * HOMESCREEN_DIM) as u32;
|
|
|
|
let r = (d & 0xF800) >> 8;
|
|
let g = (d & 0x07E0) >> 3;
|
|
let b = (d & 0x001F) << 3;
|
|
|
|
let r = (((coef * r as u32) >> 8) & 0xF800) as u16;
|
|
let g = (((coef * g as u32) >> 13) & 0x07E0) as u16;
|
|
let b = (((coef * b as u32) >> 19) & 0x001F) as u16;
|
|
r | g | b
|
|
} else {
|
|
d
|
|
};
|
|
|
|
let j = 2 * x as usize;
|
|
img_buffer.buffer[j + 1] = (c >> 8) as u8;
|
|
img_buffer.buffer[j] = (c & 0xFF) as u8;
|
|
}
|
|
|
|
let done = homescreen_get_fg_text(y, text_info, text_buffer, fg_buffer);
|
|
homescreen_get_fg_icon(y, text_info, icon_data, fg_buffer);
|
|
|
|
dma2d_wait_for_transfer();
|
|
dma2d_setup_4bpp_over_16bpp(text_info.text_color.into());
|
|
unsafe {
|
|
dma2d_start_blend(&fg_buffer.buffer, &img_buffer.buffer, WIDTH);
|
|
}
|
|
|
|
done
|
|
}
|
|
|
|
fn homescreen_next_text(
|
|
texts: &[HomescreenText],
|
|
text_buffer: &mut BufferText,
|
|
icon_data: &mut [u8],
|
|
text_info: HomescreenTextInfo,
|
|
text_idx: usize,
|
|
) -> (HomescreenTextInfo, usize) {
|
|
let mut next_text_idx = text_idx;
|
|
let mut next_text_info = text_info;
|
|
|
|
if next_text_idx < texts.len() {
|
|
if let Some(txt) = texts.get(next_text_idx) {
|
|
text_buffer.buffer.fill(0);
|
|
next_text_info = homescreen_position_text(txt, text_buffer, icon_data);
|
|
next_text_idx += 1;
|
|
}
|
|
}
|
|
|
|
(next_text_info, next_text_idx)
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn update_accs_add(data: &[u16], idx: usize, acc_r: &mut u16, acc_g: &mut u16, acc_b: &mut u16) {
|
|
let d = data[idx];
|
|
let r = (d & 0xF800) >> 8;
|
|
let g = (d & 0x07E0) >> 3;
|
|
let b = (d & 0x001F) << 3;
|
|
*acc_r += r;
|
|
*acc_g += g;
|
|
*acc_b += b;
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn update_accs_sub(data: &[u16], idx: usize, acc_r: &mut u16, acc_g: &mut u16, acc_b: &mut u16) {
|
|
let d = data[idx];
|
|
let r = (d & 0xF800) >> 8;
|
|
let g = (d & 0x07E0) >> 3;
|
|
let b = (d & 0x001F) << 3;
|
|
*acc_r -= r;
|
|
*acc_g -= g;
|
|
*acc_b -= b;
|
|
}
|
|
|
|
struct BlurringContext {
|
|
mem: BufferBlurring,
|
|
pub totals: BufferBlurringTotals,
|
|
line_num: i16,
|
|
add_idx: usize,
|
|
rem_idx: usize,
|
|
}
|
|
|
|
impl BlurringContext {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
mem: BufferBlurring::get_cleared(),
|
|
totals: BufferBlurringTotals::get_cleared(),
|
|
line_num: 0,
|
|
add_idx: 0,
|
|
rem_idx: 0,
|
|
}
|
|
}
|
|
|
|
fn clear(&mut self) {
|
|
let lines = &mut self.mem.buffer[0..DECOMP_LINES];
|
|
for (i, total) in self.totals.buffer.iter_mut().enumerate() {
|
|
for line in lines.iter_mut() {
|
|
line[i].fill(0);
|
|
}
|
|
total.fill(0);
|
|
}
|
|
}
|
|
|
|
// computes color averages for one line of image data
|
|
fn compute_line_avgs(&mut self, buffer: &mut BufferJpeg, mcu_height: i16) {
|
|
let lines = &mut self.mem.buffer[0..DECOMP_LINES];
|
|
let mut acc_r = 0;
|
|
let mut acc_g = 0;
|
|
let mut acc_b = 0;
|
|
let data = get_data(buffer, self.line_num, mcu_height);
|
|
|
|
for i in -BLUR_RADIUS..=BLUR_RADIUS {
|
|
let ic = i.clamp(0, HOMESCREEN_IMAGE_WIDTH - 1) as usize;
|
|
update_accs_add(data, ic, &mut acc_r, &mut acc_g, &mut acc_b);
|
|
}
|
|
|
|
for i in 0..HOMESCREEN_IMAGE_WIDTH {
|
|
lines[self.add_idx][RED_IDX][i as usize] = acc_r;
|
|
lines[self.add_idx][GREEN_IDX][i as usize] = acc_g;
|
|
lines[self.add_idx][BLUE_IDX][i as usize] = acc_b;
|
|
|
|
// clamping handles left and right edges
|
|
let ic = (i - BLUR_RADIUS).clamp(0, HOMESCREEN_IMAGE_WIDTH - 1) as usize;
|
|
let ic2 =
|
|
(i + BLUR_SIZE as i16 - BLUR_RADIUS).clamp(0, HOMESCREEN_IMAGE_WIDTH - 1) as usize;
|
|
update_accs_add(data, ic2, &mut acc_r, &mut acc_g, &mut acc_b);
|
|
update_accs_sub(data, ic, &mut acc_r, &mut acc_g, &mut acc_b);
|
|
}
|
|
self.line_num += 1;
|
|
}
|
|
|
|
// adds one line of averages to sliding total averages
|
|
fn vertical_avg_add(&mut self) {
|
|
let lines = &mut self.mem.buffer[0..DECOMP_LINES];
|
|
for i in 0..HOMESCREEN_IMAGE_WIDTH as usize {
|
|
self.totals.buffer[RED_IDX][i] += lines[self.add_idx][RED_IDX][i];
|
|
self.totals.buffer[GREEN_IDX][i] += lines[self.add_idx][GREEN_IDX][i];
|
|
self.totals.buffer[BLUE_IDX][i] += lines[self.add_idx][BLUE_IDX][i];
|
|
}
|
|
}
|
|
|
|
// adds one line and removes one line of averages to/from sliding total averages
|
|
fn vertical_avg(&mut self) {
|
|
let lines = &mut self.mem.buffer[0..DECOMP_LINES];
|
|
for i in 0..HOMESCREEN_IMAGE_WIDTH as usize {
|
|
self.totals.buffer[RED_IDX][i] +=
|
|
lines[self.add_idx][RED_IDX][i] - lines[self.rem_idx][RED_IDX][i];
|
|
self.totals.buffer[GREEN_IDX][i] +=
|
|
lines[self.add_idx][GREEN_IDX][i] - lines[self.rem_idx][GREEN_IDX][i];
|
|
self.totals.buffer[BLUE_IDX][i] +=
|
|
lines[self.add_idx][BLUE_IDX][i] - lines[self.rem_idx][BLUE_IDX][i];
|
|
}
|
|
}
|
|
|
|
fn inc_add(&mut self) {
|
|
self.add_idx += 1;
|
|
if self.add_idx >= DECOMP_LINES {
|
|
self.add_idx = 0;
|
|
}
|
|
}
|
|
|
|
fn inc_rem(&mut self) {
|
|
self.rem_idx += 1;
|
|
if self.rem_idx >= DECOMP_LINES {
|
|
self.rem_idx = 0;
|
|
}
|
|
}
|
|
|
|
fn get_line_num(&self) -> i16 {
|
|
self.line_num
|
|
}
|
|
}
|
|
|
|
#[inline(always)]
|
|
fn get_data(buffer: &mut BufferJpeg, line_num: i16, mcu_height: i16) -> &mut [u16] {
|
|
let data_start = ((line_num % mcu_height) * WIDTH) as usize;
|
|
let data_end = (((line_num % mcu_height) + 1) * WIDTH) as usize;
|
|
&mut buffer.buffer[data_start..data_end]
|
|
}
|
|
|
|
pub fn homescreen_blurred(data: &mut dyn HomescreenDecompressor, texts: &[HomescreenText]) {
|
|
let mut icon_data = [0_u8; (HOMESCREEN_MAX_ICON_SIZE * HOMESCREEN_MAX_ICON_SIZE / 2) as usize];
|
|
|
|
let mut text_buffer = BufferText::get_cleared();
|
|
let mut fg_buffer_0 = BufferLine4bpp::get_cleared();
|
|
let mut img_buffer_0 = BufferLine16bpp::get_cleared();
|
|
let mut fg_buffer_1 = BufferLine4bpp::get_cleared();
|
|
let mut img_buffer_1 = BufferLine16bpp::get_cleared();
|
|
|
|
let mut next_text_idx = 1;
|
|
let mut text_info =
|
|
homescreen_position_text(unwrap!(texts.first()), &mut text_buffer, &mut icon_data);
|
|
|
|
let mcu_height = data.get_height();
|
|
data.decompress();
|
|
|
|
set_window(screen());
|
|
|
|
let mut blurring = BlurringContext::new();
|
|
|
|
// handling top edge case: preload the edge value N+1 times
|
|
blurring.compute_line_avgs(data.get_data(), mcu_height);
|
|
|
|
for _ in 0..=BLUR_RADIUS {
|
|
blurring.vertical_avg_add();
|
|
}
|
|
blurring.inc_add();
|
|
|
|
// load enough values to be able to compute first line averages
|
|
for _ in 0..BLUR_RADIUS {
|
|
blurring.compute_line_avgs(data.get_data(), mcu_height);
|
|
blurring.vertical_avg_add();
|
|
blurring.inc_add();
|
|
|
|
if (blurring.get_line_num() % mcu_height) == 0 {
|
|
data.decompress();
|
|
}
|
|
}
|
|
|
|
for y in 0..HEIGHT {
|
|
// several lines have been already decompressed before this loop, adjust for
|
|
// that
|
|
if y < HOMESCREEN_IMAGE_HEIGHT - (BLUR_RADIUS + 1) {
|
|
blurring.compute_line_avgs(data.get_data(), mcu_height);
|
|
}
|
|
|
|
let done = if y % 2 == 0 {
|
|
homescreen_line_blurred(
|
|
&icon_data,
|
|
&mut text_buffer,
|
|
&mut fg_buffer_0,
|
|
&mut img_buffer_0,
|
|
text_info,
|
|
&blurring,
|
|
y,
|
|
)
|
|
} else {
|
|
homescreen_line_blurred(
|
|
&icon_data,
|
|
&mut text_buffer,
|
|
&mut fg_buffer_1,
|
|
&mut img_buffer_1,
|
|
text_info,
|
|
&blurring,
|
|
y,
|
|
)
|
|
};
|
|
|
|
if done {
|
|
(text_info, next_text_idx) = homescreen_next_text(
|
|
texts,
|
|
&mut text_buffer,
|
|
&mut icon_data,
|
|
text_info,
|
|
next_text_idx,
|
|
);
|
|
}
|
|
|
|
blurring.vertical_avg();
|
|
|
|
// handling bottom edge case: stop incrementing counter, adding the edge value
|
|
// for the rest of image
|
|
// the extra -1 is to indicate that this was the last decompressed line,
|
|
// in the next pass the docompression and compute_line_avgs won't happen
|
|
if y < HOMESCREEN_IMAGE_HEIGHT - (BLUR_RADIUS + 1) - 1 {
|
|
blurring.inc_add();
|
|
}
|
|
|
|
if y == HOMESCREEN_IMAGE_HEIGHT {
|
|
// reached end of image, clear avgs (display black)
|
|
blurring.clear();
|
|
}
|
|
|
|
// only start incrementing remove index when enough lines have been loaded
|
|
if y >= (BLUR_RADIUS) {
|
|
blurring.inc_rem();
|
|
}
|
|
|
|
if (blurring.get_line_num() % mcu_height) == 0 && (blurring.get_line_num() < HEIGHT) {
|
|
data.decompress();
|
|
}
|
|
}
|
|
dma2d_wait_for_transfer();
|
|
}
|
|
|
|
pub fn homescreen(
|
|
data: &mut dyn HomescreenDecompressor,
|
|
texts: &[HomescreenText],
|
|
notification: Option<HomescreenNotification>,
|
|
notification_only: bool,
|
|
) {
|
|
let mut icon_data = [0_u8; (HOMESCREEN_MAX_ICON_SIZE * HOMESCREEN_MAX_ICON_SIZE / 2) as usize];
|
|
|
|
let mut text_buffer = BufferText::get_cleared();
|
|
let mut fg_buffer_0 = BufferLine4bpp::get_cleared();
|
|
let mut img_buffer_0 = BufferLine16bpp::get_cleared();
|
|
let mut fg_buffer_1 = BufferLine4bpp::get_cleared();
|
|
let mut img_buffer_1 = BufferLine16bpp::get_cleared();
|
|
|
|
let mut next_text_idx = 0;
|
|
let mut text_info = if let Some(notification) = notification {
|
|
rect_fill_rounded_buffer(
|
|
Rect::from_top_left_and_size(
|
|
Point::new(NOTIFICATION_BORDER, 0),
|
|
Offset::new(WIDTH - NOTIFICATION_BORDER * 2, NOTIFICATION_HEIGHT),
|
|
),
|
|
2,
|
|
&mut text_buffer,
|
|
);
|
|
let area = Rect::new(
|
|
Point::new(0, NOTIFICATION_BORDER),
|
|
Point::new(WIDTH, NOTIFICATION_HEIGHT + NOTIFICATION_BORDER),
|
|
);
|
|
HomescreenTextInfo {
|
|
text_area: area,
|
|
text_width: WIDTH,
|
|
text_color: notification.color,
|
|
icon_area: None,
|
|
}
|
|
} else {
|
|
next_text_idx += 1;
|
|
homescreen_position_text(unwrap!(texts.first()), &mut text_buffer, &mut icon_data)
|
|
};
|
|
|
|
set_window(screen());
|
|
|
|
let mcu_height = data.get_height();
|
|
|
|
for y in 0..HEIGHT {
|
|
if (y % mcu_height) == 0 {
|
|
data.decompress();
|
|
}
|
|
|
|
let done = if y % 2 == 0 {
|
|
homescreen_line(
|
|
&icon_data,
|
|
&mut text_buffer,
|
|
text_info,
|
|
data.get_data(),
|
|
&mut fg_buffer_0,
|
|
&mut img_buffer_0,
|
|
mcu_height,
|
|
y,
|
|
)
|
|
} else {
|
|
homescreen_line(
|
|
&icon_data,
|
|
&mut text_buffer,
|
|
text_info,
|
|
data.get_data(),
|
|
&mut fg_buffer_1,
|
|
&mut img_buffer_1,
|
|
mcu_height,
|
|
y,
|
|
)
|
|
};
|
|
|
|
if done {
|
|
if notification.is_some() && next_text_idx == 0 {
|
|
//finished notification area, let interrupt and draw the text
|
|
let notification = unwrap!(notification);
|
|
|
|
let style = TextStyle {
|
|
background_color: notification.color,
|
|
..theme::TEXT_BOLD
|
|
};
|
|
|
|
dma2d_wait_for_transfer();
|
|
|
|
drop(fg_buffer_0);
|
|
drop(fg_buffer_1);
|
|
|
|
icon_text_center(
|
|
text_info.text_area.center(),
|
|
notification.icon,
|
|
8,
|
|
notification.text,
|
|
style,
|
|
Offset::new(1, -2),
|
|
);
|
|
|
|
fg_buffer_0 = BufferLine4bpp::get_cleared();
|
|
fg_buffer_1 = BufferLine4bpp::get_cleared();
|
|
|
|
set_window(
|
|
screen()
|
|
.split_top(NOTIFICATION_HEIGHT + NOTIFICATION_BORDER)
|
|
.1,
|
|
);
|
|
}
|
|
|
|
if notification_only && next_text_idx == 0 {
|
|
dma2d_wait_for_transfer();
|
|
return;
|
|
}
|
|
|
|
(text_info, next_text_idx) = homescreen_next_text(
|
|
texts,
|
|
&mut text_buffer,
|
|
&mut icon_data,
|
|
text_info,
|
|
next_text_idx,
|
|
);
|
|
}
|
|
}
|
|
dma2d_wait_for_transfer();
|
|
}
|