Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add ability to incrementally add/remove nodes from bvh #99

Merged
merged 11 commits into from
Feb 17, 2024
Merged

add ability to incrementally add/remove nodes from bvh #99

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
be4f135
add ability to incrementally add/remove nodes from bvh
dbenson24 May 6, 2023
e8d5ae0
delete no longer needed fns, address various comments
dbenson24 Sep 29, 2023
b1f2a60
fix depth in pretty_print
dbenson24 Sep 29, 2023
77b548b
fix clippy errors
dbenson24 Sep 29, 2023
89af190
make shape slice non mut
dbenson24 Sep 29, 2023
e8d91a2
fix fmt
dbenson24 Sep 29, 2023
4c1be96
Rename add/remove shape and delete comments.
dbenson24 Nov 27, 2023
cee5bd1
delete unused use statement
dbenson24 Nov 28, 2023
2f3012d
Capitalization on comments
dbenson24 Dec 5, 2023
ee96860
fix clippy
dbenson24 Dec 5, 2023
18abab8
fix clippy lint
dbenson24 Dec 7, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
76 changes: 38 additions & 38 deletions src/bvh/bvh_impl.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,6 @@ pub enum BvhNode<T: Scalar + Copy, const D: usize> {
/// The node's parent.
parent_index: usize,

/// The node's depth.
depth: u32,

/// The shape contained in this leaf.
shape_index: usize,
},
Expand All @@ -41,9 +38,6 @@ pub enum BvhNode<T: Scalar + Copy, const D: usize> {
/// The node's parent.
parent_index: usize,

/// The node's depth.
depth: u32,

/// Index of the left subtree's root node.
child_l_index: usize,

Expand All @@ -65,38 +59,32 @@ impl<T: Scalar + Copy, const D: usize> PartialEq for BvhNode<T, D> {
(
&BvhNode::Node {
parent_index: self_parent_index,
depth: self_depth,
child_l_index: self_child_l_index,
child_r_index: self_child_r_index,
..
},
&BvhNode::Node {
parent_index: other_parent_index,
depth: other_depth,
child_l_index: other_child_l_index,
child_r_index: other_child_r_index,
..
},
) => {
self_parent_index == other_parent_index
&& self_depth == other_depth
&& self_child_l_index == other_child_l_index
&& self_child_r_index == other_child_r_index
}
(
&BvhNode::Leaf {
parent_index: self_parent_index,
depth: self_depth,
shape_index: self_shape_index,
},
&BvhNode::Leaf {
parent_index: other_parent_index,
depth: other_depth,
shape_index: other_shape_index,
},
) => {
self_parent_index == other_parent_index
&& self_depth == other_depth
&& self_shape_index == other_shape_index
}
_ => false,
Expand Down Expand Up @@ -134,6 +122,17 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
}
}

/// Returns the index of the left child node.
pub fn child_l_mut(&mut self) -> &mut usize {
match *self {
BvhNode::Node {
ref mut child_l_index,
..
} => child_l_index,
_ => panic!("Tried to get the left child of a leaf node."),
}
}

/// Returns the `Aabb` of the right child node.
pub fn child_l_aabb(&self) -> Aabb<T, D> {
match *self {
Expand All @@ -160,6 +159,17 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
_ => panic!("Tried to get the right child of a leaf node."),
}
}

/// Returns the index of the right child node.
pub fn child_r_mut(&mut self) -> &mut usize {
match *self {
BvhNode::Node {
ref mut child_r_index,
..
} => child_r_index,
_ => panic!("Tried to get the right child of a leaf node."),
}
}

/// Returns the [`Aabb`] of the right child node.
pub fn child_r_aabb(&self) -> Aabb<T, D> {
Expand All @@ -180,13 +190,6 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
}
}

/// Returns the depth of the node. The root node has depth `0`.
pub fn depth(&self) -> u32 {
match *self {
BvhNode::Node { depth, .. } | BvhNode::Leaf { depth, .. } => depth,
}
}

/// Gets the [`Aabb`] for a [`BvhNode`].
/// Returns the shape's [`Aabb`] for leaves, and the joined [`Aabb`] of
/// the two children's [`Aabb`]'s for non-leaves.
Expand All @@ -213,12 +216,23 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
}
}

