Netcode Bootstrap

2026-05-31 14:27:52 -07:00
parent 99d8d2d2a9
commit 7fa77ce821
1813 changed files with 2921554 additions and 84 deletions
@@ -0,0 +1,111 @@
+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;
+	}
+}
+}