use std::ops::{Add, AddAssign, Sub, SubAssign, Mul, MulAssign, Div, DivAssign, Neg}; #[macro_export] macro_rules! point { ( $x:expr, $y:expr ) => { Point2D { x: $x, y: $y } }; } #[derive(Debug, Default, Copy, Clone, PartialEq)] pub struct Point2D { pub x: T, pub y: T, } impl Point2D { pub fn length(self) -> f64 { ((self.x * self.x) + (self.y * self.y)).sqrt() } pub fn to_i32(self) -> Point2D { Point2D { x: self.x as i32, y: self.y as i32, } } } macro_rules! point_op { ($op:tt, $trait:ident($fn:ident), $trait_assign:ident($fn_assign:ident), $rhs:ident = $Rhs:ty => $x:expr, $y:expr) => { impl> $trait<$Rhs> for Point2D { type Output = Self; fn $fn(self, $rhs: $Rhs) -> Self { Self { x: self.x $op $x, y: self.y $op $y, } } } impl + Copy> $trait_assign<$Rhs> for Point2D { fn $fn_assign(&mut self, $rhs: $Rhs) { *self = Self { x: self.x $op $x, y: self.y $op $y, } } } } } point_op!(+, Add(add), AddAssign(add_assign), rhs = Point2D => rhs.x, rhs.y); point_op!(-, Sub(sub), SubAssign(sub_assign), rhs = Point2D => rhs.x, rhs.y); point_op!(*, Mul(mul), MulAssign(mul_assign), rhs = Point2D => rhs.x, rhs.y); point_op!(/, Div(div), DivAssign(div_assign), rhs = Point2D => rhs.x, rhs.y); point_op!(+, Add(add), AddAssign(add_assign), rhs = (T, T) => rhs.0, rhs.1); point_op!(-, Sub(sub), SubAssign(sub_assign), rhs = (T, T) => rhs.0, rhs.1); point_op!(*, Mul(mul), MulAssign(mul_assign), rhs = (T, T) => rhs.0, rhs.1); point_op!(/, Div(div), DivAssign(div_assign), rhs = (T, T) => rhs.0, rhs.1); ////////// multiply point with scalar ////////////////////////////////////////// impl + Copy> Mul for Point2D { type Output = Self; fn mul(self, rhs: T) -> Self { Self { x: self.x * rhs, y: self.y * rhs, } } } impl + Copy> MulAssign for Point2D { fn mul_assign(&mut self, rhs: T) { *self = Self { x: self.x * rhs, y: self.y * rhs, } } } ////////// divide point with scalar //////////////////////////////////////////// impl + Copy> Div for Point2D { type Output = Self; fn div(self, rhs: T) -> Self { Self { x: self.x / rhs, y: self.y / rhs, } } } impl + Copy> DivAssign for Point2D { fn div_assign(&mut self, rhs: T) { *self = Self { x: self.x / rhs, y: self.y / rhs, } } } impl> Neg for Point2D { type Output = Self; fn neg(self) -> Self { Self { x: -self.x, y: -self.y, } } } impl From<(T, T)> for Point2D { fn from(item: (T, T)) -> Self { Point2D { x: item.0, y: item.1, } } } impl From> for (T, T) { fn from(item: Point2D) -> Self { (item.x, item.y) } } impl From for Point2D { fn from(item: Degrees) -> Self { Point2D { x: (item.0 * std::f64::consts::PI / 180.0).cos(), y: (item.0 * std::f64::consts::PI / 180.0).sin(), } } } impl From for Point2D { fn from(item: Radians) -> Self { Point2D { x: item.0.cos(), y: item.0.sin(), } } } #[derive(Debug, Default, PartialEq, Clone, Copy)] pub struct Degrees(pub f64); #[derive(Debug, Default, PartialEq, Clone, Copy)] pub struct Radians(pub f64); impl Degrees { #[allow(dead_code)] fn to_radians(&self) -> Radians { Radians(self.0 * std::f64::consts::PI / 180.0) } } impl Radians { #[allow(dead_code)] fn to_degrees(&self) -> Degrees { Degrees(self.0 * 180.0 * std::f64::consts::FRAC_1_PI) } } #[macro_export] macro_rules! rect { ( $x:expr, $y:expr ) => { Rect { x: $x, y: $y } }; } #[derive(Default)] pub struct Rect { pub width: T, pub height: T, } impl + Copy> Rect { #[allow(dead_code)] pub fn area(&self) -> T { self.width * self.height } } impl From<(T, T)> for Rect { fn from(item: (T, T)) -> Self { Rect { width: item.0, height: item.1, } } } #[macro_export] macro_rules! hashmap { ($($k:expr => $v:expr),*) => { { let mut map = std::collections::HashMap::new(); $(map.insert($k, $v);)* map } } } #[cfg(test)] mod tests { use super::*; #[test] fn immutable_copy_of_point() { let a = point!(0, 0); let mut b = a; // Copy assert_eq!(a, b); // PartialEq b.x = 1; assert_ne!(a, b); // PartialEq } #[test] fn add_points() { let mut a = point!(1, 0); assert_eq!(a + point!(2, 2), point!(3, 2)); // Add a += point!(2, 2); // AddAssign assert_eq!(a, point!(3, 2)); assert_eq!(point!(1, 0) + (2, 3), point!(3, 3)); } #[test] fn sub_points() { let mut a = point!(1, 0); assert_eq!(a - point!(2, 2), point!(-1, -2)); a -= point!(2, 2); assert_eq!(a, point!(-1, -2)); assert_eq!(point!(1, 0) - (2, 3), point!(-1, -3)); } #[test] fn mul_points() { let mut a = point!(1, 2); assert_eq!(a * 2, point!(2, 4)); assert_eq!(a * point!(2, 3), point!(2, 6)); a *= 2; assert_eq!(a, point!(2, 4)); a *= point!(3, 1); assert_eq!(a, point!(6, 4)); assert_eq!(point!(1, 0) * (2, 3), point!(2, 0)); } #[test] fn div_points() { let mut a = point!(4, 8); assert_eq!(a / 2, point!(2, 4)); assert_eq!(a / point!(2, 4), point!(2, 2)); a /= 2; assert_eq!(a, point!(2, 4)); a /= point!(2, 4); assert_eq!(a, point!(1, 1)); assert_eq!(point!(6, 3) / (2, 3), point!(3, 1)); } #[test] fn neg_point() { assert_eq!(point!(1, 1), -point!(-1, -1)); } #[test] fn angles() { assert_eq!(Radians(0.0).to_degrees(), Degrees(0.0)); assert_eq!(Radians(std::f64::consts::PI).to_degrees(), Degrees(180.0)); assert_eq!(Degrees(180.0).to_radians(), Radians(std::f64::consts::PI)); assert!((Point2D::from(Degrees(90.0)) - point!(0.0, 1.0)).length() < 0.001); assert!((Point2D::from(Radians(std::f64::consts::FRAC_PI_2)) - point!(0.0, 1.0)).length() < 0.001); } #[test] fn area_for_rect_of_multipliable_type() { let r: Rect<_> = (30, 20).into(); // the Into trait uses the From trait assert_eq!(r.area(), 30 * 20); // let a = Rect::from(("a".to_string(), "b".to_string())).area(); // this doesn't work, because area() is not implemented for String } }