123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713 |
- /*************************************************************************/
- /* FBXDocument.cpp */
- /*************************************************************************/
- /* This file is part of: */
- /* GODOT ENGINE */
- /* https://godotengine.org */
- /*************************************************************************/
- /* Copyright (c) 2007-2021 Juan Linietsky, Ariel Manzur. */
- /* Copyright (c) 2014-2021 Godot Engine contributors (cf. AUTHORS.md). */
- /* */
- /* 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. */
- /*************************************************************************/
- /*
- Open Asset Import Library (assimp)
- ----------------------------------------------------------------------
- Copyright (c) 2006-2019, assimp team
- All rights reserved.
- Redistribution and use of this software in source and binary forms,
- with or without modification, are permitted provided that the
- following conditions are met:
- * Redistributions of source code must retain the above
- copyright notice, this list of conditions and the
- following disclaimer.
- * Redistributions in binary form must reproduce the above
- copyright notice, this list of conditions and the*
- following disclaimer in the documentation and/or other
- materials provided with the distribution.
- * Neither the name of the assimp team, nor the names of its
- contributors may be used to endorse or promote products
- derived from this software without specific prior
- written permission of the assimp team.
- THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- ----------------------------------------------------------------------
- */
- /** @file FBXDocument.cpp
- * @brief Implementation of the FBX DOM classes
- */
- #include "FBXDocument.h"
- #include "FBXDocumentUtil.h"
- #include "FBXImportSettings.h"
- #include "FBXMeshGeometry.h"
- #include "FBXParser.h"
- #include "FBXProperties.h"
- #include "FBXUtil.h"
- #include <algorithm>
- #include <functional>
- #include <iostream>
- #include <map>
- #include <memory>
- namespace FBXDocParser {
- using namespace Util;
- // ------------------------------------------------------------------------------------------------
- LazyObject::LazyObject(uint64_t id, const ElementPtr element, const Document &doc) :
- doc(doc), element(element), id(id), flags() {
- // empty
- }
- // ------------------------------------------------------------------------------------------------
- LazyObject::~LazyObject() {
- object.reset();
- }
- ObjectPtr LazyObject::LoadObject() {
- if (IsBeingConstructed() || FailedToConstruct()) {
- return nullptr;
- }
- if (object) {
- return object.get();
- }
- TokenPtr key = element->KeyToken();
- ERR_FAIL_COND_V(!key, nullptr);
- const TokenList &tokens = element->Tokens();
- if (tokens.size() < 3) {
- //DOMError("expected at least 3 tokens: id, name and class tag",&element);
- return nullptr;
- }
- const char *err = nullptr;
- std::string name = ParseTokenAsString(tokens[1], err);
- if (err) {
- DOMError(err, element);
- }
- // small fix for binary reading: binary fbx files don't use
- // prefixes such as Model:: in front of their names. The
- // loading code expects this at many places, though!
- // so convert the binary representation (a 0x0001) to the
- // double colon notation.
- if (tokens[1]->IsBinary()) {
- for (size_t i = 0; i < name.length(); ++i) {
- if (name[i] == 0x0 && name[i + 1] == 0x1) {
- name = name.substr(i + 2) + "::" + name.substr(0, i);
- }
- }
- }
- const std::string classtag = ParseTokenAsString(tokens[2], err);
- if (err) {
- DOMError(err, element);
- }
- // prevent recursive calls
- flags |= BEING_CONSTRUCTED;
- // this needs to be relatively fast since it happens a lot,
- // so avoid constructing strings all the time.
- const char *obtype = key->begin();
- const size_t length = static_cast<size_t>(key->end() - key->begin());
- if (!strncmp(obtype, "Pose", length)) {
- object.reset(new FbxPose(id, element, doc, name));
- } else if (!strncmp(obtype, "Geometry", length)) {
- if (!strcmp(classtag.c_str(), "Mesh")) {
- object.reset(new MeshGeometry(id, element, name, doc));
- }
- if (!strcmp(classtag.c_str(), "Shape")) {
- object.reset(new ShapeGeometry(id, element, name, doc));
- }
- if (!strcmp(classtag.c_str(), "Line")) {
- object.reset(new LineGeometry(id, element, name, doc));
- }
- } else if (!strncmp(obtype, "NodeAttribute", length)) {
- if (!strcmp(classtag.c_str(), "Camera")) {
- object.reset(new Camera(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "CameraSwitcher")) {
- object.reset(new CameraSwitcher(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "Light")) {
- object.reset(new Light(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "Null")) {
- object.reset(new Null(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "LimbNode")) {
- // This is an older format for bones
- // this is what blender uses I believe
- object.reset(new LimbNode(id, element, doc, name));
- }
- } else if (!strncmp(obtype, "Constraint", length)) {
- object.reset(new Constraint(id, element, doc, name));
- } else if (!strncmp(obtype, "Deformer", length)) {
- if (!strcmp(classtag.c_str(), "Cluster")) {
- object.reset(new Cluster(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "Skin")) {
- object.reset(new Skin(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "BlendShape")) {
- object.reset(new BlendShape(id, element, doc, name));
- } else if (!strcmp(classtag.c_str(), "BlendShapeChannel")) {
- object.reset(new BlendShapeChannel(id, element, doc, name));
- }
- } else if (!strncmp(obtype, "Model", length)) {
- // Model is normal node
- // LimbNode model is a 'bone' node.
- if (!strcmp(classtag.c_str(), "LimbNode")) {
- object.reset(new ModelLimbNode(id, element, doc, name));
- } else if (strcmp(classtag.c_str(), "IKEffector") && strcmp(classtag.c_str(), "FKEffector")) {
- // FK and IK effectors are not supporte
- object.reset(new Model(id, element, doc, name));
- }
- } else if (!strncmp(obtype, "Material", length)) {
- object.reset(new Material(id, element, doc, name));
- } else if (!strncmp(obtype, "Texture", length)) {
- object.reset(new Texture(id, element, doc, name));
- } else if (!strncmp(obtype, "LayeredTexture", length)) {
- object.reset(new LayeredTexture(id, element, doc, name));
- } else if (!strncmp(obtype, "Video", length)) {
- object.reset(new Video(id, element, doc, name));
- } else if (!strncmp(obtype, "AnimationStack", length)) {
- object.reset(new AnimationStack(id, element, name, doc));
- } else if (!strncmp(obtype, "AnimationLayer", length)) {
- object.reset(new AnimationLayer(id, element, name, doc));
- } else if (!strncmp(obtype, "AnimationCurve", length)) {
- object.reset(new AnimationCurve(id, element, name, doc));
- } else if (!strncmp(obtype, "AnimationCurveNode", length)) {
- object.reset(new AnimationCurveNode(id, element, name, doc));
- } else {
- ERR_FAIL_V_MSG(nullptr, "FBX contains unsupported object: " + String(obtype));
- }
- flags &= ~BEING_CONSTRUCTED;
- return object.get();
- }
- // ------------------------------------------------------------------------------------------------
- Object::Object(uint64_t id, const ElementPtr element, const std::string &name) :
- element(element), name(name), id(id) {
- }
- // ------------------------------------------------------------------------------------------------
- Object::~Object() {
- // empty
- }
- // ------------------------------------------------------------------------------------------------
- FileGlobalSettings::FileGlobalSettings(const Document &doc, const PropertyTable *props) :
- props(props), doc(doc) {
- // empty
- }
- // ------------------------------------------------------------------------------------------------
- FileGlobalSettings::~FileGlobalSettings() {
- if (props != nullptr) {
- delete props;
- props = nullptr;
- }
- }
- // ------------------------------------------------------------------------------------------------
- Document::Document(const Parser &parser, const ImportSettings &settings) :
- settings(settings), parser(parser), SafeToImport(false) {
- // Cannot use array default initialization syntax because vc8 fails on it
- for (unsigned int &timeStamp : creationTimeStamp) {
- timeStamp = 0;
- }
- // we must check if we can read the header version safely, if its outdated then drop it.
- if (ReadHeader()) {
- SafeToImport = true;
- ReadPropertyTemplates();
- ReadGlobalSettings();
- // This order is important, connections need parsed objects to check
- // whether connections are ok or not. Objects may not be evaluated yet,
- // though, since this may require valid connections.
- ReadObjects();
- ReadConnections();
- }
- }
- // ------------------------------------------------------------------------------------------------
- Document::~Document() {
- for (PropertyTemplateMap::value_type v : templates) {
- delete v.second;
- }
- for (ObjectMap::value_type &v : objects) {
- delete v.second;
- }
- for (ConnectionMap::value_type &v : src_connections) {
- delete v.second;
- }
- if (metadata_properties != nullptr) {
- delete metadata_properties;
- }
- // clear globals import pointer
- globals.reset();
- }
- // ------------------------------------------------------------------------------------------------
- static const unsigned int LowerSupportedVersion = 7300;
- static const unsigned int UpperSupportedVersion = 7700;
- bool Document::ReadHeader() {
- // Read ID objects from "Objects" section
- ScopePtr sc = parser.GetRootScope();
- ElementPtr ehead = sc->GetElement("FBXHeaderExtension");
- if (!ehead || !ehead->Compound()) {
- DOMError("no FBXHeaderExtension dictionary found");
- }
- const ScopePtr shead = ehead->Compound();
- fbxVersion = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(shead, "FBXVersion", ehead), 0));
- // While we may have some success with newer files, we don't support
- // the older 6.n fbx format
- if (fbxVersion < LowerSupportedVersion) {
- DOMWarning("unsupported, old format version, FBX 2015-2020, you must re-export in a more modern version of your original modelling application");
- return false;
- }
- if (fbxVersion > UpperSupportedVersion) {
- DOMWarning("unsupported, newer format version, supported are only FBX 2015, up to FBX 2020"
- " trying to read it nevertheless");
- }
- const ElementPtr ecreator = shead->GetElement("Creator");
- if (ecreator) {
- creator = ParseTokenAsString(GetRequiredToken(ecreator, 0));
- }
- //
- // Scene Info
- //
- const ElementPtr scene_info = shead->GetElement("SceneInfo");
- if (scene_info) {
- PropertyTable *fileExportProps = const_cast<PropertyTable *>(GetPropertyTable(*this, "", scene_info, scene_info->Compound(), true));
- if (fileExportProps) {
- metadata_properties = fileExportProps;
- }
- }
- const ElementPtr etimestamp = shead->GetElement("CreationTimeStamp");
- if (etimestamp && etimestamp->Compound()) {
- const ScopePtr stimestamp = etimestamp->Compound();
- creationTimeStamp[0] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Year"), 0));
- creationTimeStamp[1] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Month"), 0));
- creationTimeStamp[2] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Day"), 0));
- creationTimeStamp[3] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Hour"), 0));
- creationTimeStamp[4] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Minute"), 0));
- creationTimeStamp[5] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Second"), 0));
- creationTimeStamp[6] = ParseTokenAsInt(GetRequiredToken(GetRequiredElement(stimestamp, "Millisecond"), 0));
- }
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- void Document::ReadGlobalSettings() {
- ERR_FAIL_COND_MSG(globals != nullptr, "Global settings is already setup this is a serious error and should be reported");
- const ScopePtr sc = parser.GetRootScope();
- const ElementPtr ehead = sc->GetElement("GlobalSettings");
- if (nullptr == ehead || !ehead->Compound()) {
- DOMWarning("no GlobalSettings dictionary found");
- globals = std::make_shared<FileGlobalSettings>(*this, new PropertyTable());
- return;
- }
- const PropertyTable *props = GetPropertyTable(*this, "", ehead, ehead->Compound(), true);
- //double v = PropertyGet<float>( *props, std::string("UnitScaleFactor"), 1.0 );
- if (!props) {
- DOMError("GlobalSettings dictionary contains no property table");
- }
- globals = std::make_shared<FileGlobalSettings>(*this, props);
- }
- // ------------------------------------------------------------------------------------------------
- void Document::ReadObjects() {
- // read ID objects from "Objects" section
- const ScopePtr sc = parser.GetRootScope();
- const ElementPtr eobjects = sc->GetElement("Objects");
- if (!eobjects || !eobjects->Compound()) {
- DOMError("no Objects dictionary found");
- }
- // add a dummy entry to represent the Model::RootNode object (id 0),
- // which is only indirectly defined in the input file
- objects[0] = new LazyObject(0L, eobjects, *this);
- const ScopePtr sobjects = eobjects->Compound();
- for (const ElementMap::value_type &iter : sobjects->Elements()) {
- // extract ID
- const TokenList &tok = iter.second->Tokens();
- if (tok.empty()) {
- DOMError("expected ID after object key", iter.second);
- }
- const char *err;
- const uint64_t id = ParseTokenAsID(tok[0], err);
- if (err) {
- DOMError(err, iter.second);
- }
- // id=0 is normally implicit
- if (id == 0L) {
- DOMError("encountered object with implicitly defined id 0", iter.second);
- }
- if (objects.find(id) != objects.end()) {
- DOMWarning("encountered duplicate object id, ignoring first occurrence", iter.second);
- }
- objects[id] = new LazyObject(id, iter.second, *this);
- // grab all animation stacks upfront since there is no listing of them
- if (!strcmp(iter.first.c_str(), "AnimationStack")) {
- animationStacks.push_back(id);
- } else if (!strcmp(iter.first.c_str(), "Constraint")) {
- constraints.push_back(id);
- } else if (!strcmp(iter.first.c_str(), "Pose")) {
- bind_poses.push_back(id);
- } else if (!strcmp(iter.first.c_str(), "Material")) {
- materials.push_back(id);
- } else if (!strcmp(iter.first.c_str(), "Deformer")) {
- TokenPtr key = iter.second->KeyToken();
- ERR_CONTINUE_MSG(!key, "[parser bug] invalid token key for deformer");
- const TokenList &tokens = iter.second->Tokens();
- const std::string class_tag = ParseTokenAsString(tokens[2], err);
- if (err) {
- DOMError(err, iter.second);
- }
- if (class_tag == "Skin") {
- //print_verbose("registered skin:" + itos(id));
- skins.push_back(id);
- }
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void Document::ReadPropertyTemplates() {
- const ScopePtr sc = parser.GetRootScope();
- // read property templates from "Definitions" section
- const ElementPtr edefs = sc->GetElement("Definitions");
- if (!edefs || !edefs->Compound()) {
- DOMWarning("no Definitions dictionary found");
- return;
- }
- const ScopePtr sdefs = edefs->Compound();
- const ElementCollection otypes = sdefs->GetCollection("ObjectType");
- for (ElementMap::const_iterator it = otypes.first; it != otypes.second; ++it) {
- const ElementPtr el = (*it).second;
- const ScopePtr sc_2 = el->Compound();
- if (!sc_2) {
- DOMWarning("expected nested scope in ObjectType, ignoring", el);
- continue;
- }
- const TokenList &tok = el->Tokens();
- if (tok.empty()) {
- DOMWarning("expected name for ObjectType element, ignoring", el);
- continue;
- }
- const std::string &oname = ParseTokenAsString(tok[0]);
- const ElementCollection templs = sc_2->GetCollection("PropertyTemplate");
- for (ElementMap::const_iterator iter = templs.first; iter != templs.second; ++iter) {
- const ElementPtr el_2 = (*iter).second;
- const ScopePtr sc_3 = el_2->Compound();
- if (!sc_3) {
- DOMWarning("expected nested scope in PropertyTemplate, ignoring", el);
- continue;
- }
- const TokenList &tok_2 = el_2->Tokens();
- if (tok_2.empty()) {
- DOMWarning("expected name for PropertyTemplate element, ignoring", el);
- continue;
- }
- const std::string &pname = ParseTokenAsString(tok_2[0]);
- const ElementPtr Properties70 = sc_3->GetElement("Properties70");
- if (Properties70) {
- // PropertyTable(const ElementPtr element, const PropertyTable* templateProps);
- const PropertyTable *props = new PropertyTable(Properties70, nullptr);
- templates[oname + "." + pname] = props;
- }
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void Document::ReadConnections() {
- const ScopePtr sc = parser.GetRootScope();
- // read property templates from "Definitions" section
- const ElementPtr econns = sc->GetElement("Connections");
- if (!econns || !econns->Compound()) {
- DOMError("no Connections dictionary found");
- }
- uint64_t insertionOrder = 0l;
- const ScopePtr sconns = econns->Compound();
- const ElementCollection conns = sconns->GetCollection("C");
- for (ElementMap::const_iterator it = conns.first; it != conns.second; ++it) {
- const ElementPtr el = (*it).second;
- const std::string &type = ParseTokenAsString(GetRequiredToken(el, 0));
- // PP = property-property connection, ignored for now
- // (tokens: "PP", ID1, "Property1", ID2, "Property2")
- if (type == "PP") {
- continue;
- }
- const uint64_t src = ParseTokenAsID(GetRequiredToken(el, 1));
- const uint64_t dest = ParseTokenAsID(GetRequiredToken(el, 2));
- // OO = object-object connection
- // OP = object-property connection, in which case the destination property follows the object ID
- const std::string &prop = (type == "OP" ? ParseTokenAsString(GetRequiredToken(el, 3)) : "");
- if (objects.find(src) == objects.end()) {
- DOMWarning("source object for connection does not exist", el);
- continue;
- }
- // dest may be 0 (root node) but we added a dummy object before
- if (objects.find(dest) == objects.end()) {
- DOMWarning("destination object for connection does not exist", el);
- continue;
- }
- // add new connection
- const Connection *const c = new Connection(insertionOrder++, src, dest, prop, *this);
- src_connections.insert(ConnectionMap::value_type(src, c));
- dest_connections.insert(ConnectionMap::value_type(dest, c));
- }
- }
- // ------------------------------------------------------------------------------------------------
- const std::vector<const AnimationStack *> &Document::AnimationStacks() const {
- if (!animationStacksResolved.empty() || animationStacks.empty()) {
- return animationStacksResolved;
- }
- animationStacksResolved.reserve(animationStacks.size());
- for (uint64_t id : animationStacks) {
- LazyObject *lazy = GetObject(id);
- // Two things happen here:
- // We cast internally an Object PTR to an Animation Stack PTR
- // We return invalid weak_ptrs for objects which are invalid
- const AnimationStack *stack = lazy->Get<AnimationStack>();
- ERR_CONTINUE_MSG(!stack, "invalid ptr to AnimationStack - conversion failure");
- // We push back the weak reference :) to keep things simple, as ownership is on the parser side so it wont be cleaned up.
- animationStacksResolved.push_back(stack);
- }
- return animationStacksResolved;
- }
- // ------------------------------------------------------------------------------------------------
- LazyObject *Document::GetObject(uint64_t id) const {
- ObjectMap::const_iterator it = objects.find(id);
- return it == objects.end() ? nullptr : (*it).second;
- }
- #define MAX_CLASSNAMES 6
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsSequenced(uint64_t id, const ConnectionMap &conns) const {
- std::vector<const Connection *> temp;
- const std::pair<ConnectionMap::const_iterator, ConnectionMap::const_iterator> range =
- conns.equal_range(id);
- temp.reserve(std::distance(range.first, range.second));
- for (ConnectionMap::const_iterator it = range.first; it != range.second; ++it) {
- temp.push_back((*it).second);
- }
- std::sort(temp.begin(), temp.end(), std::mem_fn(&Connection::Compare));
- return temp; // NRVO should handle this
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsSequenced(uint64_t id, bool is_src,
- const ConnectionMap &conns,
- const char *const *classnames,
- size_t count) const
- {
- size_t lengths[MAX_CLASSNAMES];
- const size_t c = count;
- for (size_t i = 0; i < c; ++i) {
- lengths[i] = strlen(classnames[i]);
- }
- std::vector<const Connection *> temp;
- const std::pair<ConnectionMap::const_iterator, ConnectionMap::const_iterator> range =
- conns.equal_range(id);
- temp.reserve(std::distance(range.first, range.second));
- for (ConnectionMap::const_iterator it = range.first; it != range.second; ++it) {
- TokenPtr key = (is_src ? (*it).second->LazyDestinationObject() : (*it).second->LazySourceObject())->GetElement()->KeyToken();
- const char *obtype = key->begin();
- for (size_t i = 0; i < c; ++i) {
- //ai_assert(classnames[i]);
- if (static_cast<size_t>(std::distance(key->begin(), key->end())) == lengths[i] && !strncmp(classnames[i], obtype, lengths[i])) {
- obtype = nullptr;
- break;
- }
- }
- if (obtype) {
- continue;
- }
- temp.push_back((*it).second);
- }
- std::sort(temp.begin(), temp.end(), std::mem_fn(&Connection::Compare));
- return temp; // NRVO should handle this
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsBySourceSequenced(uint64_t source) const {
- return GetConnectionsSequenced(source, ConnectionsBySource());
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsBySourceSequenced(uint64_t src, const char *classname) const {
- const char *arr[] = { classname };
- return GetConnectionsBySourceSequenced(src, arr, 1);
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsBySourceSequenced(uint64_t source,
- const char *const *classnames, size_t count) const {
- return GetConnectionsSequenced(source, true, ConnectionsBySource(), classnames, count);
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsByDestinationSequenced(uint64_t dest,
- const char *classname) const {
- const char *arr[] = { classname };
- return GetConnectionsByDestinationSequenced(dest, arr, 1);
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsByDestinationSequenced(uint64_t dest) const {
- return GetConnectionsSequenced(dest, ConnectionsByDestination());
- }
- // ------------------------------------------------------------------------------------------------
- std::vector<const Connection *> Document::GetConnectionsByDestinationSequenced(uint64_t dest,
- const char *const *classnames, size_t count) const {
- return GetConnectionsSequenced(dest, false, ConnectionsByDestination(), classnames, count);
- }
- // ------------------------------------------------------------------------------------------------
- Connection::Connection(uint64_t insertionOrder, uint64_t src, uint64_t dest, const std::string &prop,
- const Document &doc) :
- insertionOrder(insertionOrder), prop(prop), src(src), dest(dest), doc(doc) {
- }
- // ------------------------------------------------------------------------------------------------
- Connection::~Connection() {
- // empty
- }
- // ------------------------------------------------------------------------------------------------
- LazyObject *Connection::LazySourceObject() const {
- LazyObject *const lazy = doc.GetObject(src);
- return lazy;
- }
- // ------------------------------------------------------------------------------------------------
- LazyObject *Connection::LazyDestinationObject() const {
- LazyObject *const lazy = doc.GetObject(dest);
- return lazy;
- }
- // ------------------------------------------------------------------------------------------------
- Object *Connection::SourceObject() const {
- LazyObject *lazy = doc.GetObject(src);
- //ai_assert(lazy);
- return lazy->LoadObject();
- }
- // ------------------------------------------------------------------------------------------------
- Object *Connection::DestinationObject() const {
- LazyObject *lazy = doc.GetObject(dest);
- //ai_assert(lazy);
- return lazy->LoadObject();
- }
- } // namespace FBXDocParser
|