Parking/Assets/Scripts/ParkingManager.cs

using System;
using System.Collections.Generic;
using System.Linq;
using TMPro;
using UnityEngine;

namespace Parking
{
    public class ParkingManager : MonoBehaviour
    {
        public const float Width = 68;
        public const float Height = 29;

        public static ParkingManager Instance;

        private static readonly TimeSpan StartTime = TimeSpan.FromHours(5) + TimeSpan.FromMinutes(45);
        private static readonly TimeSpan EndTime = TimeSpan.FromHours(17) + TimeSpan.FromMinutes(15);
        private static TimeSpan _currentTime = StartTime;

        [SerializeField] private int stepTime = 15;
        [SerializeField] private TextMeshProUGUI timeText;
        [SerializeField] public TextMeshProUGUI countsText;
        [SerializeField] public TextMeshProUGUI rejectedText;
        [SerializeField] public TextMeshProUGUI pathText;
        [SerializeField] public GameObject startPanel;
        [SerializeField] public GameObject notFoundPanel;
        [SerializeField] public GameObject spotPrefabA;
        [SerializeField] public GameObject spotPrefabB;
        [SerializeField] public GameObject spotPrefabC;
        [SerializeField] public GameObject spotPrefabD;
        [SerializeField] public GameObject carPrefab;
        [SerializeField] public Transform mainPlanContainer;
        [SerializeField] public Transform emergencyPlanContainer;

        private readonly float[] _spotHeights = {4f, 4.5f, 5f, 7.5f};

        private readonly List<List<Spot>> _spotMap = new()
            {new List<Spot>(), new List<Spot>(), new List<Spot>(), new List<Spot>()};

        private bool _emergencyActivated;

        private int _initialConfigurationSpotCount;
        private int[,] _initialSolution;
        private int[] _rejectedDrivers = {0, 0, 0, 0};
        private List<GameObject> _spotPrefabs;

        private void Awake()
        {
            Instance = this;
        }

        private void Start()
        {
            _spotPrefabs = new List<GameObject> {spotPrefabA, spotPrefabB, spotPrefabC, spotPrefabD};

        }

        public void ReadDataButtonClick()
        {
            if (!DataImporter.ReadFile(pathText.text.Replace("\u200B", ""))) {
                notFoundPanel.SetActive(true);
                return;
            }
            InitiateManager();
            startPanel.SetActive(false);
        }

        private void InitiateManager()
        {
            timeText.text = _currentTime.ToString();
            rejectedText.text = $"Małe: {_rejectedDrivers[0]} Średnie: {_rejectedDrivers[1]} " +
                                $"Duże: {_rejectedDrivers[2]}";
            GenerateEmergencyPlan();

            InitialConfigurationGenerator generator = new();
            _initialSolution = generator.FindSolution();
            ArrangeSpots(_initialSolution);
            ReserveInitialSpots();
        }

        public void Emergency()
        {
            ResetDrivers();
            _emergencyActivated = true;
        }

        private void ReserveInitialSpots()
        {
            for (int j = 3; j >= 0; j--)
                foreach (Spot spot in _spotMap[j]) {
                    spot.Reserved = false;
                    spot.ReservedPriority = 0;
                }

            for (int i = 0; i < Math.Min(DataImporter.Drivers.Count, _initialConfigurationSpotCount); i++) {
                bool found = false;
                // Find spot
                for (int j = 3; j >= 0 && !found; j--)
                    foreach (Spot spot in _spotMap[j])
                        if (spot.Size == DataImporter.Drivers[i].Size && !spot.Reserved) {
                            spot.Reserved = true;
                            spot.ReservedPriority = DataImporter.Drivers[i].Priority;
                            found = true;
                            break;
                        }
            }
        }

