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

@ -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);
}

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)
{