feat: Adds weighted values (#1265)
This commit is contained in:
parent
1d8e8ec0a8
commit
efaf2e947a
2 changed files with 130 additions and 0 deletions
34
Projects/Server/Random/WeightedValue.cs
Normal file
34
Projects/Server/Random/WeightedValue.cs
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
/*************************************************************************
|
||||
* ModernUO *
|
||||
* Copyright 2019-2022 - ModernUO Development Team *
|
||||
* Email: hi@modernuo.com *
|
||||
* File: WeightedValue.cs *
|
||||
* *
|
||||
* This program is free software: you can redistribute it and/or modify *
|
||||
* it under the terms of the GNU General Public License as published by *
|
||||
* the Free Software Foundation, either version 3 of the License, or *
|
||||
* (at your option) any later version. *
|
||||
* *
|
||||
* You should have received a copy of the GNU General Public License *
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
||||
*************************************************************************/
|
||||
|
||||
using System;
|
||||
|
||||
namespace Server.Random;
|
||||
|
||||
public struct WeightedValue<T>
|
||||
{
|
||||
public T Value { get; }
|
||||
public int Weight { get; }
|
||||
|
||||
public WeightedValue(int weight, T value)
|
||||
{
|
||||
Value = value;
|
||||
Weight = weight;
|
||||
}
|
||||
|
||||
public bool IsValid => Weight > 0;
|
||||
|
||||
public static implicit operator WeightedValue<T>(ValueTuple<int, T> pair) => new(pair.Item1, pair.Item2);
|
||||
}
|
||||
|
|
@ -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