Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
godotengine
GitHub Repository: godotengine/godot
Path: blob/master/scene/3d/iterate_ik_3d.cpp
14709 views
1
/**************************************************************************/
2
/* iterate_ik_3d.cpp */
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
#include "iterate_ik_3d.h"
32
33
bool IterateIK3D::_set(const StringName &p_path, const Variant &p_value) {
34
String path = p_path;
35
36
if (path.begins_with("settings/")) {
37
int which = path.get_slicec('/', 1).to_int();
38
String what = path.get_slicec('/', 2);
39
ERR_FAIL_INDEX_V(which, (int)settings.size(), false);
40
41
if (what == "target_node") {
42
set_target_node(which, p_value);
43
} else if (what == "joints") {
44
int idx = path.get_slicec('/', 3).to_int();
45
String prop = path.get_slicec('/', 4);
46
if (prop == "rotation_axis") {
47
set_joint_rotation_axis(which, idx, static_cast<RotationAxis>((int)p_value));
48
} else if (prop == "rotation_axis_vector") {
49
set_joint_rotation_axis_vector(which, idx, p_value);
50
} else if (prop == "limitation") {
51
String opt = path.get_slicec('/', 5);
52
if (opt.is_empty()) {
53
set_joint_limitation(which, idx, p_value);
54
} else if (opt == "right_axis") {
55
set_joint_limitation_right_axis(which, idx, p_value);
56
} else if (opt == "right_axis_vector") {
57
set_joint_limitation_right_axis_vector(which, idx, p_value);
58
} else if (opt == "rotation_offset") {
59
set_joint_limitation_rotation_offset(which, idx, p_value);
60
} else {
61
return false;
62
}
63
} else {
64
return false;
65
}
66
} else {
67
return false;
68
}
69
}
70
return true;
71
}
72
73
bool IterateIK3D::_get(const StringName &p_path, Variant &r_ret) const {
74
String path = p_path;
75
76
if (path.begins_with("settings/")) {
77
int which = path.get_slicec('/', 1).to_int();
78
String what = path.get_slicec('/', 2);
79
ERR_FAIL_INDEX_V(which, (int)settings.size(), false);
80
81
if (what == "target_node") {
82
r_ret = get_target_node(which);
83
} else if (what == "joints") {
84
int idx = path.get_slicec('/', 3).to_int();
85
String prop = path.get_slicec('/', 4);
86
if (prop == "rotation_axis") {
87
r_ret = (int)get_joint_rotation_axis(which, idx);
88
} else if (prop == "rotation_axis_vector") {
89
r_ret = get_joint_rotation_axis_vector(which, idx);
90
} else if (prop == "limitation") {
91
String opt = path.get_slicec('/', 5);
92
if (opt.is_empty()) {
93
r_ret = get_joint_limitation(which, idx);
94
} else if (opt == "right_axis") {
95
r_ret = get_joint_limitation_right_axis(which, idx);
96
} else if (opt == "right_axis_vector") {
97
r_ret = get_joint_limitation_right_axis_vector(which, idx);
98
} else if (opt == "rotation_offset") {
99
r_ret = get_joint_limitation_rotation_offset(which, idx);
100
} else {
101
return false;
102
}
103
} else {
104
return false;
105
}
106
} else {
107
return false;
108
}
109
}
110
return true;
111
}
112
113
void IterateIK3D::_get_property_list(List<PropertyInfo> *p_list) const {
114
LocalVector<PropertyInfo> props;
115
for (uint32_t i = 0; i < settings.size(); i++) {
116
String path = "settings/" + itos(i) + "/";
117
p_list->push_back(PropertyInfo(Variant::NODE_PATH, path + "target_node"));
118
for (uint32_t j = 0; j < iterate_settings[i]->joints.size(); j++) {
119
String joint_path = path + "joints/" + itos(j) + "/";
120
props.push_back(PropertyInfo(Variant::INT, joint_path + "rotation_axis", PROPERTY_HINT_ENUM, SkeletonModifier3D::get_hint_rotation_axis()));
121
props.push_back(PropertyInfo(Variant::VECTOR3, joint_path + "rotation_axis_vector"));
122
props.push_back(PropertyInfo(Variant::OBJECT, joint_path + "limitation", PROPERTY_HINT_RESOURCE_TYPE, "JointLimitation3D"));
123
props.push_back(PropertyInfo(Variant::INT, joint_path + "limitation/right_axis", PROPERTY_HINT_ENUM, SkeletonModifier3D::get_hint_secondary_direction()));
124
props.push_back(PropertyInfo(Variant::VECTOR3, joint_path + "limitation/right_axis_vector"));
125
props.push_back(PropertyInfo(Variant::QUATERNION, joint_path + "limitation/rotation_offset"));
126
}
127
}
128
129
ChainIK3D::get_property_list(p_list);
130
131
for (PropertyInfo &p : props) {
132
_validate_dynamic_prop(p);
133
p_list->push_back(p);
134
}
135
}
136
137
void IterateIK3D::_validate_dynamic_prop(PropertyInfo &p_property) const {
138
PackedStringArray split = p_property.name.split("/");
139
if (split.size() > 3 && split[0] == "settings") {
140
int which = split[1].to_int();
141
int joint = split[3].to_int();
142
// Joints option.
143
if (split[2] == "joints" && split.size() > 4) {
144
if (split[4] == "rotation_axis_vector" && get_joint_rotation_axis(which, joint) != ROTATION_AXIS_CUSTOM) {
145
p_property.usage = PROPERTY_USAGE_NONE;
146
}
147
if (split[4] == "limitation" && split.size() > 5) {
148
if (get_joint_limitation(which, joint).is_null()) {
149
p_property.usage = PROPERTY_USAGE_NONE;
150
} else if (split[5] == "right_axis_vector" && get_joint_limitation_right_axis(which, joint) != SECONDARY_DIRECTION_CUSTOM) {
151
p_property.usage = PROPERTY_USAGE_NONE;
152
}
153
}
154
}
155
}
156
}
157
158
PackedStringArray IterateIK3D::get_configuration_warnings() const {
159
PackedStringArray warnings = SkeletonModifier3D::get_configuration_warnings();
160
for (uint32_t i = 0; i < iterate_settings.size(); i++) {
161
if (iterate_settings[i]->target_node.is_empty()) {
162
warnings.push_back(RTR("Detecting settings with no target set! IterateIK3D must have a target to work."));
163
break;
164
}
165
}
166
return warnings;
167
}
168
169
void IterateIK3D::set_max_iterations(int p_max_iterations) {
170
max_iterations = p_max_iterations;
171
}
172
173
int IterateIK3D::get_max_iterations() const {
174
return max_iterations;
175
}
176
177
void IterateIK3D::set_min_distance(double p_min_distance) {
178
min_distance = p_min_distance;
179
}
180
181
double IterateIK3D::get_min_distance() const {
182
return min_distance;
183
}
184
185
void IterateIK3D::set_angular_delta_limit(double p_angular_delta_limit) {
186
angular_delta_limit = p_angular_delta_limit;
187
}
188
189
double IterateIK3D::get_angular_delta_limit() const {
190
return angular_delta_limit;
191
}
192
193
void IterateIK3D::set_deterministic(bool p_deterministic) {
194
deterministic = p_deterministic;
195
}
196
197
bool IterateIK3D::is_deterministic() const {
198
return deterministic;
199
}
200
201
// Setting.
202
203
void IterateIK3D::set_target_node(int p_index, const NodePath &p_node_path) {
204
ERR_FAIL_INDEX(p_index, (int)settings.size());
205
iterate_settings[p_index]->target_node = p_node_path;
206
update_configuration_warnings();
207
}
208
209
NodePath IterateIK3D::get_target_node(int p_index) const {
210
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), NodePath());
211
return iterate_settings[p_index]->target_node;
212
}
213
214
// Individual joints.
215
216
void IterateIK3D::set_joint_rotation_axis(int p_index, int p_joint, RotationAxis p_axis) {
217
ERR_FAIL_INDEX(p_index, (int)settings.size());
218
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
219
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
220
joint_settings[p_joint]->rotation_axis = p_axis;
221
Skeleton3D *sk = get_skeleton();
222
if (sk) {
223
_validate_axis(sk, p_index, p_joint);
224
}
225
notify_property_list_changed();
226
_make_simulation_dirty(p_index); // Snapping to planes is needed in the initialization, so need to restructure.
227
}
228
229
SkeletonModifier3D::RotationAxis IterateIK3D::get_joint_rotation_axis(int p_index, int p_joint) const {
230
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), ROTATION_AXIS_ALL);
231
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
232
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), ROTATION_AXIS_ALL);
233
return joint_settings[p_joint]->rotation_axis;
234
}
235
236
void IterateIK3D::set_joint_rotation_axis_vector(int p_index, int p_joint, const Vector3 &p_vector) {
237
ERR_FAIL_INDEX(p_index, (int)settings.size());
238
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
239
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
240
joint_settings[p_joint]->rotation_axis_vector = p_vector;
241
Skeleton3D *sk = get_skeleton();
242
if (sk) {
243
_validate_axis(sk, p_index, p_joint);
244
}
245
_make_simulation_dirty(p_index); // Snapping to planes is needed in the initialization, so need to restructure.
246
}
247
248
Vector3 IterateIK3D::get_joint_rotation_axis_vector(int p_index, int p_joint) const {
249
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Vector3());
250
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
251
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), Vector3());
252
return joint_settings[p_joint]->get_rotation_axis_vector();
253
}
254
255
Quaternion IterateIK3D::get_joint_limitation_space(int p_index, int p_joint, const Vector3 &p_forward) const {
256
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Quaternion());
257
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
258
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), Quaternion());
259
return joint_settings[p_joint]->get_limitation_space(p_forward);
260
}
261
262
void IterateIK3D::set_joint_limitation(int p_index, int p_joint, const Ref<JointLimitation3D> &p_limitation) {
263
ERR_FAIL_INDEX(p_index, (int)settings.size());
264
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
265
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
266
_unbind_joint_limitation(p_index, p_joint);
267
joint_settings[p_joint]->limitation = p_limitation;
268
_bind_joint_limitation(p_index, p_joint);
269
notify_property_list_changed();
270
_update_joint_limitation(p_index, p_joint);
271
}
272
273
Ref<JointLimitation3D> IterateIK3D::get_joint_limitation(int p_index, int p_joint) const {
274
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Ref<JointLimitation3D>());
275
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
276
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), Ref<JointLimitation3D>());
277
return joint_settings[p_joint]->limitation;
278
}
279
280
void IterateIK3D::set_joint_limitation_right_axis(int p_index, int p_joint, SecondaryDirection p_direction) {
281
ERR_FAIL_INDEX(p_index, (int)settings.size());
282
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
283
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
284
joint_settings[p_joint]->limitation_right_axis = p_direction;
285
notify_property_list_changed();
286
_update_joint_limitation(p_index, p_joint);
287
}
288
289
IKModifier3D::SecondaryDirection IterateIK3D::get_joint_limitation_right_axis(int p_index, int p_joint) const {
290
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), SECONDARY_DIRECTION_NONE);
291
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
292
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), SECONDARY_DIRECTION_NONE);
293
return joint_settings[p_joint]->limitation_right_axis;
294
}
295
296
void IterateIK3D::set_joint_limitation_right_axis_vector(int p_index, int p_joint, const Vector3 &p_vector) {
297
ERR_FAIL_INDEX(p_index, (int)settings.size());
298
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
299
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
300
joint_settings[p_joint]->limitation_right_axis_vector = p_vector;
301
_update_joint_limitation(p_index, p_joint);
302
}
303
304
Vector3 IterateIK3D::get_joint_limitation_right_axis_vector(int p_index, int p_joint) const {
305
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Vector3());
306
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
307
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), Vector3());
308
return joint_settings[p_joint]->get_limitation_right_axis_vector();
309
}
310
311
void IterateIK3D::set_joint_limitation_rotation_offset(int p_index, int p_joint, const Quaternion &p_offset) {
312
ERR_FAIL_INDEX(p_index, (int)settings.size());
313
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
314
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
315
joint_settings[p_joint]->limitation_rotation_offset = p_offset;
316
_update_joint_limitation(p_index, p_joint);
317
}
318
319
Quaternion IterateIK3D::get_joint_limitation_rotation_offset(int p_index, int p_joint) const {
320
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Quaternion());
321
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
322
ERR_FAIL_INDEX_V(p_joint, (int)joint_settings.size(), Quaternion());
323
return joint_settings[p_joint]->limitation_rotation_offset;
324
}
325
326
void IterateIK3D::_set_joint_count(int p_index, int p_count) {
327
_unbind_joint_limitations(p_index);
328
LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
329
int delta = p_count - joint_settings.size();
330
if (delta < 0) {
331
for (int i = delta; i < 0; i++) {
332
memdelete(joint_settings[joint_settings.size() + i]);
333
joint_settings[joint_settings.size() + i] = nullptr;
334
}
335
}
336
joint_settings.resize(p_count);
337
delta++;
338
if (delta > 1) {
339
for (int i = 1; i < delta; i++) {
340
joint_settings[p_count - i] = memnew(IterateIK3DJointSetting);
341
}
342
}
343
}
344
345
void IterateIK3D::_validate_axis(Skeleton3D *p_skeleton, int p_index, int p_joint) const {
346
RotationAxis axis = iterate_settings[p_index]->joint_settings[p_joint]->rotation_axis;
347
if (axis == ROTATION_AXIS_ALL) {
348
return;
349
}
350
Vector3 rot = get_joint_rotation_axis_vector(p_index, p_joint).normalized();
351
Vector3 fwd;
352
if (p_joint < (int)iterate_settings[p_index]->joints.size() - 1) {
353
fwd = p_skeleton->get_bone_rest(iterate_settings[p_index]->joints[p_joint + 1].bone).origin;
354
} else if (iterate_settings[p_index]->extend_end_bone) {
355
fwd = IKModifier3D::get_bone_axis(p_skeleton, iterate_settings[p_index]->end_bone.bone, iterate_settings[p_index]->end_bone_direction, mutable_bone_axes);
356
if (fwd.is_zero_approx()) {
357
return;
358
}
359
}
360
fwd.normalize();
361
if (Math::is_equal_approx(Math::abs(rot.dot(fwd)), 1)) {
362
WARN_PRINT_ED("Setting: " + itos(p_index) + " Joint: " + itos(p_joint) + ": Rotation axis and forward vector are colinear. This is not advised as it may cause unwanted rotation.");
363
}
364
}
365
366
void IterateIK3D::_bind_methods() {
367
ClassDB::bind_method(D_METHOD("set_max_iterations", "max_iterations"), &IterateIK3D::set_max_iterations);
368
ClassDB::bind_method(D_METHOD("get_max_iterations"), &IterateIK3D::get_max_iterations);
369
ClassDB::bind_method(D_METHOD("set_min_distance", "min_distance"), &IterateIK3D::set_min_distance);
370
ClassDB::bind_method(D_METHOD("get_min_distance"), &IterateIK3D::get_min_distance);
371
ClassDB::bind_method(D_METHOD("set_angular_delta_limit", "angular_delta_limit"), &IterateIK3D::set_angular_delta_limit);
372
ClassDB::bind_method(D_METHOD("get_angular_delta_limit"), &IterateIK3D::get_angular_delta_limit);
373
ClassDB::bind_method(D_METHOD("set_deterministic", "deterministic"), &IterateIK3D::set_deterministic);
374
ClassDB::bind_method(D_METHOD("is_deterministic"), &IterateIK3D::is_deterministic);
375
376
// Setting.
377
ClassDB::bind_method(D_METHOD("set_target_node", "index", "target_node"), &IterateIK3D::set_target_node);
378
ClassDB::bind_method(D_METHOD("get_target_node", "index"), &IterateIK3D::get_target_node);
379
380
// Individual joints.
381
ClassDB::bind_method(D_METHOD("set_joint_rotation_axis", "index", "joint", "axis"), &IterateIK3D::set_joint_rotation_axis);
382
ClassDB::bind_method(D_METHOD("get_joint_rotation_axis", "index", "joint"), &IterateIK3D::get_joint_rotation_axis);
383
ClassDB::bind_method(D_METHOD("set_joint_rotation_axis_vector", "index", "joint", "axis_vector"), &IterateIK3D::set_joint_rotation_axis_vector);
384
ClassDB::bind_method(D_METHOD("get_joint_rotation_axis_vector", "index", "joint"), &IterateIK3D::get_joint_rotation_axis_vector);
385
ClassDB::bind_method(D_METHOD("set_joint_limitation", "index", "joint", "limitation"), &IterateIK3D::set_joint_limitation);
386
ClassDB::bind_method(D_METHOD("get_joint_limitation", "index", "joint"), &IterateIK3D::get_joint_limitation);
387
ClassDB::bind_method(D_METHOD("set_joint_limitation_right_axis", "index", "joint", "direction"), &IterateIK3D::set_joint_limitation_right_axis);
388
ClassDB::bind_method(D_METHOD("get_joint_limitation_right_axis", "index", "joint"), &IterateIK3D::get_joint_limitation_right_axis);
389
ClassDB::bind_method(D_METHOD("set_joint_limitation_right_axis_vector", "index", "joint", "vector"), &IterateIK3D::set_joint_limitation_right_axis_vector);
390
ClassDB::bind_method(D_METHOD("get_joint_limitation_right_axis_vector", "index", "joint"), &IterateIK3D::get_joint_limitation_right_axis_vector);
391
ClassDB::bind_method(D_METHOD("set_joint_limitation_rotation_offset", "index", "joint", "offset"), &IterateIK3D::set_joint_limitation_rotation_offset);
392
ClassDB::bind_method(D_METHOD("get_joint_limitation_rotation_offset", "index", "joint"), &IterateIK3D::get_joint_limitation_rotation_offset);
393
394
ADD_PROPERTY(PropertyInfo(Variant::INT, "max_iterations", PROPERTY_HINT_RANGE, "0,100,or_greater"), "set_max_iterations", "get_max_iterations");
395
ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "min_distance", PROPERTY_HINT_RANGE, "0,1,0.001,or_greater"), "set_min_distance", "get_min_distance");
396
ADD_PROPERTY(PropertyInfo(Variant::FLOAT, "angular_delta_limit", PROPERTY_HINT_RANGE, "0,180,0.001,radians_as_degrees"), "set_angular_delta_limit", "get_angular_delta_limit");
397
ADD_PROPERTY(PropertyInfo(Variant::BOOL, "deterministic"), "set_deterministic", "is_deterministic");
398
ADD_ARRAY_COUNT("Settings", "setting_count", "set_setting_count", "get_setting_count", "settings/");
399
}
400
401
void IterateIK3D::_clear_joints(int p_index) {
402
IterateIK3DSetting *setting = iterate_settings[p_index];
403
if (!setting) {
404
return;
405
}
406
_unbind_joint_limitations(p_index);
407
for (uint32_t i = 0; i < setting->solver_info_list.size(); i++) {
408
if (setting->solver_info_list[i]) {
409
memdelete(setting->solver_info_list[i]);
410
setting->solver_info_list[i] = nullptr;
411
}
412
}
413
setting->solver_info_list.clear();
414
setting->solver_info_list.resize_initialized(setting->joints.size());
415
_bind_joint_limitations(p_index);
416
}
417
418
void IterateIK3D::_init_joints(Skeleton3D *p_skeleton, int p_index) {
419
IterateIK3DSetting *setting = iterate_settings[p_index];
420
if (!setting) {
421
return;
422
}
423
cached_space = p_skeleton->get_global_transform_interpolated();
424
if (setting->simulation_dirty) {
425
_clear_joints(p_index);
426
setting->init_joints(p_skeleton, mutable_bone_axes);
427
setting->simulation_dirty = false;
428
} else if (deterministic) {
429
setting->init_joints(p_skeleton, mutable_bone_axes);
430
}
431
432
if (mutable_bone_axes) {
433
#ifdef TOOLS_ENABLED
434
_update_mutable_info();
435
#endif // TOOLS_ENABLED
436
_update_bone_axis(p_skeleton, p_index);
437
}
438
setting->simulated = false;
439
}
440
441
void IterateIK3D::_make_simulation_dirty(int p_index) {
442
IterateIK3DSetting *setting = iterate_settings[p_index];
443
if (!setting) {
444
return;
445
}
446
setting->simulation_dirty = true;
447
#ifdef TOOLS_ENABLED
448
if (!mutable_bone_axes) {
449
_make_gizmo_dirty();
450
}
451
#endif // TOOLS_ENABLED
452
}
453
454
void IterateIK3D::_update_bone_axis(Skeleton3D *p_skeleton, int p_index) {
455
#ifdef TOOLS_ENABLED
456
bool changed = false;
457
#endif // TOOLS_ENABLED
458
IterateIK3DSetting *setting = iterate_settings[p_index];
459
const LocalVector<BoneJoint> &joints = setting->joints;
460
const LocalVector<IKModifier3DSolverInfo *> &solver_info_list = setting->solver_info_list;
461
int len = (int)solver_info_list.size() - 1;
462
for (int j = 0; j < len; j++) {
463
IterateIK3DJointSetting *joint_setting = setting->joint_settings[j];
464
if (!joint_setting || !solver_info_list[j]) {
465
continue;
466
}
467
Vector3 axis = p_skeleton->get_bone_pose(joints[j + 1].bone).origin;
468
if (axis.is_zero_approx()) {
469
continue;
470
}
471
// Less computing.
472
#ifdef TOOLS_ENABLED
473
if (!changed) {
474
Vector3 old_v = solver_info_list[j]->forward_vector;
475
solver_info_list[j]->forward_vector = snap_vector_to_plane(joint_setting->get_rotation_axis_vector(), axis.normalized());
476
changed = changed || !old_v.is_equal_approx(solver_info_list[j]->forward_vector);
477
float old_l = solver_info_list[j]->length;
478
solver_info_list[j]->length = axis.length();
479
changed = changed || !Math::is_equal_approx(old_l, solver_info_list[j]->length);
480
} else {
481
solver_info_list[j]->forward_vector = snap_vector_to_plane(joint_setting->get_rotation_axis_vector(), axis.normalized());
482
solver_info_list[j]->length = axis.length();
483
}
484
#else
485
solver_info_list[j]->forward_vector = snap_vector_to_plane(joint_setting->get_rotation_axis_vector(), axis.normalized());
486
solver_info_list[j]->length = axis.length();
487
#endif // TOOLS_ENABLED
488
}
489
if (setting->extend_end_bone && len >= 0) {
490
IterateIK3DJointSetting *joint_setting = setting->joint_settings[len];
491
if (joint_setting && solver_info_list[len]) {
492
Vector3 axis = IKModifier3D::get_bone_axis(p_skeleton, setting->end_bone.bone, setting->end_bone_direction, mutable_bone_axes);
493
if (!axis.is_zero_approx()) {
494
solver_info_list[len]->forward_vector = snap_vector_to_plane(joint_setting->get_rotation_axis_vector(), axis.normalized());
495
solver_info_list[len]->length = setting->end_bone_length;
496
}
497
}
498
}
499
#ifdef TOOLS_ENABLED
500
if (changed) {
501
_make_gizmo_dirty();
502
}
503
#endif // TOOLS_ENABLED
504
}
505
506
void IterateIK3D::_process_ik(Skeleton3D *p_skeleton, double p_delta) {
507
min_distance_squared = min_distance * min_distance;
508
for (uint32_t i = 0; i < settings.size(); i++) {
509
_init_joints(p_skeleton, i);
510
Node3D *target = Object::cast_to<Node3D>(get_node_or_null(iterate_settings[i]->target_node));
511
if (!target || iterate_settings[i]->chain.is_empty()) {
512
continue; // Abort.
513
}
514
iterate_settings[i]->cache_current_joint_rotations(p_skeleton); // Iterate over first to detect parent (outside of the chain) bone pose changes.
515
516
Vector3 destination = cached_space.affine_inverse().xform(target->get_global_transform_interpolated().origin);
517
_process_joints(p_delta, p_skeleton, iterate_settings[i], destination);
518
}
519
}
520
521
void IterateIK3D::_process_joints(double p_delta, Skeleton3D *p_skeleton, IterateIK3DSetting *p_setting, const Vector3 &p_destination) {
522
double distance_to_target_sq = INFINITY;
523
int iteration_count = 0;
524
525
// To prevent oscillation, if it has been processed at least once and target was reached, abort iterating.
526
if (p_setting->simulated) {
527
distance_to_target_sq = p_setting->chain[p_setting->chain.size() - 1].distance_squared_to(p_destination);
528
}
529
530
while (distance_to_target_sq > min_distance_squared && iteration_count < max_iterations) {
531
// Solve the IK for this iteration.
532
_solve_iteration(p_delta, p_skeleton, p_setting, p_destination);
533
534
// Update virtual bone rest/poses.
535
p_setting->cache_current_joint_rotations(p_skeleton, angular_delta_limit);
536
distance_to_target_sq = p_setting->chain[p_setting->chain.size() - 1].distance_squared_to(p_destination);
537
iteration_count++;
538
}
539
540
// Apply the virtual bone rest/poses to the actual bones.
541
for (uint32_t i = 0; i < p_setting->solver_info_list.size(); i++) {
542
IKModifier3DSolverInfo *solver_info = p_setting->solver_info_list[i];
543
if (!solver_info || Math::is_zero_approx(solver_info->length)) {
544
continue;
545
}
546
p_skeleton->set_bone_pose_rotation(p_setting->joints[i].bone, solver_info->current_lpose);
547
}
548
549
p_setting->simulated = true;
550
}
551
552
void IterateIK3D::_solve_iteration(double p_delta, Skeleton3D *p_skeleton, IterateIK3DSetting *p_setting, const Vector3 &p_destination) {
553
//
554
}
555
556
void IterateIK3D::_update_joint_limitation(int p_index, int p_joint) {
557
ERR_FAIL_INDEX(p_index, (int)iterate_settings.size());
558
iterate_settings[p_index]->simulated = false;
559
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
560
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size()); // p_joint is unused directly, but need to identify bound index.
561
#ifdef TOOLS_ENABLED
562
update_gizmos();
563
#endif // TOOLS_ENABLED
564
}
565
566
void IterateIK3D::_bind_joint_limitation(int p_index, int p_joint) {
567
ERR_FAIL_INDEX(p_index, (int)iterate_settings.size());
568
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
569
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
570
if (joint_settings[p_joint]->limitation.is_valid()) {
571
joint_settings[p_joint]->limitation->connect_changed(callable_mp(this, &IterateIK3D::_update_joint_limitation).bind(p_index, p_joint));
572
}
573
}
574
575
void IterateIK3D::_unbind_joint_limitation(int p_index, int p_joint) {
576
ERR_FAIL_INDEX(p_index, (int)iterate_settings.size());
577
const LocalVector<IterateIK3DJointSetting *> &joint_settings = iterate_settings[p_index]->joint_settings;
578
ERR_FAIL_INDEX(p_joint, (int)joint_settings.size());
579
if (joint_settings[p_joint]->limitation.is_valid()) {
580
joint_settings[p_joint]->limitation->disconnect_changed(callable_mp(this, &IterateIK3D::_update_joint_limitation).bind(p_index, p_joint));
581
}
582
}
583
584
void IterateIK3D::_bind_joint_limitations(int p_index) {
585
for (uint32_t i = 0; i < iterate_settings[p_index]->joints.size(); i++) {
586
if (iterate_settings[p_index]->joint_settings[i]->limitation.is_valid()) {
587
iterate_settings[p_index]->joint_settings[i]->limitation->connect_changed(callable_mp(this, &IterateIK3D::_update_joint_limitation).bind(p_index, i));
588
}
589
}
590
}
591
592
void IterateIK3D::_unbind_joint_limitations(int p_index) {
593
for (uint32_t i = 0; i < iterate_settings[p_index]->joint_settings.size(); i++) {
594
if (iterate_settings[p_index]->joint_settings[i]->limitation.is_valid()) {
595
iterate_settings[p_index]->joint_settings[i]->limitation->disconnect_changed(callable_mp(this, &IterateIK3D::_update_joint_limitation).bind(p_index, i));
596
}
597
}
598
}
599
600
#ifdef TOOLS_ENABLED
601
Vector3 IterateIK3D::get_bone_vector(int p_index, int p_joint) const {
602
Skeleton3D *skeleton = get_skeleton();
603
if (!skeleton) {
604
return Vector3();
605
}
606
ERR_FAIL_INDEX_V(p_index, (int)settings.size(), Vector3());
607
IterateIK3DSetting *setting = iterate_settings[p_index];
608
if (!setting) {
609
return Vector3();
610
}
611
const LocalVector<BoneJoint> &joints = setting->joints;
612
ERR_FAIL_INDEX_V(p_joint, (int)joints.size(), Vector3());
613
const LocalVector<IKModifier3DSolverInfo *> &solver_info_list = setting->solver_info_list;
614
if (p_joint >= (int)solver_info_list.size() || !solver_info_list[p_joint]) {
615
if (p_joint == (int)joints.size() - 1) {
616
return IKModifier3D::get_bone_axis(skeleton, setting->end_bone.bone, setting->end_bone_direction, mutable_bone_axes) * setting->end_bone_length;
617
}
618
return mutable_bone_axes ? skeleton->get_bone_pose(joints[p_joint + 1].bone).origin : skeleton->get_bone_rest(joints[p_joint + 1].bone).origin;
619
}
620
return solver_info_list[p_joint]->forward_vector * solver_info_list[p_joint]->length;
621
}
622
#endif // TOOLS_ENABLED
623
624
IterateIK3D::~IterateIK3D() {
625
for (uint32_t i = 0; i < iterate_settings.size(); i++) {
626
_unbind_joint_limitations(i);
627
}
628
clear_settings();
629
}
630
631