mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-08-18 22:58:43 +02:00
189 lines
6.7 KiB
C#
189 lines
6.7 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Numerics;
|
|
|
|
namespace ARMeilleure.Common
|
|
{
|
|
/// <summary>
|
|
/// Represents an expandable table of the type <typeparamref name="TEntry"/>, whose entries will remain at the same
|
|
/// address through out the table's lifetime.
|
|
/// </summary>
|
|
/// <typeparam name="TEntry">Type of the entry in the table</typeparam>
|
|
class EntryTable<TEntry> : IDisposable where TEntry : unmanaged
|
|
{
|
|
private bool _disposed;
|
|
private int _freeHint;
|
|
private readonly int _pageCapacity; // Number of entries per page.
|
|
private readonly int _pageLogCapacity;
|
|
private readonly Dictionary<int, IntPtr> _pages;
|
|
private readonly BitMap _allocated;
|
|
|
|
/// <summary>
|
|
/// Initializes a new instance of the <see cref="EntryTable{TEntry}"/> class with the desired page size in
|
|
/// bytes.
|
|
/// </summary>
|
|
/// <param name="pageSize">Desired page size in bytes</param>
|
|
/// <exception cref="ArgumentOutOfRangeException"><paramref name="pageSize"/> is less than 0</exception>
|
|
/// <exception cref="ArgumentException"><typeparamref name="TEntry"/>'s size is zero</exception>
|
|
/// <remarks>
|
|
/// The actual page size may be smaller or larger depending on the size of <typeparamref name="TEntry"/>.
|
|
/// </remarks>
|
|
public unsafe EntryTable(int pageSize = 4096)
|
|
{
|
|
if (pageSize < 0)
|
|
{
|
|
throw new ArgumentOutOfRangeException(nameof(pageSize), "Page size cannot be negative.");
|
|
}
|
|
|
|
if (sizeof(TEntry) == 0)
|
|
{
|
|
throw new ArgumentException("Size of TEntry cannot be zero.");
|
|
}
|
|
|
|
_allocated = new BitMap(NativeAllocator.Instance);
|
|
_pages = new Dictionary<int, IntPtr>();
|
|
_pageLogCapacity = BitOperations.Log2((uint)(pageSize / sizeof(TEntry)));
|
|
_pageCapacity = 1 << _pageLogCapacity;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Allocates an entry in the <see cref="EntryTable{TEntry}"/>.
|
|
/// </summary>
|
|
/// <returns>Index of entry allocated in the table</returns>
|
|
/// <exception cref="ObjectDisposedException"><see cref="EntryTable{TEntry}"/> instance was disposed</exception>
|
|
public int Allocate()
|
|
{
|
|
ObjectDisposedException.ThrowIf(_disposed, this);
|
|
|
|
lock (_allocated)
|
|
{
|
|
if (_allocated.IsSet(_freeHint))
|
|
{
|
|
_freeHint = _allocated.FindFirstUnset();
|
|
}
|
|
|
|
int index = _freeHint++;
|
|
var page = GetPage(index);
|
|
|
|
_allocated.Set(index);
|
|
|
|
GetValue(page, index) = default;
|
|
|
|
return index;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Frees the entry at the specified <paramref name="index"/>.
|
|
/// </summary>
|
|
/// <param name="index">Index of entry to free</param>
|
|
/// <exception cref="ObjectDisposedException"><see cref="EntryTable{TEntry}"/> instance was disposed</exception>
|
|
public void Free(int index)
|
|
{
|
|
ObjectDisposedException.ThrowIf(_disposed, this);
|
|
|
|
lock (_allocated)
|
|
{
|
|
if (_allocated.IsSet(index))
|
|
{
|
|
_allocated.Clear(index);
|
|
|
|
_freeHint = index;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets a reference to the entry at the specified allocated <paramref name="index"/>.
|
|
/// </summary>
|
|
/// <param name="index">Index of the entry</param>
|
|
/// <returns>Reference to the entry at the specified <paramref name="index"/></returns>
|
|
/// <exception cref="ObjectDisposedException"><see cref="EntryTable{TEntry}"/> instance was disposed</exception>
|
|
/// <exception cref="ArgumentException">Entry at <paramref name="index"/> is not allocated</exception>
|
|
public ref TEntry GetValue(int index)
|
|
{
|
|
ObjectDisposedException.ThrowIf(_disposed, this);
|
|
|
|
lock (_allocated)
|
|
{
|
|
if (!_allocated.IsSet(index))
|
|
{
|
|
throw new ArgumentException("Entry at the specified index was not allocated", nameof(index));
|
|
}
|
|
|
|
var page = GetPage(index);
|
|
|
|
return ref GetValue(page, index);
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets a reference to the entry at using the specified <paramref name="index"/> from the specified
|
|
/// <paramref name="page"/>.
|
|
/// </summary>
|
|
/// <param name="page">Page to use</param>
|
|
/// <param name="index">Index to use</param>
|
|
/// <returns>Reference to the entry</returns>
|
|
private ref TEntry GetValue(Span<TEntry> page, int index)
|
|
{
|
|
return ref page[index & (_pageCapacity - 1)];
|
|
}
|
|
|
|
/// <summary>
|
|
/// Gets the page for the specified <see cref="index"/>.
|
|
/// </summary>
|
|
/// <param name="index">Index to use</param>
|
|
/// <returns>Page for the specified <see cref="index"/></returns>
|
|
private unsafe Span<TEntry> GetPage(int index)
|
|
{
|
|
var pageIndex = (int)((uint)(index & ~(_pageCapacity - 1)) >> _pageLogCapacity);
|
|
|
|
if (!_pages.TryGetValue(pageIndex, out IntPtr page))
|
|
{
|
|
page = (IntPtr)NativeAllocator.Instance.Allocate((uint)sizeof(TEntry) * (uint)_pageCapacity);
|
|
|
|
_pages.Add(pageIndex, page);
|
|
}
|
|
|
|
return new Span<TEntry>((void*)page, _pageCapacity);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Releases all resources used by the <see cref="EntryTable{TEntry}"/> instance.
|
|
/// </summary>
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Releases all unmanaged and optionally managed resources used by the <see cref="EntryTable{TEntry}"/>
|
|
/// instance.
|
|
/// </summary>
|
|
/// <param name="disposing"><see langword="true"/> to dispose managed resources also; otherwise just unmanaged resouces</param>
|
|
protected unsafe virtual void Dispose(bool disposing)
|
|
{
|
|
if (!_disposed)
|
|
{
|
|
_allocated.Dispose();
|
|
|
|
foreach (var page in _pages.Values)
|
|
{
|
|
NativeAllocator.Instance.Free((void*)page);
|
|
}
|
|
|
|
_disposed = true;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Frees resources used by the <see cref="EntryTable{TEntry}"/> instance.
|
|
/// </summary>
|
|
~EntryTable()
|
|
{
|
|
Dispose(false);
|
|
}
|
|
}
|
|
}
|