Jakiś czas temu napisałem zestaw metod rozszerzających, które działają dla kilku różnych rodzajów Enum
. Jeden w szczególności działa dla tego, co próbujesz osiągnąć i obsługuje Enum
s z FlagsAttribute
oraz Enum
z różnymi podstawowymi typami.
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;
}
Stamtąd możesz dodać inne, bardziej szczegółowe metody rozszerzeń.
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);
}
Ten zmieni typy, Enum
tak jak próbujesz to zrobić.
public static tEnum ChangeType<tEnum>(this Enum e) where tEnum : IComparable
{
return SetFlags(e, default(tEnum), true, false);
}
Ostrzegamy jednak, że MOŻESZ konwertować między dowolnymi Enum
innymi Enum
przy użyciu tej metody, nawet tymi, które nie mają flag. Na przykład:
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>();
}
Zmienna turtle
będzie miała wartość Turtle.Blue
.
Jednak Enum
użycie tej metody zapewnia bezpieczeństwo przed niezdefiniowanymi wartościami. Na przykład:
static void Main(string[] args)
{
Turtle turtle = Turtle.Yellow;
WriteAccess access = turtle.ChangeType<WriteAccess>();
}
W tym przypadku access
zostanie ustawiony na WriteAccess.ReadWrite
, ponieważWriteAccess
Enum
maksymalna wartość wynosi 3.
Innym efektem ubocznym mieszania Enum
s z tymi FlagsAttribute
i bez niego jest to, że proces konwersji nie spowoduje dopasowania 1 do 1 między ich wartościami.
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>();
}
W tym przypadku letters
będzie miał wartość Letters.H
zamiast Letters.D
, ponieważ wartość bazowa Flavors.Peach
wynosi 8. Ponadto przyniosłaby konwersja z Flavors.Cherry | Flavors.Grape
na , co może wydawać się nieintuicyjne.Letters
Letters.C