| | | 1 | | //======================================================================= |
| | | 2 | | // ColliderRegistry.cs |
| | | 3 | | //======================================================================= |
| | | 4 | | // MIT License, Copyright (c) 2026-present David Oravsky (mrdav30) |
| | | 5 | | // See LICENSE file in the project root for full license information. |
| | | 6 | | //======================================================================= |
| | | 7 | | |
| | | 8 | | using Gravitas.Colliders; |
| | | 9 | | using SwiftCollections; |
| | | 10 | | using System.Collections.Generic; |
| | | 11 | | using System.Runtime.CompilerServices; |
| | | 12 | | |
| | | 13 | | namespace Gravitas; |
| | | 14 | | |
| | | 15 | | internal sealed class ColliderRegistry<TCollider> |
| | | 16 | | where TCollider : class, IPhysicsColliderRegistryItem |
| | | 17 | | { |
| | | 18 | | private readonly SwiftBucket<TCollider> _byId; |
| | | 19 | | private readonly SwiftList<TCollider> _liveColliders; |
| | | 20 | | private readonly SwiftList<TCollider> _replayColliders; |
| | | 21 | | private int _nextReplayOrder; |
| | | 22 | | |
| | 10170 | 23 | | public ColliderRegistry(int capacity = SwiftBucket<TCollider>.DefaultCapacity) |
| | | 24 | | { |
| | 10170 | 25 | | _byId = new SwiftBucket<TCollider>(capacity); |
| | 10170 | 26 | | _liveColliders = new SwiftList<TCollider>(capacity); |
| | 10170 | 27 | | _replayColliders = new SwiftList<TCollider>(capacity); |
| | 10170 | 28 | | } |
| | | 29 | | |
| | 13351 | 30 | | public int Count => _liveColliders.Count; |
| | | 31 | | |
| | 449 | 32 | | public int PeakCount => _byId.PeakCount; |
| | | 33 | | |
| | | 34 | | public TCollider this[int serviceIndex] |
| | | 35 | | { |
| | | 36 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | 2620 | 37 | | get => _liveColliders[serviceIndex]; |
| | | 38 | | } |
| | | 39 | | |
| | | 40 | | public int Register(TCollider collider) |
| | | 41 | | { |
| | 10855 | 42 | | SwiftThrowHelper.ThrowIfNull(collider, nameof(collider)); |
| | | 43 | | |
| | 10855 | 44 | | int id = _byId.Add(collider); |
| | 10855 | 45 | | int serviceIndex = _liveColliders.Count; |
| | 10855 | 46 | | collider.SetRegistryState(id, serviceIndex, _nextReplayOrder++); |
| | 10855 | 47 | | _liveColliders.Add(collider); |
| | 10855 | 48 | | return id; |
| | | 49 | | } |
| | | 50 | | |
| | | 51 | | public bool Remove(TCollider collider) |
| | | 52 | | { |
| | 487 | 53 | | SwiftThrowHelper.ThrowIfNull(collider, nameof(collider)); |
| | | 54 | | |
| | 487 | 55 | | int id = collider.Id; |
| | 487 | 56 | | if (!TryGetById(id, out TCollider? registered) || !ReferenceEquals(registered, collider)) |
| | 3 | 57 | | return false; |
| | | 58 | | |
| | 484 | 59 | | RemoveLiveCollider(collider); |
| | 484 | 60 | | _byId.RemoveAt(id); |
| | 484 | 61 | | collider.ClearRegistryState(); |
| | 484 | 62 | | return true; |
| | | 63 | | } |
| | | 64 | | |
| | | 65 | | public void Clear() |
| | | 66 | | { |
| | 230 | 67 | | for (int i = 0; i < _liveColliders.Count; i++) |
| | 48 | 68 | | _liveColliders[i].ClearRegistryState(); |
| | | 69 | | |
| | 67 | 70 | | _liveColliders.FastClear(); |
| | 67 | 71 | | _replayColliders.FastClear(); |
| | 67 | 72 | | _byId.Clear(); |
| | 67 | 73 | | _nextReplayOrder = 0; |
| | 67 | 74 | | } |
| | | 75 | | |
| | | 76 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 77 | | public bool TryGetById(int id, out TCollider? collider) |
| | | 78 | | { |
| | 1663451 | 79 | | if (id < 0 || !_byId.TryGetValue(id, out TCollider? value)) |
| | | 80 | | { |
| | 266 | 81 | | collider = null; |
| | 266 | 82 | | return false; |
| | | 83 | | } |
| | | 84 | | |
| | 1663185 | 85 | | collider = value; |
| | 1663185 | 86 | | return true; |
| | | 87 | | } |
| | | 88 | | |
| | | 89 | | public bool TryGetByServiceIndex(int serviceIndex, out TCollider? collider) |
| | | 90 | | { |
| | 15 | 91 | | if (serviceIndex < 0 || serviceIndex >= _liveColliders.Count) |
| | | 92 | | { |
| | 6 | 93 | | collider = null; |
| | 6 | 94 | | return false; |
| | | 95 | | } |
| | | 96 | | |
| | 9 | 97 | | collider = _liveColliders[serviceIndex]; |
| | 9 | 98 | | return true; |
| | | 99 | | } |
| | | 100 | | |
| | | 101 | | public SwiftList<TCollider> PrepareReplayColliders() |
| | | 102 | | { |
| | 2754 | 103 | | _replayColliders.FastClear(); |
| | 2754 | 104 | | _replayColliders.EnsureCapacity(_liveColliders.Count); |
| | 18300 | 105 | | for (int i = 0; i < _liveColliders.Count; i++) |
| | 6396 | 106 | | _replayColliders.Add(_liveColliders[i]); |
| | | 107 | | |
| | 2754 | 108 | | _replayColliders.SortInPlace(default(ReplayOrderComparer)); |
| | 18300 | 109 | | for (int i = 0; i < _replayColliders.Count; i++) |
| | 6396 | 110 | | _replayColliders[i].SetRegistryReplayOrdinal(i); |
| | | 111 | | |
| | 2754 | 112 | | return _replayColliders; |
| | | 113 | | } |
| | | 114 | | |
| | | 115 | | private void RemoveLiveCollider(TCollider collider) |
| | | 116 | | { |
| | 484 | 117 | | int index = collider.ServiceIndex; |
| | 484 | 118 | | int lastIndex = _liveColliders.Count - 1; |
| | 484 | 119 | | if (index != lastIndex) |
| | | 120 | | { |
| | 96 | 121 | | TCollider moved = _liveColliders[lastIndex]; |
| | 96 | 122 | | _liveColliders[index] = moved; |
| | 96 | 123 | | moved.SetRegistryServiceIndex(index); |
| | | 124 | | } |
| | | 125 | | |
| | 484 | 126 | | _liveColliders.RemoveAt(lastIndex); |
| | 484 | 127 | | } |
| | | 128 | | |
| | | 129 | | private readonly struct ReplayOrderComparer : IComparer<TCollider> |
| | | 130 | | { |
| | | 131 | | [MethodImpl(MethodImplOptions.AggressiveInlining)] |
| | | 132 | | public int Compare(TCollider? x, TCollider? y) => |
| | 3658 | 133 | | x!.ReplayOrder.CompareTo(y!.ReplayOrder); |
| | | 134 | | } |
| | | 135 | | } |