collada.cpp 73 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390
  1. /*************************************************************************/
  2. /* collada.cpp */
  3. /*************************************************************************/
  4. /* This file is part of: */
  5. /* GODOT ENGINE */
  6. /* https://godotengine.org */
  7. /*************************************************************************/
  8. /* Copyright (c) 2007-2022 Juan Linietsky, Ariel Manzur. */
  9. /* Copyright (c) 2014-2022 Godot Engine contributors (cf. AUTHORS.md). */
  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 "collada.h"
  31. #include <stdio.h>
  32. //#define DEBUG_DEFAULT_ANIMATION
  33. //#define DEBUG_COLLADA
  34. #ifdef DEBUG_COLLADA
  35. #define COLLADA_PRINT(m_what) print_line(m_what)
  36. #else
  37. #define COLLADA_PRINT(m_what)
  38. #endif
  39. #define COLLADA_IMPORT_SCALE_SCENE
  40. /* HELPERS */
  41. String Collada::Effect::get_texture_path(const String &p_source, Collada &state) const {
  42. const String &image = p_source;
  43. ERR_FAIL_COND_V(!state.state.image_map.has(image), "");
  44. return state.state.image_map[image].path;
  45. }
  46. Transform3D Collada::get_root_transform() const {
  47. Transform3D unit_scale_transform;
  48. #ifndef COLLADA_IMPORT_SCALE_SCENE
  49. unit_scale_transform.scale(Vector3(state.unit_scale, state.unit_scale, state.unit_scale));
  50. #endif
  51. return unit_scale_transform;
  52. }
  53. void Collada::Vertex::fix_unit_scale(const Collada &state) {
  54. #ifdef COLLADA_IMPORT_SCALE_SCENE
  55. vertex *= state.state.unit_scale;
  56. #endif
  57. }
  58. static String _uri_to_id(const String &p_uri) {
  59. if (p_uri.begins_with("#")) {
  60. return p_uri.substr(1, p_uri.size() - 1);
  61. } else {
  62. return p_uri;
  63. }
  64. }
  65. /** HELPER FUNCTIONS **/
  66. Transform3D Collada::fix_transform(const Transform3D &p_transform) {
  67. Transform3D tr = p_transform;
  68. #ifndef NO_UP_AXIS_SWAP
  69. if (state.up_axis != Vector3::AXIS_Y) {
  70. for (int i = 0; i < 3; i++) {
  71. SWAP(tr.basis[1][i], tr.basis[state.up_axis][i]);
  72. }
  73. for (int i = 0; i < 3; i++) {
  74. SWAP(tr.basis[i][1], tr.basis[i][state.up_axis]);
  75. }
  76. SWAP(tr.origin[1], tr.origin[state.up_axis]);
  77. tr.basis[state.up_axis][0] = -tr.basis[state.up_axis][0];
  78. tr.basis[state.up_axis][1] = -tr.basis[state.up_axis][1];
  79. tr.basis[0][state.up_axis] = -tr.basis[0][state.up_axis];
  80. tr.basis[1][state.up_axis] = -tr.basis[1][state.up_axis];
  81. tr.origin[state.up_axis] = -tr.origin[state.up_axis];
  82. }
  83. #endif
  84. //tr.scale(Vector3(state.unit_scale.unit_scale.unit_scale));
  85. return tr;
  86. //return state.matrix_fix * p_transform;
  87. }
  88. static Transform3D _read_transform_from_array(const Vector<float> &array, int ofs = 0) {
  89. Transform3D tr;
  90. // i wonder why collada matrices are transposed, given that's opposed to opengl..
  91. tr.basis.rows[0][0] = array[0 + ofs];
  92. tr.basis.rows[0][1] = array[1 + ofs];
  93. tr.basis.rows[0][2] = array[2 + ofs];
  94. tr.basis.rows[1][0] = array[4 + ofs];
  95. tr.basis.rows[1][1] = array[5 + ofs];
  96. tr.basis.rows[1][2] = array[6 + ofs];
  97. tr.basis.rows[2][0] = array[8 + ofs];
  98. tr.basis.rows[2][1] = array[9 + ofs];
  99. tr.basis.rows[2][2] = array[10 + ofs];
  100. tr.origin.x = array[3 + ofs];
  101. tr.origin.y = array[7 + ofs];
  102. tr.origin.z = array[11 + ofs];
  103. return tr;
  104. }
  105. /* STRUCTURES */
  106. Transform3D Collada::Node::compute_transform(const Collada &state) const {
  107. Transform3D xform;
  108. for (int i = 0; i < xform_list.size(); i++) {
  109. Transform3D xform_step;
  110. const XForm &xf = xform_list[i];
  111. switch (xf.op) {
  112. case XForm::OP_ROTATE: {
  113. if (xf.data.size() >= 4) {
  114. xform_step.rotate(Vector3(xf.data[0], xf.data[1], xf.data[2]), Math::deg_to_rad(xf.data[3]));
  115. }
  116. } break;
  117. case XForm::OP_SCALE: {
  118. if (xf.data.size() >= 3) {
  119. xform_step.scale(Vector3(xf.data[0], xf.data[1], xf.data[2]));
  120. }
  121. } break;
  122. case XForm::OP_TRANSLATE: {
  123. if (xf.data.size() >= 3) {
  124. xform_step.origin = Vector3(xf.data[0], xf.data[1], xf.data[2]);
  125. }
  126. } break;
  127. case XForm::OP_MATRIX: {
  128. if (xf.data.size() >= 16) {
  129. xform_step = _read_transform_from_array(xf.data, 0);
  130. }
  131. } break;
  132. default: {
  133. }
  134. }
  135. xform = xform * xform_step;
  136. }
  137. #ifdef COLLADA_IMPORT_SCALE_SCENE
  138. xform.origin *= state.state.unit_scale;
  139. #endif
  140. return xform;
  141. }
  142. Transform3D Collada::Node::get_transform() const {
  143. return default_transform;
  144. }
  145. Transform3D Collada::Node::get_global_transform() const {
  146. if (parent) {
  147. return parent->get_global_transform() * default_transform;
  148. } else {
  149. return default_transform;
  150. }
  151. }
  152. Vector<float> Collada::AnimationTrack::get_value_at_time(float p_time) const {
  153. ERR_FAIL_COND_V(keys.size() == 0, Vector<float>());
  154. int i = 0;
  155. for (i = 0; i < keys.size(); i++) {
  156. if (keys[i].time > p_time) {
  157. break;
  158. }
  159. }
  160. if (i == 0) {
  161. return keys[0].data;
  162. }
  163. if (i == keys.size()) {
  164. return keys[keys.size() - 1].data;
  165. }
  166. switch (keys[i].interp_type) {
  167. case INTERP_BEZIER: //wait for bezier
  168. case INTERP_LINEAR: {
  169. float c = (p_time - keys[i - 1].time) / (keys[i].time - keys[i - 1].time);
  170. if (keys[i].data.size() == 16) {
  171. //interpolate a matrix
  172. Transform3D src = _read_transform_from_array(keys[i - 1].data);
  173. Transform3D dst = _read_transform_from_array(keys[i].data);
  174. Transform3D interp = c < 0.001 ? src : src.interpolate_with(dst, c);
  175. Vector<float> ret;
  176. ret.resize(16);
  177. Transform3D tr;
  178. // i wonder why collada matrices are transposed, given that's opposed to opengl..
  179. ret.write[0] = interp.basis.rows[0][0];
  180. ret.write[1] = interp.basis.rows[0][1];
  181. ret.write[2] = interp.basis.rows[0][2];
  182. ret.write[4] = interp.basis.rows[1][0];
  183. ret.write[5] = interp.basis.rows[1][1];
  184. ret.write[6] = interp.basis.rows[1][2];
  185. ret.write[8] = interp.basis.rows[2][0];
  186. ret.write[9] = interp.basis.rows[2][1];
  187. ret.write[10] = interp.basis.rows[2][2];
  188. ret.write[3] = interp.origin.x;
  189. ret.write[7] = interp.origin.y;
  190. ret.write[11] = interp.origin.z;
  191. ret.write[12] = 0;
  192. ret.write[13] = 0;
  193. ret.write[14] = 0;
  194. ret.write[15] = 1;
  195. return ret;
  196. } else {
  197. Vector<float> dest;
  198. dest.resize(keys[i].data.size());
  199. for (int j = 0; j < dest.size(); j++) {
  200. dest.write[j] = keys[i].data[j] * c + keys[i - 1].data[j] * (1.0 - c);
  201. }
  202. return dest;
  203. //interpolate one by one
  204. }
  205. } break;
  206. }
  207. ERR_FAIL_V(Vector<float>());
  208. }
  209. void Collada::_parse_asset(XMLParser &parser) {
  210. while (parser.read() == OK) {
  211. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  212. String name = parser.get_node_name();
  213. if (name == "up_axis") {
  214. parser.read();
  215. if (parser.get_node_data() == "X_UP") {
  216. state.up_axis = Vector3::AXIS_X;
  217. }
  218. if (parser.get_node_data() == "Y_UP") {
  219. state.up_axis = Vector3::AXIS_Y;
  220. }
  221. if (parser.get_node_data() == "Z_UP") {
  222. state.up_axis = Vector3::AXIS_Z;
  223. }
  224. COLLADA_PRINT("up axis: " + parser.get_node_data());
  225. } else if (name == "unit") {
  226. state.unit_scale = parser.get_attribute_value("meter").to_float();
  227. COLLADA_PRINT("unit scale: " + rtos(state.unit_scale));
  228. }
  229. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "asset") {
  230. break; //end of <asset>
  231. }
  232. }
  233. }
  234. void Collada::_parse_image(XMLParser &parser) {
  235. String id = parser.get_attribute_value("id");
  236. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  237. if (!parser.is_empty()) {
  238. parser.skip_section();
  239. }
  240. return;
  241. }
  242. Image image;
  243. if (state.version < State::Version(1, 4, 0)) {
  244. /* <1.4 */
  245. String path = parser.get_attribute_value("source").strip_edges();
  246. if (!path.contains("://") && path.is_relative_path()) {
  247. // path is relative to file being loaded, so convert to a resource path
  248. image.path = ProjectSettings::get_singleton()->localize_path(state.local_path.get_base_dir().path_join(path.uri_decode()));
  249. }
  250. } else {
  251. while (parser.read() == OK) {
  252. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  253. String name = parser.get_node_name();
  254. if (name == "init_from") {
  255. parser.read();
  256. String path = parser.get_node_data().strip_edges().uri_decode();
  257. if (!path.contains("://") && path.is_relative_path()) {
  258. // path is relative to file being loaded, so convert to a resource path
  259. path = ProjectSettings::get_singleton()->localize_path(state.local_path.get_base_dir().path_join(path));
  260. } else if (path.find("file:///") == 0) {
  261. path = path.replace_first("file:///", "");
  262. path = ProjectSettings::get_singleton()->localize_path(path);
  263. }
  264. image.path = path;
  265. } else if (name == "data") {
  266. ERR_PRINT("COLLADA Embedded image data not supported!");
  267. } else if (name == "extra" && !parser.is_empty()) {
  268. parser.skip_section();
  269. }
  270. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "image") {
  271. break; //end of <asset>
  272. }
  273. }
  274. }
  275. state.image_map[id] = image;
  276. }
  277. void Collada::_parse_material(XMLParser &parser) {
  278. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  279. if (!parser.is_empty()) {
  280. parser.skip_section();
  281. }
  282. return;
  283. }
  284. Material material;
  285. String id = parser.get_attribute_value("id");
  286. if (parser.has_attribute("name")) {
  287. material.name = parser.get_attribute_value("name");
  288. }
  289. if (state.version < State::Version(1, 4, 0)) {
  290. /* <1.4 */
  291. ERR_PRINT("Collada Materials < 1.4 are not supported (yet)");
  292. } else {
  293. while (parser.read() == OK) {
  294. if (parser.get_node_type() == XMLParser::NODE_ELEMENT && parser.get_node_name() == "instance_effect") {
  295. material.instance_effect = _uri_to_id(parser.get_attribute_value("url"));
  296. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "material") {
  297. break; //end of <asset>
  298. }
  299. }
  300. }
  301. state.material_map[id] = material;
  302. }
  303. //! reads floats from inside of xml element until end of xml element
  304. Vector<float> Collada::_read_float_array(XMLParser &parser) {
  305. if (parser.is_empty()) {
  306. return Vector<float>();
  307. }
  308. Vector<String> splitters;
  309. splitters.push_back(" ");
  310. splitters.push_back("\n");
  311. splitters.push_back("\r");
  312. splitters.push_back("\t");
  313. Vector<float> array;
  314. while (parser.read() == OK) {
  315. // TODO: check for comments inside the element
  316. // and ignore them.
  317. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  318. // parse float data
  319. String str = parser.get_node_data();
  320. array = str.split_floats_mk(splitters, false);
  321. //array=str.split_floats(" ",false);
  322. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
  323. break; // end parsing text
  324. }
  325. }
  326. return array;
  327. }
  328. Vector<String> Collada::_read_string_array(XMLParser &parser) {
  329. if (parser.is_empty()) {
  330. return Vector<String>();
  331. }
  332. Vector<String> array;
  333. while (parser.read() == OK) {
  334. // TODO: check for comments inside the element
  335. // and ignore them.
  336. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  337. // parse String data
  338. String str = parser.get_node_data();
  339. array = str.split_spaces();
  340. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
  341. break; // end parsing text
  342. }
  343. }
  344. return array;
  345. }
  346. Transform3D Collada::_read_transform(XMLParser &parser) {
  347. if (parser.is_empty()) {
  348. return Transform3D();
  349. }
  350. Vector<String> array;
  351. while (parser.read() == OK) {
  352. // TODO: check for comments inside the element
  353. // and ignore them.
  354. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  355. // parse float data
  356. String str = parser.get_node_data();
  357. array = str.split_spaces();
  358. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
  359. break; // end parsing text
  360. }
  361. }
  362. ERR_FAIL_COND_V(array.size() != 16, Transform3D());
  363. Vector<float> farr;
  364. farr.resize(16);
  365. for (int i = 0; i < 16; i++) {
  366. farr.write[i] = array[i].to_float();
  367. }
  368. return _read_transform_from_array(farr);
  369. }
  370. String Collada::_read_empty_draw_type(XMLParser &parser) {
  371. String empty_draw_type = "";
  372. if (parser.is_empty()) {
  373. return empty_draw_type;
  374. }
  375. while (parser.read() == OK) {
  376. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  377. empty_draw_type = parser.get_node_data();
  378. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
  379. break; // end parsing text
  380. }
  381. }
  382. return empty_draw_type;
  383. }
  384. Variant Collada::_parse_param(XMLParser &parser) {
  385. if (parser.is_empty()) {
  386. return Variant();
  387. }
  388. String from = parser.get_node_name();
  389. Variant data;
  390. while (parser.read() == OK) {
  391. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  392. if (parser.get_node_name() == "float") {
  393. parser.read();
  394. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  395. data = parser.get_node_data().to_float();
  396. }
  397. } else if (parser.get_node_name() == "float2") {
  398. Vector<float> v2 = _read_float_array(parser);
  399. if (v2.size() >= 2) {
  400. data = Vector2(v2[0], v2[1]);
  401. }
  402. } else if (parser.get_node_name() == "float3") {
  403. Vector<float> v3 = _read_float_array(parser);
  404. if (v3.size() >= 3) {
  405. data = Vector3(v3[0], v3[1], v3[2]);
  406. }
  407. } else if (parser.get_node_name() == "float4") {
  408. Vector<float> v4 = _read_float_array(parser);
  409. if (v4.size() >= 4) {
  410. data = Color(v4[0], v4[1], v4[2], v4[3]);
  411. }
  412. } else if (parser.get_node_name() == "sampler2D") {
  413. while (parser.read() == OK) {
  414. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  415. if (parser.get_node_name() == "source") {
  416. parser.read();
  417. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  418. data = parser.get_node_data();
  419. }
  420. }
  421. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "sampler2D") {
  422. break;
  423. }
  424. }
  425. } else if (parser.get_node_name() == "surface") {
  426. while (parser.read() == OK) {
  427. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  428. if (parser.get_node_name() == "init_from") {
  429. parser.read();
  430. if (parser.get_node_type() == XMLParser::NODE_TEXT) {
  431. data = parser.get_node_data();
  432. }
  433. }
  434. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "surface") {
  435. break;
  436. }
  437. }
  438. }
  439. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == from) {
  440. break;
  441. }
  442. }
  443. COLLADA_PRINT("newparam ending " + parser.get_node_name());
  444. return data;
  445. }
  446. void Collada::_parse_effect_material(XMLParser &parser, Effect &effect, String &id) {
  447. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  448. if (!parser.is_empty()) {
  449. parser.skip_section();
  450. }
  451. return;
  452. }
  453. while (parser.read() == OK) {
  454. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  455. // first come the tags we descend, but ignore the top-levels
  456. COLLADA_PRINT("node name: " + parser.get_node_name());
  457. if (!parser.is_empty() &&
  458. (parser.get_node_name() == "profile_COMMON" ||
  459. parser.get_node_name() == "technique" ||
  460. parser.get_node_name() == "extra")) {
  461. _parse_effect_material(parser, effect, id); // try again
  462. } else if (parser.get_node_name() == "newparam") {
  463. String name = parser.get_attribute_value("sid");
  464. Variant value = _parse_param(parser);
  465. effect.params[name] = value;
  466. COLLADA_PRINT("param: " + name + " value:" + String(value));
  467. } else if (parser.get_node_name() == "constant" ||
  468. parser.get_node_name() == "lambert" ||
  469. parser.get_node_name() == "phong" ||
  470. parser.get_node_name() == "blinn") {
  471. COLLADA_PRINT("shade model: " + parser.get_node_name());
  472. while (parser.read() == OK) {
  473. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  474. String what = parser.get_node_name();
  475. if (what == "emission" ||
  476. what == "diffuse" ||
  477. what == "specular" ||
  478. what == "reflective") {
  479. // color or texture types
  480. while (parser.read() == OK) {
  481. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  482. if (parser.get_node_name() == "color") {
  483. Vector<float> colorarr = _read_float_array(parser);
  484. COLLADA_PRINT("colorarr size: " + rtos(colorarr.size()));
  485. if (colorarr.size() >= 3) {
  486. // alpha strangely not alright? maybe it needs to be multiplied by value as a channel intensity
  487. Color color(colorarr[0], colorarr[1], colorarr[2], 1.0);
  488. if (what == "diffuse") {
  489. effect.diffuse.color = color;
  490. }
  491. if (what == "specular") {
  492. effect.specular.color = color;
  493. }
  494. if (what == "emission") {
  495. effect.emission.color = color;
  496. }
  497. COLLADA_PRINT(what + " color: " + color);
  498. }
  499. } else if (parser.get_node_name() == "texture") {
  500. String sampler = parser.get_attribute_value("texture");
  501. if (!effect.params.has(sampler)) {
  502. ERR_PRINT(String("Couldn't find sampler: " + sampler + " in material:" + id).utf8().get_data());
  503. } else {
  504. String surface = effect.params[sampler];
  505. if (!effect.params.has(surface)) {
  506. ERR_PRINT(String("Couldn't find surface: " + surface + " in material:" + id).utf8().get_data());
  507. } else {
  508. String uri = effect.params[surface];
  509. if (what == "diffuse") {
  510. effect.diffuse.texture = uri;
  511. } else if (what == "specular") {
  512. effect.specular.texture = uri;
  513. } else if (what == "emission") {
  514. effect.emission.texture = uri;
  515. } else if (what == "bump") {
  516. if (parser.has_attribute("bumptype") && parser.get_attribute_value("bumptype") != "NORMALMAP") {
  517. WARN_PRINT("'bump' texture type is not NORMALMAP, only NORMALMAP is supported.");
  518. }
  519. effect.bump.texture = uri;
  520. }
  521. COLLADA_PRINT(what + " texture: " + uri);
  522. }
  523. }
  524. } else if (!parser.is_empty()) {
  525. parser.skip_section();
  526. }
  527. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == what) {
  528. break;
  529. }
  530. }
  531. } else if (what == "shininess") {
  532. effect.shininess = _parse_param(parser);
  533. }
  534. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END &&
  535. (parser.get_node_name() == "constant" ||
  536. parser.get_node_name() == "lambert" ||
  537. parser.get_node_name() == "phong" ||
  538. parser.get_node_name() == "blinn")) {
  539. break;
  540. }
  541. }
  542. } else if (parser.get_node_name() == "double_sided" || parser.get_node_name() == "show_double_sided") { // colladamax / google earth
  543. // 3DS Max / Google Earth double sided extension
  544. parser.read();
  545. effect.found_double_sided = true;
  546. effect.double_sided = parser.get_node_data().to_int();
  547. COLLADA_PRINT("double sided: " + itos(parser.get_node_data().to_int()));
  548. } else if (parser.get_node_name() == "unshaded") {
  549. parser.read();
  550. effect.unshaded = parser.get_node_data().to_int();
  551. } else if (parser.get_node_name() == "bump") {
  552. // color or texture types
  553. while (parser.read() == OK) {
  554. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  555. if (parser.get_node_name() == "texture") {
  556. String sampler = parser.get_attribute_value("texture");
  557. if (!effect.params.has(sampler)) {
  558. ERR_PRINT(String("Couldn't find sampler: " + sampler + " in material:" + id).utf8().get_data());
  559. } else {
  560. String surface = effect.params[sampler];
  561. if (!effect.params.has(surface)) {
  562. ERR_PRINT(String("Couldn't find surface: " + surface + " in material:" + id).utf8().get_data());
  563. } else {
  564. String uri = effect.params[surface];
  565. if (parser.has_attribute("bumptype") && parser.get_attribute_value("bumptype") != "NORMALMAP") {
  566. WARN_PRINT("'bump' texture type is not NORMALMAP, only NORMALMAP is supported.");
  567. }
  568. effect.bump.texture = uri;
  569. COLLADA_PRINT(" bump: " + uri);
  570. }
  571. }
  572. } else if (!parser.is_empty()) {
  573. parser.skip_section();
  574. }
  575. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "bump") {
  576. break;
  577. }
  578. }
  579. } else if (!parser.is_empty()) {
  580. parser.skip_section();
  581. }
  582. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END &&
  583. (parser.get_node_name() == "effect" ||
  584. parser.get_node_name() == "profile_COMMON" ||
  585. parser.get_node_name() == "technique" ||
  586. parser.get_node_name() == "extra")) {
  587. break;
  588. }
  589. }
  590. }
  591. void Collada::_parse_effect(XMLParser &parser) {
  592. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  593. if (!parser.is_empty()) {
  594. parser.skip_section();
  595. }
  596. return;
  597. }
  598. String id = parser.get_attribute_value("id");
  599. Effect effect;
  600. if (parser.has_attribute("name")) {
  601. effect.name = parser.get_attribute_value("name");
  602. }
  603. _parse_effect_material(parser, effect, id);
  604. state.effect_map[id] = effect;
  605. COLLADA_PRINT("Effect ID:" + id);
  606. }
  607. void Collada::_parse_camera(XMLParser &parser) {
  608. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  609. if (!parser.is_empty()) {
  610. parser.skip_section();
  611. }
  612. return;
  613. }
  614. String id = parser.get_attribute_value("id");
  615. state.camera_data_map[id] = CameraData();
  616. CameraData &camera = state.camera_data_map[id];
  617. while (parser.read() == OK) {
  618. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  619. String name = parser.get_node_name();
  620. if (name == "perspective") {
  621. camera.mode = CameraData::MODE_PERSPECTIVE;
  622. } else if (name == "orthographic") {
  623. camera.mode = CameraData::MODE_ORTHOGONAL;
  624. } else if (name == "xfov") {
  625. parser.read();
  626. camera.perspective.x_fov = parser.get_node_data().to_float();
  627. } else if (name == "yfov") {
  628. parser.read();
  629. camera.perspective.y_fov = parser.get_node_data().to_float();
  630. } else if (name == "xmag") {
  631. parser.read();
  632. camera.orthogonal.x_mag = parser.get_node_data().to_float();
  633. } else if (name == "ymag") {
  634. parser.read();
  635. camera.orthogonal.y_mag = parser.get_node_data().to_float();
  636. } else if (name == "aspect_ratio") {
  637. parser.read();
  638. camera.aspect = parser.get_node_data().to_float();
  639. } else if (name == "znear") {
  640. parser.read();
  641. camera.z_near = parser.get_node_data().to_float();
  642. } else if (name == "zfar") {
  643. parser.read();
  644. camera.z_far = parser.get_node_data().to_float();
  645. }
  646. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "camera") {
  647. break; //end of <asset>
  648. }
  649. }
  650. COLLADA_PRINT("Camera ID:" + id);
  651. }
  652. void Collada::_parse_light(XMLParser &parser) {
  653. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  654. if (!parser.is_empty()) {
  655. parser.skip_section();
  656. }
  657. return;
  658. }
  659. String id = parser.get_attribute_value("id");
  660. state.light_data_map[id] = LightData();
  661. LightData &light = state.light_data_map[id];
  662. while (parser.read() == OK) {
  663. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  664. String name = parser.get_node_name();
  665. if (name == "ambient") {
  666. light.mode = LightData::MODE_AMBIENT;
  667. } else if (name == "directional") {
  668. light.mode = LightData::MODE_DIRECTIONAL;
  669. } else if (name == "point") {
  670. light.mode = LightData::MODE_OMNI;
  671. } else if (name == "spot") {
  672. light.mode = LightData::MODE_SPOT;
  673. } else if (name == "color") {
  674. parser.read();
  675. Vector<float> colorarr = _read_float_array(parser);
  676. COLLADA_PRINT("colorarr size: " + rtos(colorarr.size()));
  677. if (colorarr.size() >= 4) {
  678. // alpha strangely not alright? maybe it needs to be multiplied by value as a channel intensity
  679. Color color(colorarr[0], colorarr[1], colorarr[2], 1.0);
  680. light.color = color;
  681. }
  682. } else if (name == "constant_attenuation") {
  683. parser.read();
  684. light.constant_att = parser.get_node_data().to_float();
  685. } else if (name == "linear_attenuation") {
  686. parser.read();
  687. light.linear_att = parser.get_node_data().to_float();
  688. } else if (name == "quadratic_attenuation") {
  689. parser.read();
  690. light.quad_att = parser.get_node_data().to_float();
  691. } else if (name == "falloff_angle") {
  692. parser.read();
  693. light.spot_angle = parser.get_node_data().to_float();
  694. } else if (name == "falloff_exponent") {
  695. parser.read();
  696. light.spot_exp = parser.get_node_data().to_float();
  697. }
  698. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "light") {
  699. break; //end of <asset>
  700. }
  701. }
  702. COLLADA_PRINT("Light ID:" + id);
  703. }
  704. void Collada::_parse_curve_geometry(XMLParser &parser, String p_id, String p_name) {
  705. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  706. if (!parser.is_empty()) {
  707. parser.skip_section();
  708. }
  709. return;
  710. }
  711. //load everything into a pre dictionary
  712. state.curve_data_map[p_id] = CurveData();
  713. CurveData &curvedata = state.curve_data_map[p_id];
  714. curvedata.name = p_name;
  715. COLLADA_PRINT("curve name: " + p_name);
  716. String current_source;
  717. // handles geometry node and the curve children in this loop
  718. // read sources with arrays and accessor for each curve
  719. if (parser.is_empty()) {
  720. return;
  721. }
  722. while (parser.read() == OK) {
  723. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  724. String section = parser.get_node_name();
  725. if (section == "source") {
  726. String id = parser.get_attribute_value("id");
  727. curvedata.sources[id] = CurveData::Source();
  728. current_source = id;
  729. COLLADA_PRINT("source data: " + id);
  730. } else if (section == "float_array" || section == "array") {
  731. // create a new array and read it.
  732. if (curvedata.sources.has(current_source)) {
  733. curvedata.sources[current_source].array = _read_float_array(parser);
  734. COLLADA_PRINT("section: " + current_source + " read " + itos(curvedata.sources[current_source].array.size()) + " values.");
  735. }
  736. } else if (section == "Name_array") {
  737. // create a new array and read it.
  738. if (curvedata.sources.has(current_source)) {
  739. curvedata.sources[current_source].sarray = _read_string_array(parser);
  740. COLLADA_PRINT("section: " + current_source + " read " + itos(curvedata.sources[current_source].array.size()) + " values.");
  741. }
  742. } else if (section == "technique_common") {
  743. //skip it
  744. } else if (section == "accessor") { // child of source (below a technique tag)
  745. if (curvedata.sources.has(current_source)) {
  746. curvedata.sources[current_source].stride = parser.get_attribute_value("stride").to_int();
  747. COLLADA_PRINT("section: " + current_source + " stride " + itos(curvedata.sources[current_source].stride));
  748. }
  749. } else if (section == "control_vertices") {
  750. while (parser.read() == OK) {
  751. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  752. if (parser.get_node_name() == "input") {
  753. String semantic = parser.get_attribute_value("semantic");
  754. String source = _uri_to_id(parser.get_attribute_value("source"));
  755. curvedata.control_vertices[semantic] = source;
  756. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
  757. }
  758. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  759. break;
  760. }
  761. }
  762. } else if (!parser.is_empty()) {
  763. parser.skip_section();
  764. }
  765. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "spline") {
  766. break;
  767. }
  768. }
  769. }
  770. void Collada::_parse_mesh_geometry(XMLParser &parser, String p_id, String p_name) {
  771. if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
  772. if (!parser.is_empty()) {
  773. parser.skip_section();
  774. }
  775. return;
  776. }
  777. //load everything into a pre dictionary
  778. state.mesh_data_map[p_id] = MeshData();
  779. MeshData &meshdata = state.mesh_data_map[p_id];
  780. meshdata.name = p_name;
  781. COLLADA_PRINT("mesh name: " + p_name);
  782. String current_source;
  783. // handles geometry node and the mesh children in this loop
  784. // read sources with arrays and accessor for each mesh
  785. if (parser.is_empty()) {
  786. return;
  787. }
  788. while (parser.read() == OK) {
  789. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  790. String section = parser.get_node_name();
  791. if (section == "source") {
  792. String id = parser.get_attribute_value("id");
  793. meshdata.sources[id] = MeshData::Source();
  794. current_source = id;
  795. COLLADA_PRINT("source data: " + id);
  796. } else if (section == "float_array" || section == "array") {
  797. // create a new array and read it.
  798. if (meshdata.sources.has(current_source)) {
  799. meshdata.sources[current_source].array = _read_float_array(parser);
  800. COLLADA_PRINT("section: " + current_source + " read " + itos(meshdata.sources[current_source].array.size()) + " values.");
  801. }
  802. } else if (section == "technique_common") {
  803. //skip it
  804. } else if (section == "accessor") { // child of source (below a technique tag)
  805. if (meshdata.sources.has(current_source)) {
  806. meshdata.sources[current_source].stride = parser.get_attribute_value("stride").to_int();
  807. COLLADA_PRINT("section: " + current_source + " stride " + itos(meshdata.sources[current_source].stride));
  808. }
  809. } else if (section == "vertices") {
  810. MeshData::Vertices vert;
  811. String id = parser.get_attribute_value("id");
  812. int last_ref = 0;
  813. while (parser.read() == OK) {
  814. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  815. if (parser.get_node_name() == "input") {
  816. String semantic = parser.get_attribute_value("semantic");
  817. String source = _uri_to_id(parser.get_attribute_value("source"));
  818. if (semantic == "TEXCOORD") {
  819. semantic = "TEXCOORD" + itos(last_ref++);
  820. }
  821. vert.sources[semantic] = source;
  822. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
  823. }
  824. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  825. break;
  826. }
  827. }
  828. meshdata.vertices[id] = vert;
  829. } else if (section == "triangles" || section == "polylist" || section == "polygons") {
  830. bool polygons = (section == "polygons");
  831. if (polygons) {
  832. WARN_PRINT("Primitive type \"polygons\" is not well supported (concave shapes may fail). To ensure that the geometry is properly imported, please re-export using \"triangles\" or \"polylist\".");
  833. }
  834. MeshData::Primitives prim;
  835. if (parser.has_attribute("material")) {
  836. prim.material = parser.get_attribute_value("material");
  837. }
  838. prim.count = parser.get_attribute_value("count").to_int();
  839. prim.vertex_size = 0;
  840. int last_ref = 0;
  841. while (parser.read() == OK) {
  842. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  843. if (parser.get_node_name() == "input") {
  844. String semantic = parser.get_attribute_value("semantic");
  845. String source = _uri_to_id(parser.get_attribute_value("source"));
  846. if (semantic == "TEXCOORD") {
  847. semantic = "TEXCOORD" + itos(last_ref++);
  848. }
  849. int offset = parser.get_attribute_value("offset").to_int();
  850. MeshData::Primitives::SourceRef sref;
  851. sref.source = source;
  852. sref.offset = offset;
  853. prim.sources[semantic] = sref;
  854. prim.vertex_size = MAX(prim.vertex_size, offset + 1);
  855. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source + " offset: " + itos(offset));
  856. } else if (parser.get_node_name() == "p") { //indices
  857. Vector<float> values = _read_float_array(parser);
  858. if (polygons) {
  859. ERR_CONTINUE(prim.vertex_size == 0);
  860. prim.polygons.push_back(values.size() / prim.vertex_size);
  861. int from = prim.indices.size();
  862. prim.indices.resize(from + values.size());
  863. for (int i = 0; i < values.size(); i++) {
  864. prim.indices.write[from + i] = values[i];
  865. }
  866. } else if (prim.vertex_size > 0) {
  867. prim.indices = values;
  868. }
  869. COLLADA_PRINT("read " + itos(values.size()) + " index values");
  870. } else if (parser.get_node_name() == "vcount") { // primitive
  871. Vector<float> values = _read_float_array(parser);
  872. prim.polygons = values;
  873. COLLADA_PRINT("read " + itos(values.size()) + " polygon values");
  874. }
  875. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  876. break;
  877. }
  878. }
  879. meshdata.primitives.push_back(prim);
  880. } else if (parser.get_node_name() == "double_sided") {
  881. parser.read();
  882. meshdata.found_double_sided = true;
  883. meshdata.double_sided = parser.get_node_data().to_int();
  884. } else if (parser.get_node_name() == "polygons") {
  885. ERR_PRINT("Primitive type \"polygons\" not supported, re-export using \"polylist\" or \"triangles\".");
  886. } else if (!parser.is_empty()) {
  887. parser.skip_section();
  888. }
  889. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "mesh") {
  890. break;
  891. }
  892. }
  893. }
  894. void Collada::_parse_skin_controller(XMLParser &parser, String p_id) {
  895. state.skin_controller_data_map[p_id] = SkinControllerData();
  896. SkinControllerData &skindata = state.skin_controller_data_map[p_id];
  897. skindata.base = _uri_to_id(parser.get_attribute_value("source"));
  898. String current_source;
  899. while (parser.read() == OK) {
  900. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  901. String section = parser.get_node_name();
  902. if (section == "bind_shape_matrix") {
  903. skindata.bind_shape = _read_transform(parser);
  904. #ifdef COLLADA_IMPORT_SCALE_SCENE
  905. skindata.bind_shape.origin *= state.unit_scale;
  906. #endif
  907. COLLADA_PRINT("skeleton bind shape transform: " + skindata.bind_shape);
  908. } else if (section == "source") {
  909. String id = parser.get_attribute_value("id");
  910. skindata.sources[id] = SkinControllerData::Source();
  911. current_source = id;
  912. COLLADA_PRINT("source data: " + id);
  913. } else if (section == "float_array" || section == "array") {
  914. // create a new array and read it.
  915. if (skindata.sources.has(current_source)) {
  916. skindata.sources[current_source].array = _read_float_array(parser);
  917. COLLADA_PRINT("section: " + current_source + " read " + itos(skindata.sources[current_source].array.size()) + " values.");
  918. }
  919. } else if (section == "Name_array" || section == "IDREF_array") {
  920. // create a new array and read it.
  921. if (section == "IDREF_array") {
  922. skindata.use_idrefs = true;
  923. }
  924. if (skindata.sources.has(current_source)) {
  925. skindata.sources[current_source].sarray = _read_string_array(parser);
  926. if (section == "IDREF_array") {
  927. Vector<String> sa = skindata.sources[current_source].sarray;
  928. for (int i = 0; i < sa.size(); i++) {
  929. state.idref_joints.insert(sa[i]);
  930. }
  931. }
  932. COLLADA_PRINT("section: " + current_source + " read " + itos(skindata.sources[current_source].array.size()) + " values.");
  933. }
  934. } else if (section == "technique_common") {
  935. //skip it
  936. } else if (section == "accessor") { // child of source (below a technique tag)
  937. if (skindata.sources.has(current_source)) {
  938. int stride = 1;
  939. if (parser.has_attribute("stride")) {
  940. stride = parser.get_attribute_value("stride").to_int();
  941. }
  942. skindata.sources[current_source].stride = stride;
  943. COLLADA_PRINT("section: " + current_source + " stride " + itos(skindata.sources[current_source].stride));
  944. }
  945. } else if (section == "joints") {
  946. SkinControllerData::Joints joint;
  947. while (parser.read() == OK) {
  948. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  949. if (parser.get_node_name() == "input") {
  950. String semantic = parser.get_attribute_value("semantic");
  951. String source = _uri_to_id(parser.get_attribute_value("source"));
  952. joint.sources[semantic] = source;
  953. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
  954. }
  955. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  956. break;
  957. }
  958. }
  959. skindata.joints = joint;
  960. } else if (section == "vertex_weights") {
  961. SkinControllerData::Weights weights;
  962. weights.count = parser.get_attribute_value("count").to_int();
  963. while (parser.read() == OK) {
  964. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  965. if (parser.get_node_name() == "input") {
  966. String semantic = parser.get_attribute_value("semantic");
  967. String source = _uri_to_id(parser.get_attribute_value("source"));
  968. int offset = parser.get_attribute_value("offset").to_int();
  969. SkinControllerData::Weights::SourceRef sref;
  970. sref.source = source;
  971. sref.offset = offset;
  972. weights.sources[semantic] = sref;
  973. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source + " offset: " + itos(offset));
  974. } else if (parser.get_node_name() == "v") { //indices
  975. Vector<float> values = _read_float_array(parser);
  976. weights.indices = values;
  977. COLLADA_PRINT("read " + itos(values.size()) + " index values");
  978. } else if (parser.get_node_name() == "vcount") { // weightsitive
  979. Vector<float> values = _read_float_array(parser);
  980. weights.sets = values;
  981. COLLADA_PRINT("read " + itos(values.size()) + " polygon values");
  982. }
  983. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  984. break;
  985. }
  986. }
  987. skindata.weights = weights;
  988. }
  989. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "skin") {
  990. break;
  991. }
  992. }
  993. /* STORE REST MATRICES */
  994. Vector<Transform3D> rests;
  995. ERR_FAIL_COND(!skindata.joints.sources.has("JOINT"));
  996. ERR_FAIL_COND(!skindata.joints.sources.has("INV_BIND_MATRIX"));
  997. String joint_arr = skindata.joints.sources["JOINT"];
  998. String ibm = skindata.joints.sources["INV_BIND_MATRIX"];
  999. ERR_FAIL_COND(!skindata.sources.has(joint_arr));
  1000. ERR_FAIL_COND(!skindata.sources.has(ibm));
  1001. SkinControllerData::Source &joint_source = skindata.sources[joint_arr];
  1002. SkinControllerData::Source &ibm_source = skindata.sources[ibm];
  1003. ERR_FAIL_COND(joint_source.sarray.size() != ibm_source.array.size() / 16);
  1004. for (int i = 0; i < joint_source.sarray.size(); i++) {
  1005. String name = joint_source.sarray[i];
  1006. Transform3D xform = _read_transform_from_array(ibm_source.array, i * 16); //<- this is a mistake, it must be applied to vertices
  1007. xform.affine_invert(); // inverse for rest, because it's an inverse
  1008. #ifdef COLLADA_IMPORT_SCALE_SCENE
  1009. xform.origin *= state.unit_scale;
  1010. #endif
  1011. skindata.bone_rest_map[name] = xform;
  1012. }
  1013. }
  1014. void Collada::_parse_morph_controller(XMLParser &parser, String p_id) {
  1015. state.morph_controller_data_map[p_id] = MorphControllerData();
  1016. MorphControllerData &morphdata = state.morph_controller_data_map[p_id];
  1017. morphdata.mesh = _uri_to_id(parser.get_attribute_value("source"));
  1018. morphdata.mode = parser.get_attribute_value("method");
  1019. String current_source;
  1020. while (parser.read() == OK) {
  1021. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1022. String section = parser.get_node_name();
  1023. if (section == "source") {
  1024. String id = parser.get_attribute_value("id");
  1025. morphdata.sources[id] = MorphControllerData::Source();
  1026. current_source = id;
  1027. COLLADA_PRINT("source data: " + id);
  1028. } else if (section == "float_array" || section == "array") {
  1029. // create a new array and read it.
  1030. if (morphdata.sources.has(current_source)) {
  1031. morphdata.sources[current_source].array = _read_float_array(parser);
  1032. COLLADA_PRINT("section: " + current_source + " read " + itos(morphdata.sources[current_source].array.size()) + " values.");
  1033. }
  1034. } else if (section == "Name_array" || section == "IDREF_array") {
  1035. // create a new array and read it.
  1036. if (morphdata.sources.has(current_source)) {
  1037. morphdata.sources[current_source].sarray = _read_string_array(parser);
  1038. COLLADA_PRINT("section: " + current_source + " read " + itos(morphdata.sources[current_source].array.size()) + " values.");
  1039. }
  1040. } else if (section == "technique_common") {
  1041. //skip it
  1042. } else if (section == "accessor") { // child of source (below a technique tag)
  1043. if (morphdata.sources.has(current_source)) {
  1044. int stride = 1;
  1045. if (parser.has_attribute("stride")) {
  1046. stride = parser.get_attribute_value("stride").to_int();
  1047. }
  1048. morphdata.sources[current_source].stride = stride;
  1049. COLLADA_PRINT("section: " + current_source + " stride " + itos(morphdata.sources[current_source].stride));
  1050. }
  1051. } else if (section == "targets") {
  1052. while (parser.read() == OK) {
  1053. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1054. if (parser.get_node_name() == "input") {
  1055. String semantic = parser.get_attribute_value("semantic");
  1056. String source = _uri_to_id(parser.get_attribute_value("source"));
  1057. morphdata.targets[semantic] = source;
  1058. COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
  1059. }
  1060. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
  1061. break;
  1062. }
  1063. }
  1064. }
  1065. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "morph") {
  1066. break;
  1067. }
  1068. }
  1069. if (morphdata.targets.has("MORPH_WEIGHT")) {
  1070. state.morph_name_map[morphdata.targets["MORPH_WEIGHT"]] = p_id;
  1071. }
  1072. }
  1073. void Collada::_parse_controller(XMLParser &parser) {
  1074. String id = parser.get_attribute_value("id");
  1075. if (parser.is_empty()) {
  1076. return;
  1077. }
  1078. while (parser.read() == OK) {
  1079. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1080. String section = parser.get_node_name();
  1081. if (section == "skin") {
  1082. _parse_skin_controller(parser, id);
  1083. } else if (section == "morph") {
  1084. _parse_morph_controller(parser, id);
  1085. }
  1086. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "controller") {
  1087. break;
  1088. }
  1089. }
  1090. }
  1091. Collada::Node *Collada::_parse_visual_instance_geometry(XMLParser &parser) {
  1092. String type = parser.get_node_name();
  1093. NodeGeometry *geom = memnew(NodeGeometry);
  1094. geom->controller = type == "instance_controller";
  1095. geom->source = _uri_to_id(parser.get_attribute_value_safe("url"));
  1096. if (parser.is_empty()) { //nothing else to parse...
  1097. return geom;
  1098. }
  1099. // try to find also many materials and skeletons!
  1100. while (parser.read() == OK) {
  1101. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1102. if (parser.get_node_name() == "instance_material") {
  1103. String symbol = parser.get_attribute_value("symbol");
  1104. String target = _uri_to_id(parser.get_attribute_value("target"));
  1105. NodeGeometry::Material mat;
  1106. mat.target = target;
  1107. geom->material_map[symbol] = mat;
  1108. COLLADA_PRINT("uses material: '" + target + "' on primitive'" + symbol + "'");
  1109. } else if (parser.get_node_name() == "skeleton") {
  1110. parser.read();
  1111. String uri = _uri_to_id(parser.get_node_data());
  1112. if (!uri.is_empty()) {
  1113. geom->skeletons.push_back(uri);
  1114. }
  1115. }
  1116. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == type) {
  1117. break;
  1118. }
  1119. }
  1120. if (geom->controller) {
  1121. if (geom->skeletons.is_empty()) {
  1122. //XSI style
  1123. if (state.skin_controller_data_map.has(geom->source)) {
  1124. SkinControllerData *skin = &state.skin_controller_data_map[geom->source];
  1125. //case where skeletons reference bones with IDREF (XSI)
  1126. ERR_FAIL_COND_V(!skin->joints.sources.has("JOINT"), geom);
  1127. String joint_arr = skin->joints.sources["JOINT"];
  1128. ERR_FAIL_COND_V(!skin->sources.has(joint_arr), geom);
  1129. Collada::SkinControllerData::Source &joint_source = skin->sources[joint_arr];
  1130. geom->skeletons = joint_source.sarray; //quite crazy, but should work.
  1131. }
  1132. }
  1133. }
  1134. return geom;
  1135. }
  1136. Collada::Node *Collada::_parse_visual_instance_camera(XMLParser &parser) {
  1137. NodeCamera *cam = memnew(NodeCamera);
  1138. cam->camera = _uri_to_id(parser.get_attribute_value_safe("url"));
  1139. if (state.up_axis == Vector3::AXIS_Z) { //collada weirdness
  1140. cam->post_transform.basis.rotate(Vector3(1, 0, 0), -Math_PI * 0.5);
  1141. }
  1142. if (parser.is_empty()) { //nothing else to parse...
  1143. return cam;
  1144. }
  1145. while (parser.read() == OK) {
  1146. if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "instance_camera") {
  1147. break;
  1148. }
  1149. }
  1150. return cam;
  1151. }
  1152. Collada::Node *Collada::_parse_visual_instance_light(XMLParser &parser) {
  1153. NodeLight *cam = memnew(NodeLight);
  1154. cam->light = _uri_to_id(parser.get_attribute_value_safe("url"));
  1155. if (state.up_axis == Vector3::AXIS_Z) { //collada weirdness
  1156. cam->post_transform.basis.rotate(Vector3(1, 0, 0), -Math_PI * 0.5);
  1157. }
  1158. if (parser.is_empty()) { //nothing else to parse...
  1159. return cam;
  1160. }
  1161. while (parser.read() == OK) {
  1162. if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "instance_light") {
  1163. break;
  1164. }
  1165. }
  1166. return cam;
  1167. }
  1168. Collada::Node *Collada::_parse_visual_node_instance_data(XMLParser &parser) {
  1169. String instance_type = parser.get_node_name();
  1170. if (instance_type == "instance_geometry" || instance_type == "instance_controller") {
  1171. return _parse_visual_instance_geometry(parser);
  1172. } else if (instance_type == "instance_camera") {
  1173. return _parse_visual_instance_camera(parser);
  1174. } else if (instance_type == "instance_light") {
  1175. return _parse_visual_instance_light(parser);
  1176. }
  1177. if (parser.is_empty()) { //nothing else to parse...
  1178. return nullptr;
  1179. }
  1180. while (parser.read() == OK) {
  1181. if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == instance_type) {
  1182. break;
  1183. }
  1184. }
  1185. return nullptr;
  1186. }
  1187. Collada::Node *Collada::_parse_visual_scene_node(XMLParser &parser) {
  1188. String name;
  1189. String id = parser.get_attribute_value_safe("id");
  1190. bool found_name = false;
  1191. if (id.is_empty()) {
  1192. id = "%NODEID%" + itos(Math::rand());
  1193. } else {
  1194. found_name = true;
  1195. }
  1196. Vector<Node::XForm> xform_list;
  1197. Vector<Node *> children;
  1198. String empty_draw_type = "";
  1199. Node *node = nullptr;
  1200. name = parser.has_attribute("name") ? parser.get_attribute_value_safe("name") : parser.get_attribute_value_safe("id");
  1201. if (name.is_empty()) {
  1202. name = id;
  1203. } else {
  1204. found_name = true;
  1205. }
  1206. if ((parser.has_attribute("type") && parser.get_attribute_value("type") == "JOINT") || state.idref_joints.has(name)) {
  1207. // handle a bone
  1208. NodeJoint *joint = memnew(NodeJoint);
  1209. if (parser.has_attribute("sid")) { //bones may not have sid
  1210. joint->sid = parser.get_attribute_value("sid");
  1211. //state.bone_map[joint->sid]=joint;
  1212. } else if (state.idref_joints.has(name)) {
  1213. joint->sid = name; //kind of a cheat but..
  1214. } else if (parser.has_attribute("name")) {
  1215. joint->sid = parser.get_attribute_value_safe("name");
  1216. }
  1217. if (!joint->sid.is_empty()) {
  1218. state.sid_to_node_map[joint->sid] = id;
  1219. }
  1220. node = joint;
  1221. }
  1222. while (parser.read() == OK) {
  1223. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1224. String section = parser.get_node_name();
  1225. if (section == "translate") {
  1226. Node::XForm xf;
  1227. if (parser.has_attribute("sid")) {
  1228. xf.id = parser.get_attribute_value("sid");
  1229. }
  1230. xf.op = Node::XForm::OP_TRANSLATE;
  1231. Vector<float> xlt = _read_float_array(parser);
  1232. xf.data = xlt;
  1233. xform_list.push_back(xf);
  1234. } else if (section == "rotate") {
  1235. Node::XForm xf;
  1236. if (parser.has_attribute("sid")) {
  1237. xf.id = parser.get_attribute_value("sid");
  1238. }
  1239. xf.op = Node::XForm::OP_ROTATE;
  1240. Vector<float> rot = _read_float_array(parser);
  1241. xf.data = rot;
  1242. xform_list.push_back(xf);
  1243. } else if (section == "scale") {
  1244. Node::XForm xf;
  1245. if (parser.has_attribute("sid")) {
  1246. xf.id = parser.get_attribute_value("sid");
  1247. }
  1248. xf.op = Node::XForm::OP_SCALE;
  1249. Vector<float> scale = _read_float_array(parser);
  1250. xf.data = scale;
  1251. xform_list.push_back(xf);
  1252. } else if (section == "matrix") {
  1253. Node::XForm xf;
  1254. if (parser.has_attribute("sid")) {
  1255. xf.id = parser.get_attribute_value("sid");
  1256. }
  1257. xf.op = Node::XForm::OP_MATRIX;
  1258. Vector<float> matrix = _read_float_array(parser);
  1259. xf.data = matrix;
  1260. String mtx;
  1261. for (int i = 0; i < matrix.size(); i++) {
  1262. mtx += " " + rtos(matrix[i]);
  1263. }
  1264. xform_list.push_back(xf);
  1265. } else if (section == "visibility") {
  1266. Node::XForm xf;
  1267. if (parser.has_attribute("sid")) {
  1268. xf.id = parser.get_attribute_value("sid");
  1269. }
  1270. xf.op = Node::XForm::OP_VISIBILITY;
  1271. Vector<float> visible = _read_float_array(parser);
  1272. xf.data = visible;
  1273. xform_list.push_back(xf);
  1274. } else if (section == "empty_draw_type") {
  1275. empty_draw_type = _read_empty_draw_type(parser);
  1276. } else if (section == "technique" || section == "extra") {
  1277. } else if (section != "node") {
  1278. //usually what defines the type of node
  1279. if (section.begins_with("instance_")) {
  1280. if (!node) {
  1281. node = _parse_visual_node_instance_data(parser);
  1282. } else {
  1283. ERR_PRINT("Multiple instance_* not supported.");
  1284. }
  1285. }
  1286. } else {
  1287. /* Found a child node!! what to do..*/
  1288. Node *child = _parse_visual_scene_node(parser);
  1289. children.push_back(child);
  1290. }
  1291. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "node") {
  1292. break;
  1293. }
  1294. }
  1295. if (!node) {
  1296. node = memnew(Node); //generic node, nothing of relevance found
  1297. }
  1298. node->noname = !found_name;
  1299. node->xform_list = xform_list;
  1300. node->children = children;
  1301. for (int i = 0; i < children.size(); i++) {
  1302. node->children[i]->parent = node;
  1303. }
  1304. node->name = name;
  1305. node->id = id;
  1306. node->empty_draw_type = empty_draw_type;
  1307. if (node->children.size() == 1) {
  1308. if (node->children[0]->noname && !node->noname) {
  1309. node->children[0]->name = node->name;
  1310. node->name = node->name + "-base";
  1311. }
  1312. }
  1313. node->default_transform = node->compute_transform(*this);
  1314. state.scene_map[id] = node;
  1315. return node;
  1316. }
  1317. void Collada::_parse_visual_scene(XMLParser &parser) {
  1318. String id = parser.get_attribute_value("id");
  1319. if (parser.is_empty()) {
  1320. return;
  1321. }
  1322. state.visual_scene_map[id] = VisualScene();
  1323. VisualScene &vscene = state.visual_scene_map[id];
  1324. if (parser.has_attribute("name")) {
  1325. vscene.name = parser.get_attribute_value("name");
  1326. }
  1327. while (parser.read() == OK) {
  1328. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1329. String section = parser.get_node_name();
  1330. if (section == "node") {
  1331. vscene.root_nodes.push_back(_parse_visual_scene_node(parser));
  1332. }
  1333. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "visual_scene") {
  1334. break;
  1335. }
  1336. }
  1337. COLLADA_PRINT("Scene ID:" + id);
  1338. }
  1339. void Collada::_parse_animation(XMLParser &parser) {
  1340. if (!(state.import_flags & IMPORT_FLAG_ANIMATION)) {
  1341. if (!parser.is_empty()) {
  1342. parser.skip_section();
  1343. }
  1344. return;
  1345. }
  1346. HashMap<String, Vector<float>> float_sources;
  1347. HashMap<String, Vector<String>> string_sources;
  1348. HashMap<String, int> source_strides;
  1349. HashMap<String, HashMap<String, String>> samplers;
  1350. HashMap<String, Vector<String>> source_param_names;
  1351. HashMap<String, Vector<String>> source_param_types;
  1352. String id = "";
  1353. if (parser.has_attribute("id")) {
  1354. id = parser.get_attribute_value("id");
  1355. }
  1356. String current_source;
  1357. String current_sampler;
  1358. Vector<String> channel_sources;
  1359. Vector<String> channel_targets;
  1360. while (parser.read() == OK) {
  1361. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1362. String name = parser.get_node_name();
  1363. if (name == "source") {
  1364. current_source = parser.get_attribute_value("id");
  1365. source_param_names[current_source] = Vector<String>();
  1366. source_param_types[current_source] = Vector<String>();
  1367. } else if (name == "float_array") {
  1368. if (!current_source.is_empty()) {
  1369. float_sources[current_source] = _read_float_array(parser);
  1370. }
  1371. } else if (name == "Name_array") {
  1372. if (!current_source.is_empty()) {
  1373. string_sources[current_source] = _read_string_array(parser);
  1374. }
  1375. } else if (name == "accessor") {
  1376. if (!current_source.is_empty() && parser.has_attribute("stride")) {
  1377. source_strides[current_source] = parser.get_attribute_value("stride").to_int();
  1378. }
  1379. } else if (name == "sampler") {
  1380. current_sampler = parser.get_attribute_value("id");
  1381. samplers[current_sampler] = HashMap<String, String>();
  1382. } else if (name == "param") {
  1383. if (parser.has_attribute("name")) {
  1384. source_param_names[current_source].push_back(parser.get_attribute_value("name"));
  1385. } else {
  1386. source_param_names[current_source].push_back("");
  1387. }
  1388. if (parser.has_attribute("type")) {
  1389. source_param_types[current_source].push_back(parser.get_attribute_value("type"));
  1390. } else {
  1391. source_param_types[current_source].push_back("");
  1392. }
  1393. } else if (name == "input") {
  1394. if (!current_sampler.is_empty()) {
  1395. samplers[current_sampler][parser.get_attribute_value("semantic")] = parser.get_attribute_value("source");
  1396. }
  1397. } else if (name == "channel") {
  1398. channel_sources.push_back(parser.get_attribute_value("source"));
  1399. channel_targets.push_back(parser.get_attribute_value("target"));
  1400. }
  1401. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "animation") {
  1402. break; //end of <asset>
  1403. }
  1404. }
  1405. for (int i = 0; i < channel_sources.size(); i++) {
  1406. String source = _uri_to_id(channel_sources[i]);
  1407. String target = channel_targets[i];
  1408. ERR_CONTINUE(!samplers.has(source));
  1409. HashMap<String, String> &sampler = samplers[source];
  1410. ERR_CONTINUE(!sampler.has("INPUT")); //no input semantic? wtf?
  1411. String input_id = _uri_to_id(sampler["INPUT"]);
  1412. COLLADA_PRINT("input id is " + input_id);
  1413. ERR_CONTINUE(!float_sources.has(input_id));
  1414. ERR_CONTINUE(!sampler.has("OUTPUT"));
  1415. String output_id = _uri_to_id(sampler["OUTPUT"]);
  1416. ERR_CONTINUE(!float_sources.has(output_id));
  1417. ERR_CONTINUE(!source_param_names.has(output_id));
  1418. Vector<String> &names = source_param_names[output_id];
  1419. for (int l = 0; l < names.size(); l++) {
  1420. String name = names[l];
  1421. Vector<float> &time_keys = float_sources[input_id];
  1422. int key_count = time_keys.size();
  1423. AnimationTrack track; //begin crating track
  1424. track.id = id;
  1425. track.keys.resize(key_count);
  1426. for (int j = 0; j < key_count; j++) {
  1427. track.keys.write[j].time = time_keys[j];
  1428. state.animation_length = MAX(state.animation_length, time_keys[j]);
  1429. }
  1430. //now read actual values
  1431. int stride = 1;
  1432. if (source_strides.has(output_id)) {
  1433. stride = source_strides[output_id];
  1434. }
  1435. int output_len = stride / names.size();
  1436. ERR_CONTINUE(output_len == 0);
  1437. ERR_CONTINUE(!float_sources.has(output_id));
  1438. Vector<float> &output = float_sources[output_id];
  1439. ERR_CONTINUE_MSG((output.size() / stride) != key_count, "Wrong number of keys in output.");
  1440. for (int j = 0; j < key_count; j++) {
  1441. track.keys.write[j].data.resize(output_len);
  1442. for (int k = 0; k < output_len; k++) {
  1443. track.keys.write[j].data.write[k] = output[l + j * stride + k]; //super weird but should work:
  1444. }
  1445. }
  1446. if (sampler.has("INTERPOLATION")) {
  1447. String interp_id = _uri_to_id(sampler["INTERPOLATION"]);
  1448. ERR_CONTINUE(!string_sources.has(interp_id));
  1449. Vector<String> &interps = string_sources[interp_id];
  1450. ERR_CONTINUE(interps.size() != key_count);
  1451. for (int j = 0; j < key_count; j++) {
  1452. if (interps[j] == "BEZIER") {
  1453. track.keys.write[j].interp_type = AnimationTrack::INTERP_BEZIER;
  1454. } else {
  1455. track.keys.write[j].interp_type = AnimationTrack::INTERP_LINEAR;
  1456. }
  1457. }
  1458. }
  1459. if (sampler.has("IN_TANGENT") && sampler.has("OUT_TANGENT")) {
  1460. //bezier control points..
  1461. String intangent_id = _uri_to_id(sampler["IN_TANGENT"]);
  1462. ERR_CONTINUE(!float_sources.has(intangent_id));
  1463. Vector<float> &intangents = float_sources[intangent_id];
  1464. ERR_CONTINUE(intangents.size() != key_count * 2 * names.size());
  1465. String outangent_id = _uri_to_id(sampler["OUT_TANGENT"]);
  1466. ERR_CONTINUE(!float_sources.has(outangent_id));
  1467. Vector<float> &outangents = float_sources[outangent_id];
  1468. ERR_CONTINUE(outangents.size() != key_count * 2 * names.size());
  1469. for (int j = 0; j < key_count; j++) {
  1470. track.keys.write[j].in_tangent = Vector2(intangents[j * 2 * names.size() + 0 + l * 2], intangents[j * 2 * names.size() + 1 + l * 2]);
  1471. track.keys.write[j].out_tangent = Vector2(outangents[j * 2 * names.size() + 0 + l * 2], outangents[j * 2 * names.size() + 1 + l * 2]);
  1472. }
  1473. }
  1474. if (target.contains("/")) { //transform component
  1475. track.target = target.get_slicec('/', 0);
  1476. track.param = target.get_slicec('/', 1);
  1477. if (track.param.contains(".")) {
  1478. track.component = track.param.get_slice(".", 1).to_upper();
  1479. }
  1480. track.param = track.param.get_slice(".", 0);
  1481. if (names.size() > 1 && track.component.is_empty()) {
  1482. //this is a guess because the collada spec is ambiguous here...
  1483. //i suppose if you have many names (outputs) you can't use a component and i should abide to that.
  1484. track.component = name;
  1485. }
  1486. } else {
  1487. track.target = target;
  1488. }
  1489. state.animation_tracks.push_back(track);
  1490. if (!state.referenced_tracks.has(target)) {
  1491. state.referenced_tracks[target] = Vector<int>();
  1492. }
  1493. state.referenced_tracks[target].push_back(state.animation_tracks.size() - 1);
  1494. if (!id.is_empty()) {
  1495. if (!state.by_id_tracks.has(id)) {
  1496. state.by_id_tracks[id] = Vector<int>();
  1497. }
  1498. state.by_id_tracks[id].push_back(state.animation_tracks.size() - 1);
  1499. }
  1500. COLLADA_PRINT("loaded animation with " + itos(key_count) + " keys");
  1501. }
  1502. }
  1503. }
  1504. void Collada::_parse_animation_clip(XMLParser &parser) {
  1505. if (!(state.import_flags & IMPORT_FLAG_ANIMATION)) {
  1506. if (!parser.is_empty()) {
  1507. parser.skip_section();
  1508. }
  1509. return;
  1510. }
  1511. AnimationClip clip;
  1512. if (parser.has_attribute("name")) {
  1513. clip.name = parser.get_attribute_value("name");
  1514. } else if (parser.has_attribute("id")) {
  1515. clip.name = parser.get_attribute_value("id");
  1516. }
  1517. if (parser.has_attribute("start")) {
  1518. clip.begin = parser.get_attribute_value("start").to_float();
  1519. }
  1520. if (parser.has_attribute("end")) {
  1521. clip.end = parser.get_attribute_value("end").to_float();
  1522. }
  1523. while (parser.read() == OK) {
  1524. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1525. String name = parser.get_node_name();
  1526. if (name == "instance_animation") {
  1527. String url = _uri_to_id(parser.get_attribute_value("url"));
  1528. clip.tracks.push_back(url);
  1529. }
  1530. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "animation_clip") {
  1531. break; //end of <asset>
  1532. }
  1533. }
  1534. state.animation_clips.push_back(clip);
  1535. }
  1536. void Collada::_parse_scene(XMLParser &parser) {
  1537. if (parser.is_empty()) {
  1538. return;
  1539. }
  1540. while (parser.read() == OK) {
  1541. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1542. String name = parser.get_node_name();
  1543. if (name == "instance_visual_scene") {
  1544. state.root_visual_scene = _uri_to_id(parser.get_attribute_value("url"));
  1545. } else if (name == "instance_physics_scene") {
  1546. state.root_physics_scene = _uri_to_id(parser.get_attribute_value("url"));
  1547. }
  1548. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "scene") {
  1549. break; //end of <asset>
  1550. }
  1551. }
  1552. }
  1553. void Collada::_parse_library(XMLParser &parser) {
  1554. if (parser.is_empty()) {
  1555. return;
  1556. }
  1557. while (parser.read() == OK) {
  1558. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1559. String name = parser.get_node_name();
  1560. COLLADA_PRINT("library name is: " + name);
  1561. if (name == "image") {
  1562. _parse_image(parser);
  1563. } else if (name == "material") {
  1564. _parse_material(parser);
  1565. } else if (name == "effect") {
  1566. _parse_effect(parser);
  1567. } else if (name == "camera") {
  1568. _parse_camera(parser);
  1569. } else if (name == "light") {
  1570. _parse_light(parser);
  1571. } else if (name == "geometry") {
  1572. String id = parser.get_attribute_value("id");
  1573. String name2 = parser.get_attribute_value_safe("name");
  1574. while (parser.read() == OK) {
  1575. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1576. if (parser.get_node_name() == "mesh") {
  1577. state.mesh_name_map[id] = (!name2.is_empty()) ? name2 : id;
  1578. _parse_mesh_geometry(parser, id, name2);
  1579. } else if (parser.get_node_name() == "spline") {
  1580. state.mesh_name_map[id] = (!name2.is_empty()) ? name2 : id;
  1581. _parse_curve_geometry(parser, id, name2);
  1582. } else if (!parser.is_empty()) {
  1583. parser.skip_section();
  1584. }
  1585. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "geometry") {
  1586. break;
  1587. }
  1588. }
  1589. } else if (name == "controller") {
  1590. _parse_controller(parser);
  1591. } else if (name == "animation") {
  1592. _parse_animation(parser);
  1593. } else if (name == "animation_clip") {
  1594. _parse_animation_clip(parser);
  1595. } else if (name == "visual_scene") {
  1596. COLLADA_PRINT("visual scene");
  1597. _parse_visual_scene(parser);
  1598. } else if (!parser.is_empty()) {
  1599. parser.skip_section();
  1600. }
  1601. } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name().begins_with("library_")) {
  1602. break; //end of <asset>
  1603. }
  1604. }
  1605. }
  1606. void Collada::_joint_set_owner(Collada::Node *p_node, NodeSkeleton *p_owner) {
  1607. if (p_node->type == Node::TYPE_JOINT) {
  1608. NodeJoint *nj = static_cast<NodeJoint *>(p_node);
  1609. nj->owner = p_owner;
  1610. for (int i = 0; i < nj->children.size(); i++) {
  1611. _joint_set_owner(nj->children.write[i], p_owner);
  1612. }
  1613. }
  1614. }
  1615. void Collada::_create_skeletons(Collada::Node **p_node, NodeSkeleton *p_skeleton) {
  1616. Node *node = *p_node;
  1617. if (node->type == Node::TYPE_JOINT) {
  1618. if (!p_skeleton) {
  1619. // ohohohoohoo it's a joint node, time to work!
  1620. NodeSkeleton *sk = memnew(NodeSkeleton);
  1621. *p_node = sk;
  1622. sk->children.push_back(node);
  1623. sk->parent = node->parent;
  1624. node->parent = sk;
  1625. p_skeleton = sk;
  1626. }
  1627. NodeJoint *nj = static_cast<NodeJoint *>(node);
  1628. nj->owner = p_skeleton;
  1629. } else {
  1630. p_skeleton = nullptr;
  1631. }
  1632. for (int i = 0; i < node->children.size(); i++) {
  1633. _create_skeletons(&node->children.write[i], p_skeleton);
  1634. }
  1635. }
  1636. bool Collada::_remove_node(Node *p_parent, Node *p_node) {
  1637. for (int i = 0; i < p_parent->children.size(); i++) {
  1638. if (p_parent->children[i] == p_node) {
  1639. p_parent->children.remove_at(i);
  1640. return true;
  1641. }
  1642. if (_remove_node(p_parent->children[i], p_node)) {
  1643. return true;
  1644. }
  1645. }
  1646. return false;
  1647. }
  1648. void Collada::_remove_node(VisualScene *p_vscene, Node *p_node) {
  1649. for (int i = 0; i < p_vscene->root_nodes.size(); i++) {
  1650. if (p_vscene->root_nodes[i] == p_node) {
  1651. p_vscene->root_nodes.remove_at(i);
  1652. return;
  1653. }
  1654. if (_remove_node(p_vscene->root_nodes[i], p_node)) {
  1655. return;
  1656. }
  1657. }
  1658. ERR_PRINT("ERROR: Not found node to remove?");
  1659. }
  1660. void Collada::_merge_skeletons(VisualScene *p_vscene, Node *p_node) {
  1661. if (p_node->type == Node::TYPE_GEOMETRY) {
  1662. NodeGeometry *gnode = static_cast<NodeGeometry *>(p_node);
  1663. if (gnode->controller) {
  1664. // recount skeletons used
  1665. HashSet<NodeSkeleton *> skeletons;
  1666. for (int i = 0; i < gnode->skeletons.size(); i++) {
  1667. String nodeid = gnode->skeletons[i];
  1668. ERR_CONTINUE(!state.scene_map.has(nodeid)); //weird, it should have it...
  1669. #ifdef NO_SAFE_CAST
  1670. NodeJoint *nj = static_cast<NodeJoint *>(state.scene_map[nodeid]);
  1671. #else
  1672. NodeJoint *nj = dynamic_cast<NodeJoint *>(state.scene_map[nodeid]);
  1673. #endif
  1674. ERR_CONTINUE(!nj); //broken collada
  1675. ERR_CONTINUE(!nj->owner); //weird, node should have a skeleton owner
  1676. skeletons.insert(nj->owner);
  1677. }
  1678. if (skeletons.size() > 1) {
  1679. //do the merger!!
  1680. HashSet<NodeSkeleton *>::Iterator E = skeletons.begin();
  1681. NodeSkeleton *base = *E;
  1682. for (++E; E; ++E) {
  1683. NodeSkeleton *merged = *E;
  1684. _remove_node(p_vscene, merged);
  1685. for (int i = 0; i < merged->children.size(); i++) {
  1686. _joint_set_owner(merged->children[i], base);
  1687. base->children.push_back(merged->children[i]);
  1688. merged->children[i]->parent = base;
  1689. }
  1690. merged->children.clear(); //take children from it
  1691. memdelete(merged);
  1692. }
  1693. }
  1694. }
  1695. }
  1696. for (int i = 0; i < p_node->children.size(); i++) {
  1697. _merge_skeletons(p_vscene, p_node->children[i]);
  1698. }
  1699. }
  1700. void Collada::_merge_skeletons2(VisualScene *p_vscene) {
  1701. for (KeyValue<String, SkinControllerData> &E : state.skin_controller_data_map) {
  1702. SkinControllerData &cd = E.value;
  1703. NodeSkeleton *skeleton = nullptr;
  1704. for (const KeyValue<String, Transform3D> &F : cd.bone_rest_map) {
  1705. String name;
  1706. if (!state.sid_to_node_map.has(F.key)) {
  1707. continue;
  1708. }
  1709. name = state.sid_to_node_map[F.key];
  1710. ERR_CONTINUE(!state.scene_map.has(name));
  1711. Node *node = state.scene_map[name];
  1712. ERR_CONTINUE(node->type != Node::TYPE_JOINT);
  1713. NodeSkeleton *sk = nullptr;
  1714. while (node && !sk) {
  1715. if (node->type == Node::TYPE_SKELETON) {
  1716. sk = static_cast<NodeSkeleton *>(node);
  1717. }
  1718. node = node->parent;
  1719. }
  1720. ERR_CONTINUE(!sk);
  1721. if (!skeleton) {
  1722. skeleton = sk;
  1723. continue;
  1724. }
  1725. if (skeleton != sk) {
  1726. //whoa.. wtf, merge.
  1727. _remove_node(p_vscene, sk);
  1728. for (int i = 0; i < sk->children.size(); i++) {
  1729. _joint_set_owner(sk->children[i], skeleton);
  1730. skeleton->children.push_back(sk->children[i]);
  1731. sk->children[i]->parent = skeleton;
  1732. }
  1733. sk->children.clear(); //take children from it
  1734. memdelete(sk);
  1735. }
  1736. }
  1737. }
  1738. }
  1739. bool Collada::_optimize_skeletons(VisualScene *p_vscene, Node *p_node) {
  1740. Node *node = p_node;
  1741. if (node->type == Node::TYPE_SKELETON && node->parent && node->parent->type == Node::TYPE_NODE && node->parent->children.size() == 1) {
  1742. //replace parent by this...
  1743. Node *parent = node->parent;
  1744. //i wonder if this is alright.. i think it is since created skeleton (first joint) is already animated by bone..
  1745. node->id = parent->id;
  1746. node->name = parent->name;
  1747. node->xform_list = parent->xform_list;
  1748. node->default_transform = parent->default_transform;
  1749. state.scene_map[node->id] = node;
  1750. node->parent = parent->parent;
  1751. if (parent->parent) {
  1752. Node *gp = parent->parent;
  1753. bool found = false;
  1754. for (int i = 0; i < gp->children.size(); i++) {
  1755. if (gp->children[i] == parent) {
  1756. gp->children.write[i] = node;
  1757. found = true;
  1758. break;
  1759. }
  1760. }
  1761. if (!found) {
  1762. ERR_PRINT("BUG");
  1763. }
  1764. } else {
  1765. bool found = false;
  1766. for (int i = 0; i < p_vscene->root_nodes.size(); i++) {
  1767. if (p_vscene->root_nodes[i] == parent) {
  1768. p_vscene->root_nodes.write[i] = node;
  1769. found = true;
  1770. break;
  1771. }
  1772. }
  1773. if (!found) {
  1774. ERR_PRINT("BUG");
  1775. }
  1776. }
  1777. parent->children.clear();
  1778. memdelete(parent);
  1779. return true;
  1780. }
  1781. for (int i = 0; i < node->children.size(); i++) {
  1782. if (_optimize_skeletons(p_vscene, node->children[i])) {
  1783. return false; //stop processing, go up
  1784. }
  1785. }
  1786. return false;
  1787. }
  1788. bool Collada::_move_geometry_to_skeletons(VisualScene *p_vscene, Node *p_node, List<Node *> *p_mgeom) {
  1789. // Bind Shape Matrix scales the bones and makes them gigantic, so the matrix then shrinks the model?
  1790. // Solution: apply the Bind Shape Matrix to the VERTICES, and if the object comes scaled, it seems to be left alone!
  1791. if (p_node->type == Node::TYPE_GEOMETRY) {
  1792. NodeGeometry *ng = static_cast<NodeGeometry *>(p_node);
  1793. if (ng->ignore_anim) {
  1794. return false; //already made child of skeleton and processeg
  1795. }
  1796. if (ng->controller && ng->skeletons.size()) {
  1797. String nodeid = ng->skeletons[0];
  1798. ERR_FAIL_COND_V(!state.scene_map.has(nodeid), false); //weird, it should have it...
  1799. #ifdef NO_SAFE_CAST
  1800. NodeJoint *nj = static_cast<NodeJoint *>(state.scene_map[nodeid]);
  1801. #else
  1802. NodeJoint *nj = dynamic_cast<NodeJoint *>(state.scene_map[nodeid]);
  1803. #endif
  1804. ERR_FAIL_COND_V(!nj, false);
  1805. ERR_FAIL_COND_V(!nj->owner, false); //weird, node should have a skeleton owner
  1806. NodeSkeleton *sk = nj->owner;
  1807. Node *p = sk->parent;
  1808. bool node_is_parent_of_skeleton = false;
  1809. while (p) {
  1810. if (p == p_node) {
  1811. node_is_parent_of_skeleton = true;
  1812. break;
  1813. }
  1814. p = p->parent; // try again
  1815. }
  1816. ERR_FAIL_COND_V(node_is_parent_of_skeleton, false);
  1817. //this should be correct
  1818. ERR_FAIL_COND_V(!state.skin_controller_data_map.has(ng->source), false);
  1819. SkinControllerData &skin = state.skin_controller_data_map[ng->source];
  1820. Transform3D skel_inv = sk->get_global_transform().affine_inverse();
  1821. p_node->default_transform = skel_inv * (skin.bind_shape /* p_node->get_global_transform()*/); // i honestly have no idea what to do with a previous model xform.. most exporters ignore it
  1822. //make rests relative to the skeleton (they seem to be always relative to world)
  1823. for (KeyValue<String, Transform3D> &E : skin.bone_rest_map) {
  1824. E.value = skel_inv * E.value; //make the bone rest local to the skeleton
  1825. state.bone_rest_map[E.key] = E.value; // make it remember where the bone is globally, now that it's relative
  1826. }
  1827. //but most exporters seem to work only if i do this..
  1828. //p_node->default_transform = p_node->get_global_transform();
  1829. //p_node->default_transform=Transform3D(); //this seems to be correct, because bind shape makes the object local to the skeleton
  1830. p_node->ignore_anim = true; // collada may animate this later, if it does, then this is not supported (redo your original asset and don't animate the base mesh)
  1831. p_node->parent = sk;
  1832. //sk->children.push_back(0,p_node); //avoid INFINITE loop
  1833. p_mgeom->push_back(p_node);
  1834. return true;
  1835. }
  1836. }
  1837. for (int i = 0; i < p_node->children.size(); i++) {
  1838. if (_move_geometry_to_skeletons(p_vscene, p_node->children[i], p_mgeom)) {
  1839. p_node->children.remove_at(i);
  1840. i--;
  1841. }
  1842. }
  1843. return false;
  1844. }
  1845. void Collada::_find_morph_nodes(VisualScene *p_vscene, Node *p_node) {
  1846. if (p_node->type == Node::TYPE_GEOMETRY) {
  1847. NodeGeometry *nj = static_cast<NodeGeometry *>(p_node);
  1848. if (nj->controller) {
  1849. String base = nj->source;
  1850. while (!base.is_empty() && !state.mesh_data_map.has(base)) {
  1851. if (state.skin_controller_data_map.has(base)) {
  1852. SkinControllerData &sk = state.skin_controller_data_map[base];
  1853. base = sk.base;
  1854. } else if (state.morph_controller_data_map.has(base)) {
  1855. state.morph_ownership_map[base] = nj->id;
  1856. break;
  1857. } else {
  1858. ERR_FAIL_MSG("Invalid scene.");
  1859. }
  1860. }
  1861. }
  1862. }
  1863. for (int i = 0; i < p_node->children.size(); i++) {
  1864. _find_morph_nodes(p_vscene, p_node->children[i]);
  1865. }
  1866. }
  1867. void Collada::_optimize() {
  1868. for (KeyValue<String, VisualScene> &E : state.visual_scene_map) {
  1869. VisualScene &vs = E.value;
  1870. for (int i = 0; i < vs.root_nodes.size(); i++) {
  1871. _create_skeletons(&vs.root_nodes.write[i]);
  1872. }
  1873. for (int i = 0; i < vs.root_nodes.size(); i++) {
  1874. _merge_skeletons(&vs, vs.root_nodes[i]);
  1875. }
  1876. _merge_skeletons2(&vs);
  1877. for (int i = 0; i < vs.root_nodes.size(); i++) {
  1878. _optimize_skeletons(&vs, vs.root_nodes[i]);
  1879. }
  1880. for (int i = 0; i < vs.root_nodes.size(); i++) {
  1881. List<Node *> mgeom;
  1882. if (_move_geometry_to_skeletons(&vs, vs.root_nodes[i], &mgeom)) {
  1883. vs.root_nodes.remove_at(i);
  1884. i--;
  1885. }
  1886. while (!mgeom.is_empty()) {
  1887. Node *n = mgeom.front()->get();
  1888. n->parent->children.push_back(n);
  1889. mgeom.pop_front();
  1890. }
  1891. }
  1892. for (int i = 0; i < vs.root_nodes.size(); i++) {
  1893. _find_morph_nodes(&vs, vs.root_nodes[i]);
  1894. }
  1895. }
  1896. }
  1897. int Collada::get_uv_channel(String p_name) {
  1898. if (!channel_map.has(p_name)) {
  1899. ERR_FAIL_COND_V(channel_map.size() == 2, 0);
  1900. channel_map[p_name] = channel_map.size();
  1901. }
  1902. return channel_map[p_name];
  1903. }
  1904. Error Collada::load(const String &p_path, int p_flags) {
  1905. Ref<XMLParser> parserr = memnew(XMLParser);
  1906. XMLParser &parser = *parserr.ptr();
  1907. Error err = parser.open(p_path);
  1908. ERR_FAIL_COND_V_MSG(err, err, "Cannot open Collada file '" + p_path + "'.");
  1909. state.local_path = ProjectSettings::get_singleton()->localize_path(p_path);
  1910. state.import_flags = p_flags;
  1911. /* Skip headers */
  1912. while ((err = parser.read()) == OK) {
  1913. if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
  1914. if (parser.get_node_name() == "COLLADA") {
  1915. break;
  1916. } else if (!parser.is_empty()) {
  1917. parser.skip_section(); // unknown section, likely headers
  1918. }
  1919. }
  1920. }
  1921. ERR_FAIL_COND_V_MSG(err != OK, ERR_FILE_CORRUPT, "Corrupted Collada file '" + p_path + "'.");
  1922. /* Start loading Collada */
  1923. {
  1924. //version
  1925. String version = parser.get_attribute_value("version");
  1926. state.version.major = version.get_slice(".", 0).to_int();
  1927. state.version.minor = version.get_slice(".", 1).to_int();
  1928. state.version.rev = version.get_slice(".", 2).to_int();
  1929. COLLADA_PRINT("Collada VERSION: " + version);
  1930. }
  1931. while ((err = parser.read()) == OK) {
  1932. /* Read all the main sections.. */
  1933. if (parser.get_node_type() != XMLParser::NODE_ELEMENT) {
  1934. continue; //no idea what this may be, but skipping anyway
  1935. }
  1936. String section = parser.get_node_name();
  1937. COLLADA_PRINT("section: " + section);
  1938. if (section == "asset") {
  1939. _parse_asset(parser);
  1940. } else if (section.begins_with("library_")) {
  1941. _parse_library(parser);
  1942. } else if (section == "scene") {
  1943. _parse_scene(parser);
  1944. } else if (!parser.is_empty()) {
  1945. parser.skip_section(); // unknown section, likely headers
  1946. }
  1947. }
  1948. _optimize();
  1949. return OK;
  1950. }
  1951. Collada::Collada() {
  1952. }