chlorospingus/chloroway
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use structs with Drawable trait instead of methods on shm_pool

Browse Source
This commit is contained in:
Chlorospingus
2025-05-13 21:54:59 -07:00
parent 7dcf3d80bb
commit 9d0b9b97b5
5 changed files with 111 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/graphics/circle.rs Normal file
View File

@@ -0,0 +1,39 @@
use crate::wayland::shm::ShmPool;
use super::drawable::{Drawable, color_blend};
// x and y are center of circle
pub struct Circle {
x: usize,
y: usize,
radius: usize,
color: u32,
}
impl Circle {
pub fn new(x: usize, y: usize, radius: usize, color: u32) -> Self {
Circle { x, y, radius, color }
}
}
impl Drawable for Circle {
fn draw(&self, shm_pool: &mut ShmPool) {
for l_row in 1..self.radius {
let inner_diff = (((self.radius-1).pow(2) - l_row.pow(2)) as f64).sqrt();
let outer_diff = ((self.radius.pow(2) - l_row.pow(2)) as f64).sqrt();
let row: Vec<u32> = vec![self.color; 2*(inner_diff.floor() as usize)];
shm_pool.write(&row, (self.y-l_row)*shm_pool.width + self.x - inner_diff.floor() as usize);
shm_pool.write(&row, (self.y+l_row-1)*shm_pool.width + self.x - inner_diff.floor() as usize);
for l_col in (inner_diff.floor() as usize+1)..(outer_diff.ceil() as usize) {
let distance = ((l_row.pow(2) + l_col.pow(2)) as f64).sqrt();
let offset = (self.y-l_row)*shm_pool.width + self.x - l_col;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y-l_row)*shm_pool.width + self.x + l_col-1;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+l_row-1)*shm_pool.width + self.x - l_col;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+l_row-1)*shm_pool.width + self.x + l_col-1;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
}
}
}
}

31
src/graphics/drawable.rs Normal file
View File

@@ -0,0 +1,31 @@
use crate::wayland::shm::ShmPool;
pub fn color_blend(col1: u32, col2: u32, diff: f64) -> u32 {
// TODO: Account for alpha channel
let r1 = (col1 & 0x00ff0000) >> 16;
let g1 = (col1 & 0x0000ff00) >> 8;
let b1 = col1 & 0x000000ff;
let r2 = (col2 & 0x00ff0000) >> 16;
let g2 = (col2 & 0x0000ff00) >> 8;
let b2 = col2 & 0x000000ff;
let r3 = if r1 < r2 {
r1 + ((r2 - r1) as f64 * diff) as u32
} else {
r1 - ((r1 - r2) as f64 * diff) as u32
};
let g3 = if g1 < g2 {
g1 + ((g2 - g1) as f64 * diff) as u32
} else {
g1 - ((g1 - g2) as f64 * diff) as u32
};
let b3 = if b1 < b2 {
b1 + ((b2 - b1) as f64 * diff) as u32
} else {
b1 - ((b1 - b2) as f64 * diff) as u32
};
return 0xff000000 + (r3 << 16) + (g3 << 8) + b3;
}
pub trait Drawable {
fn draw(&self, shm_pool: &mut ShmPool);
}

2
src/graphics/mod.rs
View File

@@ -1,2 +1,4 @@
pub mod drawable;
pub mod rectangle;
pub mod circle;

115
src/graphics/rectangle.rs
View File

