remove gui

This commit is contained in:
saji 2024-07-16 18:55:59 -05:00
parent 227147e411
commit a161fdb011
6 changed files with 85 additions and 30 deletions

40
Cargo.lock generated
View file

@ -2109,6 +2109,16 @@ dependencies = [
"libc", "libc",
] ]
[[package]]
name = "matrixmultiply"
version = "0.3.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "7574c1cf36da4798ab73da5b215bbf444f50718207754cb522201d78d1cd0ff2"
dependencies = [
"autocfg",
"rawpointer",
]
[[package]] [[package]]
name = "maybe-rayon" name = "maybe-rayon"
version = "0.1.1" version = "0.1.1"
@ -2219,6 +2229,19 @@ version = "1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8d5439c4ad607c3c23abf66de8c8bf57ba8adcd1f129e699851a6e43935d339d" checksum = "8d5439c4ad607c3c23abf66de8c8bf57ba8adcd1f129e699851a6e43935d339d"
[[package]]
name = "ndarray"
version = "0.15.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "adb12d4e967ec485a5f71c6311fe28158e9d6f4bc4a447b474184d0f91a8fa32"
dependencies = [
"matrixmultiply",
"num-complex",
"num-integer",
"num-traits",
"rawpointer",
]
[[package]] [[package]]
name = "ndk" name = "ndk"
version = "0.8.0" version = "0.8.0"
@ -2325,6 +2348,15 @@ dependencies = [
"num-traits", "num-traits",
] ]
[[package]]
name = "num-complex"
version = "0.4.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
dependencies = [
"num-traits",
]
[[package]] [[package]]
name = "num-derive" name = "num-derive"
version = "0.4.2" version = "0.4.2"
@ -2684,10 +2716,10 @@ dependencies = [
"anyhow", "anyhow",
"clap", "clap",
"eframe", "eframe",
"egui",
"epd-waveshare", "epd-waveshare",
"image", "image",
"linux-embedded-hal", "linux-embedded-hal",
"ndarray",
"num-traits", "num-traits",
"palette", "palette",
"rayon", "rayon",
@ -2951,6 +2983,12 @@ version = "0.6.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "20675572f6f24e9e76ef639bc5552774ed45f1c30e2951e1e99c59888861c539" checksum = "20675572f6f24e9e76ef639bc5552774ed45f1c30e2951e1e99c59888861c539"
[[package]]
name = "rawpointer"
version = "0.2.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "60a357793950651c4ed0f3f52338f53b2f809f32d83a07f72909fa13e4c6c1e3"
[[package]] [[package]]
name = "rayon" name = "rayon"
version = "1.10.0" version = "1.10.0"

View file

@ -7,13 +7,18 @@ edition = "2021"
gui = [ "dep:eframe" ] gui = [ "dep:eframe" ]
eink = [ "dep:epd-waveshare", "dep:linux-embedded-hal" ] eink = [ "dep:epd-waveshare", "dep:linux-embedded-hal" ]
[[bin]]
name = "inspector"
required-features = ["gui"]
[dependencies] [dependencies]
anyhow = "1.0.86" anyhow = "1.0.86"
clap = { version = "4.5.7", features = ["derive"] } clap = { version = "4.5.7", features = ["derive"] }
eframe = "0.28.0" eframe = { version = "0.28.0", optional = true }
epd-waveshare = { git = "https://github.com/caemor/epd-waveshare.git" } epd-waveshare = { git = "https://github.com/caemor/epd-waveshare.git", optional = true }
image = "0.25.1" image = "0.25.1"
linux-embedded-hal = { version = "0.4.0" } linux-embedded-hal = { version = "0.4.0", optional = true }
ndarray = "0.15.6"
num-traits = "0.2.19" num-traits = "0.2.19"
palette = "0.7.6" palette = "0.7.6"
rayon = "1.10.0" rayon = "1.10.0"

View file

@ -6,15 +6,15 @@ use linux_embedded_hal::{CdevPin, Delay, SpidevDevice};
use linux_embedded_hal::gpio_cdev::{Chip, LineRequestFlags}; use linux_embedded_hal::gpio_cdev::{Chip, LineRequestFlags};
use anyhow::Result; use anyhow::Result;
pub struct DisplayWrapper { pub struct Wrapper {
spi: SpidevDevice, spi: SpidevDevice,
gpiochip: Chip, gpiochip: Chip,
delay: Delay, delay: Delay,
panel: Epd7in3f<SpidevDevice, CdevPin, CdevPin, CdevPin, Delay> panel: Epd7in3f<SpidevDevice, CdevPin, CdevPin, CdevPin, Delay>
} }
impl DisplayWrapper { impl Wrapper {
pub fn new() -> Result<DisplayWrapper> { pub fn new() -> Result<Wrapper> {
let mut spi = SpidevDevice::open("/dev/spidev0.0")?; let mut spi = SpidevDevice::open("/dev/spidev0.0")?;
let spi_options = SpidevOptions::new() let spi_options = SpidevOptions::new()
.bits_per_word(8) .bits_per_word(8)
@ -42,7 +42,7 @@ impl DisplayWrapper {
let mut delay = Delay {}; let mut delay = Delay {};
let panel = Epd7in3f::new(&mut spi, busy_pin, dc_pin, rst_pin, &mut delay, None)?; let panel = Epd7in3f::new(&mut spi, busy_pin, dc_pin, rst_pin, &mut delay, None)?;
Ok(DisplayWrapper { Ok(Wrapper {
spi, spi,
gpiochip, gpiochip,
delay, delay,

View file

@ -1,7 +1,6 @@
use image::RgbImage; use image::RgbImage;
use palette::{cast::FromComponents, color_difference::Ciede2000, IntoColor, Lab, Srgb}; use palette::{cast::FromComponents, color_difference::Ciede2000, IntoColor, Lab, Oklch, Srgb};
/// Palette used on the display; pixels can be one of these colors. /// Palette used on the display; pixels can be one of these colors.
/// ///
@ -56,7 +55,7 @@ impl DisplayColor {
4 => Self::Red, 4 => Self::Red,
5 => Self::Yellow, 5 => Self::Yellow,
6 => Self::Orange, 6 => Self::Orange,
_ => panic!("unexpected DisplayColor {}", value), _ => panic!("unexpected DisplayColor {value}"),
} }
} }
@ -68,16 +67,17 @@ impl DisplayColor {
} }
impl From<DisplayColor> for u8 { impl From<DisplayColor> for u8 {
fn from(value: DisplayColor) -> Self { fn from(value: DisplayColor) -> Self {
value as u8 value as Self
} }
} }
/// Buffer to be sent to the EInk display. /// Buffer to be sent to the ``EInk`` display.
#[derive(Debug)] #[derive(Debug)]
pub struct EInkBuffer(Vec<DisplayColor>); pub struct EInkBuffer(Vec<DisplayColor>);
impl EInkBuffer { impl EInkBuffer {
#[must_use]
pub fn into_display_buffer(&self) -> Vec<u8> { pub fn into_display_buffer(&self) -> Vec<u8> {
let mut buf = Vec::with_capacity(self.0.len()/2); let mut buf = Vec::with_capacity(self.0.len() / 2);
for colors in self.0.chunks_exact(2) { for colors in self.0.chunks_exact(2) {
buf.push(DisplayColor::into_byte(colors[0], colors[1])); buf.push(DisplayColor::into_byte(colors[0], colors[1]));
@ -85,9 +85,10 @@ impl EInkBuffer {
buf buf
} }
#[must_use]
pub fn new(width: usize, height: usize) -> Self { pub fn new(width: usize, height: usize) -> Self {
let v = vec![DisplayColor::Black; width * height]; let v = vec![DisplayColor::Black; width * height];
EInkBuffer(v) Self(v)
} }
} }
@ -114,7 +115,7 @@ impl EInkBuffer {
// } // }
pub trait Ditherer { pub trait Ditherer {
fn dither(&self, img: &RgbImage, output: &mut EInkBuffer); fn dither(&mut self, img: &RgbImage, output: &mut EInkBuffer);
} }
// fn color_distance(c1: LinSrgb, c2: LinSrgb) -> f32 { // fn color_distance(c1: LinSrgb, c2: LinSrgb) -> f32 {
@ -127,25 +128,23 @@ pub trait Ditherer {
/// Find the closest approximate palette color to the given sRGB value. /// Find the closest approximate palette color to the given sRGB value.
/// This uses euclidian distance in linear space. /// This uses euclidian distance in linear space.
pub fn nearest_neighbor(color: Srgb) -> (DisplayColor, f32) { pub fn nearest_neighbor(input_color: Lab) -> (DisplayColor, Lab) {
let input: Lab = color.into_color(); let (nearest, _, color_diff) = DISPLAY_PALETTE
let (nearest, dist) = DISPLAY_PALETTE
.iter() .iter()
.enumerate() .enumerate()
.map(|(idx, p_color)| { .map(|(idx, p_color)| {
let c: Lab = (*p_color).into_color(); let c: Lab = (*p_color).into_color();
(idx, input.difference(c)) (idx, input_color.difference(c), input_color - c)
}) })
.min_by(|(_, a), (_, b)| a.total_cmp(b)) .min_by(|(_, a, _), (_, b, _)| a.total_cmp(b))
.unwrap(); .unwrap();
(DisplayColor::from_u8(nearest as u8), dist) (DisplayColor::from_u8(nearest as u8), color_diff)
} }
pub struct NNDither(); pub struct NNDither();
impl Ditherer for NNDither { impl Ditherer for NNDither {
fn dither(&self, img: &RgbImage, output: &mut EInkBuffer) { fn dither(&mut self, img: &RgbImage, output: &mut EInkBuffer) {
assert!(img.width() == 800); assert!(img.width() == 800);
assert!(img.height() == 480); assert!(img.height() == 480);
@ -153,10 +152,22 @@ impl Ditherer for NNDither {
let srgb = <&[Srgb<u8>]>::from_components(&**img); let srgb = <&[Srgb<u8>]>::from_components(&**img);
for (idx, pixel) in srgb.iter().enumerate() { for (idx, pixel) in srgb.iter().enumerate() {
let (n, _) = nearest_neighbor(pixel.into_format()); let (n, _) = nearest_neighbor(pixel.into_format().into_color());
output.0[idx] = n; output.0[idx] = n;
} }
} }
} }
pub struct FloydSteinbergDither();
impl Ditherer for FloydSteinbergDither {
fn dither(&mut self, img: &RgbImage, output: &mut EInkBuffer) {
// create a copy of the image in Lab space, mutable.
let srgb = <&[Srgb<u8>]>::from_components(&**img);
let mut temp_img: Vec<Lab> = Vec::new();
for pix in srgb {
temp_img.push(pix.into_format().into_color());
}
}
}

View file

@ -1,7 +1,8 @@
#[cfg(feature = "eink")]
pub mod display; pub mod display;
pub mod imageproc; pub mod imageproc;
use crate::display::DisplayWrapper; use crate::display::Wrapper;
use crate::imageproc::{Ditherer, EInkBuffer, NNDither}; use crate::imageproc::{Ditherer, EInkBuffer, NNDither};
use clap::{Parser, Subcommand}; use clap::{Parser, Subcommand};
use image::RgbImage; use image::RgbImage;
@ -28,7 +29,7 @@ fn main() -> anyhow::Result<()> {
if let Command::Show = cli.command { if let Command::Show = cli.command {
let img: RgbImage = image::io::Reader::open("myimage.png")?.decode()?.into(); let img: RgbImage = image::io::Reader::open("myimage.png")?.decode()?.into();
let mut display = DisplayWrapper::new()?; let mut display = Wrapper::new()?;
let mut eink_buf = EInkBuffer::new(800, 480); let mut eink_buf = EInkBuffer::new(800, 480);
let dither = NNDither {}; let dither = NNDither {};

View file