bendn's repos / fimg
fast image operations
Diffstat (limited to 'src/show.rs')
-rw-r--r--src/show.rs165
1 files changed, 165 insertions, 0 deletions
diff --git a/src/show.rs b/src/show.rs
new file mode 100644
index 0000000..dc1ee4f
--- /dev/null
+++ b/src/show.rs
@@ -0,0 +1,165 @@
+use crate::Image;
+
+#[cfg(feature = "real-show")]
+mod real {
+ use crate::Image;
+ use minifb::{Key, Window};
+
+ pub fn show<const CHANNELS: usize>(i: Image<&[u8], CHANNELS>)
+ where
+ [u8; CHANNELS]: crate::Pack,
+ {
+ let mut win = Window::new(
+ "show",
+ i.width() as usize,
+ i.height() as usize,
+ Default::default(),
+ )
+ .unwrap();
+ win.set_title("image viewer");
+ let font = fontdue::Font::from_bytes(
+ &include_bytes!("../data/CascadiaCode.ttf")[..],
+ fontdue::FontSettings {
+ scale: 12.0,
+ ..Default::default()
+ },
+ )
+ .unwrap();
+ while win.is_open() && !win.is_key_down(Key::Q) && !win.is_key_down(Key::Escape) {
+ let mut buf = Image::<Box<[u32]>, 1>::from(i.as_ref());
+
+ if !win.is_key_down(Key::H)
+ && let Some((x, y)) = win
+ .get_mouse_pos(minifb::MouseMode::Discard)
+ .map(|(x, y)| (x.round() as u32, y.round() as u32))
+ .map(|(x, y)| (x.min(i.width()), y.min(i.height())))
+ {
+ // SAFETY: ctor
+ let pix = unsafe { i.pixel(x, y) };
+ unsafe { Image::new(buf.width, buf.height, &mut *buf.buffer) }.text(
+ 5,
+ i.height() - 20,
+ 12.0,
+ &font,
+ &format!(
+ "P ({x}, {y}), {}",
+ // SAFETY: clampd
+ match &pix[..] {
+ [y] => format!("(Y {y})"),
+ [y, a] => format!("(Y {y} A {a})"),
+ [r, g, b] => format!("(R {r} G {g} B {b})"),
+ [r, g, b, a] => format!("(R {r} G {g} B {b} A {a})"),
+ _ => unreachable!(),
+ }
+ ),
+ [238, 232, 213, 255],
+ )
+ }
+ win.update_with_buffer(&buf.buffer, i.width() as usize, i.height() as usize)
+ .expect("window update fail");
+ }
+ }
+}
+
+#[cfg(not(feature = "real-show"))]
+mod fake {
+ use std::process::{Command, Stdio};
+
+ macro_rules! c {
+ ($p:literal) => {
+ std::process::Command::new($p)
+ };
+ ($p:literal $($args:expr)+) => {
+ std::process::Command::new($p).args([$($args,)+])
+ }
+ }
+ pub(crate) use c;
+
+ pub fn has(c: &'static str) -> bool {
+ complete(c!("which").arg(c))
+ }
+
+ pub fn complete(c: &mut Command) -> bool {
+ c.stdout(Stdio::null())
+ .stderr(Stdio::null())
+ .status()
+ .expect("ok")
+ .success()
+ }
+
+ macro_rules! show {
+ ($me:expr) => {
+ let file = std::env::temp_dir().join("viewing.png");
+ $me.save(&file);
+ #[cfg(target_family = "windows")]
+ assert!(fake::complete(fake::c!("start" "%Temp%/viewing.png")), "command should complete successfully.");
+ #[cfg(target_family = "unix")]
+ assert!(
+ if fake::has("feh") { fake::complete(fake::c!("feh" file))
+ } else if fake::has("xdg-open") { fake::complete(fake::c!("xdg-open" file))
+ } else if fake::has("gio") { fake::complete(fake::c!("gio" file))
+ } else if fake::has("gnome-open") { fake::complete(fake::c!("gnome-open" file))
+ } else if fake::has("kde-open") { fake::complete(fake::c!("kde-open" file))
+ } else if fake::has("open") { fake::complete(fake::c!("open" file))
+ } else { panic!("no image viewer found, please enable the `real-show` feature.") },
+ "command should complete successfully.");
+ };
+
+ }
+ pub(crate) use show;
+}
+
+fn r(i: &Image<Box<[u32]>, 1>) -> Image<&[u32], 1> {
+ // SAFETY: ctor
+ unsafe { Image::new(i.width, i.height, &*i.buffer) }
+}
+
+macro_rules! show {
+ ($buf:ty) => {
+ show!($buf, 1);
+ show!($buf, 2);
+ show!($buf, 3);
+ show!($buf, 4);
+ };
+ ($buf:ty, $n:literal) => {
+ impl Image<$buf, $n> {
+ /// Open a window showing this image.
+ /// Blocks until the window finishes.
+ ///
+ /// This is like [`dbg!`] for images.
+ ///
+ /// # Panics
+ ///
+ /// if the window is un creatable
+ pub fn show(self) -> Self {
+ #[cfg(feature = "real-show")]
+ real::show(self.as_ref());
+ #[cfg(not(feature = "real-show"))]
+ fake::show!(self);
+ self
+ }
+ }
+ };
+}
+
+show!(Vec<u8>);
+show!(Box<[u8]>);
+show!(&[u8]);
+
+impl Image<Box<[u32]>, 1> {
+ /// Open a window showing this image.
+ /// Blocks until the window finishes.
+ ///
+ /// This is like [`dbg!`] for images.
+ ///
+ /// # Panics
+ ///
+ /// if the window is un creatable
+ pub fn show(self) -> Self {
+ #[cfg(feature = "real-show")]
+ real::show(Image::<Box<[u8]>, 4>::from(r(&self)).as_ref());
+ #[cfg(not(feature = "real-show"))]
+ fake::show!(Image::<Box<[u8]>, 4>::from(r(&self)));
+ self
+ }
+}