WUT_Computer_Science/Programming/E-PSYOPS/theProject/Assets/Scripts/Squad.cs

using System;
using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class Squad : MonoBehaviour
{
	private Entity.Team ownTeam;
	public Entity.Team GetOwnTeam() { return ownTeam; }
	public void SetOwnTeam(Entity.Team newTeam) {ownTeam = newTeam; }
	#region Orders
	public abstract class Order // generic order (to keep in queue)
	{
		public virtual void Execute(Squad squad) // "translate" the order to the soldier
		{// depending on implementation, for example call soldier's method to execute/plan this task
			Debug.LogWarning($"Generic order execution not overriden\n All soldiers received generic order");
		}
    }

	public class MovementOrder : Order // example how to add new types of orders
	{
		public MovementOrder(Vector2Int movementTargetCoords)
		{
			targetCoords = movementTargetCoords;
		}
		public readonly Vector2Int targetCoords;
		public override void Execute(Squad squad)
		{// here we would set soldier's target position for example
			Debug.Log(squad);
			Dictionary<Entity, Vector2Int> targetCoordsDictionary = squad.
				GetFormation().
				CalculatePositions(
				targetCoords);
			foreach (Soldier soldier in targetCoordsDictionary.Keys)
			{
				Vector2Int targetPositionForSoldier = targetCoordsDictionary[soldier];
				soldier.HandleMovementOrder(targetPositionForSoldier);
				Debug.Log($"Soldier {soldier.name} received movement order towards coordinates {targetPositionForSoldier.x},{targetPositionForSoldier.y}");
			}
		}
	}
	#endregion
	[SerializeField] private Formation formation;
	public Formation GetFormation() { return formation; }
	//[SerializeField] private List<Entity> soldiers = new List<Entity>(); // soldiers belonging to the squad
	public List<Entity> GetSoldiers() { return soldiers.GetTList(); }
    private Queue<Order> orders = new Queue<Order>(); // orders given to the squad

	[SerializeField] private TTLList<Entity> soldiers = new TTLList<Entity>(); // soldiers belonging to the squad
	[SerializeField] private TTLList<Entity> enemiesSpotted = new TTLList<Entity>(); // enemies currently viewed by squad
	[SerializeField] private int basicSoldierTTL = 5;
	[SerializeField] private int basicEnemyTTL = 5;

	public void AddSoldierToSquad(Entity soldier)
	{
		soldiers.AddToList(soldier);
		((Soldier)soldier).MySquad = this;
		soldier.OnDeath.AddListener(RemoveSoldierFromSquad);
	}

	public void RemoveSoldierFromSquad(Entity soldier)
	{
		soldiers.RemoveFromList(soldier);
		((Soldier)soldier).MySquad = null;
		soldier.OnDeath.RemoveListener(RemoveSoldierFromSquad);

		if (soldiers.GetCount() == 0)
        {
			Soldier[] objs = FindObjectsOfType<Soldier>();

			foreach (Soldier obj in objs)
            {
				if (obj != soldier && obj.GetOwnTeam() == ownTeam)
                {
					AddSoldierToSquad(obj);
                }
            }
        }
	}

	public void AddSpottedEnemy(Entity enemy)
    {
		enemiesSpotted.AddToList(enemy);
    }

	private void Awake()
	{
		TickSystem.OnTick += HandleTick;
		soldiers.Initialize(true, basicSoldierTTL);
		TickSystem.OnTick += soldiers.OnTick;
		enemiesSpotted.Initialize(false, basicEnemyTTL);
		TickSystem.OnTick += enemiesSpotted.OnTick;

		formation = GetComponent<Formation>();
	}

	private void OnDestroy()
	{
		TickSystem.OnTick -= HandleTick;
	}

	private void HandleTick(TickSystem.OnTickEventArgs eventArgs)
	{// pass a single order to all soldiers
		if (orders.Count < 1)
			return; // for now nothing to do here

		Order currentOrder = orders.Dequeue();
		Debug.Log($"Passing order {currentOrder.ToString()} on tick #{eventArgs.tickNumber}");
		currentOrder.Execute(this);
	}

	public void EnqueueOrder(Order orderToEnqueue)
    {
		orders.Enqueue(orderToEnqueue);
    }
}