LayoutPack.cs

using MPewsey.Common.Mathematics;
using MPewsey.ManiaMap;
using System.Collections.Generic;
using UnityEngine;
 
namespace MPewsey.ManiaMapUnity
{
    public class LayoutPack
    {
        public Layout Layout { get; }
 
        public LayoutState LayoutState { get; }
 
        public ManiaMapSettings Settings { get; }
 
        private Dictionary<Uid, List<DoorConnection>> RoomConnections { get; } = new Dictionary<Uid, List<DoorConnection>>();
 
        private Dictionary<int, List<Room>> RoomsByLayer { get; } = new Dictionary<int, List<Room>>();
 
        private Dictionary<Uid, List<DoorPosition>> RoomDoors { get; } = new Dictionary<Uid, List<DoorPosition>>();
 
        public RectangleInt LayoutBounds { get; }
 
        public LayoutPack(Layout layout, LayoutState layoutState, ManiaMapSettings settings = null)
        {
            if (layout == null)
                throw new System.ArgumentException("Layout cannot be null.");
            if (layoutState == null)
                throw new System.ArgumentException("Layout state cannot be null.");
            if (layout.Id != layoutState.Id)
                throw new System.ArgumentException($"Layout and layout state ID's do not match: (Layout ID = {layout.Id}, Layout State ID = {layoutState.Id})");
 
            Layout = layout;
            LayoutState = layoutState;
            Settings = settings != null ? settings : ScriptableObject.CreateInstance<ManiaMapSettings>();
            RoomConnections = layout.GetRoomConnections();
            RoomsByLayer = layout.GetRoomsByLayer();
            RoomDoors = layout.GetRoomDoors();
            LayoutBounds = layout.GetBounds();
        }
 
        public IReadOnlyList<DoorConnection> GetDoorConnections(Uid id)
        {
            if (RoomConnections.TryGetValue(id, out var connections))
                return connections;
            return System.Array.Empty<DoorConnection>();
        }
 
        public IReadOnlyList<Room> GetRoomsInLayer(int z)
        {
            if (RoomsByLayer.TryGetValue(z, out var rooms))
                return rooms;
            return System.Array.Empty<Room>();
        }
 
        public IReadOnlyList<DoorPosition> GetRoomDoors(Uid id)
        {
            if (RoomDoors.TryGetValue(id, out var doors))
                return doors;
            return System.Array.Empty<DoorPosition>();
        }
 
        public bool DoorExists(Uid id, Vector2DInt position, DoorDirection direction)
        {
            foreach (var door in GetRoomDoors(id))
            {
                if (door.Matches(position, direction))
                    return true;
            }
 
            return false;
        }
 
        public IEnumerable<int> GetLayerCoordinates()
        {
            return RoomsByLayer.Keys;
        }
 
        public DoorConnection FindDoorConnection(Uid id, Vector2DInt position, DoorDirection direction)
        {
            foreach (var connection in GetDoorConnections(id))
            {
                if (connection.ContainsDoor(id, position, direction))
                    return connection;
            }
 
            return null;
        }
    }
}