X-Git-Url: http://git.dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fcommon.rs;h=6306bf6d6205db897d6207a0e993aaf7d4dbc928;hb=bf7b5671bb386ccd3d325ae3dea33046342d129c;hp=f3b3373fcf4c553018cce9926aac751f3fe8f26e;hpb=296187ca027364405cf2fd95e727d5d1eaaec4fc;p=kaka%2Frust-sdl-test.git diff --git a/src/common.rs b/src/common.rs index f3b3373..6306bf6 100644 --- a/src/common.rs +++ b/src/common.rs @@ -1,10 +1,15 @@ -use std::ops::{Add, AddAssign}; +use std::ops::{Add, AddAssign, Sub, SubAssign, Mul, MulAssign, Div, DivAssign, Neg}; +pub type Nanoseconds = u64; + +#[macro_export] macro_rules! point { - ( $x:expr, $y:expr ) => { Point2D { x:$x, y:$y } }; + ( $x:expr, $y:expr ) => { + Point2D { x: $x, y: $y } + }; } -#[derive(Debug, Copy, Clone, PartialEq)] +#[derive(Debug, Default, Copy, Clone, PartialEq)] pub struct Point2D { pub x: T, pub y: T, @@ -14,20 +19,179 @@ 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(), + } + } } -impl> Add for Point2D { - type Output = Point2D; +#[derive(Debug, Default, PartialEq, Clone, Copy)] +pub struct Degrees(pub f64); +#[derive(Debug, Default, PartialEq, Clone, Copy)] +pub struct Radians(pub f64); - fn add(self, rhs: Point2D) -> Self::Output { - Point2D { x: self.x + rhs.x, y: self.y + rhs.y } +impl Degrees { + #[allow(dead_code)] + fn to_radians(&self) -> Radians { + Radians(self.0 * std::f64::consts::PI / 180.0) } } -impl AddAssign for Point2D { - fn add_assign(&mut self, rhs: Point2D) { - self.x += rhs.x; - self.y += rhs.y; +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, + } } } @@ -50,5 +214,60 @@ mod tests { 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 } }