123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872 |
- using NUnit.Framework;
- using UnityEngine;
- using Unity.Collections.LowLevel.Unsafe;
- using Unity.PerformanceTesting;
- using Unity.PerformanceTesting.Benchmark;
- using System.Runtime.CompilerServices;
- using System.Threading;
- using Unity.Mathematics;
- using UnityEditor;
- using Random = Unity.Mathematics.Random;
-
- namespace Unity.Collections.PerformanceTests
- {
- static class HashMapUtil
- {
- internal const uint K_RANDOM_SEED_1 = 2210602657;
- internal const uint K_RANDOM_SEED_2 = 2210602658;
- internal const uint K_RANDOM_SEED_3 = 2210602659;
- static public void AllocInt(ref NativeHashMap<int, int> container, int capacity, bool addValues)
- {
- if (capacity >= 0)
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_1);
- container = new NativeHashMap<int, int>(capacity, Allocator.Persistent);
- if (addValues)
- {
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (container.TryAdd(randKey, keysAdded))
- {
- ++keysAdded;
- }
- }
- }
- }
- else
- container.Dispose();
- }
- static public void AllocInt(ref UnsafeHashMap<int, int> container, int capacity, bool addValues)
- {
- if (capacity >= 0)
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_1);
- container = new UnsafeHashMap<int, int>(capacity, Allocator.Persistent);
- if (addValues)
- {
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (container.TryAdd(randKey, keysAdded))
- {
- ++keysAdded;
- }
- }
- }
- }
- else
- container.Dispose();
- }
- static public object AllocBclContainer(int capacity, bool addValues)
- {
- if (capacity < 0)
- return null;
-
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_1);
- var bclContainer = new System.Collections.Generic.Dictionary<int, int>(capacity);
- if (addValues)
- {
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (bclContainer.TryAdd(randKey, keysAdded))
- {
- ++keysAdded;
- }
- }
- }
- return bclContainer;
- }
- static public void CreateRandomKeys(int capacity, ref UnsafeList<int> keys)
- {
- if (capacity >= 0)
- {
- keys = new UnsafeList<int>(capacity, Allocator.Persistent);
- using (UnsafeHashSet<int> randomFilter = new UnsafeHashSet<int>(capacity, Allocator.Persistent))
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_2);
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (randomFilter.Add(randKey))
- {
- keys.Add(randKey);
- ++keysAdded;
- }
- }
- }
-
- }
- else
- keys.Dispose();
- }
-
- static public void CreateRandomKeys(int capacity, ref UnsafeList<int> keys, ref UnsafeHashMap<int, int> hashMap)
- {
- if (capacity >= 0)
- {
- keys = new UnsafeList<int>(capacity, Allocator.Persistent);
- using (UnsafeHashSet<int> randomFilter = new UnsafeHashSet<int>(capacity, Allocator.Persistent))
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_2);
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (randomFilter.Add(randKey) && !hashMap.ContainsKey(randKey))
- {
- keys.Add(randKey);
- ++keysAdded;
- }
- }
- }
-
- }
- else
- keys.Dispose();
- }
-
- static public void CreateRandomKeys(int capacity, ref UnsafeList<int> keys, ref System.Collections.Generic.Dictionary<int, int> hashMap)
- {
- if (capacity >= 0)
- {
- keys = new UnsafeList<int>(capacity, Allocator.Persistent);
- using (UnsafeHashSet<int> randomFilter = new UnsafeHashSet<int>(capacity, Allocator.Persistent))
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_2);
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (randomFilter.Add(randKey) && !hashMap.ContainsKey(randKey))
- {
- keys.Add(randKey);
- ++keysAdded;
- }
- }
- }
-
- }
- else
- keys.Dispose();
- }
-
- static public void CreateRandomKeys(int capacity, ref UnsafeList<int> keys, ref NativeHashMap<int, int> hashMap)
- {
- if (capacity >= 0)
- {
- keys = new UnsafeList<int>(capacity, Allocator.Persistent);
- using (UnsafeHashSet<int> randomFilter = new UnsafeHashSet<int>(capacity, Allocator.Persistent))
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_2);
- int keysAdded = 0;
-
- while (keysAdded < capacity)
- {
- int randKey = random.NextInt();
- if (randomFilter.Add(randKey) && !hashMap.ContainsKey(randKey))
- {
- keys.Add(randKey);
- ++keysAdded;
- }
- }
- }
-
- }
- else
- keys.Dispose();
- }
-
- static public void RandomlyShuffleKeys(int capacity, ref UnsafeList<int> keys)
- {
- if (capacity >= 0)
- {
- Unity.Mathematics.Random random = new Mathematics.Random(K_RANDOM_SEED_3);
- for (int i = 0; i < capacity; i++)
- {
- int keyAt = keys[i];
- int randomIndex = random.NextInt(0, capacity - 1);
- keys[i] = keys[randomIndex];
- keys[randomIndex] = keyAt;
- }
- }
- }
- }
-
- struct HashMapIsEmpty100k : IBenchmarkContainer
- {
- const int kIterations = 100_000;
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
-
- public void AllocNativeContainer(int capacity) => HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- public void AllocUnsafeContainer(int capacity) => HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- public object AllocBclContainer(int capacity) => HashMapUtil.AllocBclContainer(capacity, true);
-
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureNativeContainer()
- {
- var reader = nativeContainer.AsReadOnly();
- for (int i = 0; i < kIterations; i++)
- _ = reader.IsEmpty;
- }
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureUnsafeContainer()
- {
- var reader = unsafeContainer.AsReadOnly();
- for (int i = 0; i < kIterations; i++)
- _ = reader.IsEmpty;
- }
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- for (int i = 0; i < kIterations; i++)
- _ = bclContainer.Count == 0;
- }
- }
-
- struct HashMapCount100k : IBenchmarkContainer
- {
- const int kIterations = 100_000;
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
-
- public void AllocNativeContainer(int capacity) => HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- public void AllocUnsafeContainer(int capacity) => HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- public object AllocBclContainer(int capacity) => HashMapUtil.AllocBclContainer(capacity, true);
-
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureNativeContainer()
- {
- var reader = nativeContainer.AsReadOnly();
- for (int i = 0; i < kIterations; i++)
- _ = reader.Count;
- }
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureUnsafeContainer()
- {
- var reader = unsafeContainer.AsReadOnly();
- for (int i = 0; i < kIterations; i++)
- _ = reader.Count;
- }
- [MethodImpl(MethodImplOptions.NoOptimization)]
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- for (int i = 0; i < kIterations; i++)
- _ = bclContainer.Count;
- }
- }
-
- struct HashMapToNativeArrayKeys : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
-
- public void AllocNativeContainer(int capacity) => HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- public void AllocUnsafeContainer(int capacity) => HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- public object AllocBclContainer(int capacity) => HashMapUtil.AllocBclContainer(capacity, true);
-
- public void MeasureNativeContainer()
- {
- var asArray = nativeContainer.GetKeyArray(Allocator.Temp);
- asArray.Dispose();
- }
- public void MeasureUnsafeContainer()
- {
- var asArray = unsafeContainer.GetKeyArray(Allocator.Temp);
- asArray.Dispose();
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int[] asArray = new int[bclContainer.Count];
- bclContainer.Keys.CopyTo(asArray, 0);
- }
- }
-
- struct HashMapToNativeArrayValues : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
-
- public void AllocNativeContainer(int capacity) => HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- public void AllocUnsafeContainer(int capacity) => HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- public object AllocBclContainer(int capacity) => HashMapUtil.AllocBclContainer(capacity, true);
-
- public void MeasureNativeContainer()
- {
- var asArray = nativeContainer.GetValueArray(Allocator.Temp);
- asArray.Dispose();
- }
- public void MeasureUnsafeContainer()
- {
- var asArray = unsafeContainer.GetValueArray(Allocator.Temp);
- asArray.Dispose();
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int[] asArray = new int[bclContainer.Count];
- bclContainer.Values.CopyTo(asArray, 0);
- }
- }
-
- struct HashMapInsert : IBenchmarkContainer
- {
- int capacity;
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- void IBenchmarkContainer.SetParams(int capacity, params int[] args) => this.capacity = capacity;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys, ref nativeContainer);
- }
-
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys, ref unsafeContainer);
- }
-
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys, ref bclContainer);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- for (int i = 0; i < capacity; i++)
- nativeContainer.Add(keys[i], i);
- }
- public void MeasureUnsafeContainer()
- {
- for (int i = 0; i < capacity; i++)
- unsafeContainer.Add(keys[i], i);
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- for (int i = 0; i < capacity; i++)
- bclContainer.Add(keys[i], i);
- }
- }
-
- struct HashMapAddGrow : IBenchmarkContainer
- {
- int capacity;
- int toAdd;
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- void IBenchmarkContainer.SetParams(int capacity, params int[] args)
- {
- this.capacity = capacity;
- toAdd = args[0];
- }
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- int toAddCount = capacity < 0 ? -1 : toAdd;
- HashMapUtil.CreateRandomKeys(toAddCount, ref keys, ref nativeContainer);
- }
-
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- int toAddCount = capacity < 0 ? -1 : toAdd;
- HashMapUtil.CreateRandomKeys(toAddCount, ref keys, ref unsafeContainer);
- }
-
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, true);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int toAddCount = capacity < 0 ? -1 : toAdd;
- HashMapUtil.CreateRandomKeys(toAddCount, ref keys, ref bclContainer);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- // Intentionally setting capacity small and growing by adding more items
- for (int i = 0; i < toAdd; i++)
- nativeContainer.Add(keys[i], i);
- }
- public void MeasureUnsafeContainer()
- {
- // Intentionally setting capacity small and growing by adding more items
- for (int i = 0; i < toAdd; i++)
- unsafeContainer.Add(keys[i], i);
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- // Intentionally setting capacity small and growing by adding more items
- for (int i = 0; i < toAdd; i++)
- bclContainer.Add(keys[i], i);
- }
- }
-
- struct HashMapContains : IBenchmarkContainer
- {
- int capacity;
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
- void IBenchmarkContainer.SetParams(int capacity, params int[] args) => this.capacity = capacity;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- nativeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- unsafeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- bclContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- var reader = nativeContainer.AsReadOnly();
- bool data = false;
- for (int i = 0; i < capacity; i++)
- Volatile.Write(ref data, reader.ContainsKey(keys[i]));
- }
- public void MeasureUnsafeContainer()
- {
- var reader = unsafeContainer.AsReadOnly();
- bool data = false;
- for (int i = 0; i < capacity; i++)
- Volatile.Write(ref data, reader.ContainsKey(keys[i]));
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- bool data = false;
- for (int i = 0; i < capacity; i++)
- Volatile.Write(ref data, bclContainer.ContainsKey(keys[i]));
- }
- }
-
- struct HashMapIndexedRead : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- nativeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- unsafeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- bclContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- var reader = nativeContainer.AsReadOnly();
- int insertions = keys.Length;
- int value = 0;
- for (int i = 0; i < insertions; i++)
- Volatile.Write(ref value, reader[keys[i]]);
- }
- public void MeasureUnsafeContainer()
- {
- var reader = unsafeContainer.AsReadOnly();
- int insertions = keys.Length;
- int value = 0;
- for (int i = 0; i < insertions; i++)
- Volatile.Write(ref value, reader[keys[i]]);
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int insertions = keys.Length;
- int value = 0;
- for (int i = 0; i < insertions; i++)
- Volatile.Write(ref value, bclContainer[keys[i]]);
- }
- }
-
- struct HashMapIndexedWrite : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- nativeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- unsafeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public object AllocBclContainer(int capacity)
- {
- var container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- bclContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- nativeContainer[keys[i]] = i;
- }
- public void MeasureUnsafeContainer()
- {
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- unsafeContainer[keys[i]] = i;
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- bclContainer[keys[i]] = i;
- }
- }
-
- struct HashMapTryGetValue : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- nativeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- unsafeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- bclContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- var reader = nativeContainer.AsReadOnly();
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- {
- reader.TryGetValue(keys[i], out var value);
- Volatile.Read(ref value);
- }
- }
- public void MeasureUnsafeContainer()
- {
- var reader = unsafeContainer.AsReadOnly();
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- {
- reader.TryGetValue(keys[i], out var value);
- Volatile.Read(ref value);
- }
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- {
- bclContainer.TryGetValue(keys[i], out var value);
- Volatile.Read(ref value);
- }
- }
- }
-
- struct HashMapRemove : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- UnsafeList<int> keys;
-
- public void AllocNativeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref nativeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- nativeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public void AllocUnsafeContainer(int capacity)
- {
- HashMapUtil.AllocInt(ref unsafeContainer, capacity, false);
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- unsafeContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- }
- public object AllocBclContainer(int capacity)
- {
- object container = HashMapUtil.AllocBclContainer(capacity, false);
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- HashMapUtil.CreateRandomKeys(capacity, ref keys);
- for (int i = 0; i < capacity; i++)
- bclContainer.TryAdd(keys[i], i);
- HashMapUtil.RandomlyShuffleKeys(capacity, ref keys);
- return container;
- }
-
- public void MeasureNativeContainer()
- {
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- nativeContainer.Remove(keys[i]);
- }
- public void MeasureUnsafeContainer()
- {
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- unsafeContainer.Remove(keys[i]);
- }
- public void MeasureBclContainer(object container)
- {
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- int insertions = keys.Length;
- for (int i = 0; i < insertions; i++)
- bclContainer.Remove(keys[i]);
- }
- }
-
- struct HashMapForEach : IBenchmarkContainer
- {
- NativeHashMap<int, int> nativeContainer;
- UnsafeHashMap<int, int> unsafeContainer;
- public int total;
-
- public void AllocNativeContainer(int capacity) => HashMapUtil.AllocInt(ref nativeContainer, capacity, true);
- public void AllocUnsafeContainer(int capacity) => HashMapUtil.AllocInt(ref unsafeContainer, capacity, true);
- public object AllocBclContainer(int capacity) => HashMapUtil.AllocBclContainer(capacity, true);
-
- public void MeasureNativeContainer()
- {
- foreach (var pair in nativeContainer)
- Volatile.Read(ref pair.Value);
- }
- public void MeasureUnsafeContainer()
- {
- foreach (var pair in unsafeContainer)
- Volatile.Read(ref pair.Value);
- }
- public void MeasureBclContainer(object container)
- {
- int value = 0;
- var bclContainer = (System.Collections.Generic.Dictionary<int, int>)container;
- foreach (var pair in bclContainer)
- Volatile.Write(ref value, pair.Value);
- }
- }
-
-
- [Benchmark(typeof(BenchmarkContainerType))]
- [BenchmarkNameOverride(BenchmarkContainerConfig.BCL, "Dictionary")]
- class HashMap
- {
- #if UNITY_EDITOR
- [UnityEditor.MenuItem(BenchmarkContainerConfig.kMenuItemIndividual + nameof(HashMap))]
- static void RunIndividual()
- => BenchmarkContainerConfig.RunBenchmark(typeof(HashMap));
- #endif
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void IsEmpty_x_100k(
- [Values(0, 100)] int capacity,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapIsEmpty100k>.Run(capacity, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void Count_x_100k(
- [Values(0, 100)] int capacity,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapCount100k>.Run(capacity, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void ToNativeArrayKeys(
- [Values(10000, 100000, 1000000)] int capacity,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapToNativeArrayKeys>.Run(capacity, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void ToNativeArrayValues(
- [Values(10000, 100000, 1000000)] int capacity,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapToNativeArrayValues>.Run(capacity, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void Insert(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapInsert>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- [BenchmarkTestFootnote("Incrementally grows from `capacity` until reaching size of `growTo`")]
- public unsafe void AddGrow(
- [Values(4, 65536)] int capacity,
- [Values(1024 * 1024)] int growTo,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapAddGrow>.Run(capacity, type, growTo);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void Contains(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapContains>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void IndexedRead(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapIndexedRead>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void IndexedWrite(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapIndexedWrite>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void TryGetValue(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapTryGetValue>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void Remove(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapRemove>.Run(insertions, type);
- }
-
- [Test, Performance]
- [Category("Performance")]
- public unsafe void Foreach(
- [Values(10000, 100000, 1000000)] int insertions,
- [Values] BenchmarkContainerType type)
- {
- BenchmarkContainerRunner<HashMapForEach>.Run(insertions, type);
- }
- }
- }
|