wipe/src/vec.rs
2022-04-15 07:56:37 +02:00

119 lines
2.5 KiB
Rust

use std::ops::Sub;
/// A vector with a x and y axis.
#[derive(Copy, Clone)]
pub struct Vector {
pub x: f32,
pub y: f32
}
impl Vector {
pub fn new(x: f32, y: f32) -> Self {
Self { x, y }
}
/// Returns the halfway point.
pub fn center(self) -> Self {
Self::new(self.x / 2.0, self.y / 2.0)
}
/// Returns the length.
pub fn length(self) -> f32 {
(self.x * self.x + self.y * self.y).sqrt()
}
/// Returns the angle.
pub fn angle(self) -> f32 {
self.y.atan2(self.x)
}
/// Returns the value of the smaller axis.
pub fn smaller(self) -> f32 {
self.x.min(self.y)
}
/// Converts all axis into positive values.
pub fn abs(self) -> Vector {
Self::new(self.x.abs(), self.y.abs())
}
/// Returns the sum of all axis.
pub fn sum(self) -> f32 {
self.x + self.y
}
/// Creates a vector with the on screen coordinates based on the terminal coordinates.
/// # Arguments
/// * `x`: The x axis of the terminal character.
/// * `y`: The y axis of the terminal character.
pub fn from_terminal(x: usize, y: usize) -> Self {
Self::new(x as f32, y as f32 * 2.0)
}
}
impl Sub for Vector {
type Output = Vector;
fn sub(self, rhs: Self) -> Self::Output {
Vector::new(self.x - rhs.x, self.y - rhs.y)
}
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn new() {
let vec = Vector::new(3.0, 5.0);
assert_eq!(3.0, vec.x);
assert_eq!(5.0, vec.y);
}
#[test]
fn center() {
let vec = Vector::new(3.0, 8.0);
assert_eq!(1.5, vec.center().x);
assert_eq!(4.0, vec.center().y);
}
#[test]
fn length() {
let vec = Vector::new(3.0, 6.0);
assert!(6.7 < vec.length() && vec.length() < 6.8);
}
#[test]
fn angle() {
let vec = Vector::new(3.0, 6.0);
assert!(1.1 < vec.angle() && vec.angle() < 1.2);
}
#[test]
fn smaller() {
assert_eq!(4.0, Vector::new(7.0, 4.0).smaller());
assert_eq!(2.0, Vector::new(2.0, 9.0).smaller());
}
#[test]
fn from_terminal() {
let vec = Vector::from_terminal(2, 4);
assert_eq!(2.0, vec.x);
assert_eq!(8.0, vec.y);
}
#[test]
fn sub() {
let left = Vector::new(8.0, 15.0);
let right = Vector::new(2.0, 4.0);
let result = left - right;
assert_eq!(6.0, result.x);
assert_eq!(11.0, result.y);
}
}