forked from panda3d/panda3d
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_egg_transform.py
More file actions
134 lines (109 loc) · 3.93 KB
/
test_egg_transform.py
File metadata and controls
134 lines (109 loc) · 3.93 KB
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
import pytest
from panda3d import core
# Skip these tests if we can't import egg.
egg = pytest.importorskip("panda3d.egg")
EGG_TRANSFORM_MISSING = """
<CoordinateSystem> { %s }
<Group> {
}
"""
EGG_TRANSFORM_EMPTY = """
<CoordinateSystem> { %s }
<Group> {
<Transform> {
}
}
"""
EGG_TRANSFORM_IDENT = """
<CoordinateSystem> { %s }
<Group> {
<Transform> {
<Matrix4> {
1 0 0 0
0 1 0 0
0 0 1 0
0 0 0 1
}
}
}
"""
EGG_TRANSFORM_MATRIX = """
<CoordinateSystem> { %s }
<Group> {
<Transform> {
<Matrix4> {
5 2 -3 4
5 6 7 8
9 1 -3 2
5 2 5 2
}
}
}
"""
COORD_SYSTEMS = {
core.CS_zup_right: "zup-right",
core.CS_yup_right: "yup-right",
core.CS_zup_left: "zup-left",
core.CS_yup_left: "yup-left",
}
def read_egg_string(string):
"""Reads an EggData from a string."""
stream = core.StringStream(string.encode('utf-8'))
data = egg.EggData()
assert data.read(stream)
return data
@pytest.mark.parametrize("coordsys", COORD_SYSTEMS.keys())
def test_egg_transform_missing(coordsys):
data = read_egg_string(EGG_TRANSFORM_MISSING % COORD_SYSTEMS[coordsys])
assert data.get_coordinate_system() == coordsys
child, = data.get_children()
assert not child.has_transform3d()
assert child.transform_is_identity()
assert child.get_vertex_frame() == core.Mat4D.ident_mat()
assert child.get_node_frame() == core.Mat4D.ident_mat()
assert child.get_vertex_frame_inv() == core.Mat4D.ident_mat()
assert child.get_node_frame_inv() == core.Mat4D.ident_mat()
assert child.get_vertex_to_node() == core.Mat4D.ident_mat()
assert child.get_node_to_vertex() == core.Mat4D.ident_mat()
@pytest.mark.parametrize("coordsys", COORD_SYSTEMS.keys())
def test_egg_transform_empty(coordsys):
data = read_egg_string(EGG_TRANSFORM_EMPTY % COORD_SYSTEMS[coordsys])
assert data.get_coordinate_system() == coordsys
child, = data.get_children()
assert not child.has_transform3d()
assert child.transform_is_identity()
assert child.get_vertex_frame() == core.Mat4D.ident_mat()
assert child.get_node_frame() == core.Mat4D.ident_mat()
assert child.get_vertex_frame_inv() == core.Mat4D.ident_mat()
assert child.get_node_frame_inv() == core.Mat4D.ident_mat()
assert child.get_vertex_to_node() == core.Mat4D.ident_mat()
assert child.get_node_to_vertex() == core.Mat4D.ident_mat()
@pytest.mark.parametrize("coordsys", COORD_SYSTEMS.keys())
def test_egg_transform_ident(coordsys):
data = read_egg_string(EGG_TRANSFORM_IDENT % COORD_SYSTEMS[coordsys])
assert data.get_coordinate_system() == coordsys
child, = data.get_children()
assert child.has_transform3d()
assert child.transform_is_identity()
assert child.get_vertex_frame() == core.Mat4D.ident_mat()
assert child.get_node_frame() == core.Mat4D.ident_mat()
assert child.get_vertex_frame_inv() == core.Mat4D.ident_mat()
assert child.get_node_frame_inv() == core.Mat4D.ident_mat()
assert child.get_vertex_to_node() == core.Mat4D.ident_mat()
assert child.get_node_to_vertex() == core.Mat4D.ident_mat()
@pytest.mark.parametrize("coordsys", COORD_SYSTEMS.keys())
def test_egg_transform_matrix(coordsys):
data = read_egg_string(EGG_TRANSFORM_MATRIX % COORD_SYSTEMS[coordsys])
assert data.get_coordinate_system() == coordsys
mat = core.Mat4D(5, 2, -3, 4, 5, 6, 7, 8, 9, 1, -3, 2, 5, 2, 5, 2)
mat_inv = core.invert(mat)
child, = data.get_children()
assert child.has_transform3d()
assert not child.transform_is_identity()
assert child.get_transform3d() == mat
assert child.get_vertex_frame() == core.Mat4D.ident_mat()
assert child.get_node_frame() == mat
assert child.get_vertex_frame_inv() == core.Mat4D.ident_mat()
assert child.get_node_frame_inv() == mat_inv
assert child.get_vertex_to_node() == mat_inv
assert child.get_node_to_vertex() == mat