From 4f6f91b69b17b318a5e327192804499cac77d177 Mon Sep 17 00:00:00 2001
From: John Breaux <j@soft.fish>
Date: Fri, 14 Apr 2023 20:58:28 -0500
Subject: [PATCH] bus.rs: Debug print screen with drawille, if enabled

---
 Cargo.toml | 15 +++++++++------
 src/bus.rs | 32 ++++++++++++++++++++++++++++++--
 2 files changed, 39 insertions(+), 8 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 75e6eb6..05145dc 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -6,8 +6,9 @@ authors = ["John Breaux"]
 license = "MIT"
 
 [features]
-default = ["unstable"]
+default = ["unstable", "drawille"]
 unstable = []
+drawille = ["dep:drawille"]
 iced = ["dep:iced"]
 rhexdump = ["dep:rhexdump"]
 serde = ["dep:serde"]
@@ -26,12 +27,14 @@ overflow-checks = false
 
 
 [dependencies]
-gumdrop = "^0.8.1"
+drawille = {version = "0.3.0", optional = true}
 iced = {version = "0.8.0", optional = true}
-imperative-rs = "0.3.1"
-minifb = { version = "^0.24.0", features = ["wayland"] }
-owo-colors = "^3"
-rand = "^0.8.5"
 rhexdump = {version = "^0.1.1", optional = true }
 serde = { version = "^1.0", features = ["derive"], optional = true }
+
+gumdrop = "^0.8.1"
+imperative-rs = "0.3.1"
+minifb = { version = "^0.24.0" }
+owo-colors = "^3"
+rand = "^0.8.5"
 thiserror = "^1.0.39"
diff --git a/src/bus.rs b/src/bus.rs
index 2228ddd..8efb185 100644
--- a/src/bus.rs
+++ b/src/bus.rs
@@ -324,15 +324,43 @@ impl Bus {
     pub fn print_screen(&self) -> Result<()> {
         const REGION: Region = Region::Screen;
         if let Some(screen) = self.get_region(REGION) {
+            let len_log2 = screen.len().ilog2() / 2;
+            #[allow(unused_variables)]
+            let (width, height) = (2u32.pow(len_log2 - 1), 2u32.pow(len_log2));
+            // draw with the drawille library, if available
+            #[cfg(feature = "drawille")]
+            {
+                use drawille::Canvas;
+                let mut canvas = Canvas::new(width * 8, height);
+                let width = width * 8;
+                screen
+                    .iter()
+                    .enumerate()
+                    .flat_map(|(bytei, byte)| {
+                        (0..8)
+                            .into_iter()
+                            .enumerate()
+                            .filter_map(move |(biti, bit)| {
+                                if (byte << bit) & 0x80 != 0 {
+                                    Some(bytei * 8 + biti)
+                                } else {
+                                    None
+                                }
+                            })
+                    })
+                    .for_each(|index| canvas.set(index as u32 % (width), index as u32 / (width)));
+                println!("{}", canvas.frame());
+            }
+            #[cfg(not(feature = "drawille"))]
             for (index, byte) in screen.iter().enumerate() {
-                if index % 16 == 0 {
+                if index % width as usize == 0 {
                     print!("{index:03x}|");
                 }
                 print!(
                     "{}",
                     format!("{byte:08b}").replace('0', " ").replace('1', "█")
                 );
-                if index % 16 == 15 {
+                if index % width as usize == width as usize - 1 {
                     println!("|");
                 }
             }