initial commit, 4.5 stable

thirdparty/jolt_physics/Jolt/Skeleton/SkeletonPose.h

@@ -0,0 +1,82 @@
+// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
+// SPDX-FileCopyrightText: 2021 Jorrit Rouwe
+// SPDX-License-Identifier: MIT
+
+#pragma once
+
+#include <Jolt/Skeleton/Skeleton.h>
+#include <Jolt/Skeleton/SkeletalAnimation.h>
+
+JPH_NAMESPACE_BEGIN
+
+#ifdef JPH_DEBUG_RENDERER
+class DebugRenderer;
+#endif // JPH_DEBUG_RENDERER
+
+/// Instance of a skeleton, contains the pose the current skeleton is in
+class JPH_EXPORT SkeletonPose
+{
+public:
+	JPH_OVERRIDE_NEW_DELETE
+
+	using JointState = SkeletalAnimation::JointState;
+	using JointStateVector = Array<JointState>;
+	using Mat44Vector = Array<Mat44>;
+
+	///@name Skeleton
+	///@{
+	void						SetSkeleton(const Skeleton *inSkeleton);
+	const Skeleton *			GetSkeleton() const														{ return mSkeleton; }
+	///@}
+
+	/// Extra offset applied to the root (and therefore also to all of its children)
+	void						SetRootOffset(RVec3Arg inOffset)										{ mRootOffset = inOffset; }
+	RVec3						GetRootOffset() const													{ return mRootOffset; }
+
+	///@name Properties of the joints
+	///@{
+	uint						GetJointCount() const													{ return (uint)mJoints.size(); }
+	const JointStateVector &	GetJoints() const														{ return mJoints; }
+	JointStateVector &			GetJoints()																{ return mJoints; }
+	const JointState &			GetJoint(int inJoint) const												{ return mJoints[inJoint]; }
+	JointState &				GetJoint(int inJoint)													{ return mJoints[inJoint]; }
+	///@}
+
+	///@name Joint matrices
+	///@{
+	const Mat44Vector &			GetJointMatrices() const												{ return mJointMatrices; }
+	Mat44Vector &				GetJointMatrices()														{ return mJointMatrices; }
+	const Mat44 &				GetJointMatrix(int inJoint) const										{ return mJointMatrices[inJoint]; }
+	Mat44 &						GetJointMatrix(int inJoint)												{ return mJointMatrices[inJoint]; }
+	///@}
+
+	/// Convert the joint states to joint matrices
+	void						CalculateJointMatrices();
+
+	/// Convert joint matrices to joint states
+	void						CalculateJointStates();
+
+	/// Outputs the joint matrices in local space (ensure that outMatrices has GetJointCount() elements, assumes that values in GetJoints() is up to date)
+	void						CalculateLocalSpaceJointMatrices(Mat44 *outMatrices) const;
+
+#ifdef JPH_DEBUG_RENDERER
+	/// Draw settings
+	struct DrawSettings
+	{
+		bool					mDrawJoints = true;
+		bool					mDrawJointOrientations = true;
+		bool					mDrawJointNames = false;
+	};
+
+	/// Draw current pose
+	void						Draw(const DrawSettings &inDrawSettings, DebugRenderer *inRenderer, RMat44Arg inOffset = RMat44::sIdentity()) const;
+#endif // JPH_DEBUG_RENDERER
+
+private:
+	RefConst<Skeleton>			mSkeleton;																///< Skeleton definition
+	RVec3						mRootOffset { RVec3::sZero() };											///< Extra offset applied to the root (and therefore also to all of its children)
+	JointStateVector			mJoints;																///< Local joint orientations (local to parent Joint)
+	Mat44Vector					mJointMatrices;															///< Local joint matrices (local to world matrix)
+};
+
+JPH_NAMESPACE_END