Code/Sandbox/Plugins/MeshImporter/SkeletonPoser/SkeletonPoser.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "SkeletonPoser.h"
   5
   6 CSkeletonPoseModifier::SJoint::SJoint()
   7         : m_crc(0)
   8         , m_transform(IDENTITY)
   9         , m_id(-1)
  10 {}
  11
  12 CSkeletonPoseModifier::SJoint::SJoint(const string& name, const QuatT& transform)
  13         : m_name(name)
  14         , m_transform(transform)
  15         , m_id(-1)
  16 {
  17         m_crc = m_name.length() ? CCrc32::ComputeLowercase(m_name.c_str()) : 0;
  18 }
  19
  20 void CSkeletonPoseModifier::PoseJoint(const string& name, const QuatT& transform)
  21 {
  22         auto it = std::find_if(m_posedJoints.begin(), m_posedJoints.end(), [name](const SJoint& joint)
  23         {
  24                 return joint.m_name == name;
  25         });
  26         if (it != m_posedJoints.end())
  27         {
  28                 it->m_transform = transform;
  29         }
  30         else
  31         {
  32                 m_posedJoints.emplace_back(name, transform);
  33         }
  34 }
  35
  36 bool CSkeletonPoseModifier::Prepare(const SAnimationPoseModifierParams& params)
  37 {
  38         const IDefaultSkeleton& skeleton = params.GetIDefaultSkeleton();
  39
  40         for (auto& joint : m_posedJoints)
  41         {
  42                 if (joint.m_id == -1)
  43                 {
  44                         joint.m_id = skeleton.GetJointIDByCRC32(joint.m_crc);
  45                 }
  46         }
  47
  48         return true;
  49 }
  50
  51 void ComputeAbsolutJointTransformationsRecursive(int joint, const IDefaultSkeleton& skel, IAnimationPoseData* pPoseData)
  52 {
  53         const int parent = skel.GetJointParentIDByID(joint);
  54         if (parent > -1)
  55         {
  56                 pPoseData->SetJointAbsolute(joint, pPoseData->GetJointAbsolute(parent) * pPoseData->GetJointRelative(joint));
  57         }
  58
  59         for (int i = 0, N = skel.GetJointChildrenCountByID(joint); i < N; ++i)
  60         {
  61                 ComputeAbsolutJointTransformationsRecursive(skel.GetJointChildIDAtIndexByID(joint, i), skel, pPoseData);
  62         }
  63 }
  64
  65 void ComputeAbsolutJointTransformations(const std::vector<int>& dirtyRoots, const IDefaultSkeleton& skel, IAnimationPoseData* pPoseData)
  66 {
  67         for (int dirtyRoot : dirtyRoots)
  68         {
  69                 ComputeAbsolutJointTransformationsRecursive(dirtyRoot, skel, pPoseData);
  70         }
  71 }
  72
  73 bool CSkeletonPoseModifier::Execute(const SAnimationPoseModifierParams& params)
  74 {
  75         IAnimationPoseData* const pPoseData = params.pPoseData;
  76
  77         const IDefaultSkeleton& skel = params.GetIDefaultSkeleton();
  78
  79         const int jointCount = skel.GetJointCount();
  80
  81         std::vector<SJoint> validPosedJoints;
  82         std::copy_if(m_posedJoints.begin(), m_posedJoints.end(), std::back_inserter(validPosedJoints), [jointCount](const SJoint& joint)
  83         {
  84                 return joint.m_id > -1 && joint.m_id < jointCount;
  85         });
  86
  87         std::vector<bool> dirty(jointCount, false);
  88         for (auto& joint : validPosedJoints)
  89         {
  90                 const QuatT rel = pPoseData->GetJointRelative(joint.m_id);
  91                 pPoseData->SetJointRelative(joint.m_id, rel * joint.m_transform);
  92                 dirty[joint.m_id] = true;
  93         }
  94
  95         std::vector<bool> visited(jointCount, false);
  96         std::vector<int> dirtyRoots;
  97         for (auto& joint : validPosedJoints)
  98         {
  99                 if (visited[joint.m_id])
 100                 {
 101                         continue;
 102                 }
 103
 104                 visited[joint.m_id] = true;
 105
 106                 int dirtyRoot = joint.m_id;
 107                 int pathIt = skel.GetJointParentIDByID(joint.m_id);
 108                 while (pathIt > -1 && !visited[pathIt])
 109                 {
 110                         if (dirty[pathIt])
 111                         {
 112                                 dirtyRoot = pathIt;
 113                         }
 114                         visited[pathIt] = true;
 115                         pathIt = skel.GetJointParentIDByID(pathIt);
 116                 }
 117                 if (pathIt == -1)
 118                 {
 119                         dirtyRoots.push_back(dirtyRoot);
 120                 }
 121         }
 122
 123         ComputeAbsolutJointTransformations(dirtyRoots, skel, pPoseData);
 124
 125         return true;
 126 }
 127
 128 CRYREGISTER_CLASS(CSkeletonPoseModifier)