Factor out local poses into their own vector

To make it easier and more efficient to send all the local poses for
a skeleton between aspects.

Change-Id: I68bf374a7d466f606f8979fdbf84cb697a47a8a7
Reviewed-by: Paul Lemire <paul.lemire@kdab.com>

5 files changed, 25 insertions, 16 deletions

diff --git a/src/render/geometry/gltfskeletonloader.cpp b/src/render/geometry/gltfskeletonloader.cpp
index 428f7bc74..76601836a 100644
--- a/src/render/geometry/gltfskeletonloader.cpp
+++ b/src/render/geometry/gltfskeletonloader.cpp
@@ -373,13 +373,13 @@ SkeletonData GLTFSkeletonLoader::createSkeletonFromSkin(Skin *skin) const
         jointIndexMap.insert(node, i);
 
         JointInfo joint;
-        joint.localPose = node->localTransform;
         joint.inverseBindPose = inverseBindMatrix(skin, i);
         joint.parentIndex = jointIndexMap.value(&m_nodes[node->parentNodeIndex], -1);
         if (joint.parentIndex == -1 && i != 0)
             qCDebug(Jobs) << "Cannot find parent joint for joint" << i;
 
         skel.joints.push_back(joint);
+        skel.localPoses.push_back(node->localTransform);
         skel.jointNames.push_back(node->name);
     }
 
