117 lines
3.2 KiB
C#
117 lines
3.2 KiB
C#
using Ghost.Test.Core;
|
|
using Misaki.HighPerformance.Jobs;
|
|
using Misaki.HighPerformance.LowLevel.Buffer;
|
|
using Misaki.HighPerformance.Mathematics;
|
|
|
|
namespace Ghost.Entities.Test;
|
|
|
|
internal struct TestEntityQueryJob : IJobEntity<Transform>
|
|
{
|
|
public readonly void Execute(Entity entity, ref Transform transform, int threadIndex)
|
|
{
|
|
transform.position += new float3(5, 5, 5);
|
|
}
|
|
}
|
|
|
|
internal struct TestChunkQueryJob : IJobChunk
|
|
{
|
|
public readonly void Execute(ChunkView view, int threadIndex)
|
|
{
|
|
var random = new random((uint)threadIndex + 1u);
|
|
|
|
var transforms = view.GetComponentDataRW<Transform>();
|
|
for (var i = 0; i < view.Count; i++)
|
|
{
|
|
transforms[i].position += random.NextFloat3();
|
|
}
|
|
}
|
|
}
|
|
|
|
public partial class EntityQueryTest : ITest
|
|
{
|
|
private JobScheduler _jobScheduler = null!;
|
|
private World _world = null!;
|
|
|
|
public void Setup()
|
|
{
|
|
_jobScheduler = new JobScheduler(4);
|
|
_world = World.Create(_jobScheduler);
|
|
}
|
|
|
|
public void Run()
|
|
{
|
|
var entities = (Span<Entity>)stackalloc Entity[1000];
|
|
|
|
using var scope = AllocationManager.CreateStackScope();
|
|
using var set = new ComponentSet(scope.AllocationHandle, ComponentTypeID<Transform>.Value);
|
|
|
|
_world.EntityManager.CreateEntities(entities, set);
|
|
|
|
var queryID = new QueryBuilder().WithAllRW<Transform>().Build(_world);
|
|
ref var query = ref _world.GetEntityQueryReference(queryID);
|
|
|
|
_world.AdvanceVersion();
|
|
|
|
var testJob = new TestChunkQueryJob();
|
|
var handle = query.ScheduleChunkParallel(testJob, 64, JobHandle.Invalid);
|
|
_jobScheduler.WaitComplete(handle);
|
|
|
|
query.ForEach<Transform>((e, ref t) =>
|
|
{
|
|
Console.WriteLine($"Entity {e} Has Position: {t.position}");
|
|
});
|
|
|
|
foreach (var (entity, transform) in query.GetEntityComponentIterator<Transform>())
|
|
{
|
|
Console.WriteLine($"Entity {entity} Updated Position: {transform.Get().position}");
|
|
}
|
|
|
|
foreach (var chunk in query.GetChunkIterator())
|
|
{
|
|
var transforms = chunk.GetComponentData<Transform>();
|
|
var chunkEntities = chunk.GetEntities();
|
|
|
|
// if (chunk.HasChanged<Transform>(0))
|
|
{
|
|
// var bits = chunk.GetEnableBits<Transform>();
|
|
|
|
// var it = bits.GetIterator();
|
|
// while (it.Next(out var index) && index < chunk.Count)
|
|
for (var index = 0; index < chunk.Count; index++)
|
|
{
|
|
Console.WriteLine($"Entity {chunkEntities[index]} Updated Position: {transforms[index].position}");
|
|
}
|
|
}
|
|
}
|
|
|
|
_world.EntityManager.DestroyEntities(entities);
|
|
}
|
|
|
|
public void Cleanup()
|
|
{
|
|
_world.Dispose();
|
|
_jobScheduler.Dispose();
|
|
JobScheduler.ReleaseTempAllocator();
|
|
}
|
|
}
|
|
|
|
public struct Transform : IEnableableComponent
|
|
{
|
|
public float3 position;
|
|
|
|
public override string ToString()
|
|
{
|
|
return $"Position: {position}";
|
|
}
|
|
}
|
|
|
|
public struct Mesh : IComponent
|
|
{
|
|
public int index;
|
|
|
|
public override string ToString()
|
|
{
|
|
return $"Index: {index}";
|
|
}
|
|
}
|