X-Git-Url: http://www.dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fcommon.rs;h=f91ec8325908a48efdebf4223e9e4ee535d2b34f;hb=2836f506d013d21591ddbc14c6168fd929aa5e95;hp=52b732482f6ee4ea6b8013786de9783feefcb9f9;hpb=b0566120fd4a404355b6e6e1d75b472bbe3f57fe;p=kaka%2Frust-sdl-test.git diff --git a/src/common.rs b/src/common.rs index 52b7324..f91ec83 100644 --- a/src/common.rs +++ b/src/common.rs @@ -1,4 +1,4 @@ -use std::ops::{Add, AddAssign, Mul}; +use std::ops::{Add, AddAssign, Sub, SubAssign, Mul, MulAssign, Div, DivAssign, Neg}; pub type Nanoseconds = u64; @@ -21,21 +21,152 @@ impl Point2D { } } +////////// add point to point ////////////////////////////////////////////////// impl> Add for Point2D { - type Output = Point2D; + type Output = Self; - fn add(self, rhs: Point2D) -> Self::Output { - Point2D { + fn add(self, rhs: Self) -> Self { + Self { + x: self.x + rhs.x, + y: self.y + rhs.y, + } + } +} + +impl + Copy> AddAssign for Point2D { + fn add_assign(&mut self, rhs: Self) { + *self = Self { x: self.x + rhs.x, y: self.y + rhs.y, + } + } +} + +////////// add tuple to point ////////////////////////////////////////////////// +impl> Add<(T, T)> for Point2D { + type Output = Self; + + fn add(self, rhs: (T, T)) -> Self { + Self { + x: self.x + rhs.0, + y: self.y + rhs.1, + } + } +} + +////////// subtract point from point /////////////////////////////////////////// +impl> Sub for Point2D { + type Output = Self; + + fn sub(self, rhs: Self) -> Self { + Self { + x: self.x - rhs.x, + y: self.y - rhs.y, } } } -impl AddAssign for Point2D { - fn add_assign(&mut self, rhs: Point2D) { - self.x += rhs.x; - self.y += rhs.y; +impl + Copy> SubAssign for Point2D { + fn sub_assign(&mut self, rhs: Self) { + *self = Self { + x: self.x - rhs.x, + y: self.y - rhs.y, + } + } +} + +////////// 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, + } + } +} + +////////// multiply components of two points /////////////////////////////////// +impl> Mul for Point2D { + type Output = Self; + + fn mul(self, rhs: Self) -> Self { + Self { + x: self.x * rhs.x, + y: self.y * rhs.y, + } + } +} + +impl + Copy> MulAssign for Point2D { + fn mul_assign(&mut self, rhs: Self) { + *self = Self { + x: self.x * rhs.x, + y: self.y * rhs.y, + } + } +} + +////////// 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, + } + } +} + +////////// divide components of two points ///////////////////////////////////// +impl> Div for Point2D { + type Output = Self; + + fn div(self, rhs: Self) -> Self { + Self { + x: self.x / rhs.x, + y: self.y / rhs.y, + } + } +} + +impl + Copy> DivAssign for Point2D { + fn div_assign(&mut self, rhs: Self) { + *self = Self { + x: self.x / rhs.x, + y: self.y / rhs.y, + } + } +} + +impl> Neg for Point2D { + type Output = Self; + + fn neg(self) -> Self { + Self { + x: -self.x, + y: -self.y, + } } } @@ -96,6 +227,42 @@ 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)); + } + + #[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)); + } + + #[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)); + } + + #[test] + fn neg_point() { + assert_eq!(point!(1, 1), -point!(-1, -1)); } #[test]