def test_compute_collisions_tree_3d():
references = [[
set([18, 19, 20, 21, 22, 23, 42, 43, 44, 45, 46, 47]),
set([0, 1, 2, 3, 4, 5, 24, 25, 26, 27, 28, 29])
],
[
set([6, 7, 8, 9, 10, 11, 30, 31, 32, 33, 34, 35]),
set([12, 13, 14, 15, 16, 17, 36, 37, 38, 39, 40, 41])
]]
points = [Point(0.52, 0.51, 0.3), Point(0.9, -0.9, 0.3)]
for i, point in enumerate(points):
mesh_A = UnitCubeMesh(MPI.comm_world, 2, 2, 2)
mesh_B = UnitCubeMesh(MPI.comm_world, 2, 2, 2)
bgeom = mesh_B.geometry.points
bgeom += point.array()
tree_A = BoundingBoxTree(3)
tree_A.build_mesh(mesh_A, 3)
tree_B = BoundingBoxTree(3)
tree_B.build_mesh(mesh_B, 3)
entities_A, entities_B = tree_A.compute_collisions_bb(tree_B)
assert set(entities_A) == references[i][0]
assert set(entities_B) == references[i][1]
def test_compute_collisions_tree_1d():
references = [[
set([8, 9, 10, 11, 12, 13, 14, 15]),
set([0, 1, 2, 3, 4, 5, 6, 7])
], [set([14, 15]), set([0, 1])]]
points = [Point(0.52), Point(0.9)]
for i, point in enumerate(points):
mesh_A = UnitIntervalMesh(MPI.comm_world, 16)
mesh_B = UnitIntervalMesh(MPI.comm_world, 16)
bgeom = mesh_B.geometry.points
bgeom += point[0]
tree_A = BoundingBoxTree(1)
tree_A.build_mesh(mesh_A, 1)
tree_B = BoundingBoxTree(1)
tree_B.build_mesh(mesh_B, 1)
entities_A, entities_B = tree_A.compute_collisions_bb(tree_B)
assert set(entities_A) == references[i][0]
assert set(entities_B) == references[i][1]
def test_compute_collisions_tree_2d():
references = [[
set([20, 21, 22, 23, 28, 29, 30, 31]),
set([0, 1, 2, 3, 8, 9, 10, 11])
], [set([6, 7]), set([24, 25])]]
points = [Point(0.52, 0.51), Point(0.9, -0.9)]
for i, point in enumerate(points):
mesh_A = UnitSquareMesh(MPI.comm_world, 4, 4)
mesh_B = UnitSquareMesh(MPI.comm_world, 4, 4)
bgeom = mesh_B.geometry.points
bgeom += point.array()[:2]
tree_A = BoundingBoxTree(2)
tree_A.build_mesh(mesh_A, 2)
tree_B = BoundingBoxTree(2)
tree_B.build_mesh(mesh_B, 2)
entities_A, entities_B = tree_A.compute_collisions_bb(tree_B)
assert set(entities_A) == references[i][0]
assert set(entities_B) == references[i][1]
