Masuit.Tools/Masuit.Tools.Abstractions/RandomSelector/MultipleSelector.cs

using System;
using System.Collections.Generic;

namespace Masuit.Tools.RandomSelector;

/// <summary>
/// 多选器
/// </summary>
/// <typeparam name="T"></typeparam>
internal class MultipleSelector<T> : SelectorBase<T>
{
    internal MultipleSelector(WeightedSelector<T> weightedSelector) : base(weightedSelector)
    {
    }

    internal IEnumerable<T> Select(int count)
    {
        Validate(ref count);
        var items = new List<WeightedItem<T>>(WeightedSelector.Items);
        int result = 0;
        do
        {
            var item = WeightedSelector.Option.AllowDuplicate ? BinarySelect(items) : LinearSelect(items);
            yield return item.Value;
            result++;
            if (!WeightedSelector.Option.AllowDuplicate)
            {
                items.Remove(item);
            }
        } while (result < count);
    }

    private void Validate(ref int count)
    {
        if (count <= 0)
        {
            throw new InvalidOperationException("筛选个数必须大于0");
        }

        var items = WeightedSelector.Items;

        if (items.Count == 0)
        {
            count = 0;
            return;
        }

        if (!WeightedSelector.Option.AllowDuplicate && items.Count < count)
        {
            count = items.Count;
        }
    }
}