spring_bone_simulator_3d.cpp 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779
  1. /**************************************************************************/
  2. /* spring_bone_simulator_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. #include "spring_bone_simulator_3d.h"
  31. #include "scene/3d/spring_bone_collision_3d.h"
  32. // Original VRM Spring Bone movement logic was distributed by (c) VRM Consortium. Licensed under the MIT license.
  33. bool SpringBoneSimulator3D::_set(const StringName &p_path, const Variant &p_value) {
  34. String path = p_path;
  35. if (path.begins_with("settings/")) {
  36. int which = path.get_slicec('/', 1).to_int();
  37. String what = path.get_slicec('/', 2);
  38. ERR_FAIL_INDEX_V(which, settings.size(), false);
  39. if (what == "root_bone_name") {
  40. set_root_bone_name(which, p_value);
  41. } else if (what == "root_bone") {
  42. set_root_bone(which, p_value);
  43. } else if (what == "end_bone_name") {
  44. set_end_bone_name(which, p_value);
  45. } else if (what == "end_bone") {
  46. String opt = path.get_slicec('/', 3);
  47. if (opt.is_empty()) {
  48. set_end_bone(which, p_value);
  49. } else if (opt == "direction") {
  50. set_end_bone_direction(which, static_cast<BoneDirection>((int)p_value));
  51. } else if (opt == "length") {
  52. set_end_bone_length(which, p_value);
  53. } else {
  54. return false;
  55. }
  56. } else if (what == "extend_end_bone") {
  57. set_extend_end_bone(which, p_value);
  58. } else if (what == "center_from") {
  59. set_center_from(which, static_cast<CenterFrom>((int)p_value));
  60. } else if (what == "center_node") {
  61. set_center_node(which, p_value);
  62. } else if (what == "center_bone") {
  63. set_center_bone(which, p_value);
  64. } else if (what == "center_bone_name") {
  65. set_center_bone_name(which, p_value);
  66. } else if (what == "individual_config") {
  67. set_individual_config(which, p_value);
  68. } else if (what == "rotation_axis") {
  69. set_rotation_axis(which, static_cast<RotationAxis>((int)p_value));
  70. } else if (what == "rotation_axis_vector") {
  71. set_rotation_axis_vector(which, p_value);
  72. } else if (what == "radius") {
  73. String opt = path.get_slicec('/', 3);
  74. if (opt == "value") {
  75. set_radius(which, p_value);
  76. } else if (opt == "damping_curve") {
  77. set_radius_damping_curve(which, p_value);
  78. } else {
  79. return false;
  80. }
  81. } else if (what == "stiffness") {
  82. String opt = path.get_slicec('/', 3);
  83. if (opt == "value") {
  84. set_stiffness(which, p_value);
  85. } else if (opt == "damping_curve") {
  86. set_stiffness_damping_curve(which, p_value);
  87. } else {
  88. return false;
  89. }
  90. } else if (what == "drag") {
  91. String opt = path.get_slicec('/', 3);
  92. if (opt == "value") {
  93. set_drag(which, p_value);
  94. } else if (opt == "damping_curve") {
  95. set_drag_damping_curve(which, p_value);
  96. } else {
  97. return false;
  98. }
  99. } else if (what == "gravity") {
  100. String opt = path.get_slicec('/', 3);
  101. if (opt == "value") {
  102. set_gravity(which, p_value);
  103. } else if (opt == "damping_curve") {
  104. set_gravity_damping_curve(which, p_value);
  105. } else if (opt == "direction") {
  106. set_gravity_direction(which, p_value);
  107. } else {
  108. return false;
  109. }
  110. } else if (what == "enable_all_child_collisions") {
  111. set_enable_all_child_collisions(which, p_value);
  112. } else if (what == "joint_count") {
  113. set_joint_count(which, p_value);
  114. } else if (what == "joints") {
  115. int idx = path.get_slicec('/', 3).to_int();
  116. String prop = path.get_slicec('/', 4);
  117. if (prop == "bone_name") {
  118. set_joint_bone_name(which, idx, p_value);
  119. } else if (prop == "bone") {
  120. set_joint_bone(which, idx, p_value);
  121. } else if (prop == "rotation_axis") {
  122. set_joint_rotation_axis(which, idx, static_cast<RotationAxis>((int)p_value));
  123. } else if (prop == "rotation_axis_vector") {
  124. set_joint_rotation_axis_vector(which, idx, p_value);
  125. } else if (prop == "radius") {
  126. set_joint_radius(which, idx, p_value);
  127. } else if (prop == "stiffness") {
  128. set_joint_stiffness(which, idx, p_value);
  129. } else if (prop == "drag") {
  130. set_joint_drag(which, idx, p_value);
  131. } else if (prop == "gravity") {
  132. set_joint_gravity(which, idx, p_value);
  133. } else if (prop == "gravity_direction") {
  134. set_joint_gravity_direction(which, idx, p_value);
  135. } else {
  136. return false;
  137. }
  138. } else if (what == "exclude_collision_count") {
  139. set_exclude_collision_count(which, p_value);
  140. } else if (what == "exclude_collisions") {
  141. int idx = path.get_slicec('/', 3).to_int();
  142. set_exclude_collision_path(which, idx, p_value);
  143. } else if (what == "collision_count") {
  144. set_collision_count(which, p_value);
  145. } else if (what == "collisions") {
  146. int idx = path.get_slicec('/', 3).to_int();
  147. set_collision_path(which, idx, p_value);
  148. } else {
  149. return false;
  150. }
  151. }
  152. return true;
  153. }
  154. bool SpringBoneSimulator3D::_get(const StringName &p_path, Variant &r_ret) const {
  155. String path = p_path;
  156. if (path.begins_with("settings/")) {
  157. int which = path.get_slicec('/', 1).to_int();
  158. String what = path.get_slicec('/', 2);
  159. ERR_FAIL_INDEX_V(which, settings.size(), false);
  160. if (what == "root_bone_name") {
  161. r_ret = get_root_bone_name(which);
  162. } else if (what == "root_bone") {
  163. r_ret = get_root_bone(which);
  164. } else if (what == "end_bone_name") {
  165. r_ret = get_end_bone_name(which);
  166. } else if (what == "end_bone") {
  167. String opt = path.get_slicec('/', 3);
  168. if (opt.is_empty()) {
  169. r_ret = get_end_bone(which);
  170. } else if (opt == "direction") {
  171. r_ret = (int)get_end_bone_direction(which);
  172. } else if (opt == "length") {
  173. r_ret = get_end_bone_length(which);
  174. } else {
  175. return false;
  176. }
  177. } else if (what == "extend_end_bone") {
  178. r_ret = is_end_bone_extended(which);
  179. } else if (what == "center_from") {
  180. r_ret = (int)get_center_from(which);
  181. } else if (what == "center_node") {
  182. r_ret = get_center_node(which);
  183. } else if (what == "center_bone") {
  184. r_ret = get_center_bone(which);
  185. } else if (what == "center_bone_name") {
  186. r_ret = get_center_bone_name(which);
  187. } else if (what == "individual_config") {
  188. r_ret = is_config_individual(which);
  189. } else if (what == "rotation_axis") {
  190. r_ret = (int)get_rotation_axis(which);
  191. } else if (what == "rotation_axis_vector") {
  192. r_ret = get_rotation_axis_vector(which);
  193. } else if (what == "radius") {
  194. String opt = path.get_slicec('/', 3);
  195. if (opt == "value") {
  196. r_ret = get_radius(which);
  197. } else if (opt == "damping_curve") {
  198. r_ret = get_radius_damping_curve(which);
  199. } else {
  200. return false;
  201. }
  202. } else if (what == "stiffness") {
  203. String opt = path.get_slicec('/', 3);
  204. if (opt == "value") {
  205. r_ret = get_stiffness(which);
  206. } else if (opt == "damping_curve") {
  207. r_ret = get_stiffness_damping_curve(which);
  208. } else {
  209. return false;
  210. }
  211. } else if (what == "drag") {
  212. String opt = path.get_slicec('/', 3);
  213. if (opt == "value") {
  214. r_ret = get_drag(which);
  215. } else if (opt == "damping_curve") {
  216. r_ret = get_drag_damping_curve(which);
  217. } else {
  218. return false;
  219. }
  220. } else if (what == "gravity") {
  221. String opt = path.get_slicec('/', 3);
  222. if (opt == "value") {
  223. r_ret = get_gravity(which);
  224. } else if (opt == "damping_curve") {
  225. r_ret = get_gravity_damping_curve(which);
  226. } else if (opt == "direction") {
  227. r_ret = get_gravity_direction(which);
  228. } else {
  229. return false;
  230. }
  231. } else if (what == "enable_all_child_collisions") {
  232. r_ret = are_all_child_collisions_enabled(which);
  233. } else if (what == "joint_count") {
  234. r_ret = get_joint_count(which);
  235. } else if (what == "joints") {
  236. int idx = path.get_slicec('/', 3).to_int();
  237. String prop = path.get_slicec('/', 4);
  238. if (prop == "bone_name") {
  239. r_ret = get_joint_bone_name(which, idx);
  240. } else if (prop == "bone") {
  241. r_ret = get_joint_bone(which, idx);
  242. } else if (prop == "rotation_axis") {
  243. r_ret = (int)get_joint_rotation_axis(which, idx);
  244. } else if (prop == "rotation_axis_vector") {
  245. r_ret = get_joint_rotation_axis_vector(which, idx);
  246. } else if (prop == "radius") {
  247. r_ret = get_joint_radius(which, idx);
  248. } else if (prop == "stiffness") {
  249. r_ret = get_joint_stiffness(which, idx);
  250. } else if (prop == "drag") {
  251. r_ret = get_joint_drag(which, idx);
  252. } else if (prop == "gravity") {
  253. r_ret = get_joint_gravity(which, idx);
  254. } else if (prop == "gravity_direction") {
  255. r_ret = get_joint_gravity_direction(which, idx);
  256. } else {
  257. return false;
  258. }
  259. } else if (what == "exclude_collision_count") {
  260. r_ret = get_exclude_collision_count(which);
  261. } else if (what == "exclude_collisions") {
  262. int idx = path.get_slicec('/', 3).to_int();
  263. r_ret = get_exclude_collision_path(which, idx);
  264. } else if (what == "collision_count") {
  265. r_ret = get_collision_count(which);
  266. } else if (what == "collisions") {
  267. int idx = path.get_slicec('/', 3).to_int();
  268. r_ret = get_collision_path(which, idx);
  269. } else {
  270. return false;
  271. }
  272. }
  273. return true;
  274. }
  275. void SpringBoneSimulator3D::_get_property_list(List<PropertyInfo> *p_list) const {
  276. String enum_hint;
  277. Skeleton3D *skeleton = get_skeleton();
  278. if (skeleton) {
  279. enum_hint = skeleton->get_concatenated_bone_names();
  280. }
  281. LocalVector<PropertyInfo> props;
  282. for (int i = 0; i < settings.size(); i++) {
  283. String path = "settings/" + itos(i) + "/";
  284. props.push_back(PropertyInfo(Variant::STRING, path + "root_bone_name", PROPERTY_HINT_ENUM_SUGGESTION, enum_hint));
  285. props.push_back(PropertyInfo(Variant::INT, path + "root_bone", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR));
  286. props.push_back(PropertyInfo(Variant::STRING, path + "end_bone_name", PROPERTY_HINT_ENUM_SUGGESTION, enum_hint));
  287. props.push_back(PropertyInfo(Variant::INT, path + "end_bone", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR));
  288. props.push_back(PropertyInfo(Variant::BOOL, path + "extend_end_bone"));
  289. props.push_back(PropertyInfo(Variant::INT, path + "end_bone/direction", PROPERTY_HINT_ENUM, "+X,-X,+Y,-Y,+Z,-Z,FromParent"));
  290. props.push_back(PropertyInfo(Variant::FLOAT, path + "end_bone/length", PROPERTY_HINT_RANGE, "0,1,0.001,or_greater,suffix:m"));
  291. props.push_back(PropertyInfo(Variant::INT, path + "center_from", PROPERTY_HINT_ENUM, "WorldOrigin,Node,Bone"));
  292. props.push_back(PropertyInfo(Variant::NODE_PATH, path + "center_node"));
  293. props.push_back(PropertyInfo(Variant::STRING, path + "center_bone_name", PROPERTY_HINT_ENUM_SUGGESTION, enum_hint));
  294. props.push_back(PropertyInfo(Variant::INT, path + "center_bone", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR));
  295. props.push_back(PropertyInfo(Variant::BOOL, path + "individual_config"));
  296. props.push_back(PropertyInfo(Variant::INT, path + "rotation_axis", PROPERTY_HINT_ENUM, "X,Y,Z,All,Custom"));
  297. props.push_back(PropertyInfo(Variant::VECTOR3, path + "rotation_axis_vector"));
  298. props.push_back(PropertyInfo(Variant::FLOAT, path + "radius/value", PROPERTY_HINT_RANGE, "0,1,0.001,or_greater,suffix:m"));
  299. props.push_back(PropertyInfo(Variant::OBJECT, path + "radius/damping_curve", PROPERTY_HINT_RESOURCE_TYPE, "Curve"));
  300. props.push_back(PropertyInfo(Variant::FLOAT, path + "stiffness/value", PROPERTY_HINT_RANGE, "0,4,0.01,or_greater"));
  301. props.push_back(PropertyInfo(Variant::OBJECT, path + "stiffness/damping_curve", PROPERTY_HINT_RESOURCE_TYPE, "Curve"));
  302. props.push_back(PropertyInfo(Variant::FLOAT, path + "drag/value", PROPERTY_HINT_RANGE, "0,1,0.01,or_greater"));
  303. props.push_back(PropertyInfo(Variant::OBJECT, path + "drag/damping_curve", PROPERTY_HINT_RESOURCE_TYPE, "Curve"));
  304. props.push_back(PropertyInfo(Variant::FLOAT, path + "gravity/value", PROPERTY_HINT_RANGE, "0,1,0.01,or_greater,or_less,suffix:m/s"));
  305. props.push_back(PropertyInfo(Variant::OBJECT, path + "gravity/damping_curve", PROPERTY_HINT_RESOURCE_TYPE, "Curve"));
  306. props.push_back(PropertyInfo(Variant::VECTOR3, path + "gravity/direction"));
  307. props.push_back(PropertyInfo(Variant::INT, path + "joint_count", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_ARRAY, "Joints," + path + "joints/,static,const"));
  308. for (int j = 0; j < settings[i]->joints.size(); j++) {
  309. String joint_path = path + "joints/" + itos(j) + "/";
  310. props.push_back(PropertyInfo(Variant::STRING, joint_path + "bone_name", PROPERTY_HINT_ENUM_SUGGESTION, enum_hint, PROPERTY_USAGE_EDITOR | PROPERTY_USAGE_READ_ONLY | PROPERTY_USAGE_STORAGE));
  311. props.push_back(PropertyInfo(Variant::INT, joint_path + "bone", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_NO_EDITOR | PROPERTY_USAGE_READ_ONLY));
  312. props.push_back(PropertyInfo(Variant::INT, joint_path + "rotation_axis", PROPERTY_HINT_ENUM, "X,Y,Z,All,Custom"));
  313. props.push_back(PropertyInfo(Variant::VECTOR3, joint_path + "rotation_axis_vector"));
  314. props.push_back(PropertyInfo(Variant::FLOAT, joint_path + "radius", PROPERTY_HINT_RANGE, "0,1,0.001,or_greater,suffix:m"));
  315. props.push_back(PropertyInfo(Variant::FLOAT, joint_path + "stiffness", PROPERTY_HINT_RANGE, "0,4,0.01,or_greater"));
  316. props.push_back(PropertyInfo(Variant::FLOAT, joint_path + "drag", PROPERTY_HINT_RANGE, "0,1,0.01,or_greater"));
  317. props.push_back(PropertyInfo(Variant::FLOAT, joint_path + "gravity", PROPERTY_HINT_RANGE, "0,1,0.01,or_greater,or_less,suffix:m/s"));
  318. props.push_back(PropertyInfo(Variant::VECTOR3, joint_path + "gravity_direction"));
  319. }
  320. props.push_back(PropertyInfo(Variant::BOOL, path + "enable_all_child_collisions"));
  321. props.push_back(PropertyInfo(Variant::INT, path + "exclude_collision_count", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_ARRAY, "Exclude Collisions," + path + "exclude_collisions/"));
  322. for (int j = 0; j < settings[i]->exclude_collisions.size(); j++) {
  323. String collision_path = path + "exclude_collisions/" + itos(j);
  324. props.push_back(PropertyInfo(Variant::NODE_PATH, collision_path, PROPERTY_HINT_NODE_PATH_VALID_TYPES, "SpringBoneCollision3D"));
  325. }
  326. props.push_back(PropertyInfo(Variant::INT, path + "collision_count", PROPERTY_HINT_NONE, "", PROPERTY_USAGE_DEFAULT | PROPERTY_USAGE_ARRAY, "Collisions," + path + "collisions/"));
  327. for (int j = 0; j < settings[i]->collisions.size(); j++) {
  328. String collision_path = path + "collisions/" + itos(j);
  329. props.push_back(PropertyInfo(Variant::NODE_PATH, collision_path, PROPERTY_HINT_NODE_PATH_VALID_TYPES, "SpringBoneCollision3D"));
  330. }
  331. }
  332. for (PropertyInfo &p : props) {
  333. _validate_dynamic_prop(p);
  334. p_list->push_back(p);
  335. }
  336. }
  337. void SpringBoneSimulator3D::_validate_dynamic_prop(PropertyInfo &p_property) const {
  338. PackedStringArray split = p_property.name.split("/");
  339. if (split.size() > 2 && split[0] == "settings") {
  340. int which = split[1].to_int();
  341. // Extended end bone option.
  342. if (split[2] == "end_bone" && !is_end_bone_extended(which) && split.size() > 3) {
  343. p_property.usage = PROPERTY_USAGE_NONE;
  344. }
  345. // Center option.
  346. if (get_center_from(which) != CENTER_FROM_BONE && (split[2] == "center_bone" || split[2] == "center_bone_name")) {
  347. p_property.usage = PROPERTY_USAGE_NONE;
  348. }
  349. if (get_center_from(which) != CENTER_FROM_NODE && split[2] == "center_node") {
  350. p_property.usage = PROPERTY_USAGE_NONE;
  351. }
  352. // Joints option.
  353. if (is_config_individual(which)) {
  354. if (split[2] == "rotation_axis" || split[2] == "rotation_axis_vector" || split[2] == "radius" || split[2] == "radius_damping_curve" ||
  355. split[2] == "stiffness" || split[2] == "stiffness_damping_curve" ||
  356. split[2] == "drag" || split[2] == "drag_damping_curve" ||
  357. split[2] == "gravity" || split[2] == "gravity_damping_curve" || split[2] == "gravity_direction") {
  358. p_property.usage = PROPERTY_USAGE_NONE;
  359. }
  360. } else {
  361. if (split[2] == "joints" || split[2] == "joint_count") {
  362. // Don't storage them since they are overridden by _update_joints().
  363. p_property.usage ^= PROPERTY_USAGE_STORAGE;
  364. p_property.usage |= PROPERTY_USAGE_READ_ONLY;
  365. }
  366. if (split[2] == "rotation_axis_vector" && get_rotation_axis(which) != ROTATION_AXIS_CUSTOM) {
  367. p_property.usage = PROPERTY_USAGE_NONE;
  368. }
  369. }
  370. // Collisions option.
  371. if (are_all_child_collisions_enabled(which)) {
  372. if (split[2] == "collisions" || split[2] == "collision_count") {
  373. p_property.usage = PROPERTY_USAGE_NONE;
  374. }
  375. } else {
  376. if (split[2] == "exclude_collisions" || split[2] == "exclude_collision_count") {
  377. p_property.usage = PROPERTY_USAGE_NONE;
  378. }
  379. }
  380. }
  381. if (split.size() > 3 && split[0] == "settings") {
  382. int which = split[1].to_int();
  383. int joint = split[3].to_int();
  384. // Joints option.
  385. if (split[2] == "joints" && split.size() > 4) {
  386. if (split[4] == "rotation_axis_vector" && get_joint_rotation_axis(which, joint) != ROTATION_AXIS_CUSTOM) {
  387. p_property.usage = PROPERTY_USAGE_NONE;
  388. }
  389. }
  390. }
  391. }
  392. void SpringBoneSimulator3D::_notification(int p_what) {
  393. switch (p_what) {
  394. case NOTIFICATION_ENTER_TREE: {
  395. #ifdef TOOLS_ENABLED
  396. if (Engine::get_singleton()->is_editor_hint()) {
  397. set_notify_local_transform(true); // Used for updating gizmo in editor.
  398. }
  399. #endif // TOOLS_ENABLED
  400. _make_collisions_dirty();
  401. _make_all_joints_dirty();
  402. } break;
  403. #ifdef TOOLS_ENABLED
  404. case NOTIFICATION_LOCAL_TRANSFORM_CHANGED: {
  405. update_gizmos();
  406. } break;
  407. case NOTIFICATION_EDITOR_PRE_SAVE: {
  408. saving = true;
  409. } break;
  410. case NOTIFICATION_EDITOR_POST_SAVE: {
  411. saving = false;
  412. } break;
  413. #endif // TOOLS_ENABLED
  414. }
  415. }
  416. // Setting.
  417. void SpringBoneSimulator3D::set_root_bone_name(int p_index, const String &p_bone_name) {
  418. ERR_FAIL_INDEX(p_index, settings.size());
  419. settings[p_index]->root_bone_name = p_bone_name;
  420. Skeleton3D *sk = get_skeleton();
  421. if (sk) {
  422. set_root_bone(p_index, sk->find_bone(settings[p_index]->root_bone_name));
  423. }
  424. }
  425. String SpringBoneSimulator3D::get_root_bone_name(int p_index) const {
  426. ERR_FAIL_INDEX_V(p_index, settings.size(), String());
  427. return settings[p_index]->root_bone_name;
  428. }
  429. void SpringBoneSimulator3D::set_root_bone(int p_index, int p_bone) {
  430. ERR_FAIL_INDEX(p_index, settings.size());
  431. bool changed = settings[p_index]->root_bone != p_bone;
  432. settings[p_index]->root_bone = p_bone;
  433. Skeleton3D *sk = get_skeleton();
  434. if (sk) {
  435. if (settings[p_index]->root_bone <= -1 || settings[p_index]->root_bone >= sk->get_bone_count()) {
  436. WARN_PRINT("Root bone index out of range!");
  437. settings[p_index]->root_bone = -1;
  438. } else {
  439. settings[p_index]->root_bone_name = sk->get_bone_name(settings[p_index]->root_bone);
  440. }
  441. }
  442. if (changed) {
  443. _update_joint_array(p_index);
  444. }
  445. }
  446. int SpringBoneSimulator3D::get_root_bone(int p_index) const {
  447. ERR_FAIL_INDEX_V(p_index, settings.size(), -1);
  448. return settings[p_index]->root_bone;
  449. }
  450. void SpringBoneSimulator3D::set_end_bone_name(int p_index, const String &p_bone_name) {
  451. ERR_FAIL_INDEX(p_index, settings.size());
  452. settings[p_index]->end_bone_name = p_bone_name;
  453. Skeleton3D *sk = get_skeleton();
  454. if (sk) {
  455. set_end_bone(p_index, sk->find_bone(settings[p_index]->end_bone_name));
  456. }
  457. }
  458. String SpringBoneSimulator3D::get_end_bone_name(int p_index) const {
  459. ERR_FAIL_INDEX_V(p_index, settings.size(), String());
  460. return settings[p_index]->end_bone_name;
  461. }
  462. void SpringBoneSimulator3D::set_end_bone(int p_index, int p_bone) {
  463. ERR_FAIL_INDEX(p_index, settings.size());
  464. bool changed = settings[p_index]->end_bone != p_bone;
  465. settings[p_index]->end_bone = p_bone;
  466. Skeleton3D *sk = get_skeleton();
  467. if (sk) {
  468. if (settings[p_index]->end_bone <= -1 || settings[p_index]->end_bone >= sk->get_bone_count()) {
  469. WARN_PRINT("End bone index out of range!");
  470. settings[p_index]->end_bone = -1;
  471. } else {
  472. settings[p_index]->end_bone_name = sk->get_bone_name(settings[p_index]->end_bone);
  473. }
  474. }
  475. if (changed) {
  476. _update_joint_array(p_index);
  477. }
  478. }
  479. int SpringBoneSimulator3D::get_end_bone(int p_index) const {
  480. ERR_FAIL_INDEX_V(p_index, settings.size(), -1);
  481. return settings[p_index]->end_bone;
  482. }
  483. void SpringBoneSimulator3D::set_extend_end_bone(int p_index, bool p_enabled) {
  484. ERR_FAIL_INDEX(p_index, settings.size());
  485. settings[p_index]->extend_end_bone = p_enabled;
  486. _make_joints_dirty(p_index);
  487. notify_property_list_changed();
  488. }
  489. bool SpringBoneSimulator3D::is_end_bone_extended(int p_index) const {
  490. ERR_FAIL_INDEX_V(p_index, settings.size(), false);
  491. return settings[p_index]->extend_end_bone;
  492. }
  493. void SpringBoneSimulator3D::set_end_bone_direction(int p_index, BoneDirection p_bone_direction) {
  494. ERR_FAIL_INDEX(p_index, settings.size());
  495. settings[p_index]->end_bone_direction = p_bone_direction;
  496. _make_joints_dirty(p_index);
  497. }
  498. SpringBoneSimulator3D::BoneDirection SpringBoneSimulator3D::get_end_bone_direction(int p_index) const {
  499. ERR_FAIL_INDEX_V(p_index, settings.size(), BONE_DIRECTION_FROM_PARENT);
  500. return settings[p_index]->end_bone_direction;
  501. }
  502. void SpringBoneSimulator3D::set_end_bone_length(int p_index, float p_length) {
  503. ERR_FAIL_INDEX(p_index, settings.size());
  504. settings[p_index]->end_bone_length = p_length;
  505. _make_joints_dirty(p_index);
  506. }
  507. float SpringBoneSimulator3D::get_end_bone_length(int p_index) const {
  508. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  509. return settings[p_index]->end_bone_length;
  510. }
  511. Vector3 SpringBoneSimulator3D::get_end_bone_axis(int p_end_bone, BoneDirection p_direction) const {
  512. Vector3 axis;
  513. if (p_direction == BONE_DIRECTION_FROM_PARENT) {
  514. Skeleton3D *sk = get_skeleton();
  515. if (sk) {
  516. axis = sk->get_bone_rest(p_end_bone).basis.xform_inv(sk->get_bone_rest(p_end_bone).origin);
  517. axis.normalize();
  518. }
  519. } else {
  520. axis = get_vector_from_bone_axis(static_cast<BoneAxis>((int)p_direction));
  521. }
  522. return axis;
  523. }
  524. void SpringBoneSimulator3D::set_center_from(int p_index, CenterFrom p_center_from) {
  525. ERR_FAIL_INDEX(p_index, settings.size());
  526. bool center_changed = settings[p_index]->center_from != p_center_from;
  527. settings[p_index]->center_from = p_center_from;
  528. if (center_changed) {
  529. reset();
  530. }
  531. notify_property_list_changed();
  532. }
  533. SpringBoneSimulator3D::CenterFrom SpringBoneSimulator3D::get_center_from(int p_index) const {
  534. ERR_FAIL_INDEX_V(p_index, settings.size(), CENTER_FROM_WORLD_ORIGIN);
  535. return settings[p_index]->center_from;
  536. }
  537. void SpringBoneSimulator3D::set_center_node(int p_index, const NodePath &p_node_path) {
  538. ERR_FAIL_INDEX(p_index, settings.size());
  539. bool center_changed = settings[p_index]->center_node != p_node_path;
  540. settings[p_index]->center_node = p_node_path;
  541. if (center_changed) {
  542. reset();
  543. }
  544. }
  545. NodePath SpringBoneSimulator3D::get_center_node(int p_index) const {
  546. ERR_FAIL_INDEX_V(p_index, settings.size(), NodePath());
  547. return settings[p_index]->center_node;
  548. }
  549. void SpringBoneSimulator3D::set_center_bone_name(int p_index, const String &p_bone_name) {
  550. ERR_FAIL_INDEX(p_index, settings.size());
  551. settings[p_index]->center_bone_name = p_bone_name;
  552. Skeleton3D *sk = get_skeleton();
  553. if (sk) {
  554. set_center_bone(p_index, sk->find_bone(settings[p_index]->center_bone_name));
  555. }
  556. }
  557. String SpringBoneSimulator3D::get_center_bone_name(int p_index) const {
  558. ERR_FAIL_INDEX_V(p_index, settings.size(), String());
  559. return settings[p_index]->center_bone_name;
  560. }
  561. void SpringBoneSimulator3D::set_center_bone(int p_index, int p_bone) {
  562. ERR_FAIL_INDEX(p_index, settings.size());
  563. bool center_changed = settings[p_index]->center_bone != p_bone;
  564. settings[p_index]->center_bone = p_bone;
  565. Skeleton3D *sk = get_skeleton();
  566. if (sk) {
  567. if (settings[p_index]->center_bone <= -1 || settings[p_index]->center_bone >= sk->get_bone_count()) {
  568. WARN_PRINT("Center bone index out of range!");
  569. settings[p_index]->center_bone = -1;
  570. } else {
  571. settings[p_index]->center_bone_name = sk->get_bone_name(settings[p_index]->center_bone);
  572. }
  573. }
  574. if (center_changed) {
  575. reset();
  576. }
  577. }
  578. int SpringBoneSimulator3D::get_center_bone(int p_index) const {
  579. ERR_FAIL_INDEX_V(p_index, settings.size(), -1);
  580. return settings[p_index]->center_bone;
  581. }
  582. void SpringBoneSimulator3D::set_radius(int p_index, float p_radius) {
  583. ERR_FAIL_INDEX(p_index, settings.size());
  584. if (is_config_individual(p_index)) {
  585. return; // Joint config is individual mode.
  586. }
  587. settings[p_index]->radius = p_radius;
  588. _make_joints_dirty(p_index);
  589. }
  590. float SpringBoneSimulator3D::get_radius(int p_index) const {
  591. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  592. return settings[p_index]->radius;
  593. }
  594. void SpringBoneSimulator3D::set_radius_damping_curve(int p_index, const Ref<Curve> &p_damping_curve) {
  595. ERR_FAIL_INDEX(p_index, settings.size());
  596. if (is_config_individual(p_index)) {
  597. return; // Joint config is individual mode.
  598. }
  599. if (settings[p_index]->radius_damping_curve.is_valid()) {
  600. settings[p_index]->radius_damping_curve->disconnect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty));
  601. }
  602. settings[p_index]->radius_damping_curve = p_damping_curve;
  603. if (settings[p_index]->radius_damping_curve.is_valid()) {
  604. settings[p_index]->radius_damping_curve->connect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty).bind(p_index));
  605. }
  606. _make_joints_dirty(p_index);
  607. }
  608. Ref<Curve> SpringBoneSimulator3D::get_radius_damping_curve(int p_index) const {
  609. ERR_FAIL_INDEX_V(p_index, settings.size(), Ref<Curve>());
  610. return settings[p_index]->radius_damping_curve;
  611. }
  612. void SpringBoneSimulator3D::set_stiffness(int p_index, float p_stiffness) {
  613. ERR_FAIL_INDEX(p_index, settings.size());
  614. if (is_config_individual(p_index)) {
  615. return; // Joint config is individual mode.
  616. }
  617. settings[p_index]->stiffness = p_stiffness;
  618. _make_joints_dirty(p_index);
  619. }
  620. float SpringBoneSimulator3D::get_stiffness(int p_index) const {
  621. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  622. return settings[p_index]->stiffness;
  623. }
  624. void SpringBoneSimulator3D::set_stiffness_damping_curve(int p_index, const Ref<Curve> &p_damping_curve) {
  625. ERR_FAIL_INDEX(p_index, settings.size());
  626. if (is_config_individual(p_index)) {
  627. return; // Joint config is individual mode.
  628. }
  629. if (settings[p_index]->stiffness_damping_curve.is_valid()) {
  630. settings[p_index]->stiffness_damping_curve->disconnect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty));
  631. }
  632. settings[p_index]->stiffness_damping_curve = p_damping_curve;
  633. if (settings[p_index]->stiffness_damping_curve.is_valid()) {
  634. settings[p_index]->stiffness_damping_curve->connect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty).bind(p_index));
  635. }
  636. _make_joints_dirty(p_index);
  637. }
  638. Ref<Curve> SpringBoneSimulator3D::get_stiffness_damping_curve(int p_index) const {
  639. ERR_FAIL_INDEX_V(p_index, settings.size(), Ref<Curve>());
  640. return settings[p_index]->stiffness_damping_curve;
  641. }
  642. void SpringBoneSimulator3D::set_drag(int p_index, float p_drag) {
  643. ERR_FAIL_INDEX(p_index, settings.size());
  644. if (is_config_individual(p_index)) {
  645. return; // Joint config is individual mode.
  646. }
  647. settings[p_index]->drag = p_drag;
  648. _make_joints_dirty(p_index);
  649. }
  650. float SpringBoneSimulator3D::get_drag(int p_index) const {
  651. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  652. return settings[p_index]->drag;
  653. }
  654. void SpringBoneSimulator3D::set_drag_damping_curve(int p_index, const Ref<Curve> &p_damping_curve) {
  655. ERR_FAIL_INDEX(p_index, settings.size());
  656. if (is_config_individual(p_index)) {
  657. return; // Joint config is individual mode.
  658. }
  659. if (settings[p_index]->drag_damping_curve.is_valid()) {
  660. settings[p_index]->drag_damping_curve->disconnect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty));
  661. }
  662. settings[p_index]->drag_damping_curve = p_damping_curve;
  663. if (settings[p_index]->drag_damping_curve.is_valid()) {
  664. settings[p_index]->drag_damping_curve->connect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty).bind(p_index));
  665. }
  666. _make_joints_dirty(p_index);
  667. }
  668. Ref<Curve> SpringBoneSimulator3D::get_drag_damping_curve(int p_index) const {
  669. ERR_FAIL_INDEX_V(p_index, settings.size(), Ref<Curve>());
  670. return settings[p_index]->drag_damping_curve;
  671. }
  672. void SpringBoneSimulator3D::set_gravity(int p_index, float p_gravity) {
  673. ERR_FAIL_INDEX(p_index, settings.size());
  674. if (is_config_individual(p_index)) {
  675. return; // Joint config is individual mode.
  676. }
  677. settings[p_index]->gravity = p_gravity;
  678. _make_joints_dirty(p_index);
  679. }
  680. float SpringBoneSimulator3D::get_gravity(int p_index) const {
  681. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  682. return settings[p_index]->gravity;
  683. }
  684. void SpringBoneSimulator3D::set_gravity_damping_curve(int p_index, const Ref<Curve> &p_damping_curve) {
  685. ERR_FAIL_INDEX(p_index, settings.size());
  686. if (is_config_individual(p_index)) {
  687. return; // Joint config is individual mode.
  688. }
  689. if (settings[p_index]->gravity_damping_curve.is_valid()) {
  690. settings[p_index]->gravity_damping_curve->disconnect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty));
  691. }
  692. settings[p_index]->gravity_damping_curve = p_damping_curve;
  693. if (settings[p_index]->gravity_damping_curve.is_valid()) {
  694. settings[p_index]->gravity_damping_curve->connect_changed(callable_mp(this, &SpringBoneSimulator3D::_make_joints_dirty).bind(p_index));
  695. }
  696. _make_joints_dirty(p_index);
  697. }
  698. Ref<Curve> SpringBoneSimulator3D::get_gravity_damping_curve(int p_index) const {
  699. ERR_FAIL_INDEX_V(p_index, settings.size(), Ref<Curve>());
  700. return settings[p_index]->gravity_damping_curve;
  701. }
  702. void SpringBoneSimulator3D::set_gravity_direction(int p_index, const Vector3 &p_gravity_direction) {
  703. ERR_FAIL_INDEX(p_index, settings.size());
  704. ERR_FAIL_COND(p_gravity_direction.is_zero_approx());
  705. if (is_config_individual(p_index)) {
  706. return; // Joint config is individual mode.
  707. }
  708. settings[p_index]->gravity_direction = p_gravity_direction;
  709. _make_joints_dirty(p_index);
  710. }
  711. Vector3 SpringBoneSimulator3D::get_gravity_direction(int p_index) const {
  712. ERR_FAIL_INDEX_V(p_index, settings.size(), Vector3(0, -1, 0));
  713. return settings[p_index]->gravity_direction;
  714. }
  715. void SpringBoneSimulator3D::set_rotation_axis(int p_index, RotationAxis p_axis) {
  716. ERR_FAIL_INDEX(p_index, settings.size());
  717. if (is_config_individual(p_index)) {
  718. return; // Joint config is individual mode.
  719. }
  720. settings[p_index]->rotation_axis = p_axis;
  721. _make_joints_dirty(p_index);
  722. notify_property_list_changed();
  723. }
  724. SpringBoneSimulator3D::RotationAxis SpringBoneSimulator3D::get_rotation_axis(int p_index) const {
  725. ERR_FAIL_INDEX_V(p_index, settings.size(), ROTATION_AXIS_ALL);
  726. return settings[p_index]->rotation_axis;
  727. }
  728. void SpringBoneSimulator3D::set_rotation_axis_vector(int p_index, const Vector3 &p_vector) {
  729. ERR_FAIL_INDEX(p_index, settings.size());
  730. if (is_config_individual(p_index) || settings[p_index]->rotation_axis != ROTATION_AXIS_CUSTOM) {
  731. return; // Joint config is individual mode.
  732. }
  733. settings[p_index]->rotation_axis_vector = p_vector;
  734. _make_joints_dirty(p_index);
  735. }
  736. Vector3 SpringBoneSimulator3D::get_rotation_axis_vector(int p_index) const {
  737. ERR_FAIL_INDEX_V(p_index, settings.size(), Vector3());
  738. Vector3 ret;
  739. switch (settings[p_index]->rotation_axis) {
  740. case ROTATION_AXIS_X:
  741. ret = Vector3(1, 0, 0);
  742. break;
  743. case ROTATION_AXIS_Y:
  744. ret = Vector3(0, 1, 0);
  745. break;
  746. case ROTATION_AXIS_Z:
  747. ret = Vector3(0, 0, 1);
  748. break;
  749. case ROTATION_AXIS_ALL:
  750. ret = Vector3(0, 0, 0);
  751. break;
  752. case ROTATION_AXIS_CUSTOM:
  753. ret = settings[p_index]->rotation_axis_vector;
  754. break;
  755. }
  756. return ret;
  757. }
  758. void SpringBoneSimulator3D::set_setting_count(int p_count) {
  759. ERR_FAIL_COND(p_count < 0);
  760. int delta = p_count - settings.size();
  761. if (delta < 0) {
  762. for (int i = delta; i < 0; i++) {
  763. memdelete(settings[settings.size() + i]);
  764. }
  765. }
  766. settings.resize(p_count);
  767. delta++;
  768. if (delta > 1) {
  769. for (int i = 1; i < delta; i++) {
  770. settings.write[p_count - i] = memnew(SpringBone3DSetting);
  771. }
  772. }
  773. notify_property_list_changed();
  774. }
  775. int SpringBoneSimulator3D::get_setting_count() const {
  776. return settings.size();
  777. }
  778. void SpringBoneSimulator3D::clear_settings() {
  779. set_setting_count(0);
  780. }
  781. // Individual joints.
  782. void SpringBoneSimulator3D::set_individual_config(int p_index, bool p_enabled) {
  783. ERR_FAIL_INDEX(p_index, settings.size());
  784. settings[p_index]->individual_config = p_enabled;
  785. _make_joints_dirty(p_index);
  786. notify_property_list_changed();
  787. }
  788. bool SpringBoneSimulator3D::is_config_individual(int p_index) const {
  789. ERR_FAIL_INDEX_V(p_index, settings.size(), false);
  790. return settings[p_index]->individual_config;
  791. }
  792. void SpringBoneSimulator3D::set_joint_bone_name(int p_index, int p_joint, const String &p_bone_name) {
  793. ERR_FAIL_INDEX(p_index, settings.size());
  794. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  795. ERR_FAIL_INDEX(p_joint, joints.size());
  796. joints[p_joint]->bone_name = p_bone_name;
  797. Skeleton3D *sk = get_skeleton();
  798. if (sk) {
  799. set_joint_bone(p_index, p_joint, sk->find_bone(joints[p_joint]->bone_name));
  800. }
  801. }
  802. String SpringBoneSimulator3D::get_joint_bone_name(int p_index, int p_joint) const {
  803. ERR_FAIL_INDEX_V(p_index, settings.size(), String());
  804. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  805. ERR_FAIL_INDEX_V(p_joint, joints.size(), String());
  806. return joints[p_joint]->bone_name;
  807. }
  808. void SpringBoneSimulator3D::set_joint_bone(int p_index, int p_joint, int p_bone) {
  809. ERR_FAIL_INDEX(p_index, settings.size());
  810. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  811. ERR_FAIL_INDEX(p_joint, joints.size());
  812. joints[p_joint]->bone = p_bone;
  813. Skeleton3D *sk = get_skeleton();
  814. if (sk) {
  815. if (joints[p_joint]->bone <= -1 || joints[p_joint]->bone >= sk->get_bone_count()) {
  816. WARN_PRINT("Joint bone index out of range!");
  817. joints[p_joint]->bone = -1;
  818. } else {
  819. joints[p_joint]->bone_name = sk->get_bone_name(joints[p_joint]->bone);
  820. }
  821. }
  822. }
  823. int SpringBoneSimulator3D::get_joint_bone(int p_index, int p_joint) const {
  824. ERR_FAIL_INDEX_V(p_index, settings.size(), -1);
  825. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  826. ERR_FAIL_INDEX_V(p_joint, joints.size(), -1);
  827. return joints[p_joint]->bone;
  828. }
  829. void SpringBoneSimulator3D::set_joint_radius(int p_index, int p_joint, float p_radius) {
  830. ERR_FAIL_INDEX(p_index, settings.size());
  831. if (!is_config_individual(p_index)) {
  832. return; // Joints are read-only.
  833. }
  834. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  835. ERR_FAIL_INDEX(p_joint, joints.size());
  836. joints[p_joint]->radius = p_radius;
  837. #ifdef TOOLS_ENABLED
  838. update_gizmos();
  839. #endif // TOOLS_ENABLED
  840. }
  841. float SpringBoneSimulator3D::get_joint_radius(int p_index, int p_joint) const {
  842. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  843. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  844. ERR_FAIL_INDEX_V(p_joint, joints.size(), 0);
  845. return joints[p_joint]->radius;
  846. }
  847. void SpringBoneSimulator3D::set_joint_stiffness(int p_index, int p_joint, float p_stiffness) {
  848. ERR_FAIL_INDEX(p_index, settings.size());
  849. if (!is_config_individual(p_index)) {
  850. return; // Joints are read-only.
  851. }
  852. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  853. ERR_FAIL_INDEX(p_joint, joints.size());
  854. joints[p_joint]->stiffness = p_stiffness;
  855. }
  856. float SpringBoneSimulator3D::get_joint_stiffness(int p_index, int p_joint) const {
  857. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  858. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  859. ERR_FAIL_INDEX_V(p_joint, joints.size(), 0);
  860. return joints[p_joint]->stiffness;
  861. }
  862. void SpringBoneSimulator3D::set_joint_drag(int p_index, int p_joint, float p_drag) {
  863. ERR_FAIL_INDEX(p_index, settings.size());
  864. if (!is_config_individual(p_index)) {
  865. return; // Joints are read-only.
  866. }
  867. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  868. ERR_FAIL_INDEX(p_joint, joints.size());
  869. joints[p_joint]->drag = p_drag;
  870. }
  871. float SpringBoneSimulator3D::get_joint_drag(int p_index, int p_joint) const {
  872. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  873. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  874. ERR_FAIL_INDEX_V(p_joint, joints.size(), 0);
  875. return joints[p_joint]->drag;
  876. }
  877. void SpringBoneSimulator3D::set_joint_gravity(int p_index, int p_joint, float p_gravity) {
  878. ERR_FAIL_INDEX(p_index, settings.size());
  879. if (!is_config_individual(p_index)) {
  880. return; // Joints are read-only.
  881. }
  882. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  883. ERR_FAIL_INDEX(p_joint, joints.size());
  884. joints[p_joint]->gravity = p_gravity;
  885. }
  886. float SpringBoneSimulator3D::get_joint_gravity(int p_index, int p_joint) const {
  887. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  888. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  889. ERR_FAIL_INDEX_V(p_joint, joints.size(), 0);
  890. return joints[p_joint]->gravity;
  891. }
  892. void SpringBoneSimulator3D::set_joint_gravity_direction(int p_index, int p_joint, const Vector3 &p_gravity_direction) {
  893. ERR_FAIL_INDEX(p_index, settings.size());
  894. ERR_FAIL_COND(p_gravity_direction.is_zero_approx());
  895. if (!is_config_individual(p_index)) {
  896. return; // Joints are read-only.
  897. }
  898. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  899. ERR_FAIL_INDEX(p_joint, joints.size());
  900. joints[p_joint]->gravity_direction = p_gravity_direction;
  901. }
  902. Vector3 SpringBoneSimulator3D::get_joint_gravity_direction(int p_index, int p_joint) const {
  903. ERR_FAIL_INDEX_V(p_index, settings.size(), Vector3(0, -1, 0));
  904. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  905. ERR_FAIL_INDEX_V(p_joint, joints.size(), Vector3(0, -1, 0));
  906. return joints[p_joint]->gravity_direction;
  907. }
  908. void SpringBoneSimulator3D::set_joint_rotation_axis(int p_index, int p_joint, RotationAxis p_axis) {
  909. ERR_FAIL_INDEX(p_index, settings.size());
  910. if (!is_config_individual(p_index)) {
  911. return; // Joints are read-only.
  912. }
  913. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  914. ERR_FAIL_INDEX(p_joint, joints.size());
  915. joints[p_joint]->rotation_axis = p_axis;
  916. Skeleton3D *sk = get_skeleton();
  917. if (sk) {
  918. _validate_rotation_axis(sk, p_index, p_joint);
  919. }
  920. notify_property_list_changed();
  921. settings[p_index]->simulation_dirty = true;
  922. #ifdef TOOLS_ENABLED
  923. update_gizmos();
  924. #endif // TOOLS_ENABLED
  925. }
  926. SpringBoneSimulator3D::RotationAxis SpringBoneSimulator3D::get_joint_rotation_axis(int p_index, int p_joint) const {
  927. ERR_FAIL_INDEX_V(p_index, settings.size(), ROTATION_AXIS_ALL);
  928. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  929. ERR_FAIL_INDEX_V(p_joint, joints.size(), ROTATION_AXIS_ALL);
  930. return joints[p_joint]->rotation_axis;
  931. }
  932. void SpringBoneSimulator3D::set_joint_rotation_axis_vector(int p_index, int p_joint, const Vector3 &p_vector) {
  933. ERR_FAIL_INDEX(p_index, settings.size());
  934. if (!is_config_individual(p_index) || settings[p_index]->rotation_axis != ROTATION_AXIS_CUSTOM) {
  935. return; // Joints are read-only.
  936. }
  937. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  938. ERR_FAIL_INDEX(p_joint, joints.size());
  939. joints[p_joint]->rotation_axis_vector = p_vector;
  940. Skeleton3D *sk = get_skeleton();
  941. if (sk) {
  942. _validate_rotation_axis(sk, p_index, p_joint);
  943. }
  944. settings[p_index]->simulation_dirty = true;
  945. #ifdef TOOLS_ENABLED
  946. update_gizmos();
  947. #endif // TOOLS_ENABLED
  948. }
  949. Vector3 SpringBoneSimulator3D::get_joint_rotation_axis_vector(int p_index, int p_joint) const {
  950. ERR_FAIL_INDEX_V(p_index, settings.size(), Vector3());
  951. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  952. ERR_FAIL_INDEX_V(p_joint, joints.size(), Vector3());
  953. return joints[p_joint]->get_rotation_axis_vector();
  954. }
  955. void SpringBoneSimulator3D::set_joint_count(int p_index, int p_count) {
  956. ERR_FAIL_INDEX(p_index, settings.size());
  957. ERR_FAIL_COND(p_count < 0);
  958. Vector<SpringBone3DJointSetting *> &joints = settings[p_index]->joints;
  959. int delta = p_count - joints.size();
  960. if (delta < 0) {
  961. for (int i = delta; i < 0; i++) {
  962. memdelete(joints[joints.size() + i]);
  963. }
  964. }
  965. joints.resize(p_count);
  966. delta++;
  967. if (delta > 1) {
  968. for (int i = 1; i < delta; i++) {
  969. joints.write[p_count - i] = memnew(SpringBone3DJointSetting);
  970. }
  971. }
  972. notify_property_list_changed();
  973. }
  974. int SpringBoneSimulator3D::get_joint_count(int p_index) const {
  975. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  976. Vector<SpringBone3DJointSetting *> joints = settings[p_index]->joints;
  977. return joints.size();
  978. }
  979. // Individual collisions.
  980. void SpringBoneSimulator3D::set_enable_all_child_collisions(int p_index, bool p_enabled) {
  981. ERR_FAIL_INDEX(p_index, settings.size());
  982. settings[p_index]->enable_all_child_collisions = p_enabled;
  983. notify_property_list_changed();
  984. }
  985. bool SpringBoneSimulator3D::are_all_child_collisions_enabled(int p_index) const {
  986. ERR_FAIL_INDEX_V(p_index, settings.size(), false);
  987. return settings[p_index]->enable_all_child_collisions;
  988. }
  989. void SpringBoneSimulator3D::set_exclude_collision_path(int p_index, int p_collision, const NodePath &p_node_path) {
  990. ERR_FAIL_INDEX(p_index, settings.size());
  991. if (!are_all_child_collisions_enabled(p_index)) {
  992. return; // Exclude collision list is disabled.
  993. }
  994. Vector<NodePath> &setting_exclude_collisions = settings[p_index]->exclude_collisions;
  995. ERR_FAIL_INDEX(p_collision, setting_exclude_collisions.size());
  996. setting_exclude_collisions.write[p_collision] = NodePath(); // Reset first.
  997. if (is_inside_tree()) {
  998. Node *node = get_node_or_null(p_node_path);
  999. if (!node) {
  1000. _make_collisions_dirty();
  1001. return;
  1002. }
  1003. node = node->get_parent();
  1004. if (!node || node != this) {
  1005. _make_collisions_dirty();
  1006. ERR_FAIL_EDMSG("Collision must be child of current SpringBoneSimulator3D.");
  1007. }
  1008. }
  1009. setting_exclude_collisions.write[p_collision] = p_node_path;
  1010. _make_collisions_dirty();
  1011. }
  1012. NodePath SpringBoneSimulator3D::get_exclude_collision_path(int p_index, int p_collision) const {
  1013. ERR_FAIL_INDEX_V(p_index, settings.size(), NodePath());
  1014. Vector<NodePath> setting_exclude_collisions = settings[p_index]->exclude_collisions;
  1015. ERR_FAIL_INDEX_V(p_collision, setting_exclude_collisions.size(), NodePath());
  1016. return setting_exclude_collisions[p_collision];
  1017. }
  1018. void SpringBoneSimulator3D::set_exclude_collision_count(int p_index, int p_count) {
  1019. ERR_FAIL_INDEX(p_index, settings.size());
  1020. if (!are_all_child_collisions_enabled(p_index)) {
  1021. return; // Exclude collision list is disabled.
  1022. }
  1023. Vector<NodePath> &setting_exclude_collisions = settings[p_index]->exclude_collisions;
  1024. setting_exclude_collisions.resize(p_count);
  1025. _make_collisions_dirty();
  1026. notify_property_list_changed();
  1027. }
  1028. int SpringBoneSimulator3D::get_exclude_collision_count(int p_index) const {
  1029. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  1030. Vector<NodePath> setting_exclude_collisions = settings[p_index]->exclude_collisions;
  1031. return setting_exclude_collisions.size();
  1032. }
  1033. void SpringBoneSimulator3D::clear_exclude_collisions(int p_index) {
  1034. ERR_FAIL_INDEX(p_index, settings.size());
  1035. if (!are_all_child_collisions_enabled(p_index)) {
  1036. return; // Exclude collision list is disabled.
  1037. }
  1038. set_exclude_collision_count(p_index, 0);
  1039. }
  1040. void SpringBoneSimulator3D::set_collision_path(int p_index, int p_collision, const NodePath &p_node_path) {
  1041. ERR_FAIL_INDEX(p_index, settings.size());
  1042. if (are_all_child_collisions_enabled(p_index)) {
  1043. return; // Collision list is disabled.
  1044. }
  1045. Vector<NodePath> &setting_collisions = settings[p_index]->collisions;
  1046. ERR_FAIL_INDEX(p_collision, setting_collisions.size());
  1047. setting_collisions.write[p_collision] = NodePath(); // Reset first.
  1048. if (is_inside_tree()) {
  1049. Node *node = get_node_or_null(p_node_path);
  1050. if (!node) {
  1051. _make_collisions_dirty();
  1052. return;
  1053. }
  1054. node = node->get_parent();
  1055. if (!node || node != this) {
  1056. _make_collisions_dirty();
  1057. ERR_FAIL_EDMSG("Collision must be child of current SpringBoneSimulator3D.");
  1058. }
  1059. }
  1060. setting_collisions.write[p_collision] = p_node_path;
  1061. _make_collisions_dirty();
  1062. }
  1063. NodePath SpringBoneSimulator3D::get_collision_path(int p_index, int p_collision) const {
  1064. ERR_FAIL_INDEX_V(p_index, settings.size(), NodePath());
  1065. Vector<NodePath> setting_collisions = settings[p_index]->collisions;
  1066. ERR_FAIL_INDEX_V(p_collision, setting_collisions.size(), NodePath());
  1067. return setting_collisions[p_collision];
  1068. }
  1069. void SpringBoneSimulator3D::set_collision_count(int p_index, int p_count) {
  1070. ERR_FAIL_INDEX(p_index, settings.size());
  1071. if (are_all_child_collisions_enabled(p_index)) {
  1072. return; // Collision list is disabled.
  1073. }
  1074. Vector<NodePath> &setting_collisions = settings[p_index]->collisions;
  1075. setting_collisions.resize(p_count);
  1076. _make_collisions_dirty();
  1077. notify_property_list_changed();
  1078. }
  1079. int SpringBoneSimulator3D::get_collision_count(int p_index) const {
  1080. ERR_FAIL_INDEX_V(p_index, settings.size(), 0);
  1081. Vector<NodePath> setting_collisions = settings[p_index]->collisions;
  1082. return setting_collisions.size();
  1083. }
  1084. void SpringBoneSimulator3D::clear_collisions(int p_index) {
  1085. ERR_FAIL_INDEX(p_index, settings.size());
  1086. if (are_all_child_collisions_enabled(p_index)) {
  1087. return; // Collision list is disabled.
  1088. }
  1089. set_collision_count(p_index, 0);
  1090. }
  1091. LocalVector<ObjectID> SpringBoneSimulator3D::get_valid_collision_instance_ids(int p_index) {
  1092. ERR_FAIL_INDEX_V(p_index, settings.size(), LocalVector<ObjectID>());
  1093. if (collisions_dirty) {
  1094. _find_collisions();
  1095. }
  1096. return settings[p_index]->cached_collisions;
  1097. }
  1098. void SpringBoneSimulator3D::set_external_force(const Vector3 &p_force) {
  1099. external_force = p_force;
  1100. }
  1101. Vector3 SpringBoneSimulator3D::get_external_force() const {
  1102. return external_force;
  1103. }
  1104. void SpringBoneSimulator3D::_bind_methods() {
  1105. // Setting.
  1106. ClassDB::bind_method(D_METHOD("set_root_bone_name", "index", "bone_name"), &SpringBoneSimulator3D::set_root_bone_name);
  1107. ClassDB::bind_method(D_METHOD("get_root_bone_name", "index"), &SpringBoneSimulator3D::get_root_bone_name);
  1108. ClassDB::bind_method(D_METHOD("set_root_bone", "index", "bone"), &SpringBoneSimulator3D::set_root_bone);
  1109. ClassDB::bind_method(D_METHOD("get_root_bone", "index"), &SpringBoneSimulator3D::get_root_bone);
  1110. ClassDB::bind_method(D_METHOD("set_end_bone_name", "index", "bone_name"), &SpringBoneSimulator3D::set_end_bone_name);
  1111. ClassDB::bind_method(D_METHOD("get_end_bone_name", "index"), &SpringBoneSimulator3D::get_end_bone_name);
  1112. ClassDB::bind_method(D_METHOD("set_end_bone", "index", "bone"), &SpringBoneSimulator3D::set_end_bone);
  1113. ClassDB::bind_method(D_METHOD("get_end_bone", "index"), &SpringBoneSimulator3D::get_end_bone);
  1114. ClassDB::bind_method(D_METHOD("set_extend_end_bone", "index", "enabled"), &SpringBoneSimulator3D::set_extend_end_bone);
  1115. ClassDB::bind_method(D_METHOD("is_end_bone_extended", "index"), &SpringBoneSimulator3D::is_end_bone_extended);
  1116. ClassDB::bind_method(D_METHOD("set_end_bone_direction", "index", "bone_direction"), &SpringBoneSimulator3D::set_end_bone_direction);
  1117. ClassDB::bind_method(D_METHOD("get_end_bone_direction", "index"), &SpringBoneSimulator3D::get_end_bone_direction);
  1118. ClassDB::bind_method(D_METHOD("set_end_bone_length", "index", "length"), &SpringBoneSimulator3D::set_end_bone_length);
  1119. ClassDB::bind_method(D_METHOD("get_end_bone_length", "index"), &SpringBoneSimulator3D::get_end_bone_length);
  1120. ClassDB::bind_method(D_METHOD("set_center_from", "index", "center_from"), &SpringBoneSimulator3D::set_center_from);
  1121. ClassDB::bind_method(D_METHOD("get_center_from", "index"), &SpringBoneSimulator3D::get_center_from);
  1122. ClassDB::bind_method(D_METHOD("set_center_node", "index", "node_path"), &SpringBoneSimulator3D::set_center_node);
  1123. ClassDB::bind_method(D_METHOD("get_center_node", "index"), &SpringBoneSimulator3D::get_center_node);
  1124. ClassDB::bind_method(D_METHOD("set_center_bone_name", "index", "bone_name"), &SpringBoneSimulator3D::set_center_bone_name);
  1125. ClassDB::bind_method(D_METHOD("get_center_bone_name", "index"), &SpringBoneSimulator3D::get_center_bone_name);
  1126. ClassDB::bind_method(D_METHOD("set_center_bone", "index", "bone"), &SpringBoneSimulator3D::set_center_bone);
  1127. ClassDB::bind_method(D_METHOD("get_center_bone", "index"), &SpringBoneSimulator3D::get_center_bone);
  1128. ClassDB::bind_method(D_METHOD("set_radius", "index", "radius"), &SpringBoneSimulator3D::set_radius);
  1129. ClassDB::bind_method(D_METHOD("get_radius", "index"), &SpringBoneSimulator3D::get_radius);
  1130. ClassDB::bind_method(D_METHOD("set_rotation_axis", "index", "axis"), &SpringBoneSimulator3D::set_rotation_axis);
  1131. ClassDB::bind_method(D_METHOD("get_rotation_axis", "index"), &SpringBoneSimulator3D::get_rotation_axis);
  1132. ClassDB::bind_method(D_METHOD("set_rotation_axis_vector", "index", "vector"), &SpringBoneSimulator3D::set_rotation_axis_vector);
  1133. ClassDB::bind_method(D_METHOD("get_rotation_axis_vector", "index"), &SpringBoneSimulator3D::get_rotation_axis_vector);
  1134. ClassDB::bind_method(D_METHOD("set_radius_damping_curve", "index", "curve"), &SpringBoneSimulator3D::set_radius_damping_curve);
  1135. ClassDB::bind_method(D_METHOD("get_radius_damping_curve", "index"), &SpringBoneSimulator3D::get_radius_damping_curve);
  1136. ClassDB::bind_method(D_METHOD("set_stiffness", "index", "stiffness"), &SpringBoneSimulator3D::set_stiffness);
  1137. ClassDB::bind_method(D_METHOD("get_stiffness", "index"), &SpringBoneSimulator3D::get_stiffness);
  1138. ClassDB::bind_method(D_METHOD("set_stiffness_damping_curve", "index", "curve"), &SpringBoneSimulator3D::set_stiffness_damping_curve);
  1139. ClassDB::bind_method(D_METHOD("get_stiffness_damping_curve", "index"), &SpringBoneSimulator3D::get_stiffness_damping_curve);
  1140. ClassDB::bind_method(D_METHOD("set_drag", "index", "drag"), &SpringBoneSimulator3D::set_drag);
  1141. ClassDB::bind_method(D_METHOD("get_drag", "index"), &SpringBoneSimulator3D::get_drag);
  1142. ClassDB::bind_method(D_METHOD("set_drag_damping_curve", "index", "curve"), &SpringBoneSimulator3D::set_drag_damping_curve);
  1143. ClassDB::bind_method(D_METHOD("get_drag_damping_curve", "index"), &SpringBoneSimulator3D::get_drag_damping_curve);
  1144. ClassDB::bind_method(D_METHOD("set_gravity", "index", "gravity"), &SpringBoneSimulator3D::set_gravity);
  1145. ClassDB::bind_method(D_METHOD("get_gravity", "index"), &SpringBoneSimulator3D::get_gravity);
  1146. ClassDB::bind_method(D_METHOD("set_gravity_damping_curve", "index", "curve"), &SpringBoneSimulator3D::set_gravity_damping_curve);
  1147. ClassDB::bind_method(D_METHOD("get_gravity_damping_curve", "index"), &SpringBoneSimulator3D::get_gravity_damping_curve);
  1148. ClassDB::bind_method(D_METHOD("set_gravity_direction", "index", "gravity_direction"), &SpringBoneSimulator3D::set_gravity_direction);
  1149. ClassDB::bind_method(D_METHOD("get_gravity_direction", "index"), &SpringBoneSimulator3D::get_gravity_direction);
  1150. ClassDB::bind_method(D_METHOD("set_setting_count", "count"), &SpringBoneSimulator3D::set_setting_count);
  1151. ClassDB::bind_method(D_METHOD("get_setting_count"), &SpringBoneSimulator3D::get_setting_count);
  1152. ClassDB::bind_method(D_METHOD("clear_settings"), &SpringBoneSimulator3D::clear_settings);
  1153. // Individual joints.
  1154. ClassDB::bind_method(D_METHOD("set_individual_config", "index", "enabled"), &SpringBoneSimulator3D::set_individual_config);
  1155. ClassDB::bind_method(D_METHOD("is_config_individual", "index"), &SpringBoneSimulator3D::is_config_individual);
  1156. ClassDB::bind_method(D_METHOD("get_joint_bone_name", "index", "joint"), &SpringBoneSimulator3D::get_joint_bone_name);
  1157. ClassDB::bind_method(D_METHOD("get_joint_bone", "index", "joint"), &SpringBoneSimulator3D::get_joint_bone);
  1158. ClassDB::bind_method(D_METHOD("set_joint_rotation_axis", "index", "joint", "axis"), &SpringBoneSimulator3D::set_joint_rotation_axis);
  1159. ClassDB::bind_method(D_METHOD("get_joint_rotation_axis", "index", "joint"), &SpringBoneSimulator3D::get_joint_rotation_axis);
  1160. ClassDB::bind_method(D_METHOD("set_joint_rotation_axis_vector", "index", "joint", "vector"), &SpringBoneSimulator3D::set_joint_rotation_axis_vector);
  1161. ClassDB::bind_method(D_METHOD("get_joint_rotation_axis_vector", "index", "joint"), &SpringBoneSimulator3D::get_joint_rotation_axis_vector);
  1162. ClassDB::bind_method(D_METHOD("set_joint_radius", "index", "joint", "radius"), &SpringBoneSimulator3D::set_joint_radius);
  1163. ClassDB::bind_method(D_METHOD("get_joint_radius", "index", "joint"), &SpringBoneSimulator3D::get_joint_radius);
  1164. ClassDB::bind_method(D_METHOD("set_joint_stiffness", "index", "joint", "stiffness"), &SpringBoneSimulator3D::set_joint_stiffness);
  1165. ClassDB::bind_method(D_METHOD("get_joint_stiffness", "index", "joint"), &SpringBoneSimulator3D::get_joint_stiffness);
  1166. ClassDB::bind_method(D_METHOD("set_joint_drag", "index", "joint", "drag"), &SpringBoneSimulator3D::set_joint_drag);
  1167. ClassDB::bind_method(D_METHOD("get_joint_drag", "index", "joint"), &SpringBoneSimulator3D::get_joint_drag);
  1168. ClassDB::bind_method(D_METHOD("set_joint_gravity", "index", "joint", "gravity"), &SpringBoneSimulator3D::set_joint_gravity);
  1169. ClassDB::bind_method(D_METHOD("get_joint_gravity", "index", "joint"), &SpringBoneSimulator3D::get_joint_gravity);
  1170. ClassDB::bind_method(D_METHOD("set_joint_gravity_direction", "index", "joint", "gravity_direction"), &SpringBoneSimulator3D::set_joint_gravity_direction);
  1171. ClassDB::bind_method(D_METHOD("get_joint_gravity_direction", "index", "joint"), &SpringBoneSimulator3D::get_joint_gravity_direction);
  1172. ClassDB::bind_method(D_METHOD("get_joint_count", "index"), &SpringBoneSimulator3D::get_joint_count);
  1173. // Individual collisions.
  1174. ClassDB::bind_method(D_METHOD("set_enable_all_child_collisions", "index", "enabled"), &SpringBoneSimulator3D::set_enable_all_child_collisions);
  1175. ClassDB::bind_method(D_METHOD("are_all_child_collisions_enabled", "index"), &SpringBoneSimulator3D::are_all_child_collisions_enabled);
  1176. ClassDB::bind_method(D_METHOD("set_exclude_collision_path", "index", "collision", "node_path"), &SpringBoneSimulator3D::set_exclude_collision_path);
  1177. ClassDB::bind_method(D_METHOD("get_exclude_collision_path", "index", "collision"), &SpringBoneSimulator3D::get_exclude_collision_path);
  1178. ClassDB::bind_method(D_METHOD("set_exclude_collision_count", "index", "count"), &SpringBoneSimulator3D::set_exclude_collision_count);
  1179. ClassDB::bind_method(D_METHOD("get_exclude_collision_count", "index"), &SpringBoneSimulator3D::get_exclude_collision_count);
  1180. ClassDB::bind_method(D_METHOD("clear_exclude_collisions", "index"), &SpringBoneSimulator3D::clear_exclude_collisions);
  1181. ClassDB::bind_method(D_METHOD("set_collision_path", "index", "collision", "node_path"), &SpringBoneSimulator3D::set_collision_path);
  1182. ClassDB::bind_method(D_METHOD("get_collision_path", "index", "collision"), &SpringBoneSimulator3D::get_collision_path);
  1183. ClassDB::bind_method(D_METHOD("set_collision_count", "index", "count"), &SpringBoneSimulator3D::set_collision_count);
  1184. ClassDB::bind_method(D_METHOD("get_collision_count", "index"), &SpringBoneSimulator3D::get_collision_count);
  1185. ClassDB::bind_method(D_METHOD("clear_collisions", "index"), &SpringBoneSimulator3D::clear_collisions);
  1186. ClassDB::bind_method(D_METHOD("set_external_force", "force"), &SpringBoneSimulator3D::set_external_force);
  1187. ClassDB::bind_method(D_METHOD("get_external_force"), &SpringBoneSimulator3D::get_external_force);
  1188. // To process manually.
  1189. ClassDB::bind_method(D_METHOD("reset"), &SpringBoneSimulator3D::reset);
  1190. ADD_PROPERTY(PropertyInfo(Variant::VECTOR3, "external_force", PROPERTY_HINT_RANGE, "-99999,99999,or_greater,or_less,hide_slider,suffix:m/s"), "set_external_force", "get_external_force");
  1191. ADD_ARRAY_COUNT("Settings", "setting_count", "set_setting_count", "get_setting_count", "settings/");
  1192. BIND_ENUM_CONSTANT(BONE_DIRECTION_PLUS_X);
  1193. BIND_ENUM_CONSTANT(BONE_DIRECTION_MINUS_X);
  1194. BIND_ENUM_CONSTANT(BONE_DIRECTION_PLUS_Y);
  1195. BIND_ENUM_CONSTANT(BONE_DIRECTION_MINUS_Y);
  1196. BIND_ENUM_CONSTANT(BONE_DIRECTION_PLUS_Z);
  1197. BIND_ENUM_CONSTANT(BONE_DIRECTION_MINUS_Z);
  1198. BIND_ENUM_CONSTANT(BONE_DIRECTION_FROM_PARENT);
  1199. BIND_ENUM_CONSTANT(CENTER_FROM_WORLD_ORIGIN);
  1200. BIND_ENUM_CONSTANT(CENTER_FROM_NODE);
  1201. BIND_ENUM_CONSTANT(CENTER_FROM_BONE);
  1202. BIND_ENUM_CONSTANT(ROTATION_AXIS_X);
  1203. BIND_ENUM_CONSTANT(ROTATION_AXIS_Y);
  1204. BIND_ENUM_CONSTANT(ROTATION_AXIS_Z);
  1205. BIND_ENUM_CONSTANT(ROTATION_AXIS_ALL);
  1206. BIND_ENUM_CONSTANT(ROTATION_AXIS_CUSTOM);
  1207. }
  1208. void SpringBoneSimulator3D::_skeleton_changed(Skeleton3D *p_old, Skeleton3D *p_new) {
  1209. if (p_old && p_old->is_connected(SNAME("rest_updated"), callable_mp(this, &SpringBoneSimulator3D::_make_all_joints_dirty))) {
  1210. p_old->disconnect(SNAME("rest_updated"), callable_mp(this, &SpringBoneSimulator3D::_make_all_joints_dirty));
  1211. }
  1212. if (p_new && !p_new->is_connected(SNAME("rest_updated"), callable_mp(this, &SpringBoneSimulator3D::_make_all_joints_dirty))) {
  1213. p_new->connect(SNAME("rest_updated"), callable_mp(this, &SpringBoneSimulator3D::_make_all_joints_dirty));
  1214. }
  1215. _make_all_joints_dirty();
  1216. }
  1217. void SpringBoneSimulator3D::_validate_bone_names() {
  1218. for (int i = 0; i < settings.size(); i++) {
  1219. // Prior bone name.
  1220. if (!settings[i]->root_bone_name.is_empty()) {
  1221. set_root_bone_name(i, settings[i]->root_bone_name);
  1222. } else if (settings[i]->root_bone != -1) {
  1223. set_root_bone(i, settings[i]->root_bone);
  1224. }
  1225. // Prior bone name.
  1226. if (!settings[i]->end_bone_name.is_empty()) {
  1227. set_end_bone_name(i, settings[i]->end_bone_name);
  1228. } else if (settings[i]->end_bone != -1) {
  1229. set_end_bone(i, settings[i]->end_bone);
  1230. }
  1231. }
  1232. }
  1233. void SpringBoneSimulator3D::_make_joints_dirty(int p_index) {
  1234. ERR_FAIL_INDEX(p_index, settings.size());
  1235. settings[p_index]->joints_dirty = true;
  1236. if (joints_dirty) {
  1237. return;
  1238. }
  1239. joints_dirty = true;
  1240. callable_mp(this, &SpringBoneSimulator3D::_update_joints).call_deferred();
  1241. }
  1242. void SpringBoneSimulator3D::_make_all_joints_dirty() {
  1243. for (int i = 0; i < settings.size(); i++) {
  1244. _update_joint_array(i);
  1245. }
  1246. }
  1247. void SpringBoneSimulator3D::add_child_notify(Node *p_child) {
  1248. if (Object::cast_to<SpringBoneCollision3D>(p_child)) {
  1249. _make_collisions_dirty();
  1250. }
  1251. }
  1252. void SpringBoneSimulator3D::move_child_notify(Node *p_child) {
  1253. if (Object::cast_to<SpringBoneCollision3D>(p_child)) {
  1254. _make_collisions_dirty();
  1255. }
  1256. }
  1257. void SpringBoneSimulator3D::remove_child_notify(Node *p_child) {
  1258. if (Object::cast_to<SpringBoneCollision3D>(p_child)) {
  1259. _make_collisions_dirty();
  1260. }
  1261. }
  1262. void SpringBoneSimulator3D::_validate_rotation_axes(Skeleton3D *p_skeleton) const {
  1263. for (int i = 0; i < settings.size(); i++) {
  1264. for (int j = 0; j < settings[i]->joints.size(); j++) {
  1265. _validate_rotation_axis(p_skeleton, i, j);
  1266. }
  1267. }
  1268. }
  1269. void SpringBoneSimulator3D::_validate_rotation_axis(Skeleton3D *p_skeleton, int p_index, int p_joint) const {
  1270. RotationAxis axis = settings[p_index]->joints[p_joint]->rotation_axis;
  1271. if (axis == ROTATION_AXIS_ALL) {
  1272. return;
  1273. }
  1274. Vector3 rot = get_joint_rotation_axis_vector(p_index, p_joint).normalized();
  1275. Vector3 fwd;
  1276. if (p_joint < settings[p_index]->joints.size() - 1) {
  1277. fwd = p_skeleton->get_bone_rest(settings[p_index]->joints[p_joint + 1]->bone).origin;
  1278. } else if (settings[p_index]->extend_end_bone) {
  1279. fwd = get_end_bone_axis(settings[p_index]->end_bone, settings[p_index]->end_bone_direction);
  1280. if (fwd.is_zero_approx()) {
  1281. return;
  1282. }
  1283. }
  1284. fwd.normalize();
  1285. if (Math::is_equal_approx(Math::abs(rot.dot(fwd)), 1)) {
  1286. WARN_PRINT_ED("Setting: " + itos(p_index) + " Joint: " + itos(p_joint) + ": Rotation axis and forward vectors are colinear. This is not advised as it may cause unwanted rotation.");
  1287. }
  1288. }
  1289. void SpringBoneSimulator3D::_find_collisions() {
  1290. if (!collisions_dirty) {
  1291. return;
  1292. }
  1293. collisions.clear();
  1294. for (int i = 0; i < get_child_count(); i++) {
  1295. SpringBoneCollision3D *c = Object::cast_to<SpringBoneCollision3D>(get_child(i));
  1296. if (c) {
  1297. collisions.push_back(c->get_instance_id());
  1298. }
  1299. }
  1300. bool setting_updated = false;
  1301. for (int i = 0; i < settings.size(); i++) {
  1302. LocalVector<ObjectID> &cache = settings[i]->cached_collisions;
  1303. cache.clear();
  1304. if (!settings[i]->enable_all_child_collisions) {
  1305. // Allow list.
  1306. Vector<NodePath> &setting_collisions = settings[i]->collisions;
  1307. for (int j = 0; j < setting_collisions.size(); j++) {
  1308. Node *n = get_node_or_null(setting_collisions[j]);
  1309. if (!n) {
  1310. continue;
  1311. }
  1312. ObjectID id = n->get_instance_id();
  1313. if (!collisions.has(id)) {
  1314. setting_collisions.write[j] = NodePath(); // Clear path if not found.
  1315. } else {
  1316. cache.push_back(id);
  1317. }
  1318. }
  1319. } else {
  1320. // Deny list.
  1321. LocalVector<uint32_t> masks;
  1322. Vector<NodePath> &setting_exclude_collisions = settings[i]->exclude_collisions;
  1323. for (int j = 0; j < setting_exclude_collisions.size(); j++) {
  1324. Node *n = get_node_or_null(setting_exclude_collisions[j]);
  1325. if (!n) {
  1326. continue;
  1327. }
  1328. ObjectID id = n->get_instance_id();
  1329. int find = collisions.find(id);
  1330. if (find < 0) {
  1331. setting_exclude_collisions.write[j] = NodePath(); // Clear path if not found.
  1332. } else {
  1333. masks.push_back((uint32_t)find);
  1334. }
  1335. }
  1336. uint32_t mask_index = 0;
  1337. for (uint32_t j = 0; j < collisions.size(); j++) {
  1338. if (mask_index < masks.size() && j == masks[mask_index]) {
  1339. mask_index++;
  1340. continue;
  1341. }
  1342. cache.push_back(collisions[j]);
  1343. }
  1344. }
  1345. }
  1346. collisions_dirty = false;
  1347. if (setting_updated) {
  1348. notify_property_list_changed();
  1349. }
  1350. }
  1351. void SpringBoneSimulator3D::_process_collisions() {
  1352. if (!is_inside_tree()) {
  1353. return;
  1354. }
  1355. for (const ObjectID &oid : collisions) {
  1356. Object *t_obj = ObjectDB::get_instance(oid);
  1357. if (!t_obj) {
  1358. continue;
  1359. }
  1360. SpringBoneCollision3D *col = Object::cast_to<SpringBoneCollision3D>(t_obj);
  1361. if (!col) {
  1362. continue;
  1363. }
  1364. col->sync_pose();
  1365. }
  1366. }
  1367. void SpringBoneSimulator3D::_make_collisions_dirty() {
  1368. collisions_dirty = true;
  1369. }
  1370. void SpringBoneSimulator3D::_update_joint_array(int p_index) {
  1371. _make_joints_dirty(p_index);
  1372. Skeleton3D *sk = get_skeleton();
  1373. int current_bone = settings[p_index]->end_bone;
  1374. int root_bone = settings[p_index]->root_bone;
  1375. if (!sk || current_bone < 0 || root_bone < 0) {
  1376. set_joint_count(p_index, 0);
  1377. return;
  1378. }
  1379. // Validation.
  1380. bool valid = false;
  1381. while (current_bone >= 0) {
  1382. if (current_bone == root_bone) {
  1383. valid = true;
  1384. break;
  1385. }
  1386. current_bone = sk->get_bone_parent(current_bone);
  1387. }
  1388. if (!valid) {
  1389. set_joint_count(p_index, 0);
  1390. ERR_FAIL_EDMSG("End bone must be the same as or a child of root bone.");
  1391. }
  1392. Vector<int> new_joints;
  1393. current_bone = settings[p_index]->end_bone;
  1394. while (current_bone != root_bone) {
  1395. new_joints.push_back(current_bone);
  1396. current_bone = sk->get_bone_parent(current_bone);
  1397. }
  1398. new_joints.push_back(current_bone);
  1399. new_joints.reverse();
  1400. set_joint_count(p_index, new_joints.size());
  1401. for (int i = 0; i < new_joints.size(); i++) {
  1402. set_joint_bone(p_index, i, new_joints[i]);
  1403. }
  1404. }
  1405. void SpringBoneSimulator3D::_update_joints() {
  1406. if (!joints_dirty) {
  1407. return;
  1408. }
  1409. for (int i = 0; i < settings.size(); i++) {
  1410. if (!settings[i]->joints_dirty) {
  1411. continue;
  1412. }
  1413. if (settings[i]->individual_config) {
  1414. settings[i]->simulation_dirty = true;
  1415. settings[i]->joints_dirty = false;
  1416. continue; // Abort.
  1417. }
  1418. Vector<SpringBone3DJointSetting *> &joints = settings[i]->joints;
  1419. float unit = joints.size() > 0 ? (1.0 / float(joints.size() - 1)) : 0.0;
  1420. for (int j = 0; j < joints.size(); j++) {
  1421. float offset = j * unit;
  1422. if (settings[i]->radius_damping_curve.is_valid()) {
  1423. joints[j]->radius = settings[i]->radius * settings[i]->radius_damping_curve->sample_baked(offset);
  1424. } else {
  1425. joints[j]->radius = settings[i]->radius;
  1426. }
  1427. if (settings[i]->stiffness_damping_curve.is_valid()) {
  1428. joints[j]->stiffness = settings[i]->stiffness * settings[i]->stiffness_damping_curve->sample_baked(offset);
  1429. } else {
  1430. joints[j]->stiffness = settings[i]->stiffness;
  1431. }
  1432. if (settings[i]->drag_damping_curve.is_valid()) {
  1433. joints[j]->drag = settings[i]->drag * settings[i]->drag_damping_curve->sample_baked(offset);
  1434. } else {
  1435. joints[j]->drag = settings[i]->drag;
  1436. }
  1437. if (settings[i]->gravity_damping_curve.is_valid()) {
  1438. joints[j]->gravity = settings[i]->gravity * settings[i]->gravity_damping_curve->sample_baked(offset);
  1439. } else {
  1440. joints[j]->gravity = settings[i]->gravity;
  1441. }
  1442. joints[j]->gravity_direction = settings[i]->gravity_direction;
  1443. joints[j]->rotation_axis = settings[i]->rotation_axis;
  1444. joints[j]->rotation_axis_vector = settings[i]->rotation_axis_vector;
  1445. }
  1446. settings[i]->simulation_dirty = true;
  1447. settings[i]->joints_dirty = false;
  1448. }
  1449. joints_dirty = false;
  1450. Skeleton3D *sk = get_skeleton();
  1451. if (sk) {
  1452. _validate_rotation_axes(sk);
  1453. }
  1454. #ifdef TOOLS_ENABLED
  1455. update_gizmos();
  1456. #endif // TOOLS_ENABLED
  1457. }
  1458. void SpringBoneSimulator3D::_set_active(bool p_active) {
  1459. if (p_active) {
  1460. reset();
  1461. }
  1462. }
  1463. void SpringBoneSimulator3D::_process_modification(double p_delta) {
  1464. if (!is_inside_tree()) {
  1465. return;
  1466. }
  1467. Skeleton3D *skeleton = get_skeleton();
  1468. if (!skeleton) {
  1469. return;
  1470. }
  1471. _find_collisions();
  1472. _process_collisions();
  1473. #ifdef TOOLS_ENABLED
  1474. if (saving) {
  1475. return; // Collision position has been reset but we don't want to process simulating on saving. Abort.
  1476. }
  1477. #endif //TOOLS_ENABLED
  1478. for (int i = 0; i < settings.size(); i++) {
  1479. _init_joints(skeleton, settings[i]);
  1480. _process_joints(p_delta, skeleton, settings[i]->joints, get_valid_collision_instance_ids(i), settings[i]->cached_center, settings[i]->cached_inverted_center, settings[i]->cached_inverted_center.basis.get_rotation_quaternion());
  1481. }
  1482. }
  1483. void SpringBoneSimulator3D::reset() {
  1484. if (!is_inside_tree()) {
  1485. return;
  1486. }
  1487. Skeleton3D *skeleton = get_skeleton();
  1488. if (!skeleton) {
  1489. return;
  1490. }
  1491. _find_collisions();
  1492. _process_collisions();
  1493. for (int i = 0; i < settings.size(); i++) {
  1494. settings[i]->simulation_dirty = true;
  1495. _init_joints(skeleton, settings[i]);
  1496. }
  1497. }
  1498. void SpringBoneSimulator3D::_init_joints(Skeleton3D *p_skeleton, SpringBone3DSetting *setting) {
  1499. if (setting->center_from == CENTER_FROM_WORLD_ORIGIN) {
  1500. setting->cached_center = p_skeleton->get_global_transform();
  1501. } else if (setting->center_from == CENTER_FROM_NODE) {
  1502. if (setting->center_node == NodePath()) {
  1503. setting->cached_center = Transform3D();
  1504. } else {
  1505. Node3D *nd = Object::cast_to<Node3D>(get_node_or_null(setting->center_node));
  1506. if (!nd) {
  1507. setting->cached_center = Transform3D();
  1508. } else {
  1509. setting->cached_center = nd->get_global_transform().affine_inverse() * p_skeleton->get_global_transform();
  1510. }
  1511. }
  1512. } else {
  1513. if (setting->center_bone >= 0) {
  1514. setting->cached_center = p_skeleton->get_bone_global_pose(setting->center_bone);
  1515. } else {
  1516. setting->cached_center = Transform3D();
  1517. }
  1518. }
  1519. setting->cached_inverted_center = setting->cached_center.affine_inverse();
  1520. if (!setting->simulation_dirty) {
  1521. return;
  1522. }
  1523. for (int i = 0; i < setting->joints.size(); i++) {
  1524. if (setting->joints[i]->verlet) {
  1525. memdelete(setting->joints[i]->verlet);
  1526. setting->joints[i]->verlet = nullptr;
  1527. }
  1528. if (i < setting->joints.size() - 1) {
  1529. setting->joints[i]->verlet = memnew(SpringBone3DVerletInfo);
  1530. Vector3 axis = p_skeleton->get_bone_rest(setting->joints[i + 1]->bone).origin;
  1531. setting->joints[i]->verlet->current_tail = setting->cached_center.xform(p_skeleton->get_bone_global_pose(setting->joints[i]->bone).xform(axis));
  1532. setting->joints[i]->verlet->prev_tail = setting->joints[i]->verlet->current_tail;
  1533. setting->joints[i]->verlet->forward_vector = snap_vector_to_plane(setting->joints[i]->get_rotation_axis_vector(), axis.normalized());
  1534. setting->joints[i]->verlet->length = axis.length();
  1535. setting->joints[i]->verlet->current_rot = Quaternion(0, 0, 0, 1);
  1536. } else if (setting->extend_end_bone && setting->end_bone_length > 0) {
  1537. Vector3 axis = get_end_bone_axis(setting->end_bone, setting->end_bone_direction);
  1538. if (axis.is_zero_approx()) {
  1539. continue;
  1540. }
  1541. setting->joints[i]->verlet = memnew(SpringBone3DVerletInfo);
  1542. setting->joints[i]->verlet->forward_vector = snap_vector_to_plane(setting->joints[i]->get_rotation_axis_vector(), axis.normalized());
  1543. setting->joints[i]->verlet->length = setting->end_bone_length;
  1544. setting->joints[i]->verlet->current_tail = setting->cached_center.xform(p_skeleton->get_bone_global_pose(setting->joints[i]->bone).xform(axis * setting->end_bone_length));
  1545. setting->joints[i]->verlet->prev_tail = setting->joints[i]->verlet->current_tail;
  1546. setting->joints[i]->verlet->current_rot = Quaternion(0, 0, 0, 1);
  1547. }
  1548. }
  1549. setting->simulation_dirty = false;
  1550. }
  1551. void SpringBoneSimulator3D::_process_joints(double p_delta, Skeleton3D *p_skeleton, Vector<SpringBone3DJointSetting *> &p_joints, const LocalVector<ObjectID> &p_collisions, const Transform3D &p_center_transform, const Transform3D &p_inverted_center_transform, const Quaternion &p_inverted_center_rotation) {
  1552. for (int i = 0; i < p_joints.size(); i++) {
  1553. SpringBone3DVerletInfo *verlet = p_joints[i]->verlet;
  1554. if (!verlet) {
  1555. continue; // Means not extended end bone.
  1556. }
  1557. Transform3D current_global_pose = p_skeleton->get_bone_global_pose(p_joints[i]->bone);
  1558. Transform3D current_world_pose = p_center_transform * current_global_pose;
  1559. Quaternion current_rot = current_global_pose.basis.get_rotation_quaternion();
  1560. Vector3 current_origin = p_center_transform.xform(current_global_pose.origin);
  1561. Vector3 external = p_inverted_center_rotation.xform((external_force + p_joints[i]->gravity_direction * p_joints[i]->gravity) * p_delta);
  1562. // Integration of velocity by verlet.
  1563. Vector3 next_tail = verlet->current_tail +
  1564. (verlet->current_tail - verlet->prev_tail) * (1.0 - p_joints[i]->drag) +
  1565. p_center_transform.basis.get_rotation_quaternion().xform(current_rot.xform(verlet->forward_vector * (p_joints[i]->stiffness * p_delta)) + external);
  1566. // Snap to plane if axis locked.
  1567. if (p_joints[i]->rotation_axis != ROTATION_AXIS_ALL) {
  1568. next_tail = current_world_pose.origin + current_world_pose.basis.get_rotation_quaternion().xform(snap_vector_to_plane(p_joints[i]->get_rotation_axis_vector(), current_world_pose.basis.get_rotation_quaternion().xform_inv(next_tail - current_world_pose.origin)));
  1569. }
  1570. // Limit bone length.
  1571. next_tail = limit_length(current_origin, next_tail, verlet->length);
  1572. // Collision movement.
  1573. for (uint32_t j = 0; j < p_collisions.size(); j++) {
  1574. Object *obj = ObjectDB::get_instance(p_collisions[j]);
  1575. if (!obj) {
  1576. continue;
  1577. }
  1578. SpringBoneCollision3D *col = Object::cast_to<SpringBoneCollision3D>(obj);
  1579. if (col) {
  1580. // Collider movement should separate from the effect of the center.
  1581. next_tail = col->collide(p_center_transform, p_joints[i]->radius, verlet->length, next_tail);
  1582. // Snap to plane if axis locked.
  1583. if (p_joints[i]->rotation_axis != ROTATION_AXIS_ALL) {
  1584. next_tail = current_world_pose.origin + current_world_pose.basis.get_rotation_quaternion().xform(snap_vector_to_plane(p_joints[i]->get_rotation_axis_vector(), current_world_pose.basis.get_rotation_quaternion().xform_inv(next_tail - current_world_pose.origin)));
  1585. }
  1586. // Limit bone length.
  1587. next_tail = limit_length(current_origin, next_tail, verlet->length);
  1588. }
  1589. }
  1590. // Store current tails for next process.
  1591. verlet->prev_tail = verlet->current_tail;
  1592. verlet->current_tail = next_tail;
  1593. // Convert position to rotation.
  1594. Vector3 from = current_rot.xform(verlet->forward_vector);
  1595. Vector3 to = p_inverted_center_transform.basis.xform(next_tail - current_origin);
  1596. from.normalize();
  1597. to.normalize();
  1598. Quaternion from_to = get_from_to_rotation(from, to, verlet->current_rot);
  1599. verlet->current_rot = from_to;
  1600. // Apply rotation.
  1601. from_to *= current_rot;
  1602. from_to = get_local_pose_rotation(p_skeleton, p_joints[i]->bone, from_to);
  1603. p_skeleton->set_bone_pose_rotation(p_joints[i]->bone, from_to);
  1604. }
  1605. }
  1606. SpringBoneSimulator3D::~SpringBoneSimulator3D() {
  1607. clear_settings();
  1608. }