micromissiles-unity/Assets/Scripts/Assignment/Assignment.cs

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

// The assignment class is an interface for assigning a target to each missile.
public interface IAssignment
{
    // Assignment item type.
    // The first element corresponds to the missile index, and the second element
    // corresponds to the target index.
    public struct AssignmentItem
    {
        public int MissileIndex;
        public int TargetIndex;

        public AssignmentItem(int missileIndex, int targetIndex)
        {
            MissileIndex = missileIndex;
            TargetIndex = targetIndex;
        }
    }

    // A list containing the missile-target assignments.

    // Assign a target to each missile that has not been assigned a target yet.
    public abstract IEnumerable<AssignmentItem> Assign(List<Agent> missiles, List<Agent> targets);

    // Get the list of assignable missile indices.
    protected static List<int> GetAssignableMissileIndices(List<Agent> missiles)
    {
        List<int> assignableMissileIndices = new List<int>();
        for (int missileIndex = 0; missileIndex < missiles.Count; missileIndex++)
        {
            if (missiles[missileIndex].IsAssignable())
            {
                assignableMissileIndices.Add(missileIndex);
            }
        }
        return assignableMissileIndices;
    }

    // Get the list of active target indices.
    protected static List<int> GetActiveTargetIndices(List<Agent> targets)
    {
        List<int> activeTargetIndices = new List<int>();
        for (int targetIndex = 0; targetIndex < targets.Count; targetIndex++)
        {
            if (!targets[targetIndex].IsHit())
            {
                activeTargetIndices.Add(targetIndex);
            }
        }
        return activeTargetIndices;
    }
}