Migration to new ANIM extract

2025-12-08 20:27:43 +00:00 · 2016-04-06 17:40:25 -10:00
parent cd8b6b6ed4
commit 9ee8840b54
29 changed files with 691 additions and 427 deletions
--- a/DataSpec/DNACommon/RigInverter.cpp
+++ b/DataSpec/DNACommon/RigInverter.cpp
@@ -0,0 +1,114 @@
+#include "RigInverter.hpp"
+#include "DataSpec/DNAMP1/CINF.hpp"
+#include "DataSpec/DNAMP2/CINF.hpp"
+#include "DataSpec/DNAMP3/CINF.hpp"
+
+namespace DataSpec
+{
+namespace DNAANIM
+{
+
+template <class CINFType>
+RigInverter<CINFType>::Bone::Bone(const CINFType& cinf, const typename CINFType::Bone& origBone)
+: m_origBone(origBone)
+{
+    atUint32 parentIdx = cinf.getInternalBoneIdxFromId(origBone.parentId);
+    zeus::CVector3f naturalTail = zeus::CVector3f(origBone.origin) + zeus::CVector3f{0.f, 0.5f, 0.f};
+    if (parentIdx != -1)
+    {
+        const typename CINFType::Bone& pBone = cinf.bones[parentIdx];
+        m_parentDelta = zeus::CVector3f(origBone.origin) - zeus::CVector3f(pBone.origin);
+    }
+    
+    size_t actualChildren = 0;
+    for (atUint32 chId : origBone.linked)
+    {
+        if (chId == origBone.parentId)
+            continue;
+        atUint32 chIdx = cinf.getInternalBoneIdxFromId(chId);
+        if (chIdx != -1)
+            ++actualChildren;
+    }
+
+    if (parentIdx == -1)
+    {
+        /* Root will always use +Y tail */
+        m_tail = naturalTail;
+    }
+    else if (actualChildren)
+    {
+        /* Position tail to average of children */
+        for (atUint32 chId : origBone.linked)
+        {
+            if (chId == origBone.parentId)
+                continue;
+            atUint32 chIdx = cinf.getInternalBoneIdxFromId(chId);
+            if (chIdx != -1)
+            {
+                const typename CINFType::Bone& chBone = cinf.bones[chIdx];
+                m_tail += chBone.origin;
+            }
+        }
+        m_tail /= float(actualChildren);
+        if (m_tail.magSquared() < 0.001f)
+            m_tail = naturalTail;
+        else
+            m_inverter = zeus::CQuaternion(m_tail, naturalTail);
+    }
+    else if (parentIdx != -1)
+    {
+        /* Extrapolate by delta with parent */
+        const typename CINFType::Bone& pBone = cinf.bones[parentIdx];
+        m_tail = zeus::CVector3f(origBone.origin) * 2.f - zeus::CVector3f(pBone.origin);
+        if (m_tail.magSquared() < 0.001f)
+            m_tail = naturalTail;
+        else
+            m_inverter = zeus::CQuaternion(m_tail, naturalTail);
+    }
+    else
+    {
+        /* Fallback to +Y tail */
+        m_tail = naturalTail;
+    }
+}
+
+template <class CINFType>
+RigInverter<CINFType>::RigInverter(const CINFType& cinf)
+: m_cinf(cinf)
+{
+    m_bones.reserve(cinf.bones.size());
+    for (const typename CINFType::Bone& b : cinf.bones)
+        m_bones.emplace_back(cinf, b);
+}
+
+template <class CINFType>
+zeus::CQuaternion
+RigInverter<CINFType>::transformRotation(atUint32 boneId, const zeus::CQuaternion& origRot) const
+{
+    for (const Bone& b : m_bones)
+        if (b.m_origBone.id == boneId)
+            return b.m_inverter * origRot;
+    return origRot;
+}
+    
+template <class CINFType>
+zeus::CVector3f
+RigInverter<CINFType>::transformPosition(atUint32 boneId, const zeus::CVector3f& origPos, bool subDelta) const
+{
+    for (const Bone& b : m_bones)
+        if (b.m_origBone.id == boneId)
+        {
+            zeus::CVector3f localPos = origPos;
+            if (subDelta)
+                localPos -= b.m_parentDelta;
+            return zeus::CQuaternion::rotate(b.m_inverter, localPos);
+        }
+    return origPos;
+}
+
+template class RigInverter<DNAMP1::CINF>;
+template class RigInverter<DNAMP2::CINF>;
+template class RigInverter<DNAMP3::CINF>;
+
+}
+}