Luis Gonzalez
60 lines
1.8 KiB
C#
60 lines
1.8 KiB
C#
using Unity.Burst;
|
|
using Unity.Collections;
|
|
using Unity.Entities;
|
|
using Unity.Mathematics;
|
|
using Unity.Transforms;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
namespace Rukhanka
|
|
{
|
|
|
|
[UpdateInGroup(typeof(RukhankaAnimationInjectionSystemGroup))]
|
|
public partial struct FABRIKSystem: ISystem
|
|
{
|
|
struct FABRIKData
|
|
{
|
|
public NativeArray<float3> chainWorldPositions;
|
|
public NativeArray<float3> chainInitialDirections;
|
|
public NativeArray<float> chainLengths;
|
|
public NativeArray<int> chainBoneIndices;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnCreate(ref SystemState ss)
|
|
{
|
|
var q = SystemAPI.QueryBuilder()
|
|
.WithAll<FABRIKComponent, AnimatorEntityRefComponent>()
|
|
.Build();
|
|
|
|
ss.RequireForUpdate(q);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnUpdate(ref SystemState ss)
|
|
{
|
|
var rigDefLookup = SystemAPI.GetComponentLookup<RigDefinitionComponent>(true);
|
|
var localTransformLookup = SystemAPI.GetComponentLookup<LocalTransform>(true);
|
|
var parentLookup = SystemAPI.GetComponentLookup<Parent>(true);
|
|
var animatorEntityRefLookup = SystemAPI.GetComponentLookup<AnimatorEntityRefComponent>(true);
|
|
|
|
ref var runtimeData = ref SystemAPI.GetSingletonRW<RuntimeAnimationData>().ValueRW;
|
|
|
|
var ikJob = new FABRIKJob()
|
|
{
|
|
parentLookup = parentLookup,
|
|
localTransformLookup = localTransformLookup,
|
|
rigDefLookup = rigDefLookup,
|
|
animatorEntityRefLookup = animatorEntityRefLookup,
|
|
runtimeData = runtimeData
|
|
};
|
|
|
|
ikJob.ScheduleParallel();
|
|
}
|
|
}
|
|
}
|