feat: Adds weighted values (#1265)
This commit is contained in:
parent
1d8e8ec0a8
commit
efaf2e947a
2 changed files with 130 additions and 0 deletions
|
|
@ -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)
|
||||
{
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue