CoCalc -- test_convert_transform_modifier

GitHub Repository: godotengine/godot
Path: blob/master/tests/scene/test_convert_transform_modifier_3d.h
¹⁰²⁷⁷ views
1
/**************************************************************************/
2
/*  test_convert_transform_modifier_3d.h                                  */
3
/**************************************************************************/
4
/*                         This file is part of:                          */
5
/*                             GODOT ENGINE                               */
6
/*                        https://godotengine.org                         */
7
/**************************************************************************/
8
/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
9
/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur.                  */
10
/*                                                                        */
11
/* Permission is hereby granted, free of charge, to any person obtaining  */
12
/* a copy of this software and associated documentation files (the        */
13
/* "Software"), to deal in the Software without restriction, including    */
14
/* without limitation the rights to use, copy, modify, merge, publish,    */
15
/* distribute, sublicense, and/or sell copies of the Software, and to     */
16
/* permit persons to whom the Software is furnished to do so, subject to  */
17
/* the following conditions:                                              */
18
/*                                                                        */
19
/* The above copyright notice and this permission notice shall be         */
20
/* included in all copies or substantial portions of the Software.        */
21
/*                                                                        */
22
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,        */
23
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF     */
24
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
25
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY   */
26
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,   */
27
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE      */
28
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                 */
29
/**************************************************************************/
30

31
#pragma once
32

33
#include "tests/test_macros.h"
34

35
#include "scene/3d/bone_attachment_3d.h"
36
#include "scene/3d/convert_transform_modifier_3d.h"
37

38
namespace TestConvertTransformModifier3D {
39

40
Transform3D make_random_transform_3d(int p_seed) {
41
	RandomNumberGenerator rng;
42
	rng.set_seed(p_seed);
43

44
	Vector3 pos;
45
	pos.x = rng.randf_range(-10.0, 10.0);
46
	rng.set_seed(++p_seed);
47
	pos.y = rng.randf_range(-10.0, 10.0);
48
	rng.set_seed(++p_seed);
49
	pos.z = rng.randf_range(-10.0, 10.0);
50
	rng.set_seed(++p_seed);
51

52
	Quaternion rot;
53
	rot.x = rng.randf_range(-1.0, 1.0);
54
	rng.set_seed(++p_seed);
55
	rot.y = rng.randf_range(-1.0, 1.0);
56
	rng.set_seed(++p_seed);
57
	rot.z = rng.randf_range(-1.0, 1.0);
58
	rng.set_seed(++p_seed);
59
	rot.w = rng.randf_range(-1.0, 1.0);
60
	rng.set_seed(++p_seed);
61
	rot = rot.normalized();
62

63
	Vector3 scl;
64
	scl.x = rng.randf_range(0.5, 2.0);
65
	rng.set_seed(++p_seed);
66
	scl.y = rng.randf_range(0.5, 2.0);
67
	rng.set_seed(++p_seed);
68
	scl.z = rng.randf_range(0.5, 2.0);
69
	rng.set_seed(++p_seed);
70

71
	return Transform3D(Basis(rot).scaled(scl), pos);
72
}
73

74
TEST_CASE("[SceneTree][ConvertTransformModifier3D]") {
75
	SceneTree *tree = SceneTree::get_singleton();
76
	int seed = 12345;
77
	Skeleton3D *skeleton = memnew(Skeleton3D);
78
	ConvertTransformModifier3D *mod = memnew(ConvertTransformModifier3D);
79

80
	// Instead of awaiting the process to wait to finish deferred process and watch "skeleton_updated" signal,
81
	// force notify NOTIFICATION_UPDATE_SKELETON and get the modified pose from the BoneAttachment's transform.
82
	BoneAttachment3D *modified = memnew(BoneAttachment3D);
83

84
	tree->get_root()->add_child(skeleton);
85

86
	int root = skeleton->add_bone("root");
87
	skeleton->set_bone_rest(root, make_random_transform_3d(++seed));
88
	skeleton->set_bone_pose(root, make_random_transform_3d(++seed));
89

90
	int apl_root = skeleton->add_bone("apl_root");
91
	skeleton->set_bone_parent(apl_root, root);
92
	skeleton->set_bone_rest(apl_root, make_random_transform_3d(++seed));
93
	skeleton->set_bone_pose(apl_root, make_random_transform_3d(++seed));
94

95
	int apl_bone = skeleton->add_bone("apl_bone");
96
	skeleton->set_bone_parent(apl_bone, apl_root);
97
	skeleton->set_bone_rest(apl_bone, make_random_transform_3d(++seed));
98
	skeleton->set_bone_pose(apl_bone, make_random_transform_3d(++seed));
99

100
	int tgt_root = skeleton->add_bone("tgt_root");
101
	skeleton->set_bone_parent(tgt_root, root);
102
	skeleton->set_bone_rest(tgt_root, make_random_transform_3d(++seed));
103
	skeleton->set_bone_pose(tgt_root, make_random_transform_3d(++seed));
104

105
	int tgt_bone = skeleton->add_bone("tgt_bone");
106
	skeleton->set_bone_parent(tgt_bone, tgt_root);
107
	skeleton->set_bone_rest(tgt_bone, make_random_transform_3d(++seed));
108
	skeleton->set_bone_pose(tgt_bone, make_random_transform_3d(++seed));
109

110
	skeleton->add_child(mod);
111
	skeleton->add_child(modified);
112
	modified->set_rotation_edit_mode(Node3D::ROTATION_EDIT_MODE_QUATERNION);
113
	modified->set_bone_idx(apl_bone);
114

115
	mod->set_setting_count(1);
116
	mod->set_reference_bone(0, tgt_bone);
117
	mod->set_apply_bone(0, apl_bone);
118

119
	mod->set_reference_axis(0, Vector3::AXIS_X);
120
	mod->set_apply_axis(0, Vector3::AXIS_Y);
121

122
	// ===== [ConvertTransformModifier3D] Position x to y =====
123
	mod->set_reference_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_POSITION);
124
	mod->set_reference_range_min(0, -100.0);
125
	mod->set_reference_range_max(0, 100.0);
126
	mod->set_apply_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_POSITION);
