Human infection system

Assets/GameCode/ComponentTypes/Human.cs

@@ -1,6 +1,8 @@
 using Unity.Entities;
+using Unity.Mathematics;
 
 public struct Human : IComponentData
 {
     public float TimeTillNextDirectionChange;
+    public int IsInfected;
 }

Assets/GameCode/HumanInfectionSystem.cs

@@ -0,0 +1,69 @@
+using Unity.Collections;
+using Unity.Entities;
+using Unity.Jobs;
+using Unity.Mathematics;
+using Unity.Transforms2D;
+using UnityEngine;
+
+class HumanInfectionSystem : JobComponentSystem
+{
+    [Inject] private HumanInfectionData humanData;
+    [Inject] private ZombiePositionData zombieTargetData;
+
+    protected override JobHandle OnUpdate(JobHandle inputDeps)
+    {
+        var job = new HumanInfectionJob
+        {
+            zombieTargetData = zombieTargetData,
+            humanData = humanData,
+            infectionDistance = ZombieSettings.Instance.InfectionDistance
+        };
+
+        return job.Schedule(humanData.Length, 64, inputDeps);
+    }
+}
+
+
+struct HumanInfectionJob : IJobParallelFor
+{
+    public HumanInfectionData humanData;
+
+    [NativeDisableParallelForRestriction]
+    public ZombiePositionData zombieTargetData;
+    public float infectionDistance;
+
+    public void Execute(int index)
+    {
+        float2 humanPosition = humanData.Positions[index].Value;
+
+        for (int i = 0; i < zombieTargetData.Length; i++)
+        {
+            float2 zombiePosition = zombieTargetData.Positions[i].Value;
+
+            float2 delta = zombiePosition - humanPosition;
+            float distSquared = math.dot(delta, delta);
+
+            if (distSquared < infectionDistance)
+            {
+                var human = humanData.Humans[i];
+                human.IsInfected = 1;
+                humanData.Humans[i] = human;
+            }
+
+        }
+    }
+}
+
+public struct HumanInfectionData
+{
+    public int Length;
+    [ReadOnly] public ComponentDataArray<Position2D> Positions;
+    public ComponentDataArray<Human> Humans;
+}
+
+public struct ZombiePositionData
+{
+    public int Length;
+    [ReadOnly] public ComponentDataArray<Position2D> Positions;
+    [ReadOnly] public ComponentDataArray<Zombie> Zombies;
+}

Assets/GameCode/ZombieNavigationSystem.cs

@@ -26,7 +26,6 @@ protected override JobHandle OnUpdate(JobHandle inputDeps)
 
 public struct ZombieNavigationJob : IJobParallelFor
 {
-
     public ZombieData zombieDatum;
 
     [NativeDisableParallelForRestriction]
@@ -74,5 +73,5 @@ public struct ZombieTargetData
 {
     public int Length;
     [ReadOnly] public ComponentDataArray<Position2D> Positions;
-    [ReadOnly] public ComponentDataArray<Human> Human;
+    public ComponentDataArray<Human> Humans;
 }

Assets/GameCode/ZombieSettings.cs

@@ -5,9 +5,10 @@ internal class ZombieSettings : MonoBehaviour
     public float HumanSpeed = 10;
     public int HumanCount = 1000;
     public int ZombieCount = 10;
+    public float InfectionDistance = 1f;
 
     public Rect Playfield = new Rect { x = -30.0f, y = -30.0f, width = 60.0f, height = 60.0f };
-    
+
 
     public static ZombieSettings Instance { get; private set; }