Fix accidentally remove boundary enlarging

daoo · Mar 16, 2024 · 64924fc · 64924fc
1 parent d6a328d
commit 64924fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/kdtree/build_naive.rs b/src/kdtree/build_naive.rs
@@ -1,3 +1,5 @@
+use nalgebra::vector;
+
 use crate::{
     geometry::{
         aap::{Aap, Axis},
@@ -33,7 +35,7 @@ fn median(values: &[f32]) -> f32 {
 impl KdTreeBuilder for MedianKdTreeBuilder {
     fn starting_box(&self) -> KdBox {
         KdBox {
-            boundary: triangles_bounding_box(&self.triangles),
+            boundary: triangles_bounding_box(&self.triangles).enlarge(&vector![0.5, 0.5, 0.5]),
             triangle_indices: (0u32..self.triangles.len() as u32).collect(),
         }
     }

diff --git a/src/kdtree/build_sah.rs b/src/kdtree/build_sah.rs
@@ -1,3 +1,4 @@
+use nalgebra::vector;
 use rayon::prelude::*;
 
 use crate::{
@@ -57,7 +58,7 @@ impl SahKdTreeBuilder {
 impl KdTreeBuilder for SahKdTreeBuilder {
     fn starting_box(&self) -> KdBox {
         KdBox {
-            boundary: triangles_bounding_box(&self.triangles),
+            boundary: triangles_bounding_box(&self.triangles).enlarge(&vector![0.5, 0.5, 0.5]),
             triangle_indices: (0u32..self.triangles.len() as u32).collect(),
         }
     }
@@ -158,7 +159,6 @@ mod tests {
                 KdNode::new_leaf(vec![]),
             ),
         );
-        dbg!(&tree.root);
         assert_eq!(tree.root, expected);
     }
 }