def test_compute_first_collision_3d(): # FIXME: This test should not use facet indices as there are no guarantees # on how DOLFIN numbers facets reference = { 1: [1364], 2: [1967, 1968, 1970, 1972, 1974, 1976], 3: [876, 877, 878, 879, 880, 881] } p = Point(0.3, 0.3, 0.3) mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8) for dim in range(1, 4): tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, dim) first = tree.compute_first_collision(p) # FIXME: Face and test is excluded because it mistakingly # relies in the facet indices tdim = mesh.topology.dim if dim != tdim - 1 and dim != tdim - 2: assert first in reference[dim] # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp first = tree.compute_first_collision(p) assert first in reference[mesh.topology.dim]
def test_compute_first_collision_2d(): # FIXME: This test should not use facet indices as there are no guarantees # on how DOLFIN numbers facets reference = {1: [226], 2: [136, 137]} p = Point(0.3, 0.3) mesh = UnitSquareMesh(MPI.comm_world, 16, 16) for dim in range(1, 3): tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, dim) first = tree.compute_first_collision(p) # FIXME: Facet test is excluded because it mistakingly relies in the # facet indices if dim != mesh.topology.dim - 1: assert first in reference[dim] # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp first = tree.compute_first_collision(p) assert first in reference[mesh.topology.dim]
def test_compute_first_entity_collision_3d(): reference = [876, 877, 878, 879, 880, 881] p = Point(0.3, 0.3, 0.3) mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8) tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, mesh.topology.dim) first = tree.compute_first_entity_collision(p, mesh) assert first in reference # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp first = tree.compute_first_entity_collision(p, mesh) assert first in reference
def test_compute_first_collision_1d(): reference = {1: [4]} p = Point(0.3) mesh = UnitIntervalMesh(MPI.comm_world, 16) for dim in range(1, 2): tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, dim) first = tree.compute_first_collision(p) assert first in reference[dim] # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp first = tree.compute_first_collision(p) assert first in reference[mesh.topology.dim]
def test_compute_entity_collisions_2d(): reference = set([136, 137]) p = Point(0.3, 0.3) mesh = UnitSquareMesh(MPI.comm_world, 16, 16) tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, mesh.topology.dim) entities = tree.compute_entity_collisions_mesh(p, mesh) assert set(entities) == reference # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp entities = tree.compute_entity_collisions_mesh(p, mesh) assert set(entities) == reference
def test_manifold_point_search(): # Simple two-triangle surface in 3d vertices = [(0.0, 0.0, 1.0), (1.0, 1.0, 1.0), (1.0, 0.0, 0.0), (0.0, 1.0, 0.0)] cells = [(0, 1, 2), (0, 1, 3)] mesh = Mesh(MPI.comm_world, CellType.Type.triangle, numpy.array(vertices, dtype=numpy.float64), numpy.array(cells, dtype=numpy.int32), [], cpp.mesh.GhostMode.none) bb = BoundingBoxTree() bb_cpp = mesh.bounding_box_tree() bb._cpp_object = bb_cpp p = Point(0.5, 0.25, 0.75) assert bb.compute_first_entity_collision(p, mesh) == 0 p = Point(0.25, 0.5, 0.75) assert bb.compute_first_entity_collision(p, mesh) == 1
def test_compute_closest_entity_3d(): reference = (0, 0.1) p = Point(0.1, 0.05, -0.1) mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8) tree = BoundingBoxTree(mesh.geometry.dim) tree.build_mesh(mesh, mesh.topology.dim) entity, distance = tree.compute_closest_entity(p, mesh) assert entity == reference[0] assert round(distance - reference[1], 7) == 0 # FIXME: remove after Mesh is wrapped in Python tree_cpp = mesh.bounding_box_tree() tree = BoundingBoxTree() tree._cpp_object = tree_cpp entity, distance = tree.compute_closest_entity(p, mesh) assert entity == reference[0] assert round(distance - reference[1], 7) == 0