diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/camera.rs | 93 | ||||
-rw-r--r-- | src/canvas.rs | 4 | ||||
-rw-r--r-- | src/main.rs | 27 | ||||
-rw-r--r-- | src/vector.rs | 83 |
4 files changed, 207 insertions, 0 deletions
diff --git a/src/camera.rs b/src/camera.rs new file mode 100644 index 0000000..d2fbd5e --- /dev/null +++ b/src/camera.rs @@ -0,0 +1,93 @@ +use crate::vector::Vector; +use std::{cmp::{max, min}, f32::consts::PI, f64::MAX_EXP, fmt}; + +const WIDTH: i32 = 60; +const HEIGHT: i32 = 30; + +#[derive(Debug)] +pub struct Buffer (pub [[char; 60]; 30]); + +impl fmt::Display for Buffer { + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + for i in 0..30 { + for j in 0..60 { + write!(f, "{}", self.0[i][j])?; + } + writeln!(f)?; + } + write!(f, "") + } +} + +#[derive(Debug)] +pub struct Camera { + pub position: Vector, + pub direction: Vector, + pub angle: f32, + pub distance: f32, + pub brightness: f32, + pub aspect_ratio: f32, + pub buffer: Buffer, +} + +impl Camera { + pub fn sdf(&self, point: Vector) -> f32 { + let center = Vector { x: 3.0, y: 0.0, z: 0.0 }; + let radius = 1.0; + + return (point - center).magnitude() - radius + } + + pub fn screen(&self) -> (f32, f32) { + let width = self.distance * 2.0 * (self.angle / 2.0).tan(); + let height = width * self.aspect_ratio; + // println!("Screen {}x{} units", width, height); + (width, height) + } + pub fn render(& mut self) { + let palette = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\\|()1{}[]?-_+~<>i!lI;:,\"^`'. "; + let (screen_width, screen_height) = self.screen(); + + let h_step = self.direction.rotate_z(PI / 2.0).normalized() * screen_width / WIDTH as f32; + let v_step = Vector { x: 0.0, y: 0.0, z: -screen_height / HEIGHT as f32 }; + // println!("Steps: h{}, v{}", h_step, v_step); + + // Initialize with a corner + let point = self.position + (self.direction.normalized() * self.distance) - (h_step * (WIDTH / 2) as f32 + v_step * (HEIGHT / 2) as f32); + // println!("Corner: {}", point); + + let mut ray_dir = point - self.position; + + for i in 0..HEIGHT as usize { + ray_dir = ray_dir + v_step; + for j in 0..WIDTH as usize { + ray_dir = ray_dir + h_step; + + + let brightness = self.shoot_ray(ray_dir); + self.buffer.0[i][j] = palette.chars().nth((brightness * palette.len() as f32) as usize - 1).unwrap(); + // println!("[{}, {}]: {}", i, j, ray_dir); + } + ray_dir = ray_dir - h_step * WIDTH as f32; + } + } + + pub fn shoot_ray(&self, direction: Vector) -> f32 { + let threshold = 0.1; + + let ray = direction.normalized(); + let mut point = self.position; + let mut dist = 0.0; + + while dist < self.brightness { + dist = self.sdf(point); + if dist < threshold { + // println!("Dist: {}, point {}", dist, point); + return (point - self.position).magnitude() / self.brightness + } + point = point + ray * dist; + } + + return 1.0 + } +} diff --git a/src/canvas.rs b/src/canvas.rs new file mode 100644 index 0000000..87413a1 --- /dev/null +++ b/src/canvas.rs @@ -0,0 +1,4 @@ +pub struct Canvas { + pub width: u32, + pub height: u32, +} diff --git a/src/main.rs b/src/main.rs new file mode 100644 index 0000000..3bf9bb2 --- /dev/null +++ b/src/main.rs @@ -0,0 +1,27 @@ +mod vector; +mod camera; +mod canvas; +use std::f32::consts::PI; + +use crate::{camera::{Buffer, Camera}, vector::Vector}; + +trait Object { + fn sdf(&self, point: Vector) -> f32; +} + +fn main() { + let mut cam = Camera { + position: Vector { x: 0.0, y: 0.0, z: 0.0 }, + direction: Vector { x: 1.0, y: 0.0, z: 0.0 }, + angle: PI / 2.0, + distance: 1.0, + aspect_ratio: 1.0, + brightness: 10.0, + buffer: Buffer([['.'; 60]; 30]) + }; + + for _i in 0..60 { + cam.render(); + println!("{}", cam.buffer); + } +} diff --git a/src/vector.rs b/src/vector.rs new file mode 100644 index 0000000..d4e57cc --- /dev/null +++ b/src/vector.rs @@ -0,0 +1,83 @@ +use std::fmt; +use std::ops; + +#[derive(Debug, Clone, Copy)] +pub struct Vector { + pub x: f32, + pub y: f32, + pub z: f32, +} + +impl fmt::Display for Vector { + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + write!(f, "({:.2}, {:.2}, {:.2})", self.x, self.y, self.z) + } +} + +impl ops::Add for Vector { + type Output = Vector; + fn add(self, rhs: Self) -> Self::Output { + Vector { + x: self.x + rhs.x, + y: self.y + rhs.y, + z: self.z + rhs.z, + } + } +} + +impl ops::Sub for Vector { + type Output = Vector; + fn sub(self, rhs: Self) -> Self::Output { + Vector { + x: self.x - rhs.x, + y: self.y - rhs.y, + z: self.z - rhs.z, + } + } +} + +impl ops::Mul<f32> for Vector { + type Output = Vector; + fn mul(self, rhs: f32) -> Self::Output { + Vector { + x: self.x * rhs, + y: self.y * rhs, + z: self.z * rhs, + } + } +} + +impl ops::Div<f32> for Vector { + type Output = Vector; + fn div(self, rhs: f32) -> Self::Output { + Vector { + x: self.x / rhs, + y: self.y / rhs, + z: self.z / rhs, + } + } +} + +impl Vector { + pub fn magnitude_squared(self) -> f32 { + self.x * self.x + self.y * self.y + self.z * self.z + } + + pub fn magnitude(self) -> f32 { + self.magnitude_squared().sqrt() + } + + pub fn normalized(self) -> Vector { + let mag = self.magnitude(); + self / mag + } + + pub fn rotate_z(self, angle: f32) -> Vector { + // TODO: multiply by a matirx + Vector { + x: self.x * angle.cos() + self.y * angle.sin(), + y: self.y * angle.cos() - self.x * angle.sin(), + z: self.z, + } + } +} |