127
	mod->set_apply_range_min(0, -100.0);
128
	mod->set_apply_range_max(0, 100.0);
129

130
	SUBCASE("[ConvertTransformModifier3D] Position x to y, additive=false, relative=false") {
131
		mod->set_additive(0, false);
132
		mod->set_relative(0, false);
133
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
134
		CHECK(Math::is_equal_approx(
135
				skeleton->get_bone_pose_position(tgt_bone).x,
136
				(skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).origin.y));
137
	}
138

139
	SUBCASE("[ConvertTransformModifier3D] Position x to y, additive=true, relative=false") {
140
		mod->set_additive(0, true);
141
		mod->set_relative(0, false);
142
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
143
		CHECK(Math::is_equal_approx(
144
				skeleton->get_bone_pose_position(tgt_bone).x,
145
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).origin - skeleton->get_bone_pose_position(apl_bone)).y));
146
	}
147

148
	SUBCASE("[ConvertTransformModifier3D] Position x to y, additive=false, relative=true") {
149
		mod->set_additive(0, false);
150
		mod->set_relative(0, true);
151
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
152
		CHECK(Math::is_equal_approx(
153
				(skeleton->get_bone_pose_position(tgt_bone) - skeleton->get_bone_rest(tgt_bone).origin).x,
154
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).origin - skeleton->get_bone_rest(apl_bone).origin).y));
155
	}
156

157
	SUBCASE("[ConvertTransformModifier3D] Position x to y, additive=true, relative=true") {
158
		mod->set_additive(0, true);
159
		mod->set_relative(0, true);
160
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
161
		CHECK(Math::is_equal_approx(
162
				(skeleton->get_bone_pose_position(tgt_bone) - skeleton->get_bone_rest(tgt_bone).origin).x,
163
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).origin - skeleton->get_bone_pose_position(apl_bone)).y));
164
	}
165

166
	// ===== [ConvertTransformModifier3D] Rotation (roll) x to y =====
167
	mod->set_reference_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_ROTATION);
168
	mod->set_reference_range_min(0, -180.0);
169
	mod->set_reference_range_max(0, 180.0);
170
	mod->set_apply_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_ROTATION);
171
	mod->set_apply_range_min(0, -180.0);
172
	mod->set_apply_range_max(0, 180.0);
173

174
	SUBCASE("[ConvertTransformModifier3D] Rotation (roll) x to y, additive=false, relative=false") {
175
		mod->set_additive(0, false);
176
		mod->set_relative(0, false);
177
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
178
		CHECK(Math::is_equal_approx(
179
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_pose_rotation(tgt_bone), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_X)),
180
				BoneConstraint3D::get_roll_angle((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_rotation_quaternion(), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_Y))));