@@ -1,94 +1,45 @@
use std::usize;
use crate::wayland::shm::ShmPool;
use super::drawable::{Drawable, color_blend};
fn color_blend(col1: u32, col2: u32, diff: f64) -> u32 {
// TODO: Account for alpha channel
let r1 = (col1 & 0x00ff0000) >> 16;
let g1 = (col1 & 0x0000ff00) >> 8;
let b1 = col1 & 0x000000ff;
let r2 = (col2 & 0x00ff0000) >> 16;
let g2 = (col2 & 0x0000ff00) >> 8;
let b2 = col2 & 0x000000ff;
let r3 = if r1 < r2 {
r1 + ((r2 - r1) as f64 * diff) as u32
} else {
r1 - ((r1 - r2) as f64 * diff) as u32
};
let g3 = if g1 < g2 {
g1 + ((g2 - g1) as f64 * diff) as u32
} else {
g1 - ((g1 - g2) as f64 * diff) as u32
};
let b3 = if b1 < b2 {
b1 + ((b2 - b1) as f64 * diff) as u32
} else {
b1 - ((b1 - b2) as f64 * diff) as u32
};
return 0xff000000 + (r3 << 16) + (g3 << 8) + b3;
// x and y and topleft corner of rect
pub struct Rectangle {
x: usize,
y: usize,
width: usize,
height: usize,
radius: usize,
color: u32,
}
// l_row means local row and g_row means global row
impl ShmPool {
// x and y are topleft corner of rect
pub fn rectangle(&mut self, x: usize, y: usize, w: usize, h: usize, color: u32) {
for g_row in y..y+h {
self.write(&vec![color; w], g_row*self.width+x);
}
impl Rectangle {
pub fn new(x: usize, y: usize, width: usize, height: usize, radius: usize, color: u32) -> Self {
Rectangle {x, y, width, height, radius, color}
}
}
// x and y and topleft corner of rect
pub fn rounded_rectangle(
&mut self,
x: usize,
y: usize,
w: usize,
h: usize,
radius: usize,
color: u32
) {
for l_row in 1..radius {
let inner_diff = (((radius-1).pow(2) - l_row.pow(2)) as f64).sqrt();
let outer_diff = ((radius.pow(2) - l_row.pow(2)) as f64).sqrt();
self.write(&vec![color; w - (2*(radius - inner_diff.floor() as usize-1))], (y+radius-l_row)*self.width + x + radius - inner_diff.floor() as usize-1);
self.write(&vec![color; w - (2*(radius - inner_diff.floor() as usize-1))], (y+h-radius+l_row)*self.width + x + radius - inner_diff.floor() as usize-1);
impl Drawable for Rectangle {
fn draw(&self, shm_pool: &mut ShmPool) {
for g_row in self.y+self.radius..self.y+self.height-self.radius+1 {
shm_pool.write(&vec![self.color; self.width], g_row*shm_pool.width+self.x);
}
for l_row in 1..self.radius {
let inner_diff = (((self.radius-1).pow(2) - l_row.pow(2)) as f64).sqrt();
let outer_diff = ((self.radius.pow(2) - l_row.pow(2)) as f64).sqrt();
shm_pool.write(&vec![self.color; self.width - (2*(self.radius - inner_diff.floor() as usize-1))], (self.y+self.radius-l_row)*shm_pool.width + self.x + self.radius - inner_diff.floor() as usize-1);
shm_pool.write(&vec![self.color; self.width - (2*(self.radius - inner_diff.floor() as usize-1))], (self.y+self.height-self.radius+l_row)*shm_pool.width + self.x + self.radius - inner_diff.floor() as usize-1);
for l_col in inner_diff.floor() as usize+1..outer_diff.ceil() as usize {
let distance = ((l_row.pow(2) + l_col.pow(2)) as f64).sqrt();
let offset = (y+radius-l_row)*self.width + x + radius - l_col - 1;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y+radius-l_row)*self.width + x + w - radius + l_col;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y+h-radius+l_row)*self.width + x + radius - l_col - 1;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y+h-radius+l_row)*self.width + x + w - radius + l_col;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
}
}
for g_row in y+radius..y+h-radius+1 {
self.write(&vec![color; w], g_row*self.width+x);
}
}
// x and y are center of circle
pub fn circle(&mut self, x: usize, y: usize, radius: usize, color: u32) {
for l_row in 1..radius {
let inner_diff = (((radius-1).pow(2) - l_row.pow(2)) as f64).sqrt();
let outer_diff = ((radius.pow(2) - l_row.pow(2)) as f64).sqrt();
let row: Vec<u32> = vec![color; 2*(inner_diff.floor() as usize)];
self.write(&row, (y-l_row)*self.width + x - inner_diff.floor() as usize);
self.write(&row, (y+l_row-1)*self.width + x - inner_diff.floor() as usize);
for l_col in (inner_diff.floor() as usize+1)..(outer_diff.ceil() as usize) {
let distance = ((l_row.pow(2) + l_col.pow(2)) as f64).sqrt();
let offset = (y-l_row)*self.width + x - l_col;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y-l_row)*self.width + x + l_col-1;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y+l_row-1)*self.width + x - l_col;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (y+l_row-1)*self.width + x + l_col-1;
self.write_pixel(color_blend(color, self.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+self.radius-l_row)*shm_pool.width + self.x + self.radius - l_col - 1;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+self.radius-l_row)*shm_pool.width + self.x + self.width - self.radius + l_col;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+self.height-self.radius+l_row)*shm_pool.width + self.x + self.radius - l_col - 1;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
let offset = (self.y+self.height-self.radius+l_row)*shm_pool.width + self.x + self.width - self.radius + l_col;
shm_pool.write_pixel(color_blend(self.color, shm_pool.read_pixel(offset), distance.fract()), offset as isize);
}
}
}
}