This commit is contained in:
2025-11-04 13:43:48 -07:00
parent dd26e3528b
commit 375b5d1d47
12 changed files with 318 additions and 69 deletions

13
Cargo.lock generated
View File

@@ -18,7 +18,6 @@ version = "0.1.0"
dependencies = [ dependencies = [
"abi_sys", "abi_sys",
"embedded-graphics", "embedded-graphics",
"embedded-sdmmc",
"once_cell", "once_cell",
"rand_core 0.9.3", "rand_core 0.9.3",
] ]
@@ -31,7 +30,6 @@ dependencies = [
"cbindgen", "cbindgen",
"defmt 0.3.100", "defmt 0.3.100",
"embedded-graphics", "embedded-graphics",
"embedded-sdmmc",
"strum", "strum",
] ]
@@ -1289,6 +1287,7 @@ version = "0.1.0"
dependencies = [ dependencies = [
"abi", "abi",
"embedded-graphics", "embedded-graphics",
"selection_ui",
"tinygif", "tinygif",
] ]
@@ -2191,6 +2190,16 @@ version = "0.13.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c1257cd4248b4132760d6524d6dda4e053bc648c9070b960929bf50cfb1e7add" checksum = "c1257cd4248b4132760d6524d6dda4e053bc648c9070b960929bf50cfb1e7add"
[[package]]
name = "selection_ui"
version = "0.1.0"
dependencies = [
"abi",
"embedded-graphics",
"embedded-layout",
"embedded-text",
]
[[package]] [[package]]
name = "semver" name = "semver"
version = "0.9.0" version = "0.9.0"

View File

