12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408 |
- /**************************************************************************/
- /* collada.cpp */
- /**************************************************************************/
- /* This file is part of: */
- /* GODOT ENGINE */
- /* https://godotengine.org */
- /**************************************************************************/
- /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
- /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
- /* */
- /* Permission is hereby granted, free of charge, to any person obtaining */
- /* a copy of this software and associated documentation files (the */
- /* "Software"), to deal in the Software without restriction, including */
- /* without limitation the rights to use, copy, modify, merge, publish, */
- /* distribute, sublicense, and/or sell copies of the Software, and to */
- /* permit persons to whom the Software is furnished to do so, subject to */
- /* the following conditions: */
- /* */
- /* The above copyright notice and this permission notice shall be */
- /* included in all copies or substantial portions of the Software. */
- /* */
- /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
- /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
- /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
- /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
- /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
- /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
- /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
- /**************************************************************************/
- #include "collada.h"
- #include <stdio.h>
- //#define DEBUG_DEFAULT_ANIMATION
- //#define DEBUG_COLLADA
- #ifdef DEBUG_COLLADA
- #define COLLADA_PRINT(m_what) print_line(m_what)
- #else
- #define COLLADA_PRINT(m_what)
- #endif
- #define COLLADA_IMPORT_SCALE_SCENE
- /* HELPERS */
- String Collada::Effect::get_texture_path(const String &p_source, Collada &state) const {
- const String &image = p_source;
- ERR_FAIL_COND_V(!state.state.image_map.has(image), "");
- return state.state.image_map[image].path;
- }
- Transform Collada::get_root_transform() const {
- Transform unit_scale_transform;
- #ifndef COLLADA_IMPORT_SCALE_SCENE
- unit_scale_transform.scale(Vector3(state.unit_scale, state.unit_scale, state.unit_scale));
- #endif
- return unit_scale_transform;
- }
- void Collada::Vertex::fix_unit_scale(Collada &state) {
- #ifdef COLLADA_IMPORT_SCALE_SCENE
- vertex *= state.state.unit_scale;
- #endif
- }
- static String _uri_to_id(const String &p_uri) {
- if (p_uri.begins_with("#")) {
- return p_uri.substr(1, p_uri.size() - 1);
- } else {
- return p_uri;
- }
- }
- /** HELPER FUNCTIONS **/
- Transform Collada::fix_transform(const Transform &p_transform) {
- Transform tr = p_transform;
- #ifndef NO_UP_AXIS_SWAP
- if (state.up_axis != Vector3::AXIS_Y) {
- for (int i = 0; i < 3; i++) {
- SWAP(tr.basis[1][i], tr.basis[state.up_axis][i]);
- }
- for (int i = 0; i < 3; i++) {
- SWAP(tr.basis[i][1], tr.basis[i][state.up_axis]);
- }
- SWAP(tr.origin[1], tr.origin[state.up_axis]);
- tr.basis[state.up_axis][0] = -tr.basis[state.up_axis][0];
- tr.basis[state.up_axis][1] = -tr.basis[state.up_axis][1];
- tr.basis[0][state.up_axis] = -tr.basis[0][state.up_axis];
- tr.basis[1][state.up_axis] = -tr.basis[1][state.up_axis];
- tr.origin[state.up_axis] = -tr.origin[state.up_axis];
- }
- #endif
- //tr.scale(Vector3(state.unit_scale.unit_scale.unit_scale));
- return tr;
- //return state.matrix_fix * p_transform;
- }
- static Transform _read_transform_from_array(const Vector<float> &array, int ofs = 0) {
- Transform tr;
- // i wonder why collada matrices are transposed, given that's opposed to opengl..
- tr.basis.elements[0][0] = array[0 + ofs];
- tr.basis.elements[0][1] = array[1 + ofs];
- tr.basis.elements[0][2] = array[2 + ofs];
- tr.basis.elements[1][0] = array[4 + ofs];
- tr.basis.elements[1][1] = array[5 + ofs];
- tr.basis.elements[1][2] = array[6 + ofs];
- tr.basis.elements[2][0] = array[8 + ofs];
- tr.basis.elements[2][1] = array[9 + ofs];
- tr.basis.elements[2][2] = array[10 + ofs];
- tr.origin.x = array[3 + ofs];
- tr.origin.y = array[7 + ofs];
- tr.origin.z = array[11 + ofs];
- return tr;
- }
- /* STRUCTURES */
- Transform Collada::Node::compute_transform(Collada &state) const {
- Transform xform;
- for (int i = 0; i < xform_list.size(); i++) {
- Transform xform_step;
- const XForm &xf = xform_list[i];
- switch (xf.op) {
- case XForm::OP_ROTATE: {
- if (xf.data.size() >= 4) {
- xform_step.rotate(Vector3(xf.data[0], xf.data[1], xf.data[2]), Math::deg2rad(xf.data[3]));
- }
- } break;
- case XForm::OP_SCALE: {
- if (xf.data.size() >= 3) {
- xform_step.scale(Vector3(xf.data[0], xf.data[1], xf.data[2]));
- }
- } break;
- case XForm::OP_TRANSLATE: {
- if (xf.data.size() >= 3) {
- xform_step.origin = Vector3(xf.data[0], xf.data[1], xf.data[2]);
- }
- } break;
- case XForm::OP_MATRIX: {
- if (xf.data.size() >= 16) {
- xform_step = _read_transform_from_array(xf.data, 0);
- }
- } break;
- default: {
- }
- }
- xform = xform * xform_step;
- }
- #ifdef COLLADA_IMPORT_SCALE_SCENE
- xform.origin *= state.state.unit_scale;
- #endif
- return xform;
- }
- Transform Collada::Node::get_transform() const {
- return default_transform;
- }
- Transform Collada::Node::get_global_transform() const {
- if (parent) {
- return parent->get_global_transform() * default_transform;
- } else {
- return default_transform;
- }
- }
- Vector<float> Collada::AnimationTrack::get_value_at_time(float p_time) const {
- ERR_FAIL_COND_V(keys.size() == 0, Vector<float>());
- int i = 0;
- for (i = 0; i < keys.size(); i++) {
- if (keys[i].time > p_time) {
- break;
- }
- }
- if (i == 0) {
- return keys[0].data;
- }
- if (i == keys.size()) {
- return keys[keys.size() - 1].data;
- }
- switch (keys[i].interp_type) {
- case INTERP_BEZIER: //wait for bezier
- case INTERP_LINEAR: {
- float c = (p_time - keys[i - 1].time) / (keys[i].time - keys[i - 1].time);
- if (keys[i].data.size() == 16) {
- //interpolate a matrix
- Transform src = _read_transform_from_array(keys[i - 1].data);
- Transform dst = _read_transform_from_array(keys[i].data);
- Transform interp = c < 0.001 ? src : src.interpolate_with(dst, c);
- Vector<float> ret;
- ret.resize(16);
- Transform tr;
- // i wonder why collada matrices are transposed, given that's opposed to opengl..
- ret.write[0] = interp.basis.elements[0][0];
- ret.write[1] = interp.basis.elements[0][1];
- ret.write[2] = interp.basis.elements[0][2];
- ret.write[4] = interp.basis.elements[1][0];
- ret.write[5] = interp.basis.elements[1][1];
- ret.write[6] = interp.basis.elements[1][2];
- ret.write[8] = interp.basis.elements[2][0];
- ret.write[9] = interp.basis.elements[2][1];
- ret.write[10] = interp.basis.elements[2][2];
- ret.write[3] = interp.origin.x;
- ret.write[7] = interp.origin.y;
- ret.write[11] = interp.origin.z;
- ret.write[12] = 0;
- ret.write[13] = 0;
- ret.write[14] = 0;
- ret.write[15] = 1;
- return ret;
- } else {
- Vector<float> dest;
- dest.resize(keys[i].data.size());
- for (int j = 0; j < dest.size(); j++) {
- dest.write[j] = keys[i].data[j] * c + keys[i - 1].data[j] * (1.0 - c);
- }
- return dest;
- //interpolate one by one
- }
- } break;
- }
- ERR_FAIL_V(Vector<float>());
- }
- void Collada::_parse_asset(XMLParser &parser) {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "up_axis") {
- parser.read();
- if (parser.get_node_data() == "X_UP") {
- state.up_axis = Vector3::AXIS_X;
- }
- if (parser.get_node_data() == "Y_UP") {
- state.up_axis = Vector3::AXIS_Y;
- }
- if (parser.get_node_data() == "Z_UP") {
- state.up_axis = Vector3::AXIS_Z;
- }
- COLLADA_PRINT("up axis: " + parser.get_node_data());
- } else if (name == "unit") {
- state.unit_scale = parser.get_attribute_value("meter").to_double();
- COLLADA_PRINT("unit scale: " + rtos(state.unit_scale));
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "asset") {
- break; //end of <asset>
- }
- }
- }
- void Collada::_parse_image(XMLParser &parser) {
- String id = parser.get_attribute_value("id");
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- Image image;
- if (state.version < State::Version(1, 4, 0)) {
- /* <1.4 */
- String path = parser.get_attribute_value("source").strip_edges();
- if (path.find("://") == -1 && path.is_rel_path()) {
- // path is relative to file being loaded, so convert to a resource path
- image.path = ProjectSettings::get_singleton()->localize_path(state.local_path.get_base_dir().plus_file(path.percent_decode()));
- }
- } else {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "init_from") {
- parser.read();
- String path = parser.get_node_data().strip_edges().percent_decode();
- if (path.find("://") == -1 && path.is_rel_path()) {
- // path is relative to file being loaded, so convert to a resource path
- path = ProjectSettings::get_singleton()->localize_path(state.local_path.get_base_dir().plus_file(path));
- } else if (path.find("file:///") == 0) {
- path = path.replace_first("file:///", "");
- path = ProjectSettings::get_singleton()->localize_path(path);
- }
- image.path = path;
- } else if (name == "data") {
- ERR_PRINT("COLLADA Embedded image data not supported!");
- } else if (name == "extra" && !parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "image") {
- break; //end of <asset>
- }
- }
- }
- state.image_map[id] = image;
- }
- void Collada::_parse_material(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- Material material;
- String id = parser.get_attribute_value("id");
- if (parser.has_attribute("name")) {
- material.name = parser.get_attribute_value("name");
- }
- if (state.version < State::Version(1, 4, 0)) {
- /* <1.4 */
- ERR_PRINT("Collada Materials < 1.4 are not supported (yet)");
- } else {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT && parser.get_node_name() == "instance_effect") {
- material.instance_effect = _uri_to_id(parser.get_attribute_value("url"));
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "material") {
- break; //end of <asset>
- }
- }
- }
- state.material_map[id] = material;
- }
- //! reads floats from inside of xml element until end of xml element
- Vector<float> Collada::_read_float_array(XMLParser &parser) {
- if (parser.is_empty()) {
- return Vector<float>();
- }
- Vector<String> splitters;
- splitters.push_back(" ");
- splitters.push_back("\n");
- splitters.push_back("\r");
- splitters.push_back("\t");
- Vector<float> array;
- while (parser.read() == OK) {
- // TODO: check for comments inside the element
- // and ignore them.
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- // parse float data
- String str = parser.get_node_data();
- array = str.split_floats_mk(splitters, false);
- //array=str.split_floats(" ",false);
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
- break; // end parsing text
- }
- }
- return array;
- }
- Vector<String> Collada::_read_string_array(XMLParser &parser) {
- if (parser.is_empty()) {
- return Vector<String>();
- }
- Vector<String> array;
- while (parser.read() == OK) {
- // TODO: check for comments inside the element
- // and ignore them.
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- // parse String data
- String str = parser.get_node_data();
- array = str.split_spaces();
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
- break; // end parsing text
- }
- }
- return array;
- }
- Transform Collada::_read_transform(XMLParser &parser) {
- if (parser.is_empty()) {
- return Transform();
- }
- Vector<String> array;
- while (parser.read() == OK) {
- // TODO: check for comments inside the element
- // and ignore them.
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- // parse float data
- String str = parser.get_node_data();
- array = str.split_spaces();
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
- break; // end parsing text
- }
- }
- ERR_FAIL_COND_V(array.size() != 16, Transform());
- Vector<float> farr;
- farr.resize(16);
- for (int i = 0; i < 16; i++) {
- farr.write[i] = array[i].to_double();
- }
- return _read_transform_from_array(farr);
- }
- String Collada::_read_empty_draw_type(XMLParser &parser) {
- String empty_draw_type = "";
- if (parser.is_empty()) {
- return empty_draw_type;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- empty_draw_type = parser.get_node_data();
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END) {
- break; // end parsing text
- }
- }
- return empty_draw_type;
- }
- Variant Collada::_parse_param(XMLParser &parser) {
- if (parser.is_empty()) {
- return Variant();
- }
- String from = parser.get_node_name();
- Variant data;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "float") {
- parser.read();
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- data = parser.get_node_data().to_double();
- }
- } else if (parser.get_node_name() == "float2") {
- Vector<float> v2 = _read_float_array(parser);
- if (v2.size() >= 2) {
- data = Vector2(v2[0], v2[1]);
- }
- } else if (parser.get_node_name() == "float3") {
- Vector<float> v3 = _read_float_array(parser);
- if (v3.size() >= 3) {
- data = Vector3(v3[0], v3[1], v3[2]);
- }
- } else if (parser.get_node_name() == "float4") {
- Vector<float> v4 = _read_float_array(parser);
- if (v4.size() >= 4) {
- data = Color(v4[0], v4[1], v4[2], v4[3]);
- }
- } else if (parser.get_node_name() == "sampler2D") {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "source") {
- parser.read();
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- data = parser.get_node_data();
- }
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "sampler2D") {
- break;
- }
- }
- } else if (parser.get_node_name() == "surface") {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "init_from") {
- parser.read();
- if (parser.get_node_type() == XMLParser::NODE_TEXT) {
- data = parser.get_node_data();
- }
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "surface") {
- break;
- }
- }
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == from) {
- break;
- }
- }
- COLLADA_PRINT("newparam ending " + parser.get_node_name());
- return data;
- }
- void Collada::_parse_effect_material(XMLParser &parser, Effect &effect, String &id) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- // first come the tags we descend, but ignore the top-levels
- COLLADA_PRINT("node name: " + parser.get_node_name());
- if (!parser.is_empty() && (parser.get_node_name() == "profile_COMMON" || parser.get_node_name() == "technique" || parser.get_node_name() == "extra")) {
- _parse_effect_material(parser, effect, id); // try again
- } else if (parser.get_node_name() == "newparam") {
- String name = parser.get_attribute_value("sid");
- Variant value = _parse_param(parser);
- effect.params[name] = value;
- COLLADA_PRINT("param: " + name + " value:" + String(value));
- } else if (parser.get_node_name() == "constant" ||
- parser.get_node_name() == "lambert" ||
- parser.get_node_name() == "phong" ||
- parser.get_node_name() == "blinn") {
- COLLADA_PRINT("shade model: " + parser.get_node_name());
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String what = parser.get_node_name();
- if (what == "emission" ||
- what == "diffuse" ||
- what == "specular" ||
- what == "reflective") {
- // color or texture types
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "color") {
- Vector<float> colorarr = _read_float_array(parser);
- COLLADA_PRINT("colorarr size: " + rtos(colorarr.size()));
- if (colorarr.size() >= 3) {
- // alpha strangely not alright? maybe it needs to be multiplied by value as a channel intensity
- Color color(colorarr[0], colorarr[1], colorarr[2], 1.0);
- if (what == "diffuse") {
- effect.diffuse.color = color;
- }
- if (what == "specular") {
- effect.specular.color = color;
- }
- if (what == "emission") {
- effect.emission.color = color;
- }
- COLLADA_PRINT(what + " color: " + color);
- }
- } else if (parser.get_node_name() == "texture") {
- String sampler = parser.get_attribute_value("texture");
- if (!effect.params.has(sampler)) {
- ERR_PRINT(String("Couldn't find sampler: " + sampler + " in material:" + id).utf8().get_data());
- } else {
- String surface = effect.params[sampler];
- if (!effect.params.has(surface)) {
- ERR_PRINT(String("Couldn't find surface: " + surface + " in material:" + id).utf8().get_data());
- } else {
- String uri = effect.params[surface];
- if (what == "diffuse") {
- effect.diffuse.texture = uri;
- } else if (what == "specular") {
- effect.specular.texture = uri;
- } else if (what == "emission") {
- effect.emission.texture = uri;
- } else if (what == "bump") {
- if (parser.has_attribute("bumptype") && parser.get_attribute_value("bumptype") != "NORMALMAP") {
- WARN_PRINT("'bump' texture type is not NORMALMAP, only NORMALMAP is supported.");
- }
- effect.bump.texture = uri;
- }
- COLLADA_PRINT(what + " texture: " + uri);
- }
- }
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == what) {
- break;
- }
- }
- } else if (what == "shininess") {
- effect.shininess = _parse_param(parser);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END &&
- (parser.get_node_name() == "constant" ||
- parser.get_node_name() == "lambert" ||
- parser.get_node_name() == "phong" ||
- parser.get_node_name() == "blinn")) {
- break;
- }
- }
- } else if (parser.get_node_name() == "double_sided" || parser.get_node_name() == "show_double_sided") { // colladamax / google earth
- // 3DS Max / Google Earth double sided extension
- parser.read();
- effect.found_double_sided = true;
- effect.double_sided = parser.get_node_data().to_int();
- COLLADA_PRINT("double sided: " + itos(parser.get_node_data().to_int()));
- } else if (parser.get_node_name() == "unshaded") {
- parser.read();
- effect.unshaded = parser.get_node_data().to_int();
- } else if (parser.get_node_name() == "bump") {
- // color or texture types
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "texture") {
- String sampler = parser.get_attribute_value("texture");
- if (!effect.params.has(sampler)) {
- ERR_PRINT(String("Couldn't find sampler: " + sampler + " in material:" + id).utf8().get_data());
- } else {
- String surface = effect.params[sampler];
- if (!effect.params.has(surface)) {
- ERR_PRINT(String("Couldn't find surface: " + surface + " in material:" + id).utf8().get_data());
- } else {
- String uri = effect.params[surface];
- if (parser.has_attribute("bumptype") && parser.get_attribute_value("bumptype") != "NORMALMAP") {
- WARN_PRINT("'bump' texture type is not NORMALMAP, only NORMALMAP is supported.");
- }
- effect.bump.texture = uri;
- COLLADA_PRINT(" bump: " + uri);
- }
- }
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "bump") {
- break;
- }
- }
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END &&
- (parser.get_node_name() == "effect" ||
- parser.get_node_name() == "profile_COMMON" ||
- parser.get_node_name() == "technique" ||
- parser.get_node_name() == "extra")) {
- break;
- }
- }
- }
- void Collada::_parse_effect(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- String id = parser.get_attribute_value("id");
- Effect effect;
- if (parser.has_attribute("name")) {
- effect.name = parser.get_attribute_value("name");
- }
- _parse_effect_material(parser, effect, id);
- state.effect_map[id] = effect;
- COLLADA_PRINT("Effect ID:" + id);
- }
- void Collada::_parse_camera(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- String id = parser.get_attribute_value("id");
- state.camera_data_map[id] = CameraData();
- CameraData &camera = state.camera_data_map[id];
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "perspective") {
- camera.mode = CameraData::MODE_PERSPECTIVE;
- } else if (name == "orthographic") {
- camera.mode = CameraData::MODE_ORTHOGONAL;
- } else if (name == "xfov") {
- parser.read();
- camera.perspective.x_fov = parser.get_node_data().to_double();
- } else if (name == "yfov") {
- parser.read();
- camera.perspective.y_fov = parser.get_node_data().to_double();
- } else if (name == "xmag") {
- parser.read();
- camera.orthogonal.x_mag = parser.get_node_data().to_double();
- } else if (name == "ymag") {
- parser.read();
- camera.orthogonal.y_mag = parser.get_node_data().to_double();
- } else if (name == "aspect_ratio") {
- parser.read();
- camera.aspect = parser.get_node_data().to_double();
- } else if (name == "znear") {
- parser.read();
- camera.z_near = parser.get_node_data().to_double();
- } else if (name == "zfar") {
- parser.read();
- camera.z_far = parser.get_node_data().to_double();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "camera") {
- break; //end of <asset>
- }
- }
- COLLADA_PRINT("Camera ID:" + id);
- }
- void Collada::_parse_light(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- String id = parser.get_attribute_value("id");
- state.light_data_map[id] = LightData();
- LightData &light = state.light_data_map[id];
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "ambient") {
- light.mode = LightData::MODE_AMBIENT;
- } else if (name == "directional") {
- light.mode = LightData::MODE_DIRECTIONAL;
- } else if (name == "point") {
- light.mode = LightData::MODE_OMNI;
- } else if (name == "spot") {
- light.mode = LightData::MODE_SPOT;
- } else if (name == "color") {
- parser.read();
- Vector<float> colorarr = _read_float_array(parser);
- COLLADA_PRINT("colorarr size: " + rtos(colorarr.size()));
- if (colorarr.size() >= 4) {
- // alpha strangely not alright? maybe it needs to be multiplied by value as a channel intensity
- Color color(colorarr[0], colorarr[1], colorarr[2], 1.0);
- light.color = color;
- }
- } else if (name == "constant_attenuation") {
- parser.read();
- light.constant_att = parser.get_node_data().to_double();
- } else if (name == "linear_attenuation") {
- parser.read();
- light.linear_att = parser.get_node_data().to_double();
- } else if (name == "quadratic_attenuation") {
- parser.read();
- light.quad_att = parser.get_node_data().to_double();
- } else if (name == "falloff_angle") {
- parser.read();
- light.spot_angle = parser.get_node_data().to_double();
- } else if (name == "falloff_exponent") {
- parser.read();
- light.spot_exp = parser.get_node_data().to_double();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "light") {
- break; //end of <asset>
- }
- }
- COLLADA_PRINT("Light ID:" + id);
- }
- void Collada::_parse_curve_geometry(XMLParser &parser, String p_id, String p_name) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- //load everything into a pre dictionary
- state.curve_data_map[p_id] = CurveData();
- CurveData &curvedata = state.curve_data_map[p_id];
- curvedata.name = p_name;
- String closed = parser.get_attribute_value_safe("closed").to_lower();
- curvedata.closed = closed == "true" || closed == "1";
- COLLADA_PRINT("curve name: " + p_name);
- String current_source;
- // handles geometry node and the curve children in this loop
- // read sources with arrays and accessor for each curve
- if (parser.is_empty()) {
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "source") {
- String id = parser.get_attribute_value("id");
- curvedata.sources[id] = CurveData::Source();
- current_source = id;
- COLLADA_PRINT("source data: " + id);
- } else if (section == "float_array" || section == "array") {
- // create a new array and read it.
- if (curvedata.sources.has(current_source)) {
- curvedata.sources[current_source].array = _read_float_array(parser);
- COLLADA_PRINT("section: " + current_source + " read " + itos(curvedata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "Name_array") {
- // create a new array and read it.
- if (curvedata.sources.has(current_source)) {
- curvedata.sources[current_source].sarray = _read_string_array(parser);
- COLLADA_PRINT("section: " + current_source + " read " + itos(curvedata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "technique_common") {
- //skip it
- } else if (section == "accessor") { // child of source (below a technique tag)
- if (curvedata.sources.has(current_source)) {
- curvedata.sources[current_source].stride = parser.get_attribute_value("stride").to_int();
- COLLADA_PRINT("section: " + current_source + " stride " + itos(curvedata.sources[current_source].stride));
- }
- } else if (section == "control_vertices") {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- curvedata.control_vertices[semantic] = source;
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "spline") {
- break;
- }
- }
- }
- void Collada::_parse_mesh_geometry(XMLParser &parser, String p_id, String p_name) {
- if (!(state.import_flags & IMPORT_FLAG_SCENE)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- //load everything into a pre dictionary
- state.mesh_data_map[p_id] = MeshData();
- MeshData &meshdata = state.mesh_data_map[p_id];
- meshdata.name = p_name;
- COLLADA_PRINT("mesh name: " + p_name);
- String current_source;
- // handles geometry node and the mesh children in this loop
- // read sources with arrays and accessor for each mesh
- if (parser.is_empty()) {
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "source") {
- String id = parser.get_attribute_value("id");
- meshdata.sources[id] = MeshData::Source();
- current_source = id;
- COLLADA_PRINT("source data: " + id);
- } else if (section == "float_array" || section == "array") {
- // create a new array and read it.
- if (meshdata.sources.has(current_source)) {
- meshdata.sources[current_source].array = _read_float_array(parser);
- COLLADA_PRINT("section: " + current_source + " read " + itos(meshdata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "technique_common") {
- //skip it
- } else if (section == "accessor") { // child of source (below a technique tag)
- if (meshdata.sources.has(current_source)) {
- meshdata.sources[current_source].stride = parser.get_attribute_value("stride").to_int();
- COLLADA_PRINT("section: " + current_source + " stride " + itos(meshdata.sources[current_source].stride));
- }
- } else if (section == "vertices") {
- MeshData::Vertices vert;
- String id = parser.get_attribute_value("id");
- int last_ref = 0;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- if (semantic == "TEXCOORD") {
- semantic = "TEXCOORD" + itos(last_ref++);
- }
- vert.sources[semantic] = source;
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- meshdata.vertices[id] = vert;
- } else if (section == "triangles" || section == "polylist" || section == "polygons") {
- bool polygons = (section == "polygons");
- if (polygons) {
- 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\".");
- }
- MeshData::Primitives prim;
- if (parser.has_attribute("material")) {
- prim.material = parser.get_attribute_value("material");
- }
- prim.count = parser.get_attribute_value("count").to_int();
- prim.vertex_size = 0;
- int last_ref = 0;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- if (semantic == "TEXCOORD") {
- /*
- if (parser.has_attribute("set"))// a texcoord
- semantic+=parser.get_attribute_value("set");
- else
- semantic="TEXCOORD0";*/
- semantic = "TEXCOORD" + itos(last_ref++);
- }
- int offset = parser.get_attribute_value("offset").to_int();
- MeshData::Primitives::SourceRef sref;
- sref.source = source;
- sref.offset = offset;
- prim.sources[semantic] = sref;
- prim.vertex_size = MAX(prim.vertex_size, offset + 1);
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source + " offset: " + itos(offset));
- } else if (parser.get_node_name() == "p") { //indices
- Vector<float> values = _read_float_array(parser);
- if (polygons) {
- ERR_CONTINUE(prim.vertex_size == 0);
- prim.polygons.push_back(values.size() / prim.vertex_size);
- int from = prim.indices.size();
- prim.indices.resize(from + values.size());
- for (int i = 0; i < values.size(); i++) {
- prim.indices.write[from + i] = values[i];
- }
- } else if (prim.vertex_size > 0) {
- prim.indices = values;
- }
- COLLADA_PRINT("read " + itos(values.size()) + " index values");
- } else if (parser.get_node_name() == "vcount") { // primitive
- Vector<float> values = _read_float_array(parser);
- prim.polygons = values;
- COLLADA_PRINT("read " + itos(values.size()) + " polygon values");
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- meshdata.primitives.push_back(prim);
- } else if (parser.get_node_name() == "double_sided") {
- parser.read();
- meshdata.found_double_sided = true;
- meshdata.double_sided = parser.get_node_data().to_int();
- } else if (parser.get_node_name() == "polygons") {
- ERR_PRINT("Primitive type \"polygons\" not supported, re-export using \"polylist\" or \"triangles\".");
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "mesh") {
- break;
- }
- }
- }
- void Collada::_parse_skin_controller(XMLParser &parser, String p_id) {
- state.skin_controller_data_map[p_id] = SkinControllerData();
- SkinControllerData &skindata = state.skin_controller_data_map[p_id];
- skindata.base = _uri_to_id(parser.get_attribute_value("source"));
- String current_source;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "bind_shape_matrix") {
- skindata.bind_shape = _read_transform(parser);
- #ifdef COLLADA_IMPORT_SCALE_SCENE
- skindata.bind_shape.origin *= state.unit_scale;
- #endif
- COLLADA_PRINT("skeleton bind shape transform: " + skindata.bind_shape);
- } else if (section == "source") {
- String id = parser.get_attribute_value("id");
- skindata.sources[id] = SkinControllerData::Source();
- current_source = id;
- COLLADA_PRINT("source data: " + id);
- } else if (section == "float_array" || section == "array") {
- // create a new array and read it.
- if (skindata.sources.has(current_source)) {
- skindata.sources[current_source].array = _read_float_array(parser);
- COLLADA_PRINT("section: " + current_source + " read " + itos(skindata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "Name_array" || section == "IDREF_array") {
- // create a new array and read it.
- if (section == "IDREF_array") {
- skindata.use_idrefs = true;
- }
- if (skindata.sources.has(current_source)) {
- skindata.sources[current_source].sarray = _read_string_array(parser);
- if (section == "IDREF_array") {
- Vector<String> sa = skindata.sources[current_source].sarray;
- for (int i = 0; i < sa.size(); i++) {
- state.idref_joints.insert(sa[i]);
- }
- }
- COLLADA_PRINT("section: " + current_source + " read " + itos(skindata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "technique_common") {
- //skip it
- } else if (section == "accessor") { // child of source (below a technique tag)
- if (skindata.sources.has(current_source)) {
- int stride = 1;
- if (parser.has_attribute("stride")) {
- stride = parser.get_attribute_value("stride").to_int();
- }
- skindata.sources[current_source].stride = stride;
- COLLADA_PRINT("section: " + current_source + " stride " + itos(skindata.sources[current_source].stride));
- }
- } else if (section == "joints") {
- SkinControllerData::Joints joint;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- joint.sources[semantic] = source;
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- skindata.joints = joint;
- } else if (section == "vertex_weights") {
- SkinControllerData::Weights weights;
- weights.count = parser.get_attribute_value("count").to_int();
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- int offset = parser.get_attribute_value("offset").to_int();
- SkinControllerData::Weights::SourceRef sref;
- sref.source = source;
- sref.offset = offset;
- weights.sources[semantic] = sref;
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source + " offset: " + itos(offset));
- } else if (parser.get_node_name() == "v") { //indices
- Vector<float> values = _read_float_array(parser);
- weights.indices = values;
- COLLADA_PRINT("read " + itos(values.size()) + " index values");
- } else if (parser.get_node_name() == "vcount") { // weightsitive
- Vector<float> values = _read_float_array(parser);
- weights.sets = values;
- COLLADA_PRINT("read " + itos(values.size()) + " polygon values");
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- skindata.weights = weights;
- }
- /*
- else if (!parser.is_empty())
- parser.skip_section();
- */
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "skin") {
- break;
- }
- }
- /* STORE REST MATRICES */
- Vector<Transform> rests;
- ERR_FAIL_COND(!skindata.joints.sources.has("JOINT"));
- ERR_FAIL_COND(!skindata.joints.sources.has("INV_BIND_MATRIX"));
- String joint_arr = skindata.joints.sources["JOINT"];
- String ibm = skindata.joints.sources["INV_BIND_MATRIX"];
- ERR_FAIL_COND(!skindata.sources.has(joint_arr));
- ERR_FAIL_COND(!skindata.sources.has(ibm));
- SkinControllerData::Source &joint_source = skindata.sources[joint_arr];
- SkinControllerData::Source &ibm_source = skindata.sources[ibm];
- ERR_FAIL_COND(joint_source.sarray.size() != ibm_source.array.size() / 16);
- for (int i = 0; i < joint_source.sarray.size(); i++) {
- String name = joint_source.sarray[i];
- Transform xform = _read_transform_from_array(ibm_source.array, i * 16); //<- this is a mistake, it must be applied to vertices
- xform.affine_invert(); // inverse for rest, because it's an inverse
- #ifdef COLLADA_IMPORT_SCALE_SCENE
- xform.origin *= state.unit_scale;
- #endif
- skindata.bone_rest_map[name] = xform;
- }
- }
- void Collada::_parse_morph_controller(XMLParser &parser, String p_id) {
- state.morph_controller_data_map[p_id] = MorphControllerData();
- MorphControllerData &morphdata = state.morph_controller_data_map[p_id];
- morphdata.mesh = _uri_to_id(parser.get_attribute_value("source"));
- morphdata.mode = parser.get_attribute_value("method");
- String current_source;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "source") {
- String id = parser.get_attribute_value("id");
- morphdata.sources[id] = MorphControllerData::Source();
- current_source = id;
- COLLADA_PRINT("source data: " + id);
- } else if (section == "float_array" || section == "array") {
- // create a new array and read it.
- if (morphdata.sources.has(current_source)) {
- morphdata.sources[current_source].array = _read_float_array(parser);
- COLLADA_PRINT("section: " + current_source + " read " + itos(morphdata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "Name_array" || section == "IDREF_array") {
- // create a new array and read it.
- /*
- if (section=="IDREF_array")
- morphdata.use_idrefs=true;
- */
- if (morphdata.sources.has(current_source)) {
- morphdata.sources[current_source].sarray = _read_string_array(parser);
- /*
- if (section=="IDREF_array") {
- Vector<String> sa = morphdata.sources[current_source].sarray;
- for(int i=0;i<sa.size();i++)
- state.idref_joints.insert(sa[i]);
- }*/
- COLLADA_PRINT("section: " + current_source + " read " + itos(morphdata.sources[current_source].array.size()) + " values.");
- }
- } else if (section == "technique_common") {
- //skip it
- } else if (section == "accessor") { // child of source (below a technique tag)
- if (morphdata.sources.has(current_source)) {
- int stride = 1;
- if (parser.has_attribute("stride")) {
- stride = parser.get_attribute_value("stride").to_int();
- }
- morphdata.sources[current_source].stride = stride;
- COLLADA_PRINT("section: " + current_source + " stride " + itos(morphdata.sources[current_source].stride));
- }
- } else if (section == "targets") {
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "input") {
- String semantic = parser.get_attribute_value("semantic");
- String source = _uri_to_id(parser.get_attribute_value("source"));
- morphdata.targets[semantic] = source;
- COLLADA_PRINT(section + " input semantic: " + semantic + " source: " + source);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == section) {
- break;
- }
- }
- }
- /*
- else if (!parser.is_empty())
- parser.skip_section();
- */
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "morph") {
- break;
- }
- }
- if (morphdata.targets.has("MORPH_WEIGHT")) {
- state.morph_name_map[morphdata.targets["MORPH_WEIGHT"]] = p_id;
- }
- }
- void Collada::_parse_controller(XMLParser &parser) {
- String id = parser.get_attribute_value("id");
- if (parser.is_empty()) {
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "skin") {
- _parse_skin_controller(parser, id);
- } else if (section == "morph") {
- _parse_morph_controller(parser, id);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "controller") {
- break;
- }
- }
- }
- Collada::Node *Collada::_parse_visual_instance_geometry(XMLParser &parser) {
- String type = parser.get_node_name();
- NodeGeometry *geom = memnew(NodeGeometry);
- geom->controller = type == "instance_controller";
- geom->source = _uri_to_id(parser.get_attribute_value_safe("url"));
- if (parser.is_empty()) { //nothing else to parse...
- return geom;
- }
- // try to find also many materials and skeletons!
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "instance_material") {
- String symbol = parser.get_attribute_value("symbol");
- String target = _uri_to_id(parser.get_attribute_value("target"));
- NodeGeometry::Material mat;
- mat.target = target;
- geom->material_map[symbol] = mat;
- COLLADA_PRINT("uses material: '" + target + "' on primitive'" + symbol + "'");
- } else if (parser.get_node_name() == "skeleton") {
- parser.read();
- String uri = _uri_to_id(parser.get_node_data());
- if (uri != "") {
- geom->skeletons.push_back(uri);
- }
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == type) {
- break;
- }
- }
- if (geom->controller) {
- if (geom->skeletons.empty()) {
- //XSI style
- if (state.skin_controller_data_map.has(geom->source)) {
- SkinControllerData *skin = &state.skin_controller_data_map[geom->source];
- //case where skeletons reference bones with IDREF (XSI)
- ERR_FAIL_COND_V(!skin->joints.sources.has("JOINT"), geom);
- String joint_arr = skin->joints.sources["JOINT"];
- ERR_FAIL_COND_V(!skin->sources.has(joint_arr), geom);
- Collada::SkinControllerData::Source &joint_source = skin->sources[joint_arr];
- geom->skeletons = joint_source.sarray; //quite crazy, but should work.
- }
- }
- }
- return geom;
- }
- Collada::Node *Collada::_parse_visual_instance_camera(XMLParser &parser) {
- NodeCamera *cam = memnew(NodeCamera);
- cam->camera = _uri_to_id(parser.get_attribute_value_safe("url"));
- if (state.up_axis == Vector3::AXIS_Z) { //collada weirdness
- cam->post_transform.basis.rotate(Vector3(1, 0, 0), -Math_PI * 0.5);
- }
- if (parser.is_empty()) { //nothing else to parse...
- return cam;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "instance_camera") {
- break;
- }
- }
- return cam;
- }
- Collada::Node *Collada::_parse_visual_instance_light(XMLParser &parser) {
- NodeLight *cam = memnew(NodeLight);
- cam->light = _uri_to_id(parser.get_attribute_value_safe("url"));
- if (state.up_axis == Vector3::AXIS_Z) { //collada weirdness
- cam->post_transform.basis.rotate(Vector3(1, 0, 0), -Math_PI * 0.5);
- }
- if (parser.is_empty()) { //nothing else to parse...
- return cam;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "instance_light") {
- break;
- }
- }
- return cam;
- }
- Collada::Node *Collada::_parse_visual_node_instance_data(XMLParser &parser) {
- String instance_type = parser.get_node_name();
- if (instance_type == "instance_geometry" || instance_type == "instance_controller") {
- return _parse_visual_instance_geometry(parser);
- } else if (instance_type == "instance_camera") {
- return _parse_visual_instance_camera(parser);
- } else if (instance_type == "instance_light") {
- return _parse_visual_instance_light(parser);
- }
- if (parser.is_empty()) { //nothing else to parse...
- return nullptr;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == instance_type) {
- break;
- }
- }
- return nullptr;
- }
- Collada::Node *Collada::_parse_visual_scene_node(XMLParser &parser) {
- String name;
- String id = parser.get_attribute_value_safe("id");
- bool found_name = false;
- if (id == "") {
- id = "%NODEID%" + itos(Math::rand());
- } else {
- found_name = true;
- }
- Vector<Node::XForm> xform_list;
- Vector<Node *> children;
- String empty_draw_type = "";
- Node *node = nullptr;
- name = parser.has_attribute("name") ? parser.get_attribute_value_safe("name") : parser.get_attribute_value_safe("id");
- if (name == "") {
- name = id;
- } else {
- found_name = true;
- }
- if ((parser.has_attribute("type") && parser.get_attribute_value("type") == "JOINT") || state.idref_joints.has(name)) {
- // handle a bone
- NodeJoint *joint = memnew(NodeJoint);
- if (parser.has_attribute("sid")) { //bones may not have sid
- joint->sid = parser.get_attribute_value("sid");
- //state.bone_map[joint->sid]=joint;
- } else if (state.idref_joints.has(name)) {
- joint->sid = name; //kind of a cheat but..
- } else if (parser.has_attribute("name")) {
- joint->sid = parser.get_attribute_value_safe("name");
- }
- if (joint->sid != "") {
- state.sid_to_node_map[joint->sid] = id;
- }
- node = joint;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "translate") {
- Node::XForm xf;
- if (parser.has_attribute("sid")) {
- xf.id = parser.get_attribute_value("sid");
- }
- xf.op = Node::XForm::OP_TRANSLATE;
- Vector<float> xlt = _read_float_array(parser);
- xf.data = xlt;
- xform_list.push_back(xf);
- } else if (section == "rotate") {
- Node::XForm xf;
- if (parser.has_attribute("sid")) {
- xf.id = parser.get_attribute_value("sid");
- }
- xf.op = Node::XForm::OP_ROTATE;
- Vector<float> rot = _read_float_array(parser);
- xf.data = rot;
- xform_list.push_back(xf);
- } else if (section == "scale") {
- Node::XForm xf;
- if (parser.has_attribute("sid")) {
- xf.id = parser.get_attribute_value("sid");
- }
- xf.op = Node::XForm::OP_SCALE;
- Vector<float> scale = _read_float_array(parser);
- xf.data = scale;
- xform_list.push_back(xf);
- } else if (section == "matrix") {
- Node::XForm xf;
- if (parser.has_attribute("sid")) {
- xf.id = parser.get_attribute_value("sid");
- }
- xf.op = Node::XForm::OP_MATRIX;
- Vector<float> matrix = _read_float_array(parser);
- xf.data = matrix;
- String mtx;
- for (int i = 0; i < matrix.size(); i++) {
- mtx += " " + rtos(matrix[i]);
- }
- xform_list.push_back(xf);
- } else if (section == "visibility") {
- Node::XForm xf;
- if (parser.has_attribute("sid")) {
- xf.id = parser.get_attribute_value("sid");
- }
- xf.op = Node::XForm::OP_VISIBILITY;
- Vector<float> visible = _read_float_array(parser);
- xf.data = visible;
- xform_list.push_back(xf);
- } else if (section == "empty_draw_type") {
- empty_draw_type = _read_empty_draw_type(parser);
- } else if (section == "technique" || section == "extra") {
- } else if (section != "node") {
- //usually what defines the type of node
- if (section.begins_with("instance_")) {
- if (!node) {
- node = _parse_visual_node_instance_data(parser);
- } else {
- ERR_PRINT("Multiple instance_* not supported.");
- }
- }
- } else {
- /* Found a child node!! what to do..*/
- Node *child = _parse_visual_scene_node(parser);
- children.push_back(child);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "node") {
- break;
- }
- }
- if (!node) {
- node = memnew(Node); //generic node, nothing of relevance found
- }
- node->noname = !found_name;
- node->xform_list = xform_list;
- node->children = children;
- for (int i = 0; i < children.size(); i++) {
- node->children[i]->parent = node;
- }
- node->name = name;
- node->id = id;
- node->empty_draw_type = empty_draw_type;
- if (node->children.size() == 1) {
- if (node->children[0]->noname && !node->noname) {
- node->children[0]->name = node->name;
- node->name = node->name + "-base";
- }
- }
- node->default_transform = node->compute_transform(*this);
- state.scene_map[id] = node;
- return node;
- }
- void Collada::_parse_visual_scene(XMLParser &parser) {
- String id = parser.get_attribute_value("id");
- if (parser.is_empty()) {
- return;
- }
- state.visual_scene_map[id] = VisualScene();
- VisualScene &vscene = state.visual_scene_map[id];
- if (parser.has_attribute("name")) {
- vscene.name = parser.get_attribute_value("name");
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String section = parser.get_node_name();
- if (section == "node") {
- vscene.root_nodes.push_back(_parse_visual_scene_node(parser));
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "visual_scene") {
- break;
- }
- }
- COLLADA_PRINT("Scene ID:" + id);
- }
- void Collada::_parse_animation(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_ANIMATION)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- Map<String, Vector<float>> float_sources;
- Map<String, Vector<String>> string_sources;
- Map<String, int> source_strides;
- Map<String, Map<String, String>> samplers;
- Map<String, Vector<String>> source_param_names;
- Map<String, Vector<String>> source_param_types;
- String id = "";
- if (parser.has_attribute("id")) {
- id = parser.get_attribute_value("id");
- }
- String current_source;
- String current_sampler;
- Vector<String> channel_sources;
- Vector<String> channel_targets;
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "source") {
- current_source = parser.get_attribute_value("id");
- source_param_names[current_source] = Vector<String>();
- source_param_types[current_source] = Vector<String>();
- } else if (name == "float_array") {
- if (current_source != "") {
- float_sources[current_source] = _read_float_array(parser);
- }
- } else if (name == "Name_array") {
- if (current_source != "") {
- string_sources[current_source] = _read_string_array(parser);
- }
- } else if (name == "accessor") {
- if (current_source != "" && parser.has_attribute("stride")) {
- source_strides[current_source] = parser.get_attribute_value("stride").to_int();
- }
- } else if (name == "sampler") {
- current_sampler = parser.get_attribute_value("id");
- samplers[current_sampler] = Map<String, String>();
- } else if (name == "param") {
- if (parser.has_attribute("name")) {
- source_param_names[current_source].push_back(parser.get_attribute_value("name"));
- } else {
- source_param_names[current_source].push_back("");
- }
- if (parser.has_attribute("type")) {
- source_param_types[current_source].push_back(parser.get_attribute_value("type"));
- } else {
- source_param_types[current_source].push_back("");
- }
- } else if (name == "input") {
- if (current_sampler != "") {
- samplers[current_sampler][parser.get_attribute_value("semantic")] = parser.get_attribute_value("source");
- }
- } else if (name == "channel") {
- channel_sources.push_back(parser.get_attribute_value("source"));
- channel_targets.push_back(parser.get_attribute_value("target"));
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "animation") {
- break; //end of <asset>
- }
- }
- for (int i = 0; i < channel_sources.size(); i++) {
- String source = _uri_to_id(channel_sources[i]);
- String target = channel_targets[i];
- ERR_CONTINUE(!samplers.has(source));
- Map<String, String> &sampler = samplers[source];
- ERR_CONTINUE(!sampler.has("INPUT")); //no input semantic? wtf?
- String input_id = _uri_to_id(sampler["INPUT"]);
- COLLADA_PRINT("input id is " + input_id);
- ERR_CONTINUE(!float_sources.has(input_id));
- ERR_CONTINUE(!sampler.has("OUTPUT"));
- String output_id = _uri_to_id(sampler["OUTPUT"]);
- ERR_CONTINUE(!float_sources.has(output_id));
- ERR_CONTINUE(!source_param_names.has(output_id));
- Vector<String> &names = source_param_names[output_id];
- for (int l = 0; l < names.size(); l++) {
- String name = names[l];
- Vector<float> &time_keys = float_sources[input_id];
- int key_count = time_keys.size();
- AnimationTrack track; //begin crating track
- track.id = id;
- track.keys.resize(key_count);
- for (int j = 0; j < key_count; j++) {
- track.keys.write[j].time = time_keys[j];
- state.animation_length = MAX(state.animation_length, time_keys[j]);
- }
- //now read actual values
- int stride = 1;
- if (source_strides.has(output_id)) {
- stride = source_strides[output_id];
- }
- int output_len = stride / names.size();
- ERR_CONTINUE(output_len == 0);
- ERR_CONTINUE(!float_sources.has(output_id));
- Vector<float> &output = float_sources[output_id];
- ERR_CONTINUE_MSG((output.size() / stride) != key_count, "Wrong number of keys in output.");
- for (int j = 0; j < key_count; j++) {
- track.keys.write[j].data.resize(output_len);
- for (int k = 0; k < output_len; k++) {
- track.keys.write[j].data.write[k] = output[l + j * stride + k]; //super weird but should work:
- }
- }
- if (sampler.has("INTERPOLATION")) {
- String interp_id = _uri_to_id(sampler["INTERPOLATION"]);
- ERR_CONTINUE(!string_sources.has(interp_id));
- Vector<String> &interps = string_sources[interp_id];
- ERR_CONTINUE(interps.size() != key_count);
- for (int j = 0; j < key_count; j++) {
- if (interps[j] == "BEZIER") {
- track.keys.write[j].interp_type = AnimationTrack::INTERP_BEZIER;
- } else {
- track.keys.write[j].interp_type = AnimationTrack::INTERP_LINEAR;
- }
- }
- }
- if (sampler.has("IN_TANGENT") && sampler.has("OUT_TANGENT")) {
- //bezier control points..
- String intangent_id = _uri_to_id(sampler["IN_TANGENT"]);
- ERR_CONTINUE(!float_sources.has(intangent_id));
- Vector<float> &intangents = float_sources[intangent_id];
- ERR_CONTINUE(intangents.size() != key_count * 2 * names.size());
- String outangent_id = _uri_to_id(sampler["OUT_TANGENT"]);
- ERR_CONTINUE(!float_sources.has(outangent_id));
- Vector<float> &outangents = float_sources[outangent_id];
- ERR_CONTINUE(outangents.size() != key_count * 2 * names.size());
- for (int j = 0; j < key_count; j++) {
- track.keys.write[j].in_tangent = Vector2(intangents[j * 2 * names.size() + 0 + l * 2], intangents[j * 2 * names.size() + 1 + l * 2]);
- track.keys.write[j].out_tangent = Vector2(outangents[j * 2 * names.size() + 0 + l * 2], outangents[j * 2 * names.size() + 1 + l * 2]);
- }
- }
- if (target.find("/") != -1) { //transform component
- track.target = target.get_slicec('/', 0);
- track.param = target.get_slicec('/', 1);
- if (track.param.find(".") != -1) {
- track.component = track.param.get_slice(".", 1).to_upper();
- }
- track.param = track.param.get_slice(".", 0);
- if (names.size() > 1 && track.component == "") {
- //this is a guess because the collada spec is ambiguous here...
- //i suppose if you have many names (outputs) you can't use a component and i should abide to that.
- track.component = name;
- }
- } else {
- track.target = target;
- }
- state.animation_tracks.push_back(track);
- if (!state.referenced_tracks.has(target)) {
- state.referenced_tracks[target] = Vector<int>();
- }
- state.referenced_tracks[target].push_back(state.animation_tracks.size() - 1);
- if (id != "") {
- if (!state.by_id_tracks.has(id)) {
- state.by_id_tracks[id] = Vector<int>();
- }
- state.by_id_tracks[id].push_back(state.animation_tracks.size() - 1);
- }
- COLLADA_PRINT("loaded animation with " + itos(key_count) + " keys");
- }
- }
- }
- void Collada::_parse_animation_clip(XMLParser &parser) {
- if (!(state.import_flags & IMPORT_FLAG_ANIMATION)) {
- if (!parser.is_empty()) {
- parser.skip_section();
- }
- return;
- }
- AnimationClip clip;
- if (parser.has_attribute("name")) {
- clip.name = parser.get_attribute_value("name");
- } else if (parser.has_attribute("id")) {
- clip.name = parser.get_attribute_value("id");
- }
- if (parser.has_attribute("start")) {
- clip.begin = parser.get_attribute_value("start").to_double();
- }
- if (parser.has_attribute("end")) {
- clip.end = parser.get_attribute_value("end").to_double();
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "instance_animation") {
- String url = _uri_to_id(parser.get_attribute_value("url"));
- clip.tracks.push_back(url);
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "animation_clip") {
- break; //end of <asset>
- }
- }
- state.animation_clips.push_back(clip);
- }
- void Collada::_parse_scene(XMLParser &parser) {
- if (parser.is_empty()) {
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- if (name == "instance_visual_scene") {
- state.root_visual_scene = _uri_to_id(parser.get_attribute_value("url"));
- } else if (name == "instance_physics_scene") {
- state.root_physics_scene = _uri_to_id(parser.get_attribute_value("url"));
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "scene") {
- break; //end of <asset>
- }
- }
- }
- void Collada::_parse_library(XMLParser &parser) {
- if (parser.is_empty()) {
- return;
- }
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- String name = parser.get_node_name();
- COLLADA_PRINT("library name is: " + name);
- if (name == "image") {
- _parse_image(parser);
- } else if (name == "material") {
- _parse_material(parser);
- } else if (name == "effect") {
- _parse_effect(parser);
- } else if (name == "camera") {
- _parse_camera(parser);
- } else if (name == "light") {
- _parse_light(parser);
- } else if (name == "geometry") {
- String id = parser.get_attribute_value("id");
- String name2 = parser.get_attribute_value_safe("name");
- while (parser.read() == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "mesh") {
- state.mesh_name_map[id] = (name2 != "") ? name2 : id;
- _parse_mesh_geometry(parser, id, name2);
- } else if (parser.get_node_name() == "spline") {
- state.mesh_name_map[id] = (name2 != "") ? name2 : id;
- _parse_curve_geometry(parser, id, name2);
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name() == "geometry") {
- break;
- }
- }
- } else if (name == "controller") {
- _parse_controller(parser);
- } else if (name == "animation") {
- _parse_animation(parser);
- } else if (name == "animation_clip") {
- _parse_animation_clip(parser);
- } else if (name == "visual_scene") {
- COLLADA_PRINT("visual scene");
- _parse_visual_scene(parser);
- } else if (!parser.is_empty()) {
- parser.skip_section();
- }
- } else if (parser.get_node_type() == XMLParser::NODE_ELEMENT_END && parser.get_node_name().begins_with("library_")) {
- break; //end of <asset>
- }
- }
- }
- void Collada::_joint_set_owner(Collada::Node *p_node, NodeSkeleton *p_owner) {
- if (p_node->type == Node::TYPE_JOINT) {
- NodeJoint *nj = static_cast<NodeJoint *>(p_node);
- nj->owner = p_owner;
- for (int i = 0; i < nj->children.size(); i++) {
- _joint_set_owner(nj->children.write[i], p_owner);
- }
- }
- }
- void Collada::_create_skeletons(Collada::Node **p_node, NodeSkeleton *p_skeleton) {
- Node *node = *p_node;
- if (node->type == Node::TYPE_JOINT) {
- if (!p_skeleton) {
- // ohohohoohoo it's a joint node, time to work!
- NodeSkeleton *sk = memnew(NodeSkeleton);
- *p_node = sk;
- sk->children.push_back(node);
- sk->parent = node->parent;
- node->parent = sk;
- p_skeleton = sk;
- }
- NodeJoint *nj = static_cast<NodeJoint *>(node);
- nj->owner = p_skeleton;
- } else {
- p_skeleton = nullptr;
- }
- for (int i = 0; i < node->children.size(); i++) {
- _create_skeletons(&node->children.write[i], p_skeleton);
- }
- }
- bool Collada::_remove_node(Node *p_parent, Node *p_node) {
- for (int i = 0; i < p_parent->children.size(); i++) {
- if (p_parent->children[i] == p_node) {
- p_parent->children.remove(i);
- return true;
- }
- if (_remove_node(p_parent->children[i], p_node)) {
- return true;
- }
- }
- return false;
- }
- void Collada::_remove_node(VisualScene *p_vscene, Node *p_node) {
- for (int i = 0; i < p_vscene->root_nodes.size(); i++) {
- if (p_vscene->root_nodes[i] == p_node) {
- p_vscene->root_nodes.remove(i);
- return;
- }
- if (_remove_node(p_vscene->root_nodes[i], p_node)) {
- return;
- }
- }
- ERR_PRINT("ERROR: Not found node to remove?");
- }
- void Collada::_merge_skeletons(VisualScene *p_vscene, Node *p_node) {
- if (p_node->type == Node::TYPE_GEOMETRY) {
- NodeGeometry *gnode = static_cast<NodeGeometry *>(p_node);
- if (gnode->controller) {
- // recount skeletons used
- Set<NodeSkeleton *> skeletons;
- for (int i = 0; i < gnode->skeletons.size(); i++) {
- String nodeid = gnode->skeletons[i];
- ERR_CONTINUE(!state.scene_map.has(nodeid)); //weird, it should have it...
- NodeJoint *nj = SAFE_CAST<NodeJoint *>(state.scene_map[nodeid]);
- ERR_CONTINUE(!nj); //broken collada
- ERR_CONTINUE(!nj->owner); //weird, node should have a skeleton owner
- skeletons.insert(nj->owner);
- }
- if (skeletons.size() > 1) {
- //do the merger!!
- Set<NodeSkeleton *>::Element *E = skeletons.front();
- NodeSkeleton *base = E->get();
- for (E = E->next(); E; E = E->next()) {
- NodeSkeleton *merged = E->get();
- _remove_node(p_vscene, merged);
- for (int i = 0; i < merged->children.size(); i++) {
- _joint_set_owner(merged->children[i], base);
- base->children.push_back(merged->children[i]);
- merged->children[i]->parent = base;
- }
- merged->children.clear(); //take children from it
- memdelete(merged);
- }
- }
- }
- }
- for (int i = 0; i < p_node->children.size(); i++) {
- _merge_skeletons(p_vscene, p_node->children[i]);
- }
- }
- void Collada::_merge_skeletons2(VisualScene *p_vscene) {
- for (Map<String, SkinControllerData>::Element *E = state.skin_controller_data_map.front(); E; E = E->next()) {
- SkinControllerData &cd = E->get();
- NodeSkeleton *skeleton = nullptr;
- for (Map<String, Transform>::Element *F = cd.bone_rest_map.front(); F; F = F->next()) {
- String name;
- if (!state.sid_to_node_map.has(F->key())) {
- continue;
- }
- name = state.sid_to_node_map[F->key()];
- ERR_CONTINUE(!state.scene_map.has(name));
- Node *node = state.scene_map[name];
- ERR_CONTINUE(node->type != Node::TYPE_JOINT);
- NodeSkeleton *sk = nullptr;
- while (node && !sk) {
- if (node->type == Node::TYPE_SKELETON) {
- sk = static_cast<NodeSkeleton *>(node);
- }
- node = node->parent;
- }
- ERR_CONTINUE(!sk);
- if (!skeleton) {
- skeleton = sk;
- continue;
- }
- if (skeleton != sk) {
- //whoa.. wtf, merge.
- _remove_node(p_vscene, sk);
- for (int i = 0; i < sk->children.size(); i++) {
- _joint_set_owner(sk->children[i], skeleton);
- skeleton->children.push_back(sk->children[i]);
- sk->children[i]->parent = skeleton;
- }
- sk->children.clear(); //take children from it
- memdelete(sk);
- }
- }
- }
- }
- bool Collada::_optimize_skeletons(VisualScene *p_vscene, Node *p_node) {
- Node *node = p_node;
- if (node->type == Node::TYPE_SKELETON && node->parent && node->parent->type == Node::TYPE_NODE && node->parent->children.size() == 1) {
- //replace parent by this...
- Node *parent = node->parent;
- //i wonder if this is alright.. i think it is since created skeleton (first joint) is already animated by bone..
- node->id = parent->id;
- node->name = parent->name;
- node->xform_list = parent->xform_list;
- node->default_transform = parent->default_transform;
- state.scene_map[node->id] = node;
- node->parent = parent->parent;
- if (parent->parent) {
- Node *gp = parent->parent;
- bool found = false;
- for (int i = 0; i < gp->children.size(); i++) {
- if (gp->children[i] == parent) {
- gp->children.write[i] = node;
- found = true;
- break;
- }
- }
- if (!found) {
- ERR_PRINT("BUG");
- }
- } else {
- bool found = false;
- for (int i = 0; i < p_vscene->root_nodes.size(); i++) {
- if (p_vscene->root_nodes[i] == parent) {
- p_vscene->root_nodes.write[i] = node;
- found = true;
- break;
- }
- }
- if (!found) {
- ERR_PRINT("BUG");
- }
- }
- parent->children.clear();
- memdelete(parent);
- return true;
- }
- for (int i = 0; i < node->children.size(); i++) {
- if (_optimize_skeletons(p_vscene, node->children[i])) {
- return false; //stop processing, go up
- }
- }
- return false;
- }
- bool Collada::_move_geometry_to_skeletons(VisualScene *p_vscene, Node *p_node, List<Node *> *p_mgeom) {
- // Bind Shape Matrix scales the bones and makes them gigantic, so the matrix then shrinks the model?
- // Solution: apply the Bind Shape Matrix to the VERTICES, and if the object comes scaled, it seems to be left alone!
- if (p_node->type == Node::TYPE_GEOMETRY) {
- NodeGeometry *ng = static_cast<NodeGeometry *>(p_node);
- if (ng->ignore_anim) {
- return false; //already made child of skeleton and processeg
- }
- if (ng->controller && ng->skeletons.size()) {
- String nodeid = ng->skeletons[0];
- ERR_FAIL_COND_V(!state.scene_map.has(nodeid), false); //weird, it should have it...
- NodeJoint *nj = SAFE_CAST<NodeJoint *>(state.scene_map[nodeid]);
- ERR_FAIL_COND_V(!nj, false);
- ERR_FAIL_COND_V(!nj->owner, false); //weird, node should have a skeleton owner
- NodeSkeleton *sk = nj->owner;
- Node *p = sk->parent;
- bool node_is_parent_of_skeleton = false;
- while (p) {
- if (p == p_node) {
- node_is_parent_of_skeleton = true;
- break;
- }
- p = p->parent; // try again
- }
- ERR_FAIL_COND_V(node_is_parent_of_skeleton, false);
- //this should be correct
- ERR_FAIL_COND_V(!state.skin_controller_data_map.has(ng->source), false);
- SkinControllerData &skin = state.skin_controller_data_map[ng->source];
- Transform skel_inv = sk->get_global_transform().affine_inverse();
- 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
- //make rests relative to the skeleton (they seem to be always relative to world)
- for (Map<String, Transform>::Element *E = skin.bone_rest_map.front(); E; E = E->next()) {
- E->get() = skel_inv * E->get(); //make the bone rest local to the skeleton
- state.bone_rest_map[E->key()] = E->get(); // make it remember where the bone is globally, now that it's relative
- }
- //but most exporters seem to work only if i do this..
- //p_node->default_transform = p_node->get_global_transform();
- //p_node->default_transform=Transform(); //this seems to be correct, because bind shape makes the object local to the skeleton
- 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)
- p_node->parent = sk;
- //sk->children.push_back(0,p_node); //avoid INFINITE loop
- p_mgeom->push_back(p_node);
- return true;
- }
- }
- for (int i = 0; i < p_node->children.size(); i++) {
- if (_move_geometry_to_skeletons(p_vscene, p_node->children[i], p_mgeom)) {
- p_node->children.remove(i);
- i--;
- }
- }
- return false;
- }
- void Collada::_find_morph_nodes(VisualScene *p_vscene, Node *p_node) {
- if (p_node->type == Node::TYPE_GEOMETRY) {
- NodeGeometry *nj = static_cast<NodeGeometry *>(p_node);
- if (nj->controller) {
- String base = nj->source;
- while (base != "" && !state.mesh_data_map.has(base)) {
- if (state.skin_controller_data_map.has(base)) {
- SkinControllerData &sk = state.skin_controller_data_map[base];
- base = sk.base;
- } else if (state.morph_controller_data_map.has(base)) {
- state.morph_ownership_map[base] = nj->id;
- break;
- } else {
- ERR_FAIL_MSG("Invalid scene.");
- }
- }
- }
- }
- for (int i = 0; i < p_node->children.size(); i++) {
- _find_morph_nodes(p_vscene, p_node->children[i]);
- }
- }
- void Collada::_optimize() {
- for (Map<String, VisualScene>::Element *E = state.visual_scene_map.front(); E; E = E->next()) {
- VisualScene &vs = E->get();
- for (int i = 0; i < vs.root_nodes.size(); i++) {
- _create_skeletons(&vs.root_nodes.write[i]);
- }
- for (int i = 0; i < vs.root_nodes.size(); i++) {
- _merge_skeletons(&vs, vs.root_nodes[i]);
- }
- _merge_skeletons2(&vs);
- for (int i = 0; i < vs.root_nodes.size(); i++) {
- _optimize_skeletons(&vs, vs.root_nodes[i]);
- }
- for (int i = 0; i < vs.root_nodes.size(); i++) {
- List<Node *> mgeom;
- if (_move_geometry_to_skeletons(&vs, vs.root_nodes[i], &mgeom)) {
- vs.root_nodes.remove(i);
- i--;
- }
- while (!mgeom.empty()) {
- Node *n = mgeom.front()->get();
- n->parent->children.push_back(n);
- mgeom.pop_front();
- }
- }
- for (int i = 0; i < vs.root_nodes.size(); i++) {
- _find_morph_nodes(&vs, vs.root_nodes[i]);
- }
- }
- }
- int Collada::get_uv_channel(String p_name) {
- if (!channel_map.has(p_name)) {
- ERR_FAIL_COND_V(channel_map.size() == 2, 0);
- channel_map[p_name] = channel_map.size();
- }
- return channel_map[p_name];
- }
- Error Collada::load(const String &p_path, int p_flags) {
- Ref<XMLParser> parserr = memnew(XMLParser);
- XMLParser &parser = *parserr.ptr();
- Error err = parser.open(p_path);
- ERR_FAIL_COND_V_MSG(err, err, "Cannot open Collada file '" + p_path + "'.");
- state.local_path = ProjectSettings::get_singleton()->localize_path(p_path);
- state.import_flags = p_flags;
- /* Skip headers */
- while ((err = parser.read()) == OK) {
- if (parser.get_node_type() == XMLParser::NODE_ELEMENT) {
- if (parser.get_node_name() == "COLLADA") {
- break;
- } else if (!parser.is_empty()) {
- parser.skip_section(); // unknown section, likely headers
- }
- }
- }
- ERR_FAIL_COND_V_MSG(err != OK, ERR_FILE_CORRUPT, "Corrupted Collada file '" + p_path + "'.");
- /* Start loading Collada */
- {
- //version
- String version = parser.get_attribute_value("version");
- state.version.major = version.get_slice(".", 0).to_int();
- state.version.minor = version.get_slice(".", 1).to_int();
- state.version.rev = version.get_slice(".", 2).to_int();
- COLLADA_PRINT("Collada VERSION: " + version);
- }
- while ((err = parser.read()) == OK) {
- /* Read all the main sections.. */
- if (parser.get_node_type() != XMLParser::NODE_ELEMENT) {
- continue; //no idea what this may be, but skipping anyway
- }
- String section = parser.get_node_name();
- COLLADA_PRINT("section: " + section);
- if (section == "asset") {
- _parse_asset(parser);
- } else if (section.begins_with("library_")) {
- _parse_library(parser);
- } else if (section == "scene") {
- _parse_scene(parser);
- } else if (!parser.is_empty()) {
- parser.skip_section(); // unknown section, likely headers
- }
- }
- _optimize();
- return OK;
- }
- Collada::Collada() {
- }
|