From 1c12173b778dee5a797be2954604379fe7b969fd Mon Sep 17 00:00:00 2001
From: Empire <empire@example.com>
Date: Thu, 25 May 2023 21:27:13 +0200
Subject: [PATCH] write image

---
 client/Cargo.toml  |   1 +
 client/src/main.rs | 106 +++++++++++++++++++++++++--------------------
 2 files changed, 59 insertions(+), 48 deletions(-)

diff --git a/client/Cargo.toml b/client/Cargo.toml
index 23c0542..919d0c4 100644
--- a/client/Cargo.toml
+++ b/client/Cargo.toml
@@ -6,4 +6,5 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+bit = "0.1.1"
 embedded-graphics = "0.8.0"
diff --git a/client/src/main.rs b/client/src/main.rs
index 4b4bd6b..01b74d7 100644
--- a/client/src/main.rs
+++ b/client/src/main.rs
@@ -1,37 +1,41 @@
-use std::net::UdpSocket;
-use std::env;
+use bit::BitIndex;
 use embedded_graphics::{
     image::{Image, ImageRaw},
     pixelcolor::{BinaryColor, Rgb565},
-    prelude::*, primitives::{PrimitiveStyle, Line},
+    prelude::*,
+    primitives::{Line, PrimitiveStyle},
 };
+use std::env;
+use std::net::UdpSocket;
 
+const IMAGE_SIZE_BYTE: usize = (IMAGE_WIDTH_BYTE * IMAGE_HEIGHT) as usize; /* one byte contains 8 LEDs, one in each bit */
+const IMAGE_WIDTH: u32 = 5 * 32;
+const IMAGE_WIDTH_BYTE: u32 = IMAGE_WIDTH / 8; /* one byte contains 8 LEDs, one in each bit */
 
-const IMAGE_SIZE_BYTE : usize = (IMAGE_WIDTH_BYTE * IMAGE_HEIGHT) as usize; /* one byte contains 8 LEDs, one in each bit */
-const IMAGE_WIDTH : u32 = 5*32;
-const IMAGE_WIDTH_BYTE : u32 = IMAGE_WIDTH / 8; /* one byte contains 8 LEDs, one in each bit */
+const IMAGE_HEIGHT: u32 = 40;
+const IMAGE_HEIGHT_BYTE: u32 = 40;
+const IMAGE_LENGTH: usize = (IMAGE_WIDTH_BYTE * IMAGE_HEIGHT_BYTE) as usize;
+const PACKAGE_LENGTH: usize = (IMAGE_LENGTH + 1) as usize;
 
-const IMAGE_HEIGHT : u32 = 40;
-const IMAGE_HEIGHT_BYTE : u32 = 40;
-const IMAGE_LENGTH : usize = (IMAGE_WIDTH_BYTE * IMAGE_HEIGHT_BYTE) as usize;
-const PACKAGE_LENGTH : usize = (IMAGE_LENGTH + 1) as usize;
-
-
-struct UdpDisplay {
-    imageSlice: [u8; IMAGE_SIZE_BYTE]
+struct UdpDisplay<'a> {
+    image_slice: &'a mut[u8; IMAGE_SIZE_BYTE],
 }
 
-impl OriginDimensions for UdpDisplay {
+impl OriginDimensions for UdpDisplay<'_> {
     fn size(&self) -> Size {
         Size::new(IMAGE_WIDTH, IMAGE_HEIGHT)
     }
 }
 
