diff --git a/src/cgmath/aabb.rs b/src/cgmath/aabb.rs index 5e8ff65..93578c6 100644 --- a/src/cgmath/aabb.rs +++ b/src/cgmath/aabb.rs @@ -15,12 +15,48 @@ //! Axis-aligned bounding boxes -use point::{Point2, Point3}; +use point::{Point, Point2, Point3}; +use vector::{Vector, Vec2, Vec3}; +use array::build; +use std::num::{zero, one}; +use std::iter::Iterator; + +pub trait Aabb +< + S: Primitive, + V: Vector, + P: Point, + Slice +> { + fn new(p1: &P, p2: &P) -> Self; + fn min<'a>(&'a self) -> &'a P; + fn max<'a>(&'a self) -> &'a P; + #[inline] fn dim(&self) -> V { self.max().sub_p(self.min()) } + #[inline] fn volume(&self) -> S { self.dim().comp_mul() } + #[inline] fn center(&self) -> P { + let two = one::() + one::(); + self.min().add_v(&self.dim().div_s(two)) + } + + // Tests whether a point is cointained in the box, inclusive for min corner + // and exclusive for the max corner. + #[inline] fn contains(&self, p: &P) -> bool { + p.sub_p(self.min()).iter().all(|x| *x >= zero::()) && + self.max().sub_p(p).iter().all(|x| *x > zero::()) + } + + // Returns a new AABB that is grown to include the given point. + fn grow(&self, p: &P) -> Self { + let mn : P = build(|i| self.min().i(i).min(p.i(i))); + let mx : P = build(|i| self.max().i(i).max(p.i(i))); + Aabb::new(&mn, &mx) + } +} #[deriving(Clone, Eq)] pub struct Aabb2 { - min: Point2, - max: Point2, + mn: Point2, + mx: Point2, } impl Aabb2 { @@ -28,16 +64,22 @@ impl Aabb2 { #[inline] pub fn new(p1: &Point2, p2: &Point2) -> Aabb2 { Aabb2 { - min: Point2::new(p1.x.min(&p2.x), p1.y.min(&p2.y)), - max: Point2::new(p1.x.max(&p2.x), p1.y.max(&p2.y)), + mn: Point2::new(p1.x.min(&p2.x), p1.y.min(&p2.y)), + mx: Point2::new(p1.x.max(&p2.x), p1.y.max(&p2.y)), } } } +impl Aabb, Point2, [S, ..2]> for Aabb2 { + fn new(p1: &Point2, p2: &Point2) -> Aabb2 { Aabb2::new(p1, p2) } + #[inline] fn min<'a>(&'a self) -> &'a Point2 { &self.mn } + #[inline] fn max<'a>(&'a self) -> &'a Point2 { &self.mx } +} + #[deriving(Clone, Eq)] pub struct Aabb3 { - min: Point3, - max: Point3, + mn: Point3, + mx: Point3, } impl Aabb3 { @@ -45,8 +87,14 @@ impl Aabb3 { #[inline] pub fn new(p1: &Point3, p2: &Point3) -> Aabb3 { Aabb3 { - min: Point3::new(p1.x.min(&p2.x), p1.y.min(&p2.y), p1.z.min(&p2.z)), - max: Point3::new(p1.x.max(&p2.x), p1.y.max(&p2.y), p1.z.max(&p2.z)), + mn: Point3::new(p1.x.min(&p2.x), p1.y.min(&p2.y), p1.z.min(&p2.z)), + mx: Point3::new(p1.x.max(&p2.x), p1.y.max(&p2.y), p1.z.max(&p2.z)), } } } + +impl Aabb, Point3, [S, ..3]> for Aabb3 { + fn new(p1: &Point3, p2: &Point3) -> Aabb3 { Aabb3::new(p1, p2) } + #[inline] fn min<'a>(&'a self) -> &'a Point3 { &self.mn } + #[inline] fn max<'a>(&'a self) -> &'a Point3 { &self.mx } +} diff --git a/src/test/aabb.rs b/src/test/aabb.rs index ccbe832..bf3b926 100644 --- a/src/test/aabb.rs +++ b/src/test/aabb.rs @@ -1,13 +1,38 @@ -use cgmath::aabb::{Aabb2, Aabb3}; +use cgmath::aabb::*; use cgmath::point::{Point2, Point3}; +use cgmath::vector::{Vec2, Vec3}; #[test] fn test_aabb() { - let aabb = Aabb2::new(&Point2::new(-20f64, 30f64), &Point2::new(10f64, -10f64)); - assert_eq!(aabb.min, Point2::new(-20f64, -10f64)); - assert_eq!(aabb.max, Point2::new(10f64, 30f64)); + let aabb = Aabb2::new(&Point2::new(-20, 30), &Point2::new(10, -10)); + assert_eq!(aabb.min(), &Point2::new(-20, -10)); + assert_eq!(aabb.max(), &Point2::new(10, 30)); + assert_eq!(aabb.dim(), Vec2::new(30, 40)); + assert_eq!(aabb.volume(), 30 * 40); + assert_eq!(aabb.center(), Point2::new(-5, 10)); - let aabb = Aabb3::new(&Point3::new(-20f64, 30f64, 0f64), &Point3::new(10f64, -10f64, -5f64)); - assert_eq!(aabb.min, Point3::new(-20f64, -10f64, -5f64)); - assert_eq!(aabb.max, Point3::new(10f64, 30f64, 0f64)); + assert!(aabb.contains(&Point2::new(0, 0))); + assert!(!aabb.contains(&Point2::new(-50, -50))); + assert!(!aabb.contains(&Point2::new(50, 50))); + + assert_eq!(aabb.grow(&Point2::new(0, 0)), aabb); + assert_eq!(aabb.grow(&Point2::new(100, 100)), + Aabb2::new(&Point2::new(-20, -10), &Point2::new(100, 100))); + assert_eq!(aabb.grow(&Point2::new(-100, -100)), + Aabb2::new(&Point2::new(-100, -100), &Point2::new(10, 30))); + + let aabb = Aabb3::new(&Point3::new(-20, 30, 5), &Point3::new(10, -10, -5)); + assert_eq!(aabb.min(), &Point3::new(-20, -10, -5)); + assert_eq!(aabb.max(), &Point3::new(10, 30, 5)); + assert_eq!(aabb.dim(), Vec3::new(30, 40, 10)); + assert_eq!(aabb.volume(), 30 * 40 * 10); + assert_eq!(aabb.center(), Point3::new(-5, 10, 0)); + + assert!(aabb.contains(&Point3::new(0, 0, 0))); + assert!(!aabb.contains(&Point3::new(-100, 0, 0))); + assert!(!aabb.contains(&Point3::new(100, 0, 0))); + assert!(aabb.contains(&Point3::new(9, 29, -1))); + assert!(!aabb.contains(&Point3::new(10, 30, 5))); + assert!(aabb.contains(&Point3::new(-20, -10, -5))); + assert!(!aabb.contains(&Point3::new(-21, -11, -6))); }