- fbx: read object-object and object-property connections.

code/FBXDocument.cpp

@@ -433,6 +433,7 @@ Object::~Object()
 
 }
 
+
 // ------------------------------------------------------------------------------------------------
 Geometry::Geometry(const Element& element, const std::string& name)
 : Object(element,name)
@@ -440,6 +441,7 @@ Geometry::Geometry(const Element& element, const std::string& name)
 
 }
 
+
 // ------------------------------------------------------------------------------------------------
 Geometry::~Geometry()
 {
@@ -454,8 +456,10 @@ Document::Document(const Parser& parser, const ImportSettings& settings)
 {
 	ReadPropertyTemplates();
 	ReadObjects();
+	ReadConnections();
 }
 
+
 // ------------------------------------------------------------------------------------------------
 Document::~Document()
 {
@@ -464,6 +468,7 @@ Document::~Document()
 	}
 }
 
+
 // ------------------------------------------------------------------------------------------------
 void Document::ReadObjects()
 {
@@ -497,6 +502,7 @@ void Document::ReadObjects()
 	}
 }
 
+
 // ------------------------------------------------------------------------------------------------
 void Document::ReadPropertyTemplates()
 {
@@ -555,6 +561,94 @@ void Document::ReadPropertyTemplates()
 	}
 }
 
+
+
+// ------------------------------------------------------------------------------------------------
+void Document::ReadConnections()
+{
+	const Scope& sc = parser.GetRootScope();
+	// read property templates from "Definitions" section
+	const Element* const econns = sc["Connections"];
+	if(!econns || !econns->Compound()) {
+		DOMError("no Connections dictionary found");
+	}
+
+	uint64_t insertionOrder = 0l;
+
+	const Scope& sconns = *econns->Compound();
+	const ElementCollection conns = sconns.GetCollection("C");
+	for(ElementMap::const_iterator it = conns.first; it != conns.second; ++it) {
+		const Element& el = *(*it).second;
+		const std::string& type = ParseTokenAsString(GetRequiredToken(el,0));
+		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;
+		}
+
+		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));  
+	}
+}
+
+
+// ------------------------------------------------------------------------------------------------
+LazyObject* Document::GetObject(uint64_t id) const
+{
+	ObjectMap::const_iterator it = objects.find(id);
+	return it == objects.end() ? NULL : (*it).second;
+}
+
+
+// ------------------------------------------------------------------------------------------------
+Connection::Connection(uint64_t insertionOrder,  uint64_t src, uint64_t dest, const std::string& prop, const Document& doc)
+: insertionOrder(insertionOrder)
+, src(src)
+, dest(dest)
+, prop(prop)
+, doc(doc)
+{
+	ai_assert(doc.Objects().find(src) != doc.Objects().end());
+	ai_assert(doc.Objects().find(dest) != doc.Objects().end());
+}
+
+
+// ------------------------------------------------------------------------------------------------
+Connection::~Connection()
+{
+
+}
+
+
+// ------------------------------------------------------------------------------------------------
+const Object* Connection::SourceObject() const
+{
+	LazyObject* const lazy = doc.GetObject(src);
+	ai_assert(lazy);
+	return lazy->Get();
+}
+
+// ------------------------------------------------------------------------------------------------
+const Object* Connection::DestinationObject() const
+{
+	LazyObject* const lazy = doc.GetObject(dest);
+	ai_assert(lazy);
+	return lazy->Get();
+}
+
 } // !FBX
 } // !Assimp