-impl DrawTarget for UdpDisplay {
+impl DrawTarget for UdpDisplay<'_> {
     type Color = BinaryColor;
     type Error = core::convert::Infallible;
 
-    fn fill_contiguous<I>(&mut self, area: &embedded_graphics::primitives::Rectangle, colors: I) -> Result<(), Self::Error>
+    fn fill_contiguous<I>(
+        &mut self,
+        area: &embedded_graphics::primitives::Rectangle,
+        colors: I,
+    ) -> Result<(), Self::Error>
     where
         I: IntoIterator<Item = Self::Color>,
     {
@@ -42,7 +46,11 @@ impl DrawTarget for UdpDisplay {
         )
     }
 
-    fn fill_solid(&mut self, area: &embedded_graphics::primitives::Rectangle, color: Self::Color) -> Result<(), Self::Error> {
+    fn fill_solid(
+        &mut self,
+        area: &embedded_graphics::primitives::Rectangle,
+        color: Self::Color,
+    ) -> Result<(), Self::Error> {
         self.fill_contiguous(area, core::iter::repeat(color))
     }
 
@@ -52,49 +60,51 @@ impl DrawTarget for UdpDisplay {
 
     fn draw_iter<I>(&mut self, pixels: I) -> Result<(), Self::Error>
     where
-        I: IntoIterator<Item = Pixel<Self::Color>> {
-        todo!()
+        I: IntoIterator<Item = Pixel<Self::Color>>,
+    {
+        for pixel in pixels {
+            let x = pixel.0.x as u32;
+            let y = pixel.0.y as u32;
+            let v = pixel.1.is_on();
+            let offset: usize = (x + y * IMAGE_WIDTH) as usize;
+            
+            let subbit: usize = (offset % 8).into();
+            let byte_offset: usize = (offset / 8).into();
+
+            let current =  &mut self.image_slice[byte_offset];
+            current.set_bit(subbit, v);
+        }
+        return Ok(());
     }
-
 }
 
-
 fn send_package(ipaddress: String) {
     let mut package = [0; PACKAGE_LENGTH];
+    // Brightness
     package[0] = 128;
-    let imageSlice: &mut[u8;IMAGE_SIZE_BYTE] = &mut package[1..PACKAGE_LENGTH].try_into().unwrap();
+    let image_slice: &mut [u8; IMAGE_SIZE_BYTE] =
+        &mut package[1..PACKAGE_LENGTH].try_into().unwrap();
 
-
-    let mut display = UdpDisplay {
-        imageSlice
-    };
-
-
-    const IMAGE: &[u8] = &[0 as u8; ((IMAGE_WIDTH_BYTE * IMAGE_HEIGHT_BYTE) as usize)];
-    let raw_image = ImageRaw::<BinaryColor>::new(IMAGE, IMAGE_WIDTH);
+    let mut display = UdpDisplay { image_slice };
 
     Line::new(Point::new(50, 20), Point::new(60, 35))
-    .into_styled(PrimitiveStyle::with_stroke(BinaryColor::On, 1))
-    .draw(&mut raw_image)?;
-
-    // Brightness
-    
-    imageSlice.copy_from_slice(&IMAGE);
-
-
-    // TODO convert RgbImage into image buffer:
-    // let pixel = img[(100, 100)];::
+        .into_styled(PrimitiveStyle::with_stroke(BinaryColor::On, 1))
+        .draw(&mut display).unwrap();
 
     let socket = UdpSocket::bind("0.0.0.0:4242").expect("couldn't bind to address");
-    socket.send_to(&package, ipaddress + ":4242").expect("couldn't send data");
+    socket
+        .send_to(&package, ipaddress + ":4242")
+        .expect("couldn't send data");
     println!("Packet sent");
 }
 
 fn help() {
-    println!("usage:
-LEDboardClient <ip address>");
-println!("one argument necessary!");
-            println!("<ip address>");
+    println!(
+        "usage:
+LEDboardClient <ip address>"
+    );
+    println!("one argument necessary!");
+    println!("<ip address>");
 }
 
 fn main() {
@@ -105,12 +115,12 @@ fn main() {
         1 => {
             // show a help message
             help();
-        },
+        }
         // one argument passed
         2 => {
             let ip = &args[1];
             send_package(ip.to_string());
-        },
+        }
         // all the other cases
         _ => {
             // show a help message