1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
try:
from . import generic as g
except BaseException:
import generic as g
class IntegralMeanCurvatureTest(g.unittest.TestCase):
def test_IMCsphere(self):
# how close do we need to be - relative tolerance
tol = 1e-3
for radius in [0.1, 1.0, 3.1459, 29.20]:
m = g.trimesh.creation.icosphere(subdivisions=4, radius=radius)
IMC = m.integral_mean_curvature
ref = 4 * g.np.pi * radius
assert g.np.isclose(IMC, ref, rtol=tol)
def test_IMCcapsule(self):
# how close do we need to be - relative tolerance
tol = 1e-3
radius = 1.2
for aspect_ratio in [0.0, 0.5, 1.0, 4.0, 100]:
L = aspect_ratio * radius
m = g.trimesh.creation.capsule(height=L, radius=radius, count=[64, 64])
IMC = m.integral_mean_curvature
ref = g.np.pi * (L + 4 * radius)
assert g.np.isclose(IMC, ref, rtol=tol)
def test_IMCbox(self):
# how close do we need to be - relative tolerance
tol = 1e-3
n_tests = 4
extents = 1 - g.random((n_tests, 3))
for extent in extents:
m = g.trimesh.creation.box(extents=extent)
IMC = m.integral_mean_curvature
# only the right angle (=pi/2) edges contribute,
# edges that lie on the faces of the box do not
ref = 4 * extent.sum() * g.np.pi / 2 * 0.5
assert g.np.isclose(IMC, ref, rtol=tol)
if __name__ == "__main__":
g.trimesh.util.attach_to_log()
g.unittest.main()
|
