lsh4/Assets/DunGen/Code/Collision/DungeonCollisionManager.cs

using System.Collections.Generic;
using Unity.Profiling;
using UnityEngine;

namespace DunGen.Collision
{
	public class DungeonCollisionManager
	{
		private static readonly ProfilerMarker initPerfMarker = new ProfilerMarker("DungeonCollisionManager.Initialize");
		private static readonly ProfilerMarker preCachePerfMarker = new ProfilerMarker("DungeonCollisionManager.PreCacheCounds");
		private static readonly ProfilerMarker addTilePerMarker = new ProfilerMarker("DungeonCollisionManager.AddTile");
		private static readonly ProfilerMarker removeTilePerfMarker = new ProfilerMarker("DungeonCollisionManager.RemoveTile");
		private static readonly ProfilerMarker collisionBroadPhasePerfMarker = new ProfilerMarker("DungeonCollisionManager.BroadPhase");
		private static readonly ProfilerMarker collisionNarrowPhasePerfMarker = new ProfilerMarker("DungeonCollisionManager.NarrowPhase");

		public DungeonCollisionSettings Settings { get; set; }
		public ICollisionBroadphase Broadphase { get; private set; }

		private readonly List<Bounds> cachedBounds = new List<Bounds>();
		private readonly List<TileProxy> tiles = new List<TileProxy>();
		private List<Bounds> boundsToCheck = new List<Bounds>();


		/// <summary>
		/// Initializes the collision manager. This should be done at the beginning of the dungeon generation process
		/// </summary>
		/// <param name="dungeonGenerator">The dungeon generator we're initializing for</param>
		public virtual void Initialize(DungeonGenerator dungeonGenerator)
		{
			using(initPerfMarker.Auto())
			{
				Clear();
				PreCacheBounds(dungeonGenerator);
				InitializeBroadphase(dungeonGenerator);
			}
		}

		protected virtual void Clear()
		{
			tiles.Clear();
			cachedBounds.Clear();
			boundsToCheck.Clear();
		}

		protected virtual void PreCacheBounds(DungeonGenerator dungeonGenerator)
		{
			using (preCachePerfMarker.Auto())
			{
				cachedBounds.Clear();

				// Cache tiles from other dungeons if we need to avoid collisions with them
				if (Settings.AvoidCollisionsWithOtherDungeons || dungeonGenerator.AttachmentSettings != null)
				{
					foreach (var tile in UnityUtil.FindObjectsByType<Tile>())
						cachedBounds.Add(tile.Placement.Bounds);
				}

				// Add all additional collision bounds to the cache
				foreach (var bounds in Settings.AdditionalCollisionBounds)
					cachedBounds.Add(bounds);
			}
		}

		protected virtual void InitializeBroadphase(DungeonGenerator dungeonGenerator)
		{
			var broadphaseSettings = DunGenSettings.Instance.BroadphaseSettings;

			if (broadphaseSettings == null)
			{
				Broadphase = null;
				return;
			}

			Broadphase = broadphaseSettings.Create();
			Broadphase.Init(broadphaseSettings, dungeonGenerator);

			// Add all cached bounds to the quadtree
			foreach(var bounds in cachedBounds)
				Broadphase.Insert(bounds);
		}

		/// <summary>
		/// Adds a tile to the collision manager
		/// </summary>
		/// <param name="tile">The tile to add</param>
		public virtual void AddTile(TileProxy tile)
		{
			using(addTilePerMarker.Auto())
			{
				tiles.Add(tile);
				Broadphase?.Insert(tile.Placement.Bounds);
			}
		}

		/// <summary>
		/// Removed a tile from the collision manager
		/// </summary>
		/// <param name="tile">The tile to remove</param>
		public virtual void RemoveTile(TileProxy tile)
		{
			using (removeTilePerfMarker.Auto())
			{
				tiles.Remove(tile);
				Broadphase?.Remove(tile.Placement.Bounds);
			}
		}

		/// <summary>
		/// Checks if a tile is colliding with any other tiles in the dungeon
		/// </summary>
		/// <param name="upDirection">The up direction for the dungeon</param>
		/// <param name="prospectiveNewTile">The new tile we'd like to spawn</param>
		/// <param name="previousTile">The tile we're trying to attach to</param>
		/// <returns>True if any blocking collision occurs</returns>
		public virtual bool IsCollidingWithAnyTile(AxisDirection upDirection, TileProxy prospectiveNewTile, TileProxy previousTile)
		{
			bool isColliding = false;

			using (collisionBroadPhasePerfMarker.Auto())
			{
				UpdateBoundsToCheck(prospectiveNewTile, previousTile);
			}

			using (collisionNarrowPhasePerfMarker.Auto())
			{
				// Check for collisions with potentially colliding tiles
				for (int i = 0; i < boundsToCheck.Count; i++)
				{
					var bounds = boundsToCheck[i];

					bool isConnected = previousTile != null && i == 0;
					float maxOverlap = (isConnected) ? Settings.OverlapThreshold : -Settings.Padding;

					if (Settings.DisallowOverhangs && !isConnected)
					{
						if (UnityUtil.AreBoundsOverlappingOrOverhanging(prospectiveNewTile.Placement.Bounds,
							bounds,
							upDirection,
							maxOverlap))
						{
							isColliding = true;
							break;
						}
					}
					else
					{
						if (UnityUtil.AreBoundsOverlapping(prospectiveNewTile.Placement.Bounds, bounds, maxOverlap))
						{
							isColliding = true;
							break;
						}
					}
				}
			}

			// Process custom collision predicate if there is one
			if (Settings.AdditionalCollisionsPredicate != null)
				isColliding = Settings.AdditionalCollisionsPredicate(prospectiveNewTile.Placement.Bounds, isColliding);

			return isColliding;
		}

		protected virtual void UpdateBoundsToCheck(TileProxy prospectiveNewTile, TileProxy previousTile)
		{
			boundsToCheck.Clear();

			if (Broadphase != null)
			{
				Broadphase.Query(prospectiveNewTile.Placement.Bounds, ref boundsToCheck);

				// Ensure previous tile bounds is at the front of the list
				if (previousTile != null)
				{
					var previousBounds = previousTile.Placement.Bounds;
					int existingIndex = boundsToCheck.FindIndex(b => b.Equals(previousBounds));

					if (existingIndex != -1)
					{
						boundsToCheck.RemoveAt(existingIndex);
						boundsToCheck.Insert(0, previousBounds);
					}
					else
						boundsToCheck.Insert(0, previousBounds);
				}
			}
			else
			{
				// Ensure previous tile bounds is at the front of the list
				if (previousTile != null)
					boundsToCheck.Add(previousTile.Placement.Bounds);

				foreach (var tile in tiles)
					if (tile != previousTile)
						boundsToCheck.Add(tile.Placement.Bounds);

				boundsToCheck.AddRange(cachedBounds);
			}
		}
	}
}