143 lines
3.7 KiB
C#
143 lines
3.7 KiB
C#
using UnityEngine;
|
|
|
|
|
|
namespace UniHumanoid
|
|
{
|
|
public class HumanPoseTransfer : MonoBehaviour
|
|
{
|
|
public enum HumanPoseTransferSourceType
|
|
{
|
|
None,
|
|
HumanPoseTransfer,
|
|
HumanPoseClip,
|
|
}
|
|
|
|
[SerializeField]
|
|
public HumanPoseTransferSourceType SourceType;
|
|
|
|
[SerializeField]
|
|
public Avatar Avatar;
|
|
|
|
#region Standalone
|
|
public HumanPose CreatePose()
|
|
{
|
|
var handler = new HumanPoseHandler(Avatar, transform);
|
|
var pose = default(HumanPose);
|
|
handler.GetHumanPose(ref pose);
|
|
return pose;
|
|
}
|
|
public void SetPose(HumanPose pose)
|
|
{
|
|
SetPose(Avatar, transform, pose);
|
|
}
|
|
public static void SetPose(Avatar avatar, Transform transform, HumanPose pose)
|
|
{
|
|
var handler = new HumanPoseHandler(avatar, transform);
|
|
handler.SetHumanPose(ref pose);
|
|
}
|
|
public static void SetTPose(Avatar avatar, Transform transform)
|
|
{
|
|
var humanPoseClip = Resources.Load<HumanPoseClip>(HumanPoseClip.TPoseResourcePath);
|
|
var pose = humanPoseClip.GetPose();
|
|
HumanPoseTransfer.SetPose(avatar, transform, pose);
|
|
}
|
|
#endregion
|
|
|
|
private void Reset()
|
|
{
|
|
var animator = GetComponent<Animator>();
|
|
if (animator != null)
|
|
{
|
|
Avatar = animator.avatar;
|
|
}
|
|
}
|
|
|
|
[SerializeField]
|
|
public HumanPoseTransfer Source;
|
|
|
|
[SerializeField]
|
|
public HumanPoseClip PoseClip;
|
|
|
|
[ContextMenu("Set T-Pose")]
|
|
void SetTPose()
|
|
{
|
|
if (Avatar == null) return;
|
|
SetTPose(Avatar, transform);
|
|
}
|
|
|
|
HumanPoseHandler m_handler;
|
|
public void OnEnable()
|
|
{
|
|
var animator = GetComponent<Animator>();
|
|
if (animator != null)
|
|
{
|
|
Avatar = animator.avatar;
|
|
}
|
|
|
|
Setup();
|
|
}
|
|
|
|
public void Setup()
|
|
{
|
|
if (Avatar == null)
|
|
{
|
|
return;
|
|
}
|
|
m_handler = new HumanPoseHandler(Avatar, transform);
|
|
}
|
|
|
|
HumanPose m_pose;
|
|
|
|
int m_lastFrameCount = -1;
|
|
|
|
public bool GetPose(int frameCount, ref HumanPose pose)
|
|
{
|
|
if (PoseClip != null)
|
|
{
|
|
pose = PoseClip.GetPose();
|
|
return true;
|
|
}
|
|
|
|
if (m_handler == null)
|
|
{
|
|
pose = m_pose;
|
|
return false;
|
|
}
|
|
|
|
if (frameCount != m_lastFrameCount)
|
|
{
|
|
m_handler.GetHumanPose(ref m_pose);
|
|
m_lastFrameCount = frameCount;
|
|
}
|
|
pose = m_pose;
|
|
return true;
|
|
}
|
|
|
|
private void Update()
|
|
{
|
|
switch (SourceType)
|
|
{
|
|
case HumanPoseTransferSourceType.None:
|
|
break;
|
|
|
|
case HumanPoseTransferSourceType.HumanPoseTransfer:
|
|
if (Source != null && m_handler != null)
|
|
{
|
|
if (Source.GetPose(Time.frameCount, ref m_pose))
|
|
{
|
|
m_handler.SetHumanPose(ref m_pose);
|
|
}
|
|
}
|
|
break;
|
|
|
|
case HumanPoseTransferSourceType.HumanPoseClip:
|
|
if (PoseClip != null)
|
|
{
|
|
var pose = PoseClip.GetPose();
|
|
m_handler.SetHumanPose(ref pose);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|