@@ -4,6 +4,7 @@ members = [
"kernel", "kernel",
"abi_sys", "abi_sys",
"abi", "abi",
"selection_ui",
"user-apps/calculator", "user-apps/calculator",
"user-apps/snake", "user-apps/snake",
"user-apps/gallery", "user-apps/gallery",
@@ -26,4 +27,4 @@ panic = "abort"
[profile.dev] [profile.dev]
lto = true lto = true
opt-level = "z" opt-level = "s"

View File

@@ -4,8 +4,7 @@ version = "0.1.0"
edition = "2024" edition = "2024"
[dependencies] [dependencies]
embedded-sdmmc = { version = "0.9.0", default-features = false }
embedded-graphics = "0.8.1"
abi_sys = { path = "../abi_sys" } abi_sys = { path = "../abi_sys" }
embedded-graphics = "0.8.1"
once_cell = { version = "1", default-features = false } once_cell = { version = "1", default-features = false }
rand_core = "0.9.3" rand_core = "0.9.3"

View File

@@ -140,7 +140,9 @@ impl RngCore for Rng {
} }
pub mod fs { pub mod fs {
use embedded_sdmmc::DirEntry; use core::fmt::Display;
use alloc::{format, vec::Vec};
pub fn read_file(file: &str, read_from: usize, buf: &mut [u8]) -> usize { pub fn read_file(file: &str, read_from: usize, buf: &mut [u8]) -> usize {
abi_sys::read_file( abi_sys::read_file(
@@ -152,8 +154,85 @@ pub mod fs {
) )
} }
pub fn list_dir(path: &str, files: &mut [Option<DirEntry>]) -> usize { pub struct FileName<'a> {
abi_sys::list_dir(path.as_ptr(), path.len(), files.as_mut_ptr(), files.len()) full: &'a str,
base: &'a str,
ext: Option<&'a str>,
}
impl<'a> FileName<'a> {
pub fn full_name(&self) -> &str {
self.full
}
pub fn base(&self) -> &str {
self.base
}
pub fn extension(&self) -> Option<&str> {
self.ext
}
}
impl<'a> Display for FileName<'a> {
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
write!(f, "{}", self.full_name())
}
}
impl<'a> From<&'a str> for FileName<'a> {
fn from(s: &'a str) -> FileName<'a> {
let full = s;
// Split on last dot for extension
let (base, ext) = match s.rfind('.') {
Some(idx) => (&s[..idx], Some(&s[idx + 1..])),
None => (s, None),
};
FileName { full, base, ext }
}
}
const MAX_ENTRY_NAME_LEN: usize = 25;
const MAX_ENTRIES: usize = 25;
#[derive(Clone, Copy)]
pub struct Entries([[u8; MAX_ENTRY_NAME_LEN]; MAX_ENTRIES]);
impl Entries {
pub fn new() -> Self {
Self([[0; MAX_ENTRY_NAME_LEN]; MAX_ENTRIES])
}
/// Get list of file names after listing
pub fn entries<'a>(&'a self) -> Vec<FileName<'a>> {
self.0
.iter()
.filter_map(|buf| {
let nul_pos = buf.iter().position(|&b| b == 0).unwrap_or(buf.len());
Some(core::str::from_utf8(&buf[..nul_pos]).ok()?.into())
})
.collect()
}
fn as_ptrs(&mut self) -> [*mut u8; MAX_ENTRIES] {
let mut ptrs: [*mut u8; MAX_ENTRIES] = [core::ptr::null_mut(); MAX_ENTRIES];
for (i, buf) in self.0.iter_mut().enumerate() {
ptrs[i] = buf.as_mut_ptr();
}
ptrs
}
}
pub fn list_dir(path: &str, entries: &mut Entries) -> usize {
abi_sys::list_dir(
path.as_ptr(),
path.len(),
entries.as_ptrs().as_mut_ptr(),
MAX_ENTRIES,
MAX_ENTRY_NAME_LEN,
)
} }
pub fn file_len(str: &str) -> usize { pub fn file_len(str: &str) -> usize {

View File

@@ -12,7 +12,6 @@ defmt = ["dep:defmt"]
strum = { version = "0.27.2", default-features = false, features = ["derive"] } strum = { version = "0.27.2", default-features = false, features = ["derive"] }
bitflags = "2.9.4" bitflags = "2.9.4"
embedded-graphics = "0.8.1" embedded-graphics = "0.8.1"
embedded-sdmmc = { version = "0.9.0", default-features = false }
defmt = { version = "0.3", optional = true } defmt = { version = "0.3", optional = true }
[build-dependencies] [build-dependencies]

View File

@@ -1,14 +1,13 @@
#![no_std] #![no_std]
#[cfg(feature = "alloc")] #[cfg(feature = "alloc")]
use core::alloc::Layout; use core::alloc::Layout;
use core::ffi::c_char;
use embedded_graphics::{ use embedded_graphics::{
Pixel, Pixel,
pixelcolor::{Rgb565, raw::RawU16}, pixelcolor::{Rgb565, raw::RawU16},
prelude::{IntoStorage, Point}, prelude::{IntoStorage, Point},
}; };
use embedded_sdmmc::DirEntry;
use strum::{EnumCount, EnumIter}; use strum::{EnumCount, EnumIter};
pub type EntryFn = fn(); pub type EntryFn = fn();
@@ -396,21 +395,23 @@ pub extern "C" fn gen_rand(req: &mut RngRequest) {
pub type ListDir = extern "C" fn( pub type ListDir = extern "C" fn(
str: *const u8, str: *const u8,
len: usize, len: usize,
files: *mut Option<DirEntry>, entries: *mut *mut c_char,
file_len: usize, file_len: usize,
max_entry_str_len: usize,
) -> usize; ) -> usize;
#[unsafe(no_mangle)] #[unsafe(no_mangle)]
pub extern "C" fn list_dir( pub extern "C" fn list_dir(
str: *const u8, str: *const u8,
len: usize, len: usize,
files: *mut Option<DirEntry>, entries: *mut *mut c_char,
file_len: usize, entry_count: usize,
max_entry_str_len: usize,
) -> usize { ) -> usize {
unsafe { unsafe {
let ptr = CALL_ABI_TABLE[CallTable::ListDir as usize]; let ptr = CALL_ABI_TABLE[CallTable::ListDir as usize];
let f: ListDir = core::mem::transmute(ptr); let f: ListDir = core::mem::transmute(ptr);
f(str, len, files, file_len) f(str, len, entries, entry_count, max_entry_str_len)
} }
} }

View File

@@ -3,11 +3,11 @@ use abi_sys::{
PrintAbi, ReadFile, RngRequest, SleepMsAbi, keyboard::*, PrintAbi, ReadFile, RngRequest, SleepMsAbi, keyboard::*,
}; };
use alloc::{string::ToString, vec::Vec}; use alloc::{string::ToString, vec::Vec};
use core::{alloc::GlobalAlloc, sync::atomic::Ordering}; use core::{alloc::GlobalAlloc, ffi::c_char, ptr, sync::atomic::Ordering};
use embassy_rp::clocks::{RoscRng, clk_sys_freq}; use embassy_rp::clocks::{RoscRng, clk_sys_freq};
use embassy_time::Instant; use embassy_time::Instant;
use embedded_graphics::draw_target::DrawTarget; use embedded_graphics::draw_target::DrawTarget;
use embedded_sdmmc::{DirEntry, LfnBuffer}; use embedded_sdmmc::LfnBuffer;
use heapless::spsc::Queue; use heapless::spsc::Queue;
use crate::{ use crate::{
@@ -116,11 +116,27 @@ pub extern "C" fn gen_rand(req: &mut RngRequest) {
} }
} }
fn get_dir_entries(dir: &Dir, files: &mut [Option<DirEntry>]) -> usize { unsafe fn copy_entry_to_user_buf(name: &[u8], dest: *mut c_char, max_str_len: usize) {
if !dest.is_null() {
let len = name.len().min(max_str_len - 1);
unsafe {
ptr::copy_nonoverlapping(name.as_ptr(), dest as *mut u8, len);
*dest.add(len) = 0; // nul terminator
}
}
}
unsafe fn get_dir_entries(dir: &Dir, entries: &mut [*mut c_char], max_str_len: usize) -> usize {
let mut b = [0; 25];
let mut buf = LfnBuffer::new(&mut b);
let mut i = 0; let mut i = 0;
dir.iterate_dir(|entry| { dir.iterate_dir_lfn(&mut buf, |entry, lfn_name| {
if i < files.len() { if i < entries.len() {
files[i] = Some(entry.clone()); if let Some(name) = lfn_name {
unsafe { copy_entry_to_user_buf(name.as_bytes(), entries[i], max_str_len) };
} else {
unsafe { copy_entry_to_user_buf(entry.name.base_name(), entries[i], max_str_len) };
}
i += 1; i += 1;
} }
}) })
@@ -128,24 +144,30 @@ fn get_dir_entries(dir: &Dir, files: &mut [Option<DirEntry>]) -> usize {
i i
} }
fn recurse_dir(dir: &Dir, dirs: &[&str], files: &mut [Option<DirEntry>]) -> usize { unsafe fn recurse_dir(
dir: &Dir,
dirs: &[&str],
entries: &mut [*mut c_char],
max_str_len: usize,
) -> usize {
if dirs.is_empty() { if dirs.is_empty() {
return get_dir_entries(dir, files); return unsafe { get_dir_entries(dir, entries, max_str_len) };
} }
let dir = dir.open_dir(dirs[0]).unwrap(); let dir = dir.open_dir(dirs[0]).unwrap();
recurse_dir(&dir, &dirs[1..], files) unsafe { recurse_dir(&dir, &dirs[1..], entries, max_str_len) }
} }
const _: ListDir = list_dir; const _: ListDir = list_dir;
pub extern "C" fn list_dir( pub extern "C" fn list_dir(
dir: *const u8, dir: *const u8,
len: usize, len: usize,
files: *mut Option<DirEntry>, entries: *mut *mut c_char,
files_len: usize, files_len: usize,
max_entry_str_len: usize,
) -> usize { ) -> usize {
// SAFETY: caller guarantees `ptr` is valid for `len` bytes // SAFETY: caller guarantees `ptr` is valid for `len` bytes
let files = unsafe { core::slice::from_raw_parts_mut(files, files_len) }; let files = unsafe { core::slice::from_raw_parts_mut(entries, files_len) };
// SAFETY: caller guarantees `ptr` is valid for `len` bytes // SAFETY: caller guarantees `ptr` is valid for `len` bytes
let dir = unsafe { core::str::from_raw_parts(dir, len) }; let dir = unsafe { core::str::from_raw_parts(dir, len) };
let dirs: Vec<&str> = dir.split('/').collect(); let dirs: Vec<&str> = dir.split('/').collect();
@@ -156,10 +178,12 @@ pub extern "C" fn list_dir(
let mut wrote = 0; let mut wrote = 0;
sd.access_root_dir(|root| { sd.access_root_dir(|root| {
if dirs[0] == "" && dirs.len() >= 2 { if dirs[0] == "" && dirs.len() >= 2 {
if dir == "/" { unsafe {
wrote = get_dir_entries(&root, files); if dir == "/" {
} else { wrote = get_dir_entries(&root, files, max_entry_str_len);
wrote = recurse_dir(&root, &dirs[1..], files); } else {
wrote = recurse_dir(&root, &dirs[1..], files, max_entry_str_len);
}
} }
} }
}); });

10
selection_ui/Cargo.toml Normal file
View File

@@ -0,0 +1,10 @@
[package]
name = "selection_ui"
version = "0.1.0"
edition = "2024"
[dependencies]
abi = { path = "../abi" }
embedded-graphics = "0.8.1"
embedded-layout = "0.4.2"
embedded-text = "0.7.3"

120
selection_ui/src/lib.rs Normal file
View File

@@ -0,0 +1,120 @@
#![no_std]
extern crate alloc;
use abi::{
display::Display,
fs::{Entries, FileName},
get_key,
keyboard::{KeyCode, KeyState},
};
use alloc::vec::Vec;
use embedded_graphics::{
Drawable,
mono_font::{MonoTextStyle, ascii::FONT_10X20},
pixelcolor::Rgb565,
prelude::{Dimensions, Point, Primitive, RgbColor, Size},
primitives::{PrimitiveStyle, Rectangle},
text::Text,
};
use embedded_layout::{
align::{horizontal, vertical},
layout::linear::{FixedMargin, LinearLayout},
prelude::*,
};
use embedded_text::TextBox;
pub struct SelectionUi<'a> {
selection: usize,
items: &'a Entries,
error: &'a str,
last_bounds: Option<Rectangle>,
}
impl<'a> SelectionUi<'a> {
pub fn new(items: &'a Entries, error: &'a str) -> Self {
Self {
selection: 0,
items,
error,
last_bounds: None,
}
}
pub fn run_selection_ui(&mut self, display: &mut Display) -> Result<Option<usize>, ()> {
self.draw(display)?;
let selection;
loop {
let key = get_key();
if key.state == KeyState::Pressed {
if let Some(s) = self.update(display, key.key)? {
selection = Some(s);
break;
}
}
}
Ok(selection)
}
/// updates the display with a new keypress.
/// returns selection idx if selected
pub fn update(&mut self, display: &mut Display, key: KeyCode) -> Result<Option<usize>, ()> {
match key {
KeyCode::JoyUp => {
let _ = self.selection.saturating_sub(1);
}
KeyCode::JoyDown => {
let _ = self.selection.saturating_add(1);
}
KeyCode::Enter | KeyCode::JoyRight => return Ok(Some(self.selection)),
_ => (),
};
self.draw(display)?;
Ok(None)
}
fn draw(&mut self, display: &mut Display) -> Result<(), ()> {
let text_style = MonoTextStyle::new(&FONT_10X20, Rgb565::WHITE);
let display_area = display.bounding_box();
let entries = self.items.entries();
if entries.is_empty() {
TextBox::new(
&self.error,
Rectangle::new(
Point::new(25, 25),
Size::new(display_area.size.width - 50, display_area.size.width - 50),
),
text_style,
)
.draw(display)
.unwrap();
}
let mut views: Vec<Text<MonoTextStyle<Rgb565>>> = Vec::new();
for i in &entries {
views.push(Text::new(i.full_name(), Point::zero(), text_style));
}
let views_group = Views::new(views.as_mut_slice());
let layout = LinearLayout::vertical(views_group)
.with_alignment(horizontal::Center)
.with_spacing(FixedMargin(5))
.arrange()
.align_to(&display_area, horizontal::Center, vertical::Center);
// draw selected box
let selected_bounds = layout.inner().get(self.selection).ok_or(())?.bounding_box();
Rectangle::new(selected_bounds.top_left, selected_bounds.size)
.into_styled(PrimitiveStyle::with_stroke(Rgb565::WHITE, 1))
.draw(display)?;
self.last_bounds = Some(layout.bounds());
layout.draw(display)?;
Ok(())
}
}

View File

@@ -5,12 +5,12 @@
extern crate alloc; extern crate alloc;
use abi::{ use abi::{
display::{Display, SCREEN_HEIGHT, SCREEN_WIDTH}, display::{Display, SCREEN_HEIGHT, SCREEN_WIDTH},
fs::{list_dir, read_file}, fs::{Entries, list_dir, read_file},
get_key, get_key,
keyboard::{KeyCode, KeyState}, keyboard::{KeyCode, KeyState},
print, print,
}; };
use alloc::{format, string::ToString, vec}; use alloc::{format, vec};
use core::panic::PanicInfo; use core::panic::PanicInfo;
use embedded_graphics::{ use embedded_graphics::{
Drawable, image::Image, mono_font::MonoTextStyle, mono_font::ascii::FONT_6X10, Drawable, image::Image, mono_font::MonoTextStyle, mono_font::ascii::FONT_6X10,
@@ -34,7 +34,6 @@ pub fn main() {
let mut bmp_buf = vec![0_u8; 100_000]; let mut bmp_buf = vec![0_u8; 100_000];
let mut display = Display; let mut display = Display;
// Grid parameters
let grid_cols = 3; let grid_cols = 3;
let grid_rows = 3; let grid_rows = 3;
let cell_width = SCREEN_WIDTH as i32 / grid_cols; let cell_width = SCREEN_WIDTH as i32 / grid_cols;
@@ -42,50 +41,44 @@ pub fn main() {
let mut images_drawn = 0; let mut images_drawn = 0;
let mut files = [const { None }; 18]; let mut entries = Entries::new();
let files_num = list_dir("/images", &mut files); let files_num = list_dir("/images", &mut entries);
for file in &files[2..files_num] { for file in &entries.entries()[2..files_num] {
if images_drawn >= grid_cols * grid_rows { if images_drawn >= grid_cols * grid_rows {
break; // only draw 3x3 break; // only draw 3x3
} }
if let Some(f) = file { print!("file: {}", file);
print!("file: {}", f.name); if file.extension().unwrap_or("") == "bmp" || file.extension().unwrap_or("") == "BMP" {
if f.name.extension() == b"bmp" || f.name.extension() == b"BMP" { let file_path = format!("/images/{}", file);
let file = format!("/images/{}", f.name);
let read = read_file(&file, 0, &mut &mut bmp_buf[..]); let read = read_file(&file_path, 0, &mut &mut bmp_buf[..]);
if read > 0 { if read > 0 {
let bmp = Bmp::from_slice(&bmp_buf).expect("failed to parse bmp"); let bmp = Bmp::from_slice(&bmp_buf).expect("failed to parse bmp");
let row = images_drawn / grid_cols; let row = images_drawn / grid_cols;
let col = images_drawn % grid_cols; let col = images_drawn % grid_cols;
let cell_x = col * cell_width; let cell_x = col * cell_width;
let cell_y = row * cell_height; let cell_y = row * cell_height;
// Center image inside cell // Center image inside cell
let bmp_w = bmp.size().width as i32; let bmp_w = bmp.size().width as i32;
let bmp_h = bmp.size().height as i32; let bmp_h = bmp.size().height as i32;
let x = cell_x + (cell_width - bmp_w) / 2; let x = cell_x + (cell_width - bmp_w) / 2;
let y = cell_y + 5; // 5px top margin let y = cell_y + 5; // 5px top margin
Image::new(&bmp, Point::new(x, y)) Image::new(&bmp, Point::new(x, y))
.draw(&mut display)
.unwrap();
let text_style = MonoTextStyle::new(&FONT_6X10, Rgb565::WHITE);
let text_y = y + bmp_h + 2; // 2px gap under image
Text::new(
f.name.to_string().as_str(),
Point::new(cell_x + 2, text_y),
text_style,
)
.draw(&mut display) .draw(&mut display)
.unwrap(); .unwrap();
images_drawn += 1; let text_style = MonoTextStyle::new(&FONT_6X10, Rgb565::WHITE);
} let text_y = y + bmp_h + 2; // 2px gap under image
Text::new(&file.base(), Point::new(cell_x + 2, text_y), text_style)
.draw(&mut display)
.unwrap();
images_drawn += 1;
} }
} }
} }

View File

@@ -6,4 +6,5 @@ edition = "2024"
[dependencies] [dependencies]
abi = { path = "../../abi" } abi = { path = "../../abi" }
embedded-graphics = "0.8.1" embedded-graphics = "0.8.1"
selection_ui = { path = "../../selection_ui" }
tinygif = { git = "https://github.com/LegitCamper/tinygif" } tinygif = { git = "https://github.com/LegitCamper/tinygif" }

View File

@@ -4,16 +4,17 @@
extern crate alloc; extern crate alloc;
use abi::{ use abi::{
display::Display, display::Display,
fs::{file_len, read_file}, fs::{Entries, file_len, list_dir, read_file},
get_key, get_ms, get_key, get_ms,
keyboard::{KeyCode, KeyState}, keyboard::{KeyCode, KeyState},
print, sleep, print, sleep,
}; };
use alloc::vec; use alloc::{format, vec, vec::Vec};
use core::panic::PanicInfo; use core::panic::PanicInfo;
use embedded_graphics::{ use embedded_graphics::{
image::ImageDrawable, pixelcolor::Rgb565, prelude::Point, transform::Transform, image::ImageDrawable, pixelcolor::Rgb565, prelude::Point, transform::Transform,
}; };
use selection_ui::SelectionUi;
use tinygif::Gif; use tinygif::Gif;
#[panic_handler] #[panic_handler]
@@ -31,13 +32,25 @@ pub fn main() {
print!("Starting Gif app"); print!("Starting Gif app");
let mut display = Display; let mut display = Display;
let size = file_len("/gifs/bad_apple.gif"); let mut gifs = Entries::new();
list_dir("/gifs", &mut gifs);
gifs.entries()
.retain(|e| e.extension().unwrap_or("") == "gif");
let mut selection_ui = SelectionUi::new(&gifs, "No Gif files found in /gifs");
let selection = selection_ui
.run_selection_ui(&mut display)
.expect("failed to draw")
.expect("Failed to get user selection");
let size = file_len(&format!("/gifs/{}.gif", gifs.entries()[selection]));
let mut buf = vec![0_u8; size]; let mut buf = vec![0_u8; size];
let read = read_file("/gifs/bad_apple.gif", 0, &mut buf); let read = read_file("/gifs/bad_apple.gif", 0, &mut buf);
print!("read: {}, file size: {}", read, size); print!("read: {}, file size: {}", read, size);
assert!(read == size); assert!(read == size);
let gif = Gif::<Rgb565>::from_slice(&buf).unwrap(); let gif = Gif::<Rgb565>::from_slice(&buf).expect("Failed to parse gif");
let height = gif.height(); let height = gif.height();
let mut frame_num = 0; let mut frame_num = 0;