        public void AdvanceTime()
        {
            // Update time
            _currentTime += TimeSpan.FromMinutes(stepTime);
            if (_currentTime > EndTime) {
                _currentTime = StartTime;
                timeText.text = _currentTime.ToString();
                _rejectedDrivers = new[] {0, 0, 0, 0};
                rejectedText.text = $"Małe: {_rejectedDrivers[0]} Średnie: {_rejectedDrivers[1]} " +
                                    $"Duże: {_rejectedDrivers[2]}";
                if (_emergencyActivated) {
                    emergencyPlanContainer.gameObject.SetActive(true);
                    mainPlanContainer.gameObject.SetActive(false);
                }
                else {
                    ArrangeSpots(_initialSolution);
                    ReserveInitialSpots();
                    ResetDrivers();
                }

                return;
            }

            timeText.text = _currentTime.ToString();

            if (_emergencyActivated)
                return;

            ReconfigureSpots();

            foreach (Driver driver in DataImporter.Drivers) {
                // TODO: Check if car can stay before other reservation - may not work for priority-based reservations
                bool triesToPark = _currentTime <= driver.Times[3].TimeOfDay &&
                                   _currentTime > driver.Times[2].TimeOfDay;
                bool leftTheParking = _currentTime > driver.Times[3].TimeOfDay && driver.Parked;
                bool reservedButTimedout =
                    (_currentTime - driver.Times[0].TimeOfDay).TotalMinutes > driver.UpdateInterval &&
                    GetReservedSpotCount(driver.Priority, driver.Size) > 0
                    && !driver.Parked && !driver.Rejected && _currentTime <= driver.Times[3].TimeOfDay;
                if (triesToPark && !driver.Parked && !driver.Rejected) {
                    if (!PlaceCarOnParking(driver)) {
                        if (TryReconfigureSpotForSize(driver.Size)) {
                            if (!PlaceCarOnParking(driver)) {
                                Debug.Log($"Placing failed for car {driver.Number} size {driver.Size}");
                                driver.Rejected = true;
                                _rejectedDrivers[(int) driver.Size]++;
                                rejectedText.text = $"Małe: {_rejectedDrivers[0]} Średnie: {_rejectedDrivers[1]} " +
                                                    $"Duże: {_rejectedDrivers[2]}";
                            }
                        }
                        else {
                            Debug.Log($"Placing failed for car {driver.Number} size {driver.Size}");
                            driver.Rejected = true;
                            _rejectedDrivers[(int) driver.Size]++;
                            rejectedText.text = $"Małe: {_rejectedDrivers[0]} Średnie: {_rejectedDrivers[1]} " +
                                                $"Duże: {_rejectedDrivers[2]}";
                        }
                    }
                }
                else if (leftTheParking) {
                    driver.Spot.Reserved = false;
                    driver.Reset();
                }
                else if (reservedButTimedout) {
                    Debug.Log($"Reserved but timed out driver {driver.Number}");
                    driver.Rejected = true;
                }
            }
            
            UpdateCounts();
        }

        private void UpdateCounts()
        {

            int[] counts = {0, 0, 0, 0};
            for (int i = 0; i < _spotMap.Count; i++)
            for (int j = 0; j < _spotMap[i].Count; j++)
                counts[(int) _spotMap[i][j].Size]++;

            string countsString = $"Małe: {counts[0]} Średnie: {counts[1]} Duże: {counts[2]} " +
                                  $"Suma: {counts.Sum()}";
            UpdateText(countsString);
        }

        private void ResetDrivers()
        {
            foreach (Driver driver in DataImporter.Drivers) {
                if (driver.Spot != null) {
                    driver.Spot.Reserved = false;
                    driver.Spot.ReservedPriority = 0;
                }

                driver.Reset();
            }
        }

        private int GetReservedSpotCount(int priority, Size size)
        {
            int count = 0;
            foreach (var list in _spotMap)
            foreach (Spot spot in list)
                if (spot.Size == size && spot.Reserved && spot.ReservedPriority == priority)
                    count++;

            return count;
        }

        private bool FindReservedSpot(Driver driver, out Spot spot)
        {
            foreach (var list in _spotMap)
            foreach (Spot spot1 in list)
                if (spot1.Reserved && spot1.ReservedPriority == driver.Priority) {
                    spot = spot1;
                    return true;
                }

            spot = new Spot();

            return false;
        }

        private bool PlaceCarOnParking(Driver driver)
        {
            if (FindReservedSpot(driver, out Spot reservedSpot)) {
                if (reservedSpot.Free) {
                    PlaceDriverOnSpot(driver, reservedSpot);
                    return true;
                }

                Debug.Log("Spot should be reserved but is taken");
            }


            foreach (var t in _spotMap)
            foreach (Spot spot in t) {
                bool spotAvailable = spot.Size == driver.Size && spot.Free &&
                                     (spot.ParkingDirection == driver.ParkingPreference ||
                                      spot.ParkingDirection == ParkingPreference.Any)
                                     && !spot.Reserved;

                if (spotAvailable) {
                    PlaceDriverOnSpot(driver, spot);
                    return true;
                }
            }

            foreach (var t in _spotMap)
            foreach (Spot spot in t)
                if (spot.Size == driver.Size && spot.Free && !spot.Reserved) {
                    PlaceDriverOnSpot(driver, spot);
                    return true;
                }

            return false;
        }

