public EntityId[] Take(uint requestedCount)
{
if (requestedCount > Count)
{
throw new ArgumentOutOfRangeException(nameof(requestedCount), requestedCount, "Collection does not have enough stored IDs.");
}
var collection = new EntityId[requestedCount];
uint index = 0;
var remainingCount = requestedCount;
while (remainingCount > 0)
{
if (firstElement.Count <= remainingCount)
{
remainingCount -= firstElement.Count;
firstElement.CopyTo(collection, index);
index += firstElement.Count;
firstElement = queue.Count > 0
? queue.Dequeue()
: new EntityIdRange();
}
else
{
var(splitRange, remainder) = firstElement.Split(remainingCount);
firstElement = remainder;
splitRange.CopyTo(collection, index);
break;
}
}
Count -= requestedCount;
return(collection);
}
public (EntityIdRange part, EntityIdRange remainder) Split(uint count)
{
if (count > Count)
{
throw new ArgumentOutOfRangeException(nameof(count), count, "Split count out of range");
}
var part = new EntityIdRange(FirstEntityId, count);
var remainder = new EntityIdRange(new EntityId(FirstEntityId.Id + count), Count - count);
return(part, remainder);
}
public EntityId Dequeue()
{
if (firstElement.Count <= 0)
{
throw new InvalidOperationException($"{nameof(EntityRangeCollection)} is empty");
}
var(splitRange, remainder) = firstElement.Split(1);
firstElement = remainder.Count == 0 && queue.Count > 0
? queue.Dequeue()
: remainder;
Count -= 1;
return(splitRange.FirstEntityId);
}
internal void Add(EntityIdRange range)
{
if (!range.FirstEntityId.IsValid())
{
throw new ArgumentException("Invalid EntityId in range", nameof(range));
}
if (range.Count == 0)
{
throw new ArgumentOutOfRangeException(nameof(range), "EntityId range cannot be 0 in length.");
}
if (firstElement.Count == 0)
{
firstElement = range;
}
else
{
queue.Enqueue(range);
}
Count += range.Count;
}
public void Clear()
{
firstElement = new EntityIdRange();
queue.Clear();
Count = 0;
}
