Нещодавно я написав методи розширення набору, які працюють для декількох різних типів Enums. Зокрема, одна працює на те, що ви намагаєтеся виконати, і обробляє Enums FlagsAttributeяк і Enums з різними основними типами.
public static tEnum SetFlags<tEnum>(this Enum e, tEnum flags, bool set, bool typeCheck = true) where tEnum : IComparable
{
if (typeCheck)
{
if (e.GetType() != flags.GetType())
throw new ArgumentException("Argument is not the same type as this instance.", "flags");
}
var flagsUnderlyingType = Enum.GetUnderlyingType(typeof(tEnum));
var firstNum = Convert.ToUInt32(e);
var secondNum = Convert.ToUInt32(flags);
if (set)
firstNum |= secondNum;
else
firstNum &= ~secondNum;
var newValue = (tEnum)Convert.ChangeType(firstNum, flagsUnderlyingType);
if (!typeCheck)
{
var values = Enum.GetValues(typeof(tEnum));
var lastValue = (tEnum)values.GetValue(values.Length - 1);
if (newValue.CompareTo(lastValue) > 0)
return lastValue;
}
return newValue;
}
Звідти ви можете додати інші більш конкретні методи розширення.
public static tEnum AddFlags<tEnum>(this Enum e, tEnum flags) where tEnum : IComparable
{
SetFlags(e, flags, true);
}
public static tEnum RemoveFlags<tEnum>(this Enum e, tEnum flags) where tEnum : IComparable
{
SetFlags(e, flags, false);
}
Цей змінить типи Enums, як ви намагаєтесь зробити.
public static tEnum ChangeType<tEnum>(this Enum e) where tEnum : IComparable
{
return SetFlags(e, default(tEnum), true, false);
}
Попереджуйте, що ви МОЖЕТЕ перетворити будь-який Enumта будь-який інший Enumза допомогою цього методу, навіть тих, у яких немає прапорів. Наприклад:
public enum Turtle
{
None = 0,
Pink,
Green,
Blue,
Black,
Yellow
}
[Flags]
public enum WriteAccess : short
{
None = 0,
Read = 1,
Write = 2,
ReadWrite = 3
}
static void Main(string[] args)
{
WriteAccess access = WriteAccess.ReadWrite;
Turtle turtle = access.ChangeType<Turtle>();
}
Змінна turtleматиме значення Turtle.Blue.
Однак існує безпека від невизначених Enumзначень за допомогою цього методу. Наприклад:
static void Main(string[] args)
{
Turtle turtle = Turtle.Yellow;
WriteAccess access = turtle.ChangeType<WriteAccess>();
}
У цьому випадку accessбуде встановлено значення WriteAccess.ReadWrite, оскільки значення WriteAccess Enumмає максимальне значення 3.
Ще один побічний ефект змішування Enums з тими FlagsAttributeі без них полягає в тому, що процес перетворення не призведе до відповідності 1 до 1 між їх значеннями.
public enum Letters
{
None = 0,
A,
B,
C,
D,
E,
F,
G,
H
}
[Flags]
public enum Flavors
{
None = 0,
Cherry = 1,
Grape = 2,
Orange = 4,
Peach = 8
}
static void Main(string[] args)
{
Flavors flavors = Flavors.Peach;
Letters letters = flavors.ChangeType<Letters>();
}
У цьому випадку lettersбуде мати значення Letters.Hзамість Letters.D, оскільки значення резервного значення Flavors.Peachстановить 8. Також конверсія з Flavors.Cherry | Flavors.Grapeв Lettersотримає вихід Letters.C, що може здатися неінтуїтивним.