181
	}
182

183
	SUBCASE("[ConvertTransformModifier3D] Rotation (roll) x to y, additive=true, relative=false") {
184
		mod->set_additive(0, true);
185
		mod->set_relative(0, false);
186
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
187
		CHECK(Math::is_equal_approx(
188
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_pose_rotation(tgt_bone), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_X)),
189
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_pose_rotation(apl_bone).inverse() * (skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_rotation_quaternion(), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_Y))));
190
	}
191

192
	SUBCASE("[ConvertTransformModifier3D] Rotation (roll) x to y, additive=false, relative=true") {
193
		mod->set_additive(0, false);
194
		mod->set_relative(0, true);
195
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
196
		CHECK(Math::is_equal_approx(
197
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_rest(tgt_bone).basis.get_rotation_quaternion().inverse() * skeleton->get_bone_pose_rotation(tgt_bone), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_X)),
198
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_rest(apl_bone).basis.get_rotation_quaternion().inverse() * (skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_rotation_quaternion(), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_Y))));
199
	}
200

201
	SUBCASE("[ConvertTransformModifier3D] Rotation (roll) x to y, additive=true, relative=true") {
202
		mod->set_additive(0, true);
203
		mod->set_relative(0, true);
204
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
205
		CHECK(Math::is_equal_approx(
206
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_rest(tgt_bone).basis.get_rotation_quaternion().inverse() * skeleton->get_bone_pose_rotation(tgt_bone), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_X)),
207
				BoneConstraint3D::get_roll_angle(skeleton->get_bone_pose_rotation(apl_bone).inverse() * (skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_rotation_quaternion(), BoneConstraint3D::get_vector_from_axis(Vector3::AXIS_Y))));
208
	}
209

210
	// ===== [ConvertTransformModifier3D] Scale x to y =====
211
	mod->set_reference_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_SCALE);
212
	mod->set_reference_range_min(0, 0);
213
	mod->set_reference_range_max(0, 10.0);
214
	mod->set_apply_transform_mode(0, ConvertTransformModifier3D::TRANSFORM_MODE_SCALE);
215
	mod->set_apply_range_min(0, 0);
216
	mod->set_apply_range_max(0, 10.0);
217

218
	SUBCASE("[ConvertTransformModifier3D] Scale x to y, additive=false, relative=false") {
219
		mod->set_additive(0, false);
220
		mod->set_relative(0, false);
221
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
222
		CHECK(Math::is_equal_approx(
223
				skeleton->get_bone_pose_scale(tgt_bone).x,
224
				(skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_scale().y));
225
	}
226

227
	SUBCASE("[ConvertTransformModifier3D] Scale x to y, additive=true, relative=false") {
228
		mod->set_additive(0, true);
229
		mod->set_relative(0, false);
230
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
231
		CHECK(Math::is_equal_approx(
232
				skeleton->get_bone_pose_scale(tgt_bone).x,
233
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_scale() / skeleton->get_bone_pose_scale(apl_bone)).y));
234
	}
235

236
	SUBCASE("[ConvertTransformModifier3D] Scale x to y, additive=false, relative=true") {
237
		mod->set_additive(0, false);
238
		mod->set_relative(0, true);
239
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
240
		CHECK(Math::is_equal_approx(
241
				(skeleton->get_bone_pose_scale(tgt_bone) / skeleton->get_bone_rest(tgt_bone).basis.get_scale()).x,
242
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_scale() / skeleton->get_bone_rest(apl_bone).basis.get_scale()).y));
243
	}
244

245
	SUBCASE("[ConvertTransformModifier3D] Scale x to y, additive=true, relative=true") {
246
		mod->set_additive(0, true);
247
		mod->set_relative(0, true);
248
		skeleton->notification(Skeleton3D::NOTIFICATION_UPDATE_SKELETON);
249
		CHECK(Math::is_equal_approx(
250
				(skeleton->get_bone_pose_scale(tgt_bone) / skeleton->get_bone_rest(tgt_bone).basis.get_scale()).x,
251
				((skeleton->get_bone_global_pose(apl_root).affine_inverse() * modified->get_transform()).basis.get_scale() / skeleton->get_bone_pose_scale(apl_bone)).y));
252
	}
253

254
	memdelete(modified);
255
	memdelete(mod);
256
	memdelete(skeleton);
257
}
258
} // namespace TestConvertTransformModifier3D
259

260
Product

Resources

Company
