Project-M/Packages/com.rukhanka.animation/Rukhanka.Hybrid/IK/DynamicBoneChainBaker.cs

using System.Collections.Generic;
using Unity.Entities;
using UnityEngine;

////////////////////////////////////////////////////////////////////////////////////////

namespace Rukhanka.Hybrid
{
public class DynamicBoneChainBaker: Baker<DynamicBoneChainAuthoring>
{
	public override void Bake(DynamicBoneChainAuthoring a)
	{
		if (a.tip == null)
		{
			Debug.LogError($"DynamicBoneChainAuthoring '{a.name}': No tip transform defined!");
			return;
		}
		
		var e = GetEntity(a, TransformUsageFlags.None);
		
		var dbcc = new DynamicBoneChainComponent()
		{
			inertia = a.inertia,
			damping = a.damping,
			elasticity = a.elasticity,
			stiffness = a.stiffness,
			timeAccumulator = 0,
			prevPosition = a.transform.position
		};

		AddComponent(e, dbcc);
		var nodeList = AddBuffer<DynamicBoneChainNode>(e);
		
		var allBones = GatherChainTransforms(a.transform, a.tip);
		
		for (var i = 0; i < allBones.Count; ++i)
		{
			var currentTransform = allBones[i];
			var refLocalPos = new BoneTransform()
			{
				pos = currentTransform.localPosition,
				rot = currentTransform.localRotation,
				scale = currentTransform.localScale
			};
			
			var be = new DynamicBoneChainNode()
			{
				boneEntity = GetEntity(currentTransform, TransformUsageFlags.Dynamic),
				prevPosition = currentTransform.position,
				position = currentTransform.position,
				referenceLocalPose = refLocalPos,
				parentIndex = allBones.FindIndex(0, allBones.Count, x => x == currentTransform.parent)
			};
			nodeList.Add(be);
		}
	}
	
////////////////////////////////////////////////////////////////////////////////////////
	
	List<Transform> GatherChainTransforms(Transform root, Transform tip)
	{
		var rv = new List<Transform>();
		if (tip == null)
			return rv;
		
		var curNode = tip;
		do
		{
			rv.Add(curNode);
			curNode = curNode.parent;
		}
		while (curNode != root && curNode.parent != null);
		
		rv.Reverse();
		
		return rv;
	}
}
}