/// Returns the index of the shape contained within the node if is a leaf,
/// or `None` if it is an interior node.
pub fn shape_index_mut(&mut self) -> Option<&mut usize> {
match *self {
BvhNode::Leaf {
ref mut shape_index,
..
} => Some(shape_index),
_ => None,
}
}

/// The build function sometimes needs to add nodes while their data is not available yet.
/// A dummy cerated by this function serves the purpose of being changed later on.
fn create_dummy() -> BvhNode<T, D> {
BvhNode::Leaf {
parent_index: 0,
depth: 0,
shape_index: 0,
}
}
Expand Down Expand Up @@ -274,7 +288,6 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
let node_index = nodes.len();
nodes.push(BvhNode::Leaf {
parent_index,
depth,
shape_index,
});
// Let the shape know the index of the node that represents it.
Expand Down Expand Up @@ -375,7 +388,6 @@ impl<T: Scalar + Copy, const D: usize> BvhNode<T, D> {
assert!(!child_r_aabb.is_empty());
nodes[node_index] = BvhNode::Node {
parent_index,
depth,
child_l_aabb,
child_l_index,
child_r_aabb,
Expand Down Expand Up @@ -508,21 +520,20 @@ impl<T: Scalar + Copy, const D: usize> Bvh<T, D> {
BvhNode::Node {
child_l_index,
child_r_index,
depth,
child_l_aabb,
child_r_aabb,
..
} => {
let padding: String = " ".repeat(depth as usize);
let padding: String = " ".repeat(0 as usize); // TODO:(FIX)
dbenson24 marked this conversation as resolved.
Show resolved Hide resolved
println!("{}child_l {}", padding, child_l_aabb);
print_node(nodes, child_l_index);
println!("{}child_r {}", padding, child_r_aabb);
print_node(nodes, child_r_index);
}
BvhNode::Leaf {
shape_index, depth, ..
shape_index, ..
} => {
let padding: String = " ".repeat(depth as usize);
let padding: String = " ".repeat(0 as usize); // TODO:(FIX)
dbenson24 marked this conversation as resolved.
Show resolved Hide resolved
println!("{}shape\t{:?}", padding, shape_index);
}
}
Expand All @@ -549,14 +560,12 @@ impl<T: Scalar + Copy, const D: usize> Bvh<T, D> {
match self.nodes[node_index] {
BvhNode::Node {
parent_index,
depth,
child_l_index,
child_l_aabb,
child_r_index,
child_r_aabb,
} => {
let correct_parent_index = expected_parent_index == parent_index;
let correct_depth = expected_depth == depth;
let left_aabb_in_parent =
expected_outer_aabb.approx_contains_aabb_eps(&child_l_aabb, T::epsilon());
let right_aabb_in_parent =
Expand All @@ -579,24 +588,21 @@ impl<T: Scalar + Copy, const D: usize> Bvh<T, D> {
);

correct_parent_index
&& correct_depth
&& left_aabb_in_parent
&& right_aabb_in_parent
&& left_subtree_consistent
&& right_subtree_consistent
}
BvhNode::Leaf {
parent_index,
depth,
shape_index,
} => {
let correct_parent_index = expected_parent_index == parent_index;
let correct_depth = expected_depth == depth;
let shape_aabb = shapes[shape_index].aabb();
let shape_aabb_in_parent =
expected_outer_aabb.approx_contains_aabb_eps(&shape_aabb, T::epsilon());

correct_parent_index && correct_depth && shape_aabb_in_parent
correct_parent_index && shape_aabb_in_parent
}
}
}
Expand Down Expand Up @@ -642,12 +648,6 @@ impl<T: Scalar + Copy, const D: usize> Bvh<T, D> {
"Wrong parent index. Expected: {}; Actual: {}",
expected_parent_index, parent
);
let depth = node.depth();
assert_eq!(
expected_depth, depth,
"Wrong depth. Expected: {}; Actual: {}",
expected_depth, depth
);

match *node {
BvhNode::Node {
Expand Down
Loading