        private void PlaceDriverOnSpot(Driver driver, Spot spot)
        {
            spot.Free = false;
            spot.Reserved = false;
            driver.Spot = spot;
            driver.Parked = true;
            driver.GameObject = Instantiate(carPrefab);
            driver.GameObject.GetComponentInChildren<TextMeshProUGUI>().text = driver.Number.ToString();
            driver.GameObject.GetComponentInChildren<TextMeshProUGUI>().transform.rotation =
                Quaternion.Euler(new Vector3(0, 0, (spot.Flipped ? 180 : 0) + (!spot.Perpendicular ? -90 : 0)));
            driver.GameObject.transform.rotation =
                Quaternion.Euler(new Vector3(0, 0, (spot.Flipped ? 180 : 0) + (!spot.Perpendicular ? 90 : 0)));
            driver.GameObject.transform.position = spot.GameObject.transform.position;
            driver.GameObject.transform.SetParent(spot.GameObject.transform, true);
            driver.GameObject.transform.localPosition += new Vector3(0, 0, -1);
        }

        public void UpdateText(string text)
        {
            countsText.text = text;
        }

        private void ReconfigureSpots()
        {
            int[] freeSpots = GetFreeSpotCount();
            var nextCars = GetNextCars(1);
            int[] plannedSpots = {0, 0, 0, 0};
            foreach (Driver driver in nextCars)
                plannedSpots[(int) driver.Size]++;

            int[] neededSpots = {0, 0, 0, 0};

            for (int i = 0; i < neededSpots.Length; i++)
                neededSpots[i] = Math.Max(plannedSpots[i] - freeSpots[i], 0);
            if (neededSpots.Sum() > 1) {
                Debug.Log($"Needed spots = {neededSpots[0]} {neededSpots[1]} {neededSpots[2]} {neededSpots[3]}");
                Debug.Log("Attempting reconfiguration...");

                var spotlessCars = new List<Driver>();

                // Filter drivers by assigning free space
                foreach (Driver nextCar in nextCars)
                    if (freeSpots[(int) nextCar.Size] != 0)
                        freeSpots[(int) nextCar.Size]--;
                    else
                        spotlessCars.Add(nextCar);


                foreach (Driver nextCar in spotlessCars) {
                    TryReconfigureSpotForSize(nextCar.Size);
                }
            }
        }

        private bool TryReconfigureSpotForSize(Size newSize)
        {
            foreach (var currentLaneSpots in _spotMap)
            foreach (Spot spot in currentLaneSpots) {
                if (spot.Size > newSize && spot.Free && spot.Size != Size.D && !spot.Reserved) {
                    if (GetSpotCountOfSize(Size.A) <= 2 && spot.Size == Size.A)
                        continue;
                    if (GetSpotCountOfSize(Size.B) <= 2 && spot.Size == Size.B)
                        continue;

                    int size = (int) newSize;

                    float diff = (_spotHeights[(int) spot.Size] - _spotHeights[size]) / 2.0f;
                    if (!spot.AlignToTop)
                        diff *= -1;

                    spot.Size = (Size) size;
                    Vector3 position = Vector3.zero;
                    Quaternion rotation = Quaternion.identity;
                    if (spot.GameObject != null) {
                        position = spot.GameObject.transform.position;
                        rotation = spot.GameObject.transform.rotation;
                        Destroy(spot.GameObject);
                    }

                    spot.Reserved = false;

                    Vector3 offset = spot.Perpendicular ? new Vector3(0, diff, 0) : new Vector3(-diff, 0, 0);

                    Vector3 newPosition = position + offset;

                    if (!spot.Perpendicular) {
                        var allBorders = new List<float>();
                        foreach (Spot spot1 in currentLaneSpots) {
                            if (spot1 == spot)
                                continue;
                            Vector3 spotPosition = spot1.GameObject.transform.position;
                            float sizeOffset = _spotHeights[(int) spot1.Size] / 2.0f;
                            allBorders.Add(spotPosition.x + sizeOffset);
                            allBorders.Add(spotPosition.x - sizeOffset);
                        }

                        allBorders.Sort();
                        foreach (float adjacentBorder in allBorders)
                            if (adjacentBorder > position.x) {
                                newPosition = new Vector3(adjacentBorder -
                                                          _spotHeights[size] / 2.0f,
                                    newPosition.y, newPosition.z);
                                break;
                            }
                    }

                    // TODO: check how much space to right to push

                    spot.GameObject = Instantiate(_spotPrefabs[size],
                        newPosition, rotation, mainPlanContainer);

                    return true;
                }
            }

            return false;
        }
        
        
        private int GetSpotCountOfSize(Size size)
        {
            int count = 0;
            foreach (var t in _spotMap)
            foreach (Spot spot in t)
                if (spot.Size == size)
                    count++;
            return count;
        }

