You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
o3de/Gems/MotionMatching/Code/Source/PoseDataJointVelocities.h

61 lines
2.3 KiB
C++

/*
* Copyright (c) Contributors to the Open 3D Engine Project.
* For complete copyright and license terms please see the LICENSE at the root of this distribution.
*
* SPDX-License-Identifier: Apache-2.0 OR MIT
*
*/
#pragma once
#include <AzCore/Math/Vector3.h>
#include <AzCore/Math/Quaternion.h>
#include <EMotionFX/Source/PoseData.h>
#include <AzFramework/Entity/EntityDebugDisplayBus.h>
namespace EMotionFX::MotionMatching
{
/**
* Extends a given pose with joint-relative linear and angular velocities.
**/
class EMFX_API PoseDataJointVelocities
: public PoseData
{
public:
AZ_RTTI(PoseDataJointVelocities, "{9C082B82-7225-4550-A52C-C920CCC2482C}", PoseData)
AZ_CLASS_ALLOCATOR_DECL
PoseDataJointVelocities();
~PoseDataJointVelocities();
void Clear();
void LinkToActorInstance(const ActorInstance* actorInstance) override;
void LinkToActor(const Actor* actor) override;
void Reset() override;
void CopyFrom(const PoseData* from) override;
void Blend(const Pose* destPose, float weight) override;
void CalculateVelocity(MotionInstance* motionInstance, size_t relativeToJointIndex);
void DebugDraw(AzFramework::DebugDisplayRequests& debugDisplay, const AZ::Color& color) const override;
AZStd::vector<AZ::Vector3>& GetVelocities() { return m_velocities; }
const AZStd::vector<AZ::Vector3>& GetVelocities() const { return m_velocities; }
const AZ::Vector3& GetVelocity(size_t jointIndex) { return m_velocities[jointIndex]; }
AZStd::vector<AZ::Vector3>& GetAngularVelocities() { return m_angularVelocities; }
const AZStd::vector<AZ::Vector3>& GetAngularVelocities() const { return m_angularVelocities; }
const AZ::Vector3& GetAngularVelocity(size_t jointIndex) { return m_angularVelocities[jointIndex]; }
static void Reflect(AZ::ReflectContext* context);
void SetRelativeToJointIndex(size_t relativeToJointIndex);
private:
AZStd::vector<AZ::Vector3> m_velocities;
AZStd::vector<AZ::Vector3> m_angularVelocities;
size_t m_relativeToJointIndex = InvalidIndex;
};
} // namespace EMotionFX::MotionMatching