feat: Adds weighted values (#1265)

This commit is contained in:
Kamron Batman 2022-11-22 19:53:20 -08:00 committed by GitHub
parent 1d8e8ec0a8
commit efaf2e947a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 130 additions and 0 deletions

View file

@ -1042,6 +1042,102 @@ public static class Utility
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static T RandomElement<T>(this IList<T> list) => list.RandomElement(default);
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(this IList<WeightedValue<T>> weightedValues)
{
var totalWeight = 0;
for (var i = 0; i < weightedValues.Count; i++)
{
totalWeight += weightedValues[i].Weight;
}
return RandomWeightedElement(weightedValues, totalWeight);
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(this ReadOnlySpan<WeightedValue<T>> weightedValues, int totalWeight)
{
var random = Random(totalWeight);
for (var i = 0; i < weightedValues.Length; i++)
{
var weightedValue = weightedValues[i];
random -= weightedValue.Weight;
if (random <= 0)
{
return weightedValue;
}
}
return default;
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(this ReadOnlySpan<WeightedValue<T>> weightedValues)
{
var totalWeight = 0;
for (var i = 0; i < weightedValues.Length; i++)
{
totalWeight += weightedValues[i].Weight;
}
return RandomWeightedElement(weightedValues, totalWeight);
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(this IList<WeightedValue<T>> weightedValues, int totalWeight)
{
var random = Random(totalWeight);
for (var i = 0; i < weightedValues.Count; i++)
{
var weightedValue = weightedValues[i];
random -= weightedValue.Weight;
if (random <= 0)
{
return weightedValue;
}
}
return default;
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(this PooledRefList<WeightedValue<T>> weightedValues)
{
var totalWeight = 0;
for (var i = 0; i < weightedValues.Count; i++)
{
totalWeight += weightedValues[i].Weight;
}
return RandomWeightedElement(weightedValues, totalWeight);
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static WeightedValue<T> RandomWeightedElement<T>(
this PooledRefList<WeightedValue<T>> weightedValues,
int totalWeight
)
{
var random = Random(totalWeight);
for (var i = 0; i < weightedValues.Count; i++)
{
var weightedValue = weightedValues[i];
random -= weightedValue.Weight;
if (random <= 0)
{
return weightedValue;
}
}
return default;
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
public static T TakeRandomElement<T>(this IList<T> list)
{