| | | 1 | | using System.Collections; |
| | | 2 | | using System.Collections.Generic; |
| | | 3 | | using System.Linq; |
| | | 4 | | using UnityEngine; |
| | | 5 | | |
| | | 6 | | // The cluster class represents a collection of game objects. |
| | | 7 | | public class Cluster { |
| | | 8 | | // Coordinates of the cluster. |
| | 80 | 9 | | private Vector3 _coordinates = Vector3.zero; |
| | | 10 | | |
| | | 11 | | // List of game objects in the cluster. |
| | 80 | 12 | | private List<GameObject> _objects = new List<GameObject>(); |
| | | 13 | | |
| | 27 | 14 | | public Cluster() {} |
| | 68 | 15 | | public Cluster(in Vector3 coordinates) { |
| | 34 | 16 | | _coordinates = coordinates; |
| | 34 | 17 | | } |
| | 74 | 18 | | public Cluster(in GameObject obj) { |
| | 37 | 19 | | _coordinates = obj.transform.position; |
| | 37 | 20 | | } |
| | | 21 | | |
| | | 22 | | // Get the cluster coordinates. |
| | | 23 | | public Vector3 Coordinates { |
| | 1164 | 24 | | get { return _coordinates; } |
| | | 25 | | } |
| | | 26 | | |
| | | 27 | | // Get the list of game objects. |
| | | 28 | | public IReadOnlyList<GameObject> Objects { |
| | 15 | 29 | | get { return _objects; } |
| | | 30 | | } |
| | | 31 | | |
| | | 32 | | // Get the list of agents. |
| | | 33 | | public IReadOnlyList<Agent> Agents { |
| | 497 | 34 | | get { return _objects.Select(gameObject => gameObject.GetComponent<Agent>()).ToList(); } |
| | | 35 | | } |
| | | 36 | | |
| | | 37 | | // Get the list of interceptors. |
| | | 38 | | public IReadOnlyList<Interceptor> Interceptors { |
| | 0 | 39 | | get { |
| | 0 | 40 | | return _objects.Select(gameObject => gameObject.GetComponent<Agent>() as Interceptor) |
| | | 41 | | .ToList(); |
| | 0 | 42 | | } |
| | | 43 | | } |
| | | 44 | | |
| | | 45 | | // Get the list of threats. |
| | | 46 | | public IReadOnlyList<Threat> Threats { |
| | 2 | 47 | | get { |
| | 4 | 48 | | return _objects.Select(gameObject => gameObject.GetComponent<Agent>() as Threat).ToList(); |
| | 2 | 49 | | } |
| | | 50 | | } |
| | | 51 | | |
| | | 52 | | // Return the size of the cluster. |
| | 121 | 53 | | public int Size() { |
| | 121 | 54 | | return _objects.Count; |
| | 121 | 55 | | } |
| | | 56 | | |
| | | 57 | | // Check whether the cluster is empty. |
| | 55 | 58 | | public bool IsEmpty() { |
| | 55 | 59 | | return Size() == 0; |
| | 55 | 60 | | } |
| | | 61 | | |
| | | 62 | | // Calculate the radius of the cluster. |
| | 19 | 63 | | public float Radius() { |
| | 19 | 64 | | if (IsEmpty()) { |
| | 0 | 65 | | return 0; |
| | | 66 | | } |
| | | 67 | | |
| | 19 | 68 | | Vector3 centroid = Centroid(); |
| | 53 | 69 | | return _objects.Max(obj => Vector3.Distance(centroid, obj.transform.position)); |
| | 19 | 70 | | } |
| | | 71 | | |
| | | 72 | | // Calculate the centroid of the cluster. |
| | | 73 | | // The centroid is the mean position of all active game objects. |
| | 76 | 74 | | public Vector3 Centroid() { |
| | 76 | 75 | | Vector3 positionSum = Vector3.zero; |
| | 76 | 76 | | int activeAgentCount = 0; |
| | 999 | 77 | | foreach (var agent in Agents) { |
| | 512 | 78 | | if (agent != null && !agent.IsTerminated()) { |
| | 255 | 79 | | positionSum += agent.GetPosition(); |
| | 255 | 80 | | ++activeAgentCount; |
| | 255 | 81 | | } |
| | 257 | 82 | | } |
| | 76 | 83 | | return activeAgentCount > 0 ? positionSum / activeAgentCount : Vector3.zero; |
| | 76 | 84 | | } |
| | | 85 | | |
| | | 86 | | // Recenter the cluster's centroid to be the mean of all game objects' positions in the cluster. |
| | 13 | 87 | | public void Recenter() { |
| | 13 | 88 | | _coordinates = Centroid(); |
| | 13 | 89 | | } |
| | | 90 | | |
| | | 91 | | // Calculate the velocity of the cluster. |
| | | 92 | | // The velocity is the mean velocity of all active game objects. |
| | 2 | 93 | | public Vector3 Velocity() { |
| | 2 | 94 | | Vector3 velocitySum = Vector3.zero; |
| | 2 | 95 | | int activeAgentCount = 0; |
| | 12 | 96 | | foreach (var agent in Agents) { |
| | 3 | 97 | | if (agent != null && !agent.IsTerminated()) { |
| | 1 | 98 | | velocitySum += agent.GetVelocity(); |
| | 1 | 99 | | ++activeAgentCount; |
| | 1 | 100 | | } |
| | 2 | 101 | | } |
| | 2 | 102 | | return activeAgentCount > 0 ? velocitySum / activeAgentCount : Vector3.zero; |
| | 2 | 103 | | } |
| | | 104 | | |
| | | 105 | | // Add a game object to the cluster. |
| | | 106 | | // This function does not update the centroid of the cluster. |
| | 143 | 107 | | public void AddObject(in GameObject obj) { |
| | 143 | 108 | | _objects.Add(obj); |
| | 143 | 109 | | } |
| | | 110 | | |
| | | 111 | | // Add multiple game objects to the cluster. |
| | | 112 | | // This function does not update the centroid of the cluster. |
| | 11 | 113 | | public void AddObjects(in IReadOnlyList<GameObject> objects) { |
| | 11 | 114 | | _objects.AddRange(objects); |
| | 11 | 115 | | } |
| | | 116 | | |
| | | 117 | | // Merge another cluster into this one. |
| | | 118 | | // This function does not update the centroid of the cluster. |
| | 5 | 119 | | public void Merge(in Cluster cluster) { |
| | 5 | 120 | | AddObjects(cluster.Objects); |
| | 5 | 121 | | } |
| | | 122 | | |
| | | 123 | | // Returns true if all agents in the cluster are terminated. |
| | 1 | 124 | | public bool IsFullyTerminated() { |
| | 2 | 125 | | return Agents.All(agent => agent?.IsTerminated() ?? true); |
| | 1 | 126 | | } |
| | | 127 | | } |