working on colour support

lib/lib_framebuffer/src/colour.rs

@@ -0,0 +1,51 @@
+#[repr(u32)]
+#[derive(Copy, Clone, PartialEq, Eq, Debug)]
+pub enum Color {
+    ARGB(u8, u8, u8, u8),
+    RGB(u8, u8, u8),
+    HexARGB(u32),
+    Black = 0x000000FF,
+    Blue = 0x0000FFFF,
+    Green = 0x00FF00FF,
+    Cyan = 0x00FFFFFF,
+    Red = 0xFF0000FF,
+    Magenta = 0xFF00FFFF,
+    Yellow = 0xFFFF00FF,
+    White = 0xFFFFFFFF,
+}
+
+impl Into<u32> for Color {
+    fn into(self) -> u32 {
+        match self {
+            Color::ARGB(a, r, g, b) => (a as u32) << 24 | (r as u32) << 16 | (g as u32) << 8 | (b as u32),
+            Color::RGB(r, g, b) => (0u32) << 24 | (r as u32) << 16 | (g as u32) << 8 | (b as u32),
+            Color::HexARGB(hex) => hex,
+            Color::Black => 0xFF000000,
+            Color::Blue => 0xFF0000FF, 
+            Color::Green => 0xFF00FF00, 
+            Color::Cyan => 0xFF00FFFF,
+            Color::Red => 0xFFFF0000,   
+            Color::Magenta => 0xFFFF00FF, 
+            Color::Yellow => 0xFFFFFF00, 
+            Color::White => 0xFFFFFFFF, 
+        }
+    }
+}
+
+impl core::fmt::Display for Color {
+    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
+        match self {
+            Color::ARGB(r, g, b, a) => write!(f, "RGBA(#{:x}{:x}{:x}{:x})", r, g, b, a),
+            Color::RGB(r, g, b) => write!(f, "RGB(#{:x}{:x}{:x})", r, g, b),
+            Color::HexARGB(hex) => write!(f, "Hex(#{:x})", hex),
+            Color::Black => write!(f, "Black"),
+            Color::Blue => write!(f, "Blue"),
+            Color::Green => write!(f, "Green"),
+            Color::Cyan => write!(f, "Cyan"),
+            Color::Red => write!(f, "Red"),
+            Color::Magenta => write!(f, "Magenta"),
+            Color::Yellow => write!(f, "Yellow"),
+            Color::White => write!(f, "White"),
+        }
+    }
+}

lib/lib_framebuffer/src/draw.rs

@@ -1,57 +0,0 @@
-use core::panic;
-use limine::framebuffer::Framebuffer;
-use spin::{Mutex, Lazy};
-
-use crate::FRAMEBUFFER_REQUEST;
-
-pub static FRAMEBUFFER_WRITER: Lazy<Mutex<Option<FramebufferWriter>>> = Lazy::new(|| Mutex::new(
-    if let Some(framebuffer_response) = FRAMEBUFFER_REQUEST.get_response() {
-        let framebuffer = framebuffer_response.framebuffers().next().unwrap();
-        Some(FramebufferWriter::new(framebuffer))
-    } else {
-        panic!("Framebuffer request failed");
-    }
-));
-
-pub struct FramebufferWriter<'a> {
-    framebuffer: Framebuffer<'a>,
-}
-
-unsafe impl<'a> Send for FramebufferWriter<'a> {}
-unsafe impl<'a> Sync for FramebufferWriter<'a> {}
-impl<'a> FramebufferWriter<'a> {
-    pub fn new(framebuffer: Framebuffer<'a>) -> Self {
-        Self {
-            framebuffer,
-        }
-    }
-
-    pub fn write_pixel(&self, x: usize, y: usize, color: u32) {
-        let pitch = self.framebuffer.pitch() as usize;
-        let bpp = (self.framebuffer.bpp() / 8) as usize;
-        let pixel_offset = y * pitch + x * bpp;
-
-        unsafe {
-            *(self.framebuffer.addr().add(pixel_offset) as *mut u32) = color;
-        }
-    }
-
-    pub fn width(&self) -> u32 {
-        self.framebuffer.width() as u32
-    }
-
-    pub fn height(&self) -> u32 {
-        self.framebuffer.height() as u32
-    }
-
-    pub fn clear(&self) {
-        let width = self.framebuffer.width() as usize;
-        let height = self.framebuffer.height() as usize;
-
-        for y in 0..height {
-            for x in 0..width {
-                self.write_pixel(x, y, 0x000000);
-            }
-        }
-    }
-}

lib/lib_framebuffer/src/framebuffer.rs

@@ -0,0 +1,67 @@
+use core::panic;
+use limine::framebuffer::Framebuffer;
+use spin::{Mutex, Lazy};
+
+use crate::{colour::Color, FRAMEBUFFER_REQUEST};
+
+pub static FRAMEBUFFER_WRITER: Lazy<Mutex<Option<FramebufferWriter>>> = Lazy::new(|| Mutex::new(
+    if let Some(framebuffer_response) = FRAMEBUFFER_REQUEST.get_response() {
+        let framebuffer = framebuffer_response.framebuffers().next().unwrap();
+        Some(FramebufferWriter::new(framebuffer))
+    } else {
+        panic!("Framebuffer request failed");
+    }
+));
+
+
+
+pub struct FramebufferWriter<'a> {
+    framebuffer: Framebuffer<'a>,
+}
+
+unsafe impl<'a> Send for FramebufferWriter<'a> {}
+unsafe impl<'a> Sync for FramebufferWriter<'a> {}
+impl<'a> FramebufferWriter<'a> {
+    pub fn new(framebuffer: Framebuffer<'a>) -> Self {
+        Self {
+            framebuffer,
+        }
+    }
+
+    pub fn write_pixel(&self, x: usize, y: usize, color: Color) {
+        let pitch = self.framebuffer.pitch() as usize;
+        let bpp = (self.framebuffer.bpp() / 8) as usize;
+        let pixel_offset = y * pitch + x * bpp;
+
+        unsafe {
+            *(self.framebuffer.addr().add(pixel_offset) as *mut u32) = color.into();
+        }
+    }
+
+    pub fn render_frame(&self, buffer: &[Color; 1280 * 800]) {
+        for (y, row) in buffer.chunks(1280).enumerate() {
+            for (x, pixel) in row.iter().enumerate() {
+                self.write_pixel(x, y, *pixel);
+            }
+        }
+    }
+
+    pub fn width(&self) -> u32 {
+        self.framebuffer.width() as u32
+    }
+
+    pub fn height(&self) -> u32 {
+        self.framebuffer.height() as u32
+    }
+
+    pub fn clear(&self) {
+        let width = self.framebuffer.width() as usize;
+        let height = self.framebuffer.height() as usize;
+
+        for y in 0..height {
+            for x in 0..width {
+                self.write_pixel(x, y, Color::Black);
+            }
+        }
+    }
+}

lib/lib_framebuffer/src/lib.rs

@@ -6,8 +6,10 @@ use limine::request::FramebufferRequest;
 #[link_section = ".requests"]
 static FRAMEBUFFER_REQUEST: FramebufferRequest = FramebufferRequest::new();
 
-mod draw;
-pub use draw::FRAMEBUFFER_WRITER;
+mod framebuffer;
+mod colour;
+pub use framebuffer::FRAMEBUFFER_WRITER;
+pub use colour::Color;
 
 pub fn screensize_px() -> (u32, u32) {
     if let Some(writer) = FRAMEBUFFER_WRITER.lock().as_mut() {