有没有其他方法来测试枚举位字段?

时间:2010-09-16 20:33:02

标签: c# enums bit-manipulation

将Enums与位字段一起使用时:

   enum  ReallyBigEnum  { FirstChoice = 0x01, AnotherOption = 0x02 }
   ReallyBigEnum  flag = ReallyBigEnum.FirstChoice | ReallyBigEnum.AnotherOption;

用于测试位的代码是:

   if( (flag & ReallyBigEnum.AnotherOption) == ReallyBigEnum.AnotherOption ) { ... }

由于需要重复该位,因此看起来很冗长且容易出错 正在接受测试。

如果有某种方式可以说:

   if( flag.IsSet( ReallyBigEnum.AnotherOption ) ) { ... }

但Enums不支持实例方法。所以,我尝试了一个模板功能:

   class Enums
   {
      public static bool IsSet<T>( T flag, T bit ) { return (flag & bit) == bit; }
   }

但测试位的代码如下所示:

   if( Enums.IsSet<ReallyBigEnum>( flag, ReallyBigEnum.AnotherOption ) ) { ... }

这是很多要写的。然后我试着缩短它:

   class Enums
   {
      public static bool IsSet( int flag, int bit ) { return (flag & bit) == bit; }
   }

但是你必须将每个值转换为它的基类型:

   if( Enums.IsSet( (int)flag, (int)ReallyBigEnum.AnotherOption ) ) { ... }

这也是代码的痛苦并且失去了类型检查的好处。

可以编写相同的函数来使用'object'参数,但随后 必须测试对象类型和底层基类型。

所以,我坚持使用顶部的标准冗余方式。

有没有人对干净,简单的测试枚举位字段的方法有其他想法?

非常感谢。

3 个答案:

答案 0 :(得分:6)

最多.Net 3.5这是您唯一的选择。在.Net 4.0中,枚举上有一个HasFlag方法。

答案 1 :(得分:2)

if((flag & ReallyBigEnum.AnotherOption) != 0) { ... }

更新:

只有当你测试一个位时,上面显然才有效。如果要测试多个位,则需要其他内容,具体取决于您是检查所有位还是任何位。

测试是否设置了任何一组位

对于这种情况,只需使用单比特版本的变体。

if((flag & (ReallyBigEnum.FirstOption | ReallyBigEnum.AnotherOption)) != 0) { ... }

测试是否已设置所有位集

为了达到清晰度和可靠性,我建议创建一个包含所有位的常量。

const int ReallyBigEnum WickedAwesomeOptions = ReallyBigEnum.FirstOption | ReallyBigEnum.AnotherOption;
...
if (flag & WickedAwesomeOptions == WickedAwesomeOptions) { ... }

部分冗余仍然存在,但它并不脆弱或容易混淆,并且易于维护。

如果位组合适用广泛,那么您可以将它添加到枚举本身。

[Flags]
enum ReallyBigEnum
{
    FirstOption = 1,
    AnotherOption = 2,
    WickedAwesomeOptions = FirstOption | AnotherOption,
}
....
if (flag & ReallyBigEnum.WickedAwesomeOptions == ReallyBigEnum.WickedAwesomeOptions) { ... }

答案 2 :(得分:2)

/// <summary>
/// Taken from https://stackoverflow.com/questions/9033/hidden-features-of-c/407325#407325
/// instead of doing (enum & value) == value you can now use enum.Has(value)
/// </summary>
/// <typeparam name="T">Type of enum</typeparam>
/// <param name="type">The enum value you want to test</param>
/// <param name="value">Flag Enum Value you're looking for</param>
/// <returns>True if the type has value bit set</returns>
public static bool Has<T>(this System.Enum type, T value)
{
   return (((int)(object)type & (int)(object)value) == (int)(object)value);
} 

以下是我从above link

中获取的扩展方法

使用:

MyFlagEnum e = MyFlagEnum.First | MyFlagEnum.Second;
if(e.Has(MyFlagEnum.First))
{
   // Do stuff
}