Changeset 11153 for OpenSceneGraph/trunk/src/osgPlugins/fbx/ReaderWriterFBX.cpp

Timestamp:

03/04/10 17:27:19 (3 years ago)

Author:

mplatings

Message:

Fix for some FBX files with multiple meshes bound to a bone.

Files:

• OpenSceneGraph/trunk/src/osgPlugins/fbx/ReaderWriterFBX.cpp (modified) (4 diffs)

OpenSceneGraph/trunk/src/osgPlugins/fbx/ReaderWriterFBX.cpp

@@ -15 +15 @@
 #include <osgAnimation/AnimationManagerBase>
 #include <osgAnimation/Bone>
+#include <osgAnimation/RigGeometry>
 #include <osgAnimation/Skeleton>
+#include <osgAnimation/VertexInfluence>
 
 #if defined(_MSC_VER)
@@ -96 +98 @@
     }
 };
+
+void resolveBindMatrices(
+    osg::Node& root,
+    const BindMatrixMap& boneBindMatrices,
+    const std::map<KFbxNode*, osg::Node*>& nodeMap)
+{
+    std::set<std::string> nodeNames;
+    for (std::map<KFbxNode*, osg::Node*>::const_iterator it = nodeMap.begin(); it != nodeMap.end(); ++it)
+    {
+        nodeNames.insert(it->second->getName());
+    }
+
+    for (BindMatrixMap::const_iterator it = boneBindMatrices.begin();
+        it != boneBindMatrices.end();)
+    {
+        KFbxNode* const fbxBone = it->first.first;
+        std::map<KFbxNode*, osg::Node*>::const_iterator nodeIt = nodeMap.find(fbxBone);
+        if (nodeIt != nodeMap.end())
+        {
+            const osg::Matrix bindMatrix = it->second;
+            osgAnimation::Bone& osgBone = dynamic_cast<osgAnimation::Bone&>(*nodeIt->second);
+            osgBone.setInvBindMatrixInSkeletonSpace(bindMatrix);
+
+            ++it;
+            for (; it != boneBindMatrices.end() && it->first.first == fbxBone; ++it)
+            {
+                if (it->second != bindMatrix)
+                {
+                    std::string name;
+                    for (int i = 0;; ++i)
+                    {
+                        std::stringstream ss;
+                        ss << osgBone.getName() << '_' << i;
+                        name = ss.str();
+                        if (nodeNames.insert(name).second)
+                        {
+                            break;
+                        }
+                    }
+                    osgAnimation::Bone* newBone = new osgAnimation::Bone(name);
+                    newBone->setDefaultUpdateCallback();
+                    newBone->setInvBindMatrixInSkeletonSpace(it->second);
+                    osgBone.addChild(newBone);
+
+                    osgAnimation::RigGeometry* pRigGeometry = it->first.second;
+                    
+                    osgAnimation::VertexInfluenceMap* vertexInfluences = pRigGeometry->getInfluenceMap();
+
+                    osgAnimation::VertexInfluenceMap::iterator vimIt = vertexInfluences->find(osgBone.getName());
+                    if (vimIt != vertexInfluences->end())
+                    {
+                        osgAnimation::VertexInfluence vi;
+                        vi.swap(vimIt->second);
+                        vertexInfluences->erase(vimIt);
+                        osgAnimation::VertexInfluence& vi2 = (*vertexInfluences)[name];
+                        vi.swap(vi2);
+                        vi2.setName(name);
+                    }
+                    else
+                    {
+                        assert(0);
+                    }
+                }
+            }
+        }
+        else
+        {
+            assert(0);
+        }
+    }
+}
 
 osgDB::ReaderWriter::ReadResult
@@ -190 +263 @@
 
             std::map<KFbxNode*, osg::Node*> nodeMap;
-            std::map<KFbxNode*, osg::Matrix> boneBindMatrices;
+            BindMatrixMap boneBindMatrices;
             std::map<KFbxNode*, osgAnimation::Skeleton*> skeletonMap;
             ReadResult res = readFbxNode(*pSdkManager, pNode, pAnimationManager,
@@ -198 +271 @@
             if (res.success())
             {
-                for (std::map<KFbxNode*, osg::Matrix>::const_iterator it = boneBindMatrices.begin();
-                    it != boneBindMatrices.end(); ++it)
-                {
-                    std::map<KFbxNode*, osg::Node*>::iterator nodeIt = nodeMap.find(it->first);
-                    if (nodeIt != nodeMap.end())
-                    {
-                        osgAnimation::Bone& osgBone = dynamic_cast<osgAnimation::Bone&>(*nodeIt->second);
-                        osgBone.setInvBindMatrixInSkeletonSpace(it->second);
-                    }
-                    else
-                    {
-                        assert(0);
-                    }
-                }
+                resolveBindMatrices(*res.getNode(), boneBindMatrices, nodeMap);
 
                 osg::Node* osgNode = res.getNode();