GlobalGameJame/Library/PackageCache/com.unity.timeline@1.2.10/Editor/Utilities/SpacePartitioner.cs

128 lines
3.8 KiB
C#
Raw Normal View History

2020-01-31 15:10:04 +00:00
using System;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Timeline;
namespace UnityEditor.Timeline
{
interface IBounds
{
Rect boundingRect { get; }
}
class SpacePartitioner
{
internal class CachedList<T>
{
public static readonly List<T> Instance = new List<T>(1000);
}
struct Entry : IInterval
{
public object item { get; set; }
public long intervalStart { get; set; }
public long intervalEnd { get; set; }
public Rect bounds { get; set; }
private const float kPrecision = 100.0f;
private const float kMaxFloat = (float)long.MaxValue;
private const float kMinFloat = (float)long.MinValue;
static public Int64 FromFloat(float f)
{
if (Single.IsPositiveInfinity(f))
return long.MaxValue;
if (Single.IsNegativeInfinity(f))
return long.MinValue;
f = Mathf.Clamp(f, kMinFloat, kMaxFloat); // prevent overflow of floats
f = Mathf.Clamp(f * kPrecision, kMinFloat, kMaxFloat); // clamp to 'long' range
return (long)(f);
}
}
const EventType k_GuiEventLock = EventType.Repaint;
IntervalTree<Entry> m_Tree = new IntervalTree<Entry>();
List<Entry> m_CacheList = new List<Entry>();
public void Clear()
{
m_Tree.Clear();
}
public void AddBounds(IBounds bounds)
{
AddBounds(bounds, bounds.boundingRect);
}
public void AddBounds(object item, Rect rect)
{
if (item == null)
throw new ArgumentNullException("item");
m_Tree.Add(new Entry()
{
intervalStart = Entry.FromFloat(rect.yMin),
intervalEnd = Entry.FromFloat(rect.yMax),
bounds = rect,
item = item
}
);
}
/// <summary>
/// Get items of type T at a given position
/// </summary>
/// <param name="position"></param>
/// <param name="inClipSpace"></param>
/// <typeparam name="T"></typeparam>
/// <remarks>
/// Uses a (1,1) sized box
/// Use .ToList() or .ToArray() when not enumerating the result immediately
/// </remarks>
/// <returns></returns>
public IEnumerable<T> GetItemsAtPosition<T>(Vector2 position)
{
return GetItemsInArea<T>(new Rect(position.x, position.y, 1, 1));
}
/// <summary>
///
/// </summary>
/// <param name="area"></param>
/// <param name="inClipSpace"></param>
/// <typeparam name="T"></typeparam>
/// <returns></returns>
public IEnumerable<T> GetItemsInArea<T>(Rect area)
{
m_CacheList.Clear();
m_Tree.IntersectsWithRange(long.MinValue, long.MaxValue, m_CacheList);
var list = CachedList<T>.Instance;
list.Clear();
foreach (var i in m_CacheList)
{
if (i.item is T && i.bounds.Overlaps(area))
list.Add((T)i.item);
}
return list;
}
public void DebugDraw()
{
var kFillColor = new Color(1.0f, 1.0f, 1.0f, 0.1f);
var kOutlineColor = Color.yellow;
m_CacheList.Clear();
m_Tree.IntersectsWithRange(long.MinValue, long.MaxValue, m_CacheList);
HandleUtility.ApplyWireMaterial();
foreach (var item in m_CacheList)
{
Handles.DrawSolidRectangleWithOutline(item.bounds, kFillColor, kOutlineColor);
}
}
}
}