feat: implemented renderer
This commit is contained in:
285
src/main.rs
Normal file
285
src/main.rs
Normal file
@ -0,0 +1,285 @@
|
||||
use crossterm::cursor::MoveTo;
|
||||
use crossterm::style::Attribute::Dim;
|
||||
use crossterm::style::Print;
|
||||
use crossterm::terminal::{Clear, ClearType};
|
||||
use crossterm::{execute, QueueableCommand};
|
||||
use ndarray::{arr1, arr2, Array1, Array2};
|
||||
use std::io::{stdout, Write};
|
||||
use std::sync::Mutex;
|
||||
use std::thread;
|
||||
use std::time::Duration;
|
||||
|
||||
const X_DRAW_SCALING_FACTOR: f32 = 2.8;
|
||||
|
||||
enum Dimension {
|
||||
X,
|
||||
Y,
|
||||
Z,
|
||||
}
|
||||
|
||||
struct Mesh {
|
||||
points: Vec<Array1<f32>>,
|
||||
edges: Vec<[usize; 2]>,
|
||||
}
|
||||
|
||||
#[inline]
|
||||
#[rustfmt::skip]
|
||||
fn rotation_matrix(angle: f32, dimension: Dimension) -> ndarray::Array2<f32> {
|
||||
let sin = angle.to_radians().sin();
|
||||
let cos = angle.to_radians().cos();
|
||||
match dimension {
|
||||
Dimension::X => arr2(&[
|
||||
[1., 0., 0.],
|
||||
[0., cos, -sin],
|
||||
[0., sin, cos]
|
||||
]),
|
||||
Dimension::Y => arr2(&[
|
||||
[cos, 0., -sin],
|
||||
[0., 1., 0.],
|
||||
[sin, 0., cos]
|
||||
]),
|
||||
Dimension::Z => arr2(&[
|
||||
[cos, -sin, 0.],
|
||||
[sin, cos, 0.],
|
||||
[0., 0., 1.]
|
||||
])
|
||||
}
|
||||
}
|
||||
|
||||
#[inline]
|
||||
#[rustfmt::skip]
|
||||
fn scale(x: f32, y: f32, z: f32) -> ndarray::Array2<f32> {
|
||||
arr2(&[
|
||||
[x, 0., 0.],
|
||||
[0., y, 0.],
|
||||
[0., 0., z],
|
||||
])
|
||||
}
|
||||
|
||||
#[inline]
|
||||
#[rustfmt::skip]
|
||||
fn ortho_matrix() -> ndarray::Array2<f32> {
|
||||
arr2(&[
|
||||
[1., 0., 0.],
|
||||
[0., 1., 0.],
|
||||
])
|
||||
}
|
||||
|
||||
fn plot_line(
|
||||
origin: &[i32; 2],
|
||||
destination: &[i32; 2],
|
||||
out: &mut impl Write,
|
||||
) -> anyhow::Result<()> {
|
||||
assert!(origin[0] > 0 && destination[0] > 0);
|
||||
|
||||
let (origin, destination) = if origin[0] < destination[0] {
|
||||
(origin, destination)
|
||||
} else {
|
||||
(destination, origin)
|
||||
};
|
||||
|
||||
let (xdiff, ydiff) = (
|
||||
destination[0].abs_diff(origin[0]),
|
||||
destination[1].abs_diff(origin[1]),
|
||||
);
|
||||
if ydiff > xdiff {
|
||||
plot_line_vertical(origin, destination, out)
|
||||
} else {
|
||||
plot_line_horizontal(origin, destination, out)
|
||||
}
|
||||
}
|
||||
|
||||
/// # CONTRACT
|
||||
/// * origin is *left of* destination (`origin[0] <= destination[0]`)
|
||||
/// * inclination from origin to destination fulfills `-1 <= incl <= 1`
|
||||
/// * line is inside (u16::MAX x u16::MAX) space
|
||||
fn plot_line_horizontal(
|
||||
origin: &[i32; 2],
|
||||
destination: &[i32; 2],
|
||||
out: &mut impl Write,
|
||||
) -> anyhow::Result<()> {
|
||||
// assert!(destination[0] > origin[0]);
|
||||
|
||||
let incl = (destination[1] - origin[1]) as f32 / (destination[0] - origin[0]) as f32;
|
||||
// assert!(((-1.)..=1.).contains(&incl));
|
||||
|
||||
for i in 0..(destination[0] - origin[0]) {
|
||||
// CONTRACT: i is in u16 space
|
||||
let x = (i + origin[0]) as u16;
|
||||
// CONTRACT: line points are in u16 space
|
||||
let y = ((incl * i as f32).round() + origin[1] as f32) as u16;
|
||||
out.queue(MoveTo(x, y))?
|
||||
.queue(match (incl < -0.4, incl < 0.4) {
|
||||
(false, true) => Print("-"),
|
||||
(false, false) => Print("\\"),
|
||||
(true, true) => Print("/"),
|
||||
_ => unreachable!(),
|
||||
})?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// # CONTRACT
|
||||
/// * origin is *left of* destination (`origin[0] <= destination[0]`)
|
||||
/// * inclination from origin to destination fulfills `incl <= -1` or `incl >= 1`
|
||||
/// * line is inside (u16::MAX x u16::MAX) space
|
||||
fn plot_line_vertical(
|
||||
origin: &[i32; 2],
|
||||
destination: &[i32; 2],
|
||||
out: &mut impl Write,
|
||||
) -> anyhow::Result<()> {
|
||||
let (origin, destination) = if origin[1] < destination[1] {
|
||||
(origin, destination)
|
||||
} else {
|
||||
(destination, origin)
|
||||
};
|
||||
|
||||
let incl = (destination[0] - origin[0]) as f32 / (destination[1] - origin[1]) as f32;
|
||||
// assert!((..=(-1.)).contains(&incl) || (1.0..).contains(&incl));
|
||||
|
||||
for i in 0..destination[1] - origin[1] {
|
||||
// CONTRACT: i is in u16 space
|
||||
let y = (i + origin[1]) as u16;
|
||||
// CONTRACT: line points are in u16 space
|
||||
let x = ((incl * i as f32).round() + origin[0] as f32) as u16;
|
||||
out.queue(MoveTo(x, y))?
|
||||
.queue(match (incl > -2. && incl < 2., incl > 0.) {
|
||||
(true, false) => Print("/"),
|
||||
(true, true) => Print("\\"),
|
||||
_ => Print("|"),
|
||||
})?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
static ANGLE: Mutex<f32> = Mutex::new(0.);
|
||||
|
||||
fn draw_rotating_mesh(meshes: Vec<Mesh>) -> anyhow::Result<()> {
|
||||
let mut angle = 0.;
|
||||
|
||||
loop {
|
||||
// rotation and matmul
|
||||
let rot_matrix = rotation_matrix(angle, Dimension::Z);
|
||||
let cam_matrix = ortho_matrix();
|
||||
|
||||
let mut stdout = stdout();
|
||||
stdout.queue(Clear(ClearType::All))?;
|
||||
|
||||
for mesh in &meshes {
|
||||
let points = &mesh.points;
|
||||
let edges = &mesh.edges;
|
||||
|
||||
let projected_points = points
|
||||
.iter()
|
||||
.map(|pt| scale(1.6, 1.6, 1.6).dot(pt))
|
||||
.map(|pt| rotation_matrix((angle * 10.) % 360., Dimension::X).dot(&pt))
|
||||
.map(|pt| rotation_matrix((angle * 2.) % 360., Dimension::Y).dot(&pt))
|
||||
.map(|pt| rot_matrix.dot(&pt))
|
||||
.map(|pt| cam_matrix.dot(&pt))
|
||||
.map(|pt| pt + arr1(&[30., 30.])) // draw shift
|
||||
.collect::<Vec<Array1<_>>>();
|
||||
|
||||
for edge in edges.iter() {
|
||||
let origin = &projected_points[edge[0]];
|
||||
let dest = &projected_points[edge[1]];
|
||||
plot_line(
|
||||
&[
|
||||
(origin[0] * X_DRAW_SCALING_FACTOR).round() as i32,
|
||||
origin[1] as i32,
|
||||
],
|
||||
&[(dest[0] * X_DRAW_SCALING_FACTOR) as i32, dest[1] as i32],
|
||||
&mut stdout,
|
||||
)?;
|
||||
}
|
||||
|
||||
for pt in &projected_points {
|
||||
let pt = pt;
|
||||
stdout
|
||||
.queue(MoveTo((pt[0] * X_DRAW_SCALING_FACTOR) as u16, pt[1] as u16))?
|
||||
.queue(Print("■"))?;
|
||||
}
|
||||
}
|
||||
stdout.flush()?;
|
||||
|
||||
angle += 0.1;
|
||||
angle %= 360.;
|
||||
thread::sleep(Duration::from_millis(10));
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn main() -> anyhow::Result<()> {
|
||||
let cube = vec![
|
||||
arr1(&[-10., -10., 10.]),
|
||||
arr1(&[-10., 10., 10.]),
|
||||
arr1(&[-10., -10., -10.]),
|
||||
arr1(&[-10., 10., -10.]),
|
||||
arr1(&[10., -10., 10.]),
|
||||
arr1(&[10., 10., 10.]),
|
||||
arr1(&[10., -10., -10.]),
|
||||
arr1(&[10., 10., -10.]),
|
||||
arr1(&[-6., 0., 0.]),
|
||||
arr1(&[6., 0., 0.]),
|
||||
];
|
||||
|
||||
let edges = vec![
|
||||
// left face
|
||||
[0usize, 1],
|
||||
[0, 2],
|
||||
[1, 3],
|
||||
[2, 3],
|
||||
|
||||
// right face
|
||||
[4, 5],
|
||||
[4, 6],
|
||||
[5, 7],
|
||||
[6, 7],
|
||||
|
||||
// center edges
|
||||
[0, 8],
|
||||
[3, 8],
|
||||
|
||||
[4, 9],
|
||||
[7, 9],
|
||||
|
||||
];
|
||||
|
||||
let thing = Mesh {
|
||||
points: vec![
|
||||
arr1(&[0., 0., 6.]),
|
||||
arr1(&[0., 0., -6.]),
|
||||
arr1(&[0., 6., 0.]),
|
||||
arr1(&[0., -6., 0.]),
|
||||
arr1(&[6., 0., 0.]),
|
||||
arr1(&[-6., 0., 0.]),
|
||||
],
|
||||
edges: vec![
|
||||
[0, 2],
|
||||
[0, 3],
|
||||
[0, 4],
|
||||
[0, 5],
|
||||
[1, 2],
|
||||
[1, 3],
|
||||
[1, 4],
|
||||
[1, 5],
|
||||
[3, 4],
|
||||
[3, 5],
|
||||
[2, 4],
|
||||
[2, 5],
|
||||
],
|
||||
};
|
||||
|
||||
draw_rotating_mesh(vec![
|
||||
Mesh {
|
||||
points: cube,
|
||||
edges,
|
||||
},
|
||||
thing,
|
||||
])?;
|
||||
|
||||
loop {}
|
||||
unreachable!()
|
||||
}
|
||||
Reference in New Issue
Block a user