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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use crate::renderer::*;

///
/// A bounding box geometry used for visualising an [AxisAlignedBoundingBox].
///
pub struct BoundingBox {
    mesh: InstancedMesh,
    aabb: AxisAlignedBoundingBox,
}

impl BoundingBox {
    ///
    /// Creates a bounding box geometry from an axis aligned bounding box.
    ///
    pub fn new(context: &Context, aabb: AxisAlignedBoundingBox) -> Self {
        let size = aabb.size();
        let thickness = 0.02 * size.x.max(size.y).max(size.z);

        Self::new_with_thickness(context, aabb, thickness)
    }

    ///
    /// Creates a bounding box object from an axis aligned bounding box with a specified line
    /// thickness.
    ///
    pub fn new_with_thickness(
        context: &Context,
        aabb: AxisAlignedBoundingBox,
        thickness: f32,
    ) -> Self {
        let max = aabb.max();
        let min = aabb.min();
        let size = aabb.size();
        let translations = vec![
            min,
            vec3(min.x, max.y, max.z),
            vec3(min.x, min.y, max.z),
            vec3(min.x, max.y, min.z),
            min,
            vec3(max.x, min.y, max.z),
            vec3(min.x, min.y, max.z),
            vec3(max.x, min.y, min.z),
            min,
            vec3(max.x, max.y, min.z),
            vec3(min.x, max.y, min.z),
            vec3(max.x, min.y, min.z),
        ];

        let rotations = vec![
            Quat::zero(),
            Quat::zero(),
            Quat::zero(),
            Quat::zero(),
            Quat::from_angle_z(degrees(90.0)),
            Quat::from_angle_z(degrees(90.0)),
            Quat::from_angle_z(degrees(90.0)),
            Quat::from_angle_z(degrees(90.0)),
            Quat::from_angle_y(degrees(-90.0)),
            Quat::from_angle_y(degrees(-90.0)),
            Quat::from_angle_y(degrees(-90.0)),
            Quat::from_angle_y(degrees(-90.0)),
        ];

        let scales = vec![
            vec3(size.x, thickness, thickness),
            vec3(size.x, thickness, thickness),
            vec3(size.x, thickness, thickness),
            vec3(size.x, thickness, thickness),
            vec3(size.y, thickness, thickness),
            vec3(size.y, thickness, thickness),
            vec3(size.y, thickness, thickness),
            vec3(size.y, thickness, thickness),
            vec3(size.z, thickness, thickness),
            vec3(size.z, thickness, thickness),
            vec3(size.z, thickness, thickness),
            vec3(size.z, thickness, thickness),
        ];
        let mesh = InstancedMesh::new(
            context,
            &Instances {
                translations,
                rotations: Some(rotations),
                scales: Some(scales),
                ..Default::default()
            },
            &CpuMesh::cylinder(16),
        );
        Self { mesh, aabb }
    }
}

impl<'a> IntoIterator for &'a BoundingBox {
    type Item = &'a dyn Geometry;
    type IntoIter = std::iter::Once<&'a dyn Geometry>;

    fn into_iter(self) -> Self::IntoIter {
        std::iter::once(self)
    }
}

impl Geometry for BoundingBox {
    fn aabb(&self) -> AxisAlignedBoundingBox {
        self.aabb
    }

    fn render_with_material(
        &self,
        material: &dyn Material,
        camera: &Camera,
        lights: &[&dyn Light],
    ) {
        self.mesh.render_with_material(material, camera, lights)
    }

    fn render_with_post_material(
        &self,
        material: &dyn PostMaterial,
        camera: &Camera,
        lights: &[&dyn Light],
        color_texture: Option<ColorTexture>,
        depth_texture: Option<DepthTexture>,
    ) {
        self.mesh
            .render_with_post_material(material, camera, lights, color_texture, depth_texture)
    }
}