forked from Unity-Technologies/EntityComponentSystemSamples
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCartesianGridOnCubeTargetSystem.cs
More file actions
59 lines (52 loc) · 2.81 KB
/
Copy pathCartesianGridOnCubeTargetSystem.cs
File metadata and controls
59 lines (52 loc) · 2.81 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
using Unity.Entities;
using Unity.Jobs;
using Unity.Mathematics;
[UpdateBefore(typeof(CartesianGridChangeDirectionSystemGroup))]
public unsafe partial class CartesianGridOnCubeTargetSystem : SystemBase
{
EntityQuery m_GridQuery;
protected override void OnCreate()
{
m_GridQuery = GetEntityQuery(ComponentType.ReadOnly<CartesianGridOnCube>());
RequireForUpdate(m_GridQuery);
}
protected override void OnUpdate()
{
// Get component data from the GridPlane
var cartesianGridCube = GetSingleton<CartesianGridOnCube>();
var rowCount = cartesianGridCube.Blob.Value.RowCount;
var targetDirectionsBufferCapacity = 6 * (((rowCount + 1) / 2) * rowCount);
var targetDistancesBufferCapacity = 6 * (rowCount * rowCount);
var gridWalls = (byte*)cartesianGridCube.Blob.Value.Walls.GetUnsafePtr();
var faceLocalToLocal = (float4x4*)cartesianGridCube.Blob.Value.FaceLocalToLocal.GetUnsafePtr();
Entities
.WithName("InitializeTargets")
.WithAll<CartesianGridTarget>()
.WithAll<CartesianGridOnCubeFace>()
.WithNone<CartesianGridTargetDirection>()
.WithNone<CartesianGridTargetDistance>()
.WithStructuralChanges()
.ForEach((Entity entity) =>
{
var directionBuffer = EntityManager.AddBuffer<CartesianGridTargetDirection>(entity);
directionBuffer.ResizeUninitialized(targetDirectionsBufferCapacity);
var distanceBuffer = EntityManager.AddBuffer<CartesianGridTargetDistance>(entity);
distanceBuffer.ResizeUninitialized(targetDistancesBufferCapacity);
}).Run();
// Rebuild all the paths to the target *only* when the target's grid position changes.
Entities
.WithName("UpdateTargetPaths")
.WithNativeDisableUnsafePtrRestriction(faceLocalToLocal)
.WithAll<CartesianGridTarget>()
.ForEach((Entity entity, ref CartesianGridTargetCoordinates prevTargetPosition, in CartesianGridOnCubeFace cubeFace, in CartesianGridCoordinates targetPosition, in DynamicBuffer<CartesianGridTargetDirection> targetDirections, in DynamicBuffer<CartesianGridTargetDistance> targetDistances) =>
{
if (prevTargetPosition.Equals(targetPosition))
return;
if (targetPosition.OnGrid(rowCount, rowCount))
{
prevTargetPosition = new CartesianGridTargetCoordinates(targetPosition);
CartesianGridOnCubeShortestPath.CalculateShortestPathsToTarget(targetDirections.Reinterpret<byte>().AsNativeArray(), targetDistances.Reinterpret<int>().AsNativeArray(), rowCount, targetPosition, cubeFace, gridWalls, faceLocalToLocal);
}
}).Run();
}
}