        private int[] GetFreeSpotCount()
        {
            int[] freeSpots = {0, 0, 0, 0};
            foreach (var t in _spotMap)
            foreach (Spot spot in t)
                if (spot.Free)
                    freeSpots[(int) spot.Size]++;
            return freeSpots;
        }

        private List<Driver> GetNextCars(int steps)
        {
            var nextCars = new List<Driver>();
            TimeSpan updatedTime = _currentTime + TimeSpan.FromMinutes(stepTime * steps);
            foreach (Driver driver in DataImporter.Drivers)
                if (updatedTime <= driver.Times[1].TimeOfDay &&
                    updatedTime >= driver.Times[0].TimeOfDay &&
                    !driver.Parked && !driver.Rejected)
                    nextCars.Add(driver);

            return nextCars;
        }

        private void ArrangeSpots(int[,] spotsCreated)
        {
            var spotMap = GenerateSpotMap(spotsCreated);

            float[] spotSizes = {4, 4.5f, 5, 0};
            float maxP3 = 0;
            float maxP2 = 0;
            foreach (Spot spot in spotMap[2])
                if (spot.Size != Size.D)
                    maxP3 = Math.Max(maxP3, spot.Perpendicular ? _spotHeights[(int) spot.Size] : 2.25f);
            foreach (Spot spot in spotMap[1])
                if (spot.Size != Size.D)
                    maxP2 = Math.Max(maxP2, spot.Perpendicular ? _spotHeights[(int) spot.Size] : 2.25f);

            // float maxP3 = spotMap[2].Count == 0 ? 0 : _spotHeights[(int) spotMap[2].Max().Size];
            // float maxP2 = spotMap[1].Count == 0 ? 0 : _spotHeights[(int) spotMap[1].Max().Size];
            var prefabs = new List<GameObject> {spotPrefabA, spotPrefabB, spotPrefabC, spotPrefabD};

            for (int i = 0; i < 4; i++) {
                float currentY;
                bool parkingFromTop = i % 2 != 0;
                switch (i) {
                    case 0:
                        currentY = -Height / 2.0f;
                        break;
                    case 1:
                        currentY = Height / 2.0f - 5.5f - 5.0f - maxP3;
                        break;
                    case 2:
                        currentY = Height / 2.0f - 5.5f - 5.0f - maxP3;
                        break;
                    case 3:
                        currentY = Height / 2.0f;
                        break;
                    default:
                        currentY = -10;
                        break;
                }

                float currentX;
                if (i != 0) currentX = -Width / 2f - 2.25f / 2f + 2.25f;
                else currentX = Width / 2f; // + 5.5f - 2.25f / 2f + 2.25f + 1.75f;

                bool flipped = false;

                for (int j = 0; j < spotMap[i].Count; j++) {
                    if (!spotMap[i][j].Perpendicular && i == 0 && j == 0)
                        currentX -= spotSizes[(int) spotMap[i][j].Size] / 2.0f;
                    else if (spotMap[i][j].Perpendicular && i == 0 && j == 0)
                        currentX -= 2.25f / 2.0f;

                    spotMap[i][j].Flipped = flipped;
                    bool alignTop = i % 2 != 0;
                    spotMap[i][j].GameObject = Instantiate(prefabs[(int) spotMap[i][j].Size], mainPlanContainer);

                    if (spotMap[i][j].Perpendicular) {
                        spotMap[i][j].GameObject.transform.position =
                            new Vector3(currentX,
                                currentY + (alignTop ? -1 : 1) * spotSizes[(int) spotMap[i][j].Size] / 2.0f, 0);
                    }
                    else {
                        spotMap[i][j].GameObject.transform.position =
                            new Vector3(currentX,
                                currentY + (alignTop ? -1 : 1) * 2.25f / 2.0f, 0);
                    }

                    spotMap[i][j].AlignToTop = alignTop;


                    if (!spotMap[i][j].Perpendicular)
                        spotMap[i][j].Flipped = alignTop;

                    if (spotMap[i][j].Perpendicular) {
                        spotMap[i][j].GameObject.transform.rotation =
                            Quaternion.Euler(new Vector3(0, 0, spotMap[i][j].Flipped ? 180 : 0));
                    }
                    else {
                        spotMap[i][j].GameObject.transform.rotation =
                            Quaternion.Euler(new Vector3(0, 0, (spotMap[i][j].Flipped ? 180 : 0) + 90));
                    }

                    bool frontalParking = !parkingFromTop ^ flipped;
                    spotMap[i][j].ParkingDirection = frontalParking ? ParkingPreference.Front : ParkingPreference.Back;

                    spotMap[i][j].Lane = i;

                    currentX += (i != 0 ? 1 : -1) *
                                (spotMap[i][j].Perpendicular ? 2.25f : spotSizes[(int) spotMap[i][j].Size]);

                    flipped = !flipped;
                }
            }
        }

