Luis Gonzalez
112 lines
3.9 KiB
C#
112 lines
3.9 KiB
C#
using Rukhanka;
|
|
using Unity.Burst;
|
|
using Unity.Collections;
|
|
using Unity.Entities;
|
|
#if RUKHANKA_WITH_NETCODE
|
|
using Unity.NetCode;
|
|
#endif
|
|
using UnityEngine;
|
|
using static Rukhanka.AnimatorControllerSystemJobs;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[assembly: RegisterGenericSystemType(typeof(AnimatorControllerSystem<PredictedAnimatorControllerQuery>))]
|
|
[assembly: RegisterGenericSystemType(typeof(AnimatorControllerSystem<AnimatorControllerQuery>))]
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
namespace Rukhanka
|
|
{
|
|
|
|
[DisableAutoCreation]
|
|
public partial struct AnimatorControllerSystem<T>: ISystem where T: AnimatorControllerQueryCreator, new()
|
|
{
|
|
EntityQuery animatorControllerQuery;
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnCreate(ref SystemState ss)
|
|
{
|
|
var queryCreator = new T();
|
|
animatorControllerQuery = queryCreator.CreateQuery(ref ss);
|
|
|
|
ss.RequireForUpdate(animatorControllerQuery);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
[BurstCompile]
|
|
public void OnUpdate(ref SystemState ss)
|
|
{
|
|
var dt = SystemAPI.Time.DeltaTime;
|
|
|
|
var controllerLayersBufferHandle = SystemAPI.GetBufferTypeHandle<AnimatorControllerLayerComponent>();
|
|
var controllerParametersBufferHandle = SystemAPI.GetBufferTypeHandle<AnimatorControllerParameterComponent>();
|
|
var animatorOverrideAnimationsLookup = SystemAPI.GetComponentLookup<AnimatorOverrideAnimations>(true);
|
|
var entityTypeHandle = SystemAPI.GetEntityTypeHandle();
|
|
var controllerEventsBufferLookup = SystemAPI.GetBufferLookup<AnimatorControllerEventComponent>();
|
|
var animDBSingleton = SystemAPI.GetSingleton<BlobDatabaseSingleton>();
|
|
|
|
var internalDataSingletonQuery = SystemAPI.QueryBuilder().WithAllRW<InternalAnimatorDataSingleton>().Build();
|
|
var internalAnimatorData = FillAnimationsFromControllerSystem.GetInternalDataSingleton(internalDataSingletonQuery, ref ss);
|
|
|
|
var stateMachineProcessJob = new StateMachineProcessJob()
|
|
{
|
|
controllerLayersBufferHandle = controllerLayersBufferHandle,
|
|
controllerParametersBufferHandle = controllerParametersBufferHandle,
|
|
dt = dt,
|
|
entityTypeHandle = entityTypeHandle,
|
|
controllerEventsBufferLookup = controllerEventsBufferLookup,
|
|
animationDatabase = animDBSingleton.animations,
|
|
animatorOverrideAnimationLookup = animatorOverrideAnimationsLookup,
|
|
animatorOverrideAnimationsMap = internalAnimatorData.animatorOverrideAnimationsMap.AsParallelWriter()
|
|
|
|
};
|
|
|
|
ss.Dependency = stateMachineProcessJob.ScheduleParallel(animatorControllerQuery, ss.Dependency);
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public interface AnimatorControllerQueryCreator
|
|
{
|
|
EntityQuery CreateQuery(ref SystemState ss);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public struct PredictedAnimatorControllerQuery: AnimatorControllerQueryCreator
|
|
{
|
|
public EntityQuery CreateQuery(ref SystemState ss)
|
|
{
|
|
var eqBuilder0 = new EntityQueryBuilder(Allocator.Temp)
|
|
.WithAllRW<AnimatorControllerLayerComponent>()
|
|
#if RUKHANKA_WITH_NETCODE
|
|
.WithAll<Simulate, PredictedGhost>()
|
|
#endif
|
|
;
|
|
var animatorControllerQuery = ss.GetEntityQuery(eqBuilder0);
|
|
return animatorControllerQuery;
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
|
|
public struct AnimatorControllerQuery: AnimatorControllerQueryCreator
|
|
{
|
|
public EntityQuery CreateQuery(ref SystemState ss)
|
|
{
|
|
var eqBuilder0 = new EntityQueryBuilder(Allocator.Temp)
|
|
.WithAllRW<AnimatorControllerLayerComponent>()
|
|
#if RUKHANKA_WITH_NETCODE
|
|
.WithNone<GhostInstance>()
|
|
#endif
|
|
;
|
|
var animatorControllerQuery = ss.GetEntityQuery(eqBuilder0);
|
|
return animatorControllerQuery;
|
|
}
|
|
}
|
|
}
|