first not-great line impl

This commit is contained in:
Joe Ardent 2025-09-05 15:26:38 -07:00
parent d51a2dd5f4
commit ba91baa302
3 changed files with 99 additions and 29 deletions

View file

@ -2,9 +2,11 @@
extern crate justerror; extern crate justerror;
mod tga; mod tga;
use tga::*; use tga::*;
mod point;
use point::*;
const WHITE: TGAColor = TGAColor { const WHITE: TGAColor = TGAColor {
bgra: [255, 255, 255, 255], bgra: [255, 255, 255, 255],
}; };
@ -37,13 +39,31 @@ fn main() {
let cx = 62; let cx = 62;
let cy = 53; let cy = 53;
framebuffer.set(ax, ay, &WHITE); framebuffer.set(ax, ay, WHITE);
framebuffer.set(bx, by, &WHITE); framebuffer.set(bx, by, WHITE);
framebuffer.set(cx, cy, &WHITE); framebuffer.set(cx, cy, WHITE);
let a = Point::new(ax, ay);
let b = Point::new(bx, by);
let c = Point::new(cx, cy);
line(a, b, &mut framebuffer, BLUE);
line(c, b, &mut framebuffer, GREEN);
line(c, a, &mut framebuffer, YELLOW);
line(a, c, &mut framebuffer, RED);
framebuffer framebuffer
.write_file("framebuffer.tga", true, true) .write_file("framebuffer.tga", true, true)
.unwrap(); .unwrap();
}
let new = TGAImage::from_file("framebuffer.tga").unwrap();
fn line(a: Point, b: Point, fb: &mut TGAImage, color: TGAColor) {
let mut t = 0.0;
let step = 0.02;
while t < 1.0 {
let x = (a.x as f32 + (b.x - a.x) as f32 * t).round_ties_even() as u32;
let y = (a.y as f32 + (b.y - a.y) as f32 * t).round_ties_even() as u32;
fb.set(x, y, color);
t += step;
}
} }

49
src/point.rs Normal file
View file

@ -0,0 +1,49 @@
use std::ops::{Add, Mul, Sub};
#[derive(Default, Clone, Copy, Debug, PartialEq, Eq, PartialOrd, Ord)]
pub struct Point {
pub x: i32,
pub y: i32,
}
impl Point {
pub fn new(x: u32, y: u32) -> Self {
Self {
x: x as i32,
y: y as i32,
}
}
}
impl Add for Point {
type Output = Self;
fn add(self, rhs: Self) -> Self::Output {
Self {
x: self.x + rhs.x,
y: self.y + rhs.y,
}
}
}
impl Sub for Point {
type Output = Self;
fn sub(self, rhs: Self) -> Self::Output {
Self {
x: self.x.saturating_sub(rhs.x),
y: self.y.saturating_sub(rhs.y),
}
}
}
impl Mul for Point {
type Output = Self;
fn mul(self, rhs: Self) -> Self::Output {
Self {
x: self.x * rhs.x,
y: self.y * rhs.y,
}
}
}

View file

