warrentyexpired/ModernUO
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Optimizes FindItemsByType by removing allocations. (#1515)

Browse source 
### Summary
Container enumeration is in dire need of optimization. Thanks to @stefanomerotta for initiating this work with PR #1443. This PR handles a small part of what Stefan started. Also included are some bug fixes.

### Method Signatures

```cs
// Use with foreach without moving/deleting items
FindItemsByTypeEnumerator<T> FindItemsByType<T>(bool recurse = true, Predicate<T> predicate = null)

// Use with foreach when moving/deleting items
QueuedItemsEnumerator<T> EnumerateItemsByType<T>(bool recurse = true, Predicate<T> predicate = null)

// Use when iterating multiple times or queuing
PooledRefQueue<T> QueueItemsByType<T>(bool recurse = true, Predicate<T> predicate = null)

// Use when iterating multiples times or manipulating elements without traversing
PooledRefList<T> ListItemsByType<T>(bool recurse = true, Predicate<T> predicate = null)
```

* `FindItemsByType<T>` has changed from returning `List<T>` to `FindItemsByTypeEnumerator<T>` - This method is not safe to use in situations where an item may get consumed, deleted, or moved.

* `EnumerateItemsByType<T>` was added as a safe way to iterate and manipulate items.
  * **Note**: EnumerateItemsByType will _completely traverse the container_ before iteration starts because it uses `QueueItemsByType` under the hood.

* `QueueItemsByType<T>`  and `ListItemsByType<T>` was added to return a queue or list of items to iterate multiple times and manipulate the items. This isn't the most efficient since it uses a predicate and can result in 2 or 3 total iterations unnecessarily.

### Bug Fixes
- [X] Fishing had an error in the random check that may have caused slight bias.
This commit is contained in:
Kamron Batman 2023-09-28 19:36:45 -07:00 • committed by GitHub
parent 146abad36e
commit a4cabe2fa4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
23 changed files with 655 additions and 390 deletions
Download patch file Download diff file Expand all files Collapse all files

223
Projects/Server/Items/Container.cs
View file

@ -14,10 +14,8 @@ public delegate int CheckItemGroup(Item a, Item b);
public delegate void ContainerSnoopHandler(Container cont, Mobile from);
public class Container : Item
public partial class Container : Item
{
private static readonly List<Item> m_FindItemsList = new();
private ContainerData m_ContainerData;
private int m_DropSound;
@ -817,19 +815,26 @@ public class Container : Item
throw new ArgumentNullException(nameof(grouper));
}
var typedItems = CollectionsMarshal.AsSpan(FindItemsByType(type, recurse));
using var typedItems = PooledRefList<Item>.Create();
foreach (var item in FindItemsByType(recurse))
{
if (type.IsInstanceOfType(item))
{
typedItems.Add(item);
}
}
var groups = new List<List<Item>>();
var idx = 0;
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var a = typedItems[idx++];
var group = new List<Item>();
group.Add(a);
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var b = typedItems[idx];
var v = grouper(a, b);
@ -929,19 +934,27 @@ public class Container : Item
for (var i = 0; i < types.Length; ++i)
{
var typedItems = CollectionsMarshal.AsSpan(FindItemsByType(types[i], recurse));
var type = types[i];
using var typedItems = PooledRefList<Item>.Create();
foreach (var item in FindItemsByType(recurse))
{
if (type.IsInstanceOfType(item))
{
typedItems.Add(item);
}
}
var groups = new List<List<Item>>();
var idx = 0;
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var a = typedItems[idx++];
var group = new List<Item>();
group.Add(a);
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var b = typedItems[idx];
var v = grouper(a, b);
@ -1253,25 +1266,31 @@ public class Container : Item
public bool ConsumeTotal(Type type, int amount = 1, bool recurse = true, OnItemConsumed callback = null)
{
var items = CollectionsMarshal.AsSpan(FindItemsByType(type, recurse));
var total = 0;
using var items = PooledRefQueue<Item>.Create();
// First pass, compute total
var total = 0;
for (var i = 0; i < items.Length; ++i)
foreach (var item in FindItemsByType(recurse))
{
total += items[i].Amount;
if (type.IsInstanceOfType(item))
{
Items.Add(item);
total += item.Amount;
if (total >= amount)
{
break;
}
}
}
// We have enough, so consume it
if (total >= amount)
{
// We've enough, so consume it
var need = amount;
for (var i = 0; i < items.Length; ++i)
while (items.Count > 0)
{
var item = items[i];
var item = items.Dequeue();
var theirAmount = item.Amount;
@ -1287,7 +1306,6 @@ public class Container : Item
callback?.Invoke(item, need);
item.Consume(need);
return true;
}
}
@ -1361,12 +1379,19 @@ public class Container : Item
var best = 0;
var typedItems = CollectionsMarshal.AsSpan(FindItemsByType(type, recurse));
using var typedItems = PooledRefList<Item>.Create();
foreach (var item in FindItemsByType(recurse))
{
if (type.IsInstanceOfType(item))
{
typedItems.Add(item);
}
}
var groups = new List<List<Item>>();
var idx = 0;
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var a = typedItems[idx++];
var group = new List<Item>
@ -1374,7 +1399,7 @@ public class Container : Item
a
};
while (idx < typedItems.Length)
while (idx < typedItems.Count)
{
var b = typedItems[idx];
var v = grouper(a, b);
@ -1540,9 +1565,12 @@ public class Container : Item
public int GetAmount(Type type, bool recurse = true)
{
var total = 0;
foreach (var item in FindItemsByType(type, recurse))
foreach (var item in FindItemsByType(recurse))
{
total += item.Amount;
if (type.IsInstanceOfType(item))
{
total += item.Amount;
}
}
return total;
@ -1551,9 +1579,12 @@ public class Container : Item
public int GetAmount(Type[] types, bool recurse = true)
{
var total = 0;
foreach (var item in FindItemsByType(types, recurse))
foreach (var item in FindItemsByType(recurse))
{
total += item.Amount;
if (InTypeList(item, types))
{
total += item.Amount;
}
}
return total;
@ -1561,23 +1592,12 @@ public class Container : Item
public List<Item> FindItemsByType(Type type, bool recurse = true)
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
var items = new List<Item>();
while (queue.Count > 0)
foreach (var item in FindItemsByType(recurse))
{
var container = queue.Dequeue();
foreach (var item in container.Items)
if (type.IsInstanceOfType(item))
{
if (type.IsInstanceOfType(item))
{
items.Add(item);
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
items.Add(item);
}
}
@ -1586,23 +1606,12 @@ public class Container : Item
public List<Item> FindItemsByType(Type[] types, bool recurse = true)
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
var items = new List<Item>();
while (queue.Count > 0)
foreach (var item in FindItemsByType(recurse))
{
var container = queue.Dequeue();
foreach (var item in container.Items)
if (InTypeList(item, types))
{
if (InTypeList(item, types))
{
items.Add(item);
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
items.Add(item);
}
}
@ -1611,22 +1620,11 @@ public class Container : Item
public Item FindItemByType(Type type, bool recurse = true)
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
while (queue.Count > 0)
foreach (var item in FindItemsByType(recurse))
{
var container = queue.Dequeue();
foreach (var item in container.Items)
if (type.IsInstanceOfType(item))
{
if (type.IsInstanceOfType(item))
{
return item;
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
return item;
}
}
@ -1635,73 +1633,17 @@ public class Container : Item
public Item FindItemByType(Type[] types, bool recurse = true)
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
while (queue.Count > 0)
foreach (var item in FindItemsByType(recurse))
{
var container = queue.Dequeue();
foreach (var item in container.Items)
if (InTypeList(item, types))
{
if (InTypeList(item, types))
{
return item;
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
return item;
}
}
return null;
}
public List<T> FindItemsByType<T>(Predicate<T> predicate) where T : Item => FindItemsByType(true, predicate);
/// <summary>
/// Performs a Breadth-First search through all the <see cref="Item" />s and
/// nested <see cref="Container" />s within this <see cref="Container" />.
/// </summary>
/// <typeparam name="T">Type of objects being searched for</typeparam>
/// <param name="recurse">
/// Optional: If true, the search will recursively
/// check any nested <see cref="Container" />s; otherwise, nested
/// <see cref="Container" />s will not be searched.
/// </param>
/// <param name="predicate">
/// Optional: A predicate to check if the <see cref="Item" />
/// of type <typeparamref name="T" /> is one of the targets of the search.
/// </param>
/// <returns>
/// A list of <see cref="Item" />s of type <typeparamref name="T" /> that matche the optional
/// <paramref name="predicate" />.
/// </returns>
public List<T> FindItemsByType<T>(bool recurse = true, Predicate<T> predicate = null) where T : Item
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
var items = new List<T>();
while (queue.Count > 0)
{
var container = queue.Dequeue();
foreach (var item in container.Items)
{
if (item is T typedItem && predicate?.Invoke(typedItem) != false)
{
items.Add(typedItem);
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
}
}
return items;
}
/// <summary>
/// Performs a Breadth-First search through all the <see cref="Item" />s and
/// nested <see cref="Container" />s within this <see cref="Container" />.
@ -1722,37 +1664,14 @@ public class Container : Item
/// </returns>
public T FindItemByType<T>(bool recurse = true, Predicate<T> predicate = null) where T : Item
{
using var queue = PooledRefQueue<Container>.Create(128);
queue.Enqueue(this);
while (queue.Count > 0)
foreach (var item in FindItemsByType(recurse, predicate))
{
var container = queue.Dequeue();
foreach (var item in container.Items)
{
if (item is T typedItem && predicate?.Invoke(typedItem) != false)
{
return typedItem;
}
if (recurse && item is Container itemContainer)
{
queue.Enqueue(itemContainer);
}
}
return item;
}
return null;
}
private class GroupComparer : IComparer<Item>
{
private readonly CheckItemGroup m_Grouper;
public GroupComparer(CheckItemGroup grouper) => m_Grouper = grouper;
public int Compare(Item a, Item b) => m_Grouper(a, b);
}
[Flags]
private enum SaveFlag : byte
{