Случайное значение из перечисления Flags - PullRequest
Новая
       15

Случайное значение из перечисления Flags

5 голосов
/ 09 июня 2010

Скажем, у меня есть функция, которая принимает перечисление, украшенное атрибутом Flags. Если значение перечисления является комбинацией нескольких элементов перечисления, как я могу извлечь один из этих элементов случайным образом? У меня есть следующее, но, кажется, должен быть лучший способ. 

[Flags]
enum Colours
{
    Blue = 1,
    Red = 2,
    Green = 4
}

public static void Main()
{
    var options = Colours.Blue | Colours.Red | Colours.Green;
    var opts = options.ToString().Split(',');
    var rand = new Random();
    var selected = opts[rand.Next(opts.Length)].Trim();
    var myEnum = Enum.Parse(typeof(Colours), selected);
    Console.WriteLine(myEnum);
    Console.ReadLine();
}

Ответы [ 4 ]

9 голосов
/ 09 июня 2010 
var options = Colours.Blue | Colours.Green;

var matching = Enum.GetValues(typeof(Colours))
                   .Cast<Colours>()
                   .Where(c => (options & c) == c)    // or use HasFlag in .NET4
                   .ToArray();

var myEnum = matching[new Random().Next(matching.Length)];
9 голосов
/ 09 июня 2010

Вы можете вызвать Enum.GetValues, чтобы получить массив определенных значений перечисления, например: 

var rand = new Random();

Colors[] allValues = (Colors[])Enum.GetValues(typeof(Colors));
Colors value = allValues[rand.Next(allValues.Length)];
2 голосов
/ 09 июня 2010

Если вы не возражаете против небольшого приведения, и ваше перечисление имеет базовый тип int, следующее будет работать и быстро. 

var rand = new Random();
const int mask = (int)(Colours.Blue | Colours.Red | Colours.Green);
return (Colours)(mask & (rand.Next(mask) + 1));

Если вы хотите установить только один флаг,Вы можете сделать следующее: 

var rand = new Random();
return (Colours)(0x1 << (rand.Next(3)));
1 голос
/ 09 июня 2010

Если я правильно понимаю, вопрос состоит в том, чтобы вернуть случайное значение перечисления из значения перечисления flags, а не вернуть случайный член из перечисления flags. 

    [Flags]
    private enum Shot
    {
        Whisky = 1,
        Absynthe = 2,
        Pochin = 4,
        BrainEraser = Whisky | Absynthe | Pochin
    }

    [Test]
    public void Test()
    {
        Shot myCocktail = Shot.Absynthe | Shot.Whisky;

        Shot randomShotInCocktail = GetRandomShotFromCocktail(myCocktail);
    }

    private static Shot GetRandomShotFromCocktail(Shot cocktail)
    {
        Random random = new Random();

        Shot[] cocktailShots = Enum.GetValues(typeof(Shot)).
           Cast<Shot>().
           Where(x => cocktail.HasFlag(x)).ToArray();

        Shot randomShot = cocktailShots[random.Next(0, cocktailShots.Length)];

        return randomShot;
    }

Edit

И, очевидно, вы должны проверить, что перечисление является допустимым значением, например: 

 Shot myCocktail = (Shot)666;

Редактировать

Упрощенный

...