@ -6,6 +6,8 @@ use std::{
const HEADER_SIZE: usize = 18; const HEADER_SIZE: usize = 18;
pub type ColorBuf = [u8; 4];
#[derive(Debug, Default, Clone, PartialEq, Eq)] #[derive(Debug, Default, Clone, PartialEq, Eq)]
#[repr(C, packed)] #[repr(C, packed)]
pub struct TGAHeader { pub struct TGAHeader {
@ -48,11 +50,11 @@ impl TryFrom<u8> for TGAFormat {
#[derive(Debug, Default, Clone, Copy, PartialEq, Eq)] #[derive(Debug, Default, Clone, Copy, PartialEq, Eq)]
#[repr(transparent)] #[repr(transparent)]
pub struct TGAColor { pub struct TGAColor {
pub bgra: [u8; 4], pub bgra: ColorBuf,
} }
impl Deref for TGAColor { impl Deref for TGAColor {
type Target = [u8; 4]; type Target = ColorBuf;
fn deref(&self) -> &Self::Target { fn deref(&self) -> &Self::Target {
&self.bgra &self.bgra
@ -65,15 +67,17 @@ impl DerefMut for TGAColor {
} }
} }
impl From<[u8; 4]> for TGAColor { impl From<ColorBuf> for TGAColor {
fn from(value: [u8; 4]) -> Self { fn from(value: ColorBuf) -> Self {
Self { bgra: value } Self { bgra: value }
} }
} }
impl TGAColor { impl TGAColor {
pub fn new() -> Self { pub fn new() -> Self {
Self { bgra: [0; 4] } Self {
bgra: ColorBuf::default(),
}
} }
pub fn b(&mut self) -> &mut u8 { pub fn b(&mut self) -> &mut u8 {
@ -96,8 +100,8 @@ impl TGAColor {
#[derive(Debug, Default, Clone, PartialEq, Eq)] #[derive(Debug, Default, Clone, PartialEq, Eq)]
pub struct TGAImage { pub struct TGAImage {
pub format: TGAFormat, pub format: TGAFormat,
pub width: u16, pub width: u32,
pub height: u16, pub height: u32,
data: Vec<u8>, data: Vec<u8>,
} }
@ -128,7 +132,7 @@ impl From<std::io::Error> for TGAError {
} }
impl TGAImage { impl TGAImage {
pub fn new(width: u16, height: u16, format: TGAFormat) -> Self { pub fn new(width: u32, height: u32, format: TGAFormat) -> Self {
let size = format as usize * (width * height) as usize; let size = format as usize * (width * height) as usize;
Self { Self {
format, format,
@ -155,8 +159,8 @@ impl TGAImage {
} }
self.format = (header.bits_per_pixel >> 3).try_into()?; self.format = (header.bits_per_pixel >> 3).try_into()?;
self.width = header.width; self.width = header.width as u32;
self.height = header.height; self.height = header.height as u32;
let nbytes = self.format as usize * (self.width * self.height) as usize; let nbytes = self.format as usize * (self.width * self.height) as usize;
let mut image_data = vec![0u8; nbytes]; let mut image_data = vec![0u8; nbytes];
@ -180,8 +184,8 @@ impl TGAImage {
let mut header = TGAHeader::default(); let mut header = TGAHeader::default();
header.bits_per_pixel = self.bytes_per_pixel() << 3; header.bits_per_pixel = self.bytes_per_pixel() << 3;
header.width = self.width; header.width = self.width as u16;
header.height = self.height; header.height = self.height as u16;
header.datatype_code = match self.format { header.datatype_code = match self.format {
TGAFormat::Grayscale if do_rle => 11, TGAFormat::Grayscale if do_rle => 11,
@ -204,8 +208,8 @@ impl TGAImage {
file.flush()?; file.flush()?;
let mut developer_area = [0u8; 4]; let mut developer_area = ColorBuf::default();
let mut extension_area = [0u8; 4]; let mut extension_area = ColorBuf::default();
let mut footer: Vec<u8> = vec![ let mut footer: Vec<u8> = vec![
'T', 'R', 'U', 'E', 'V', 'I', 'S', 'I', 'O', 'N', '-', 'X', 'F', 'I', 'L', 'E', '.', 'T', 'R', 'U', 'E', 'V', 'I', 'S', 'I', 'O', 'N', '-', 'X', 'F', 'I', 'L', 'E', '.',
'\0', '\0',
@ -231,7 +235,7 @@ impl TGAImage {
todo!() todo!()
} }
pub fn set(&mut self, x: u16, y: u16, color: &TGAColor) { pub fn set(&mut self, x: u32, y: u32, color: TGAColor) {
if self.data.is_empty() || x > self.width || y > self.height { if self.data.is_empty() || x > self.width || y > self.height {
return; return;
} }
@ -239,12 +243,10 @@ impl TGAImage {
let bpp = self.bytes_per_pixel() as usize; let bpp = self.bytes_per_pixel() as usize;
let start = (x + y * self.width) as usize * bpp; let start = (x + y * self.width) as usize * bpp;
let end = start + bpp; let end = start + bpp;
for (i, b) in &mut self.data[start..end].iter_mut().enumerate() { self.data[start..end].copy_from_slice(&color[0..bpp]);
*b = color[i];
}
} }
pub fn get(&self, x: u16, y: u16) -> Option<TGAColor> { pub fn get(&self, x: u32, y: u32) -> Option<TGAColor> {
if self.data.is_empty() || x >= self.width || y >= self.height { if self.data.is_empty() || x >= self.width || y >= self.height {
return None; return None;
} }
@ -252,12 +254,11 @@ impl TGAImage {
let bpp = self.bytes_per_pixel() as usize; let bpp = self.bytes_per_pixel() as usize;
let mut color = TGAColor::new(); let mut color = TGAColor::new();
let start = (x + y * self.width) as usize; let start = bpp * (x + y * self.width) as usize;
let end = start + bpp; let end = start + bpp;
let data = &self.data[start..end]; let data = &self.data[start..end];
for i in 0..bpp { color[0..bpp].copy_from_slice(data);
color[i] = data[i];
}
Some(color) Some(color)
} }