        private void GenerateEmergencyPlan()
        {
            emergencyPlanContainer.gameObject.SetActive(false);
            for (int i = 0; i < 4; i++) {
                float currentY;
                switch (i) {
                    case 0:
                        currentY = -Height / 2.0f;
                        break;
                    case 1:
                        currentY = Height / 2.0f - 5.5f - 5.0f - 5.0f;
                        break;
                    case 2:
                        currentY = Height / 2.0f - 5.5f - 5.0f - 5.0f;
                        break;
                    case 3:
                        currentY = Height / 2.0f;
                        break;
                    default:
                        currentY = -10;
                        break;
                }

                float currentX;
                if (i != 0) currentX = -Width / 2f - 2.25f / 2f + 2.25f;
                else currentX = Width / 2f - 5.0f / 2f;

                GameObject spawnedSpot;

                int[] emergencyMap = {12, 27, 27, 20};

                for (int j = 0; j < emergencyMap[i]; j++)
                    if (i == 0) {
                        Vector3 position = new(currentX, currentY + 2.25f / 2.0f, 0);
                        Instantiate(Instance.spotPrefabC, position,
                            Quaternion.Euler(new Vector3(0, 0, 90)), emergencyPlanContainer);

                        currentX -= 5;
                    }
                    else {
                        bool alignTop = i % 2 != 0;
                        Vector3 position = new(currentX, currentY, 0);
                        spawnedSpot = Instantiate(Instance.spotPrefabC, position,
                            Quaternion.Euler(new Vector3(0, 0, 0)), emergencyPlanContainer);

                        spawnedSpot.transform.position =
                            new Vector3(currentX,
                                currentY + (alignTop ? -1 : 1) * 5.0f / 2.0f,
                                0);

                        currentX += 2.25f;
                    }
            }
        }

        private List<List<Spot>> GenerateSpotMap(int[,] spotsCreated)
        {
            foreach (var list in _spotMap) {
                foreach (Spot spot in list)
                    if (spot.GameObject != null)
                        Destroy(spot.GameObject);
                list.Clear();
            }

            for (int i = 0; i < 4; i++)
            for (int j = 0; j < spotsCreated.GetLength(1); j++)
            for (int k = 0; k < spotsCreated[i, j]; k++) {
                Spot spot = new((Size) (j % 4), false);
                spot.Perpendicular = j < 4;
                _spotMap[i].Add(spot);
                _initialConfigurationSpotCount++;
            }

            _spotMap[0].Sort((a, b) => a.Size.CompareTo(b.Size)); // ascending sort
            _spotMap[1].Sort((a, b) => b.Size.CompareTo(a.Size)); // descending sort
            _spotMap[2].Sort((a, b) => b.Size.CompareTo(a.Size)); // descending sort
            _spotMap[3].Sort((a, b) => a.Size.CompareTo(b.Size)); // ascending sort
            _spotMap[2].Add(new Spot(Size.D, false));
            return _spotMap;
        }
    }
}