diff --git a/src/render/geometry/skeleton.cpp b/src/render/geometry/skeleton.cpp
index f1761bc18..f672f867e 100644
--- a/src/render/geometry/skeleton.cpp
+++ b/src/render/geometry/skeleton.cpp
@@ -297,9 +297,10 @@ Qt3DCore::QJoint *Skeleton::createFrontendJoints(const SkeletonData &skeletonDat
     const int jointCount = skeletonData.joints.size();
     frontendJoints.reserve(jointCount);
     for (int i = 0; i < jointCount; ++i) {
-        const JointInfo &jointInfo = skeletonData.joints[i];
+        const QMatrix4x4 &inverseBindMatrix = skeletonData.joints[i].inverseBindPose;
         const QString &jointName = skeletonData.jointNames[i];
-        frontendJoints.push_back(createFrontendJoint(jointName, jointInfo));
+        const Qt3DCore::Sqt &localPose = skeletonData.localPoses[i];
+        frontendJoints.push_back(createFrontendJoint(jointName, localPose, inverseBindMatrix));
     }
 
     // Now go through and resolve the parent for each joint
@@ -317,13 +318,15 @@ Qt3DCore::QJoint *Skeleton::createFrontendJoints(const SkeletonData &skeletonDat
     return frontendJoints[0];
 }
 
-Qt3DCore::QJoint *Skeleton::createFrontendJoint(const QString &jointName, const JointInfo &jointInfo) const
+Qt3DCore::QJoint *Skeleton::createFrontendJoint(const QString &jointName,
+                                                const Qt3DCore::Sqt &localPose,
+                                                const QMatrix4x4 &inverseBindMatrix) const
 {
     auto joint = QAbstractNodeFactory::createNode<QJoint>("QJoint");
-    joint->setTranslation(jointInfo.localPose.translation);
-    joint->setRotation(jointInfo.localPose.rotation);
-    joint->setScale(jointInfo.localPose.scale);
-    joint->setInverseBindMatrix(jointInfo.inverseBindPose);
+    joint->setTranslation(localPose.translation);
+    joint->setRotation(localPose.rotation);
+    joint->setScale(localPose.scale);
+    joint->setInverseBindMatrix(inverseBindMatrix);
     joint->setName(jointName);
     return joint;
 }
@@ -338,6 +341,7 @@ void Skeleton::processJointHierarchy(Qt3DCore::QNodeId jointId,
     joint->setOwningSkeleton(m_skeletonHandle);
     const JointInfo jointInfo(joint, parentJointIndex);
     skeletonData.joints.push_back(jointInfo);
+    skeletonData.localPoses.push_back(joint->localPose());
     skeletonData.jointNames.push_back(joint->name());
 
     const int jointIndex = skeletonData.joints.size() - 1;
@@ -353,6 +357,7 @@ void Skeleton::clearData()
 {
     m_name.clear();
     m_skeletonData.joints.clear();
+    m_skeletonData.localPoses.clear();
     m_skeletonData.jointNames.clear();
     m_skeletonData.jointIndices.clear();
 }
@@ -364,19 +369,21 @@ void Skeleton::setLocalPose(HJoint jointHandle, const Qt3DCore::Sqt &localPose)
     // and set the local pose
     const int jointIndex = m_skeletonData.jointIndices.value(jointHandle, -1);
     Q_ASSERT(jointIndex != -1);
-    m_skeletonData.joints[jointIndex].localPose = localPose;
+    m_skeletonData.localPoses[jointIndex] = localPose;
 }
 
 QVector<QMatrix4x4> Skeleton::calculateSkinningMatrixPalette()
 {
+    const QVector<Sqt> &localPoses = m_skeletonData.localPoses;
+    QVector<JointInfo> &joints = m_skeletonData.joints;
     for (int i = 0; i < m_skeletonData.joints.size(); ++i) {
         // Calculate the global pose of this joint
-        JointInfo &joint = m_skeletonData.joints[i];
+        JointInfo &joint = joints[i];
         if (joint.parentIndex == -1) {
-            joint.globalPose = joint.localPose.toMatrix();
+            joint.globalPose = localPoses[i].toMatrix();
         } else {
-            JointInfo &parentJoint = m_skeletonData.joints[joint.parentIndex];
-            joint.globalPose = parentJoint.globalPose * joint.localPose.toMatrix();
+            JointInfo &parentJoint = joints[joint.parentIndex];
+            joint.globalPose = parentJoint.globalPose * localPoses[i].toMatrix();
         }
 
         m_skinningPalette[i] = joint.globalPose * joint.inverseBindPose;
diff --git a/src/render/geometry/skeleton_p.h b/src/render/geometry/skeleton_p.h
index 566304114..e1611d21a 100644
--- a/src/render/geometry/skeleton_p.h
+++ b/src/render/geometry/skeleton_p.h
@@ -123,7 +123,9 @@ private:
     void loadSkeletonFromUrl();
     void loadSkeletonFromData();
     Qt3DCore::QJoint *createFrontendJoints(const SkeletonData &skeletonData) const;
-    Qt3DCore::QJoint *createFrontendJoint(const QString &jointName, const JointInfo &jointInfo) const;
+    Qt3DCore::QJoint *createFrontendJoint(const QString &jointName,
+                                          const Qt3DCore::Sqt &localPose,
+                                          const QMatrix4x4 &inverseBindMatrix) const;
     void processJointHierarchy(Qt3DCore::QNodeId jointId, int parentJointIndex, SkeletonData &skeletonData);
     void clearData();
 
diff --git a/src/render/geometry/skeletondata.cpp b/src/render/geometry/skeletondata.cpp
index 8196856b6..bbb59e82d 100644
--- a/src/render/geometry/skeletondata.cpp
+++ b/src/render/geometry/skeletondata.cpp
@@ -48,6 +48,7 @@ SkeletonData::SkeletonData()
 void SkeletonData::reserve(int size)
 {
     joints.reserve(size);
+    localPoses.reserve(size);
     jointNames.reserve(size);
 }
 
diff --git a/src/render/geometry/skeletondata_p.h b/src/render/geometry/skeletondata_p.h
index 8de29d4ac..b30a3c6d4 100644
--- a/src/render/geometry/skeletondata_p.h
+++ b/src/render/geometry/skeletondata_p.h
@@ -71,13 +71,11 @@ struct Q_AUTOTEST_EXPORT JointInfo
 
     explicit JointInfo(Joint *joint, int parentJointIndex)
         : inverseBindPose(joint->inverseBindMatrix())
-        , localPose(joint->localPose())
         , parentIndex(parentJointIndex)
     {
     }
 
     QMatrix4x4 inverseBindPose;
-    Qt3DCore::Sqt localPose;
     QMatrix4x4 globalPose;
     int parentIndex;
 };
@@ -89,6 +87,7 @@ struct Q_AUTOTEST_EXPORT SkeletonData
     void reserve(int size);
 
     QVector<JointInfo> joints;
+    QVector<Qt3DCore::Sqt> localPoses;
     QVector<QString> jointNames;
     QHash<HJoint, int> jointIndices;
 };