class TestDistanceToFacesCubeOutsideRandom():
"""Ranom location that is always outside the cube
"""
pt_out=np.array([1, np.random.uniform(-0.4, 0.4), 0])
v_cube, f_cube=wavefront.read_obj('./integration/cube.obj')
ast = asteroid.Asteroid('castalia', 256, 'mat')
ast = ast.loadmesh(v_cube, f_cube, 'cube')
edge_vertex_map=ast.asteroid_grav['edge_vertex_map']
edge_face_map=ast.asteroid_grav['edge_face_map']
normal_face=ast.asteroid_grav['normal_face']
vf_map=ast.asteroid_grav['vertex_face_map']
D, P, V, E, F=wavefront.distance_to_faces(pt_out, v_cube, f_cube,
normal_face,
edge_vertex_map,
edge_face_map,
vf_map)
def test_distance(self):
np.testing.assert_allclose(np.isfinite(self.D), True)
def test_point(self):
np.testing.assert_allclose(len(self.P) >= 3, True)
def test_vertex(self):
np.testing.assert_allclose(np.isfinite(self.V), True)
def test_edge(self):
np.testing.assert_allclose(np.isfinite(self.E), True)
def test_face(self):
np.testing.assert_allclose(self.F.size >= 1, True)
class TestDistanceToFacesCubeSurfaceSingle():
pt_out=np.array([0.51, -0.1, 0.0])
v_cube, f_cube=wavefront.read_obj('./integration/cube.obj')
ast = asteroid.Asteroid('castalia', 256, 'mat')
ast = ast.loadmesh(v_cube, f_cube, 'cube')
edge_vertex_map=ast.asteroid_grav['edge_vertex_map']
edge_face_map=ast.asteroid_grav['edge_face_map']
normal_face=ast.asteroid_grav['normal_face']
vf_map=ast.asteroid_grav['vertex_face_map']
D, P, V, E, F=wavefront.distance_to_faces(pt_out, v_cube, f_cube,
normal_face,
edge_vertex_map,
edge_face_map,
vf_map)
D_exp = np.array(0.01)
P_exp = np.array([0.5, -0.1, 0])
V_exp = np.array([4, 7, 5])
E_exp = np.array([[7, 4],
[5, 7],
[4, 5]])
F_exp = np.array(7)
def test_distance(self):
np.testing.assert_allclose(self.D, self.D_exp)
def test_point(self):
np.testing.assert_allclose(self.P, self.P_exp)
def test_vertex(self):
np.testing.assert_allclose(self.V, self.V_exp)
def test_edge(self):
for E, E_exp in zip(self.E, self.E_exp):
np.testing.assert_allclose(E, E_exp)
def test_face(self):
np.testing.assert_allclose(self.F, self.F_exp)
def cube_closest_face(img_path):
filename = os.path.join(img_path, 'cube_closest_face.jpg')
size = (800 * 1.618, 800)
pt = np.array([0.7, 0.2, 0])
v, f = wavefront.read_obj('./integration/cube.obj')
mesh_parameters = wavefront.polyhedron_parameters(v, f)
edge_vertex_map = mesh_parameters.edge_vertex_map
edge_face_map = mesh_parameters.edge_face_map
normal_face = mesh_parameters.normal_face
vf_map = mesh_parameters.vertex_face_map
D, P, V, E, F = wavefront.distance_to_faces(pt, v, f, normal_face,
edge_vertex_map, edge_face_map,
vf_map)
mfig = plot_data(pt, v, f, D, P, V, E, F, '')
graphics.mlab.view(azimuth=view['azimuth'],
elevation=view['elevation'],
distance=view['distance'],
focalpoint=view['focalpoint'],
figure=mfig)
graphics.mlab.savefig(filename, magnification=4)
return mfig
class TestDistanceToFacesCubeOutsideFixedMultiple():
"""This point is equally close to an entire face of the cube
So there are 4 vertices equidistant from pt
"""
pt_out = np.array([1, 0, 0])
v_cube, f_cube = wavefront.read_obj('./integration/cube.obj')
ast = asteroid.Asteroid('castalia', 256, 'mat')
ast = ast.loadmesh(v_cube, f_cube, 'cube')
edge_vertex_map = ast.asteroid_grav['edge_vertex_map']
edge_face_map = ast.asteroid_grav['edge_face_map']
normal_face = ast.asteroid_grav['normal_face']
vf_map = ast.asteroid_grav['vertex_face_map']
D, P, V, E, F = wavefront.distance_to_faces(pt_out, v_cube, f_cube,
normal_face,
edge_vertex_map,
edge_face_map,
vf_map)
D_exp = np.ones_like(D) * 0.5
P_exp = np.array([[0.5, 0, 0],
[0.5, 0, 0]])
V_exp = np.array([[4, 6, 7],
[4, 7, 5]])
E_exp = np.array([[[6, 4], [7, 6], [4, 7]], [[7, 4], [5, 7], [4, 5]]])
F_exp = np.array([6, 7])
def test_distance(self):
np.testing.assert_allclose(np.absolute(self.D), self.D_exp)
def test_point(self):
np.testing.assert_allclose(self.P, self.P_exp)
def test_vertex(self):
np.testing.assert_allclose(self.V, self.V_exp)
def test_edge(self):
for E, E_exp in zip(self.E, self.E_exp):
np.testing.assert_allclose(E, E_exp)
def test_face(self):
for F, F_exp in zip(self.F, self.F_exp):
np.testing.assert_allclose(F, F_exp)
def test_closest_face_plot_asteroid(pt=np.random.uniform(0.8, 1.5) *
sphere.rand(2)):
"""Needs to be aligned with a face
i.e. not in the corners
"""
ast = asteroid.Asteroid('itokawa', 0, 'obj')
v, f = ast.asteroid_grav['V'], ast.asteroid_grav['F']
edge_vertex_map = ast.asteroid_grav['edge_vertex_map']
edge_face_map = ast.asteroid_grav['edge_face_map']
normal_face = ast.asteroid_grav['normal_face']
vf_map = ast.asteroid_grav['vertex_face_map']
D, P, V, E, F = wavefront.distance_to_faces(pt, v, f, normal_face,
edge_vertex_map, edge_face_map,
vf_map)
plot_data(pt, v, f, D, P, V, E, F, 'Closest Face')
return D, P, V, E, F
def test_closest_face_plot_cube(pt=np.random.uniform(0.8, 1.5) *
sphere.rand(2)):
"""Needs to be aligned with a face
i.e. not in the corners
"""
ast = asteroid.Asteroid('castalia', 256, 'mat')
v, f = wavefront.read_obj('./integration/cube.obj')
ast = ast.loadmesh(v, f, 'cube')
edge_vertex_map = ast.asteroid_grav['edge_vertex_map']
edge_face_map = ast.asteroid_grav['edge_face_map']
normal_face = ast.asteroid_grav['normal_face']
vf_map = ast.asteroid_grav['vertex_face_map']
D, P, V, E, F = wavefront.distance_to_faces(pt, v, f, normal_face,
edge_vertex_map, edge_face_map,
vf_map)
plot_data(pt, v, f, D, P, V, E, F, '')
return D, P, V, E, F
class TestDistanceToFacesCubeOutsideFixedSingle():
pt_out = np.array([1, 0.1, 0])
v_cube, f_cube = wavefront.read_obj('./integration/cube.obj')
ast = asteroid.Asteroid('castalia', 256, 'mat')
ast = ast.loadmesh(v_cube, f_cube, 'cube')
edge_vertex_map = ast.asteroid_grav['edge_vertex_map']
edge_face_map = ast.asteroid_grav['edge_face_map']
normal_face = ast.asteroid_grav['normal_face']
vf_map = ast.asteroid_grav['vertex_face_map']
D, P, V, E, F = wavefront.distance_to_faces(pt_out, v_cube, f_cube,
normal_face,
edge_vertex_map,
edge_face_map,
vf_map)
D_exp = 0.5
P_exp = np.array([0.5, 0.1, 0])
V_exp = [4, 6, 7]
E_exp = np.array([[6, 4],
[7, 6],
[4, 7]])
F_exp = [6]
def test_distance(self):
np.testing.assert_allclose(self.D, self.D_exp)
def test_point(self):
np.testing.assert_array_almost_equal(self.P, self.P_exp)
def test_vertex(self):
np.testing.assert_allclose(self.V, self.V_exp)
def test_edge(self):
np.testing.assert_allclose(self.E, self.E_exp)
def test_face(self):
np.testing.assert_allclose(self.F, self.F_exp)