1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
use crate::core::types::*;

pub struct AxisAlignedBoundingBox {
    pub min: Vec3,
    pub max: Vec3
}

impl AxisAlignedBoundingBox {

    pub fn new() -> Self {
        Self {min: vec3(std::f32::INFINITY, std::f32::INFINITY, std::f32::INFINITY),
            max: vec3(std::f32::NEG_INFINITY, std::f32::NEG_INFINITY, std::f32::NEG_INFINITY)}
    }

    pub fn new_from_positions(positions: &[f32]) -> Self {
        let mut aabb = Self::new();
        aabb.expand(positions);
        aabb
    }

    pub fn expand(&mut self, positions: &[f32]) {
        for i in 0..positions.len() {
            match i%3 {
                0 => {
                    self.min.x = f32::min(positions[i], self.min.x);
                    self.max.x = f32::max(positions[i], self.max.x);
                },
                1 => {
                    self.min.y = f32::min(positions[i], self.min.y);
                    self.max.y = f32::max(positions[i], self.max.y);
                },
                2 => {
                    self.min.z = f32::min(positions[i], self.min.z);
                    self.max.z = f32::max(positions[i], self.max.z);
                },
                _ => {unreachable!()}
            };
        }
    }

    pub fn add(&mut self, other: &AxisAlignedBoundingBox) {
        self.min = vec3(f32::min(self.min.x, other.min.x), f32::min(self.min.y, other.min.y), f32::min(self.min.z, other.min.z));
        self.max = vec3(f32::max(self.max.x, other.max.x), f32::max(self.max.y, other.max.y), f32::max(self.max.z, other.max.z));
    }
}