Misc code clean up in kdtree build

daoo · Mar 25, 2024 · 1a826ab · 1a826ab
1 parent 67b751f
commit 1a826ab
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/kdtree/src/build.rs b/kdtree/src/build.rs
@@ -30,22 +30,18 @@ where
     B: KdTreeBuilder,
 {
     if depth >= max_depth || parent.triangle_indices.is_empty() {
-        return Box::new(KdNode::Leaf(parent.triangle_indices));
+        return KdNode::new_leaf(parent.triangle_indices);
     }
 
     match builder.find_best_split(depth, &parent) {
-        None => Box::new(KdNode::Leaf(parent.triangle_indices)),
+        None => KdNode::new_leaf(parent.triangle_indices),
         Some(split) if builder.terminate(&parent, &split) => {
-            Box::new(KdNode::Leaf(parent.triangle_indices))
+            KdNode::new_leaf(parent.triangle_indices)
         }
         Some(split) => {
             let left = build_helper(builder, max_depth, depth + 1, split.left);
             let right = build_helper(builder, max_depth, depth + 1, split.right);
-            Box::new(KdNode::Node {
-                plane: split.plane,
-                left,
-                right,
-            })
+            KdNode::new_node(split.plane, left, right)
         }
     }
 }