如何使用枚举类作为一组标志?

时间:2013-08-31 23:43:42

标签: c++ c++11

假设我有一组标志和类似这样的类:

/// <summary>Options controlling a search for files.</summary>
enum class FindFilesOptions : unsigned char
{
    LocalSearch = 0,
    RecursiveSearch = 1,
    IncludeDotDirectories = 2
};

class FindFiles : boost::noncopyable
{
    /* omitted */
public:
    FindFiles(std::wstring const& pattern, FindFilesOptions options);
    /* omitted */
}

我希望调用者能够选择多个选项:

FindFiles handle(Append(basicRootPath, L"*"),
    FindFilesOptions::RecursiveSearch | FindFilesOptions::IncludeDotDirectories);

是否可以使用C ++ 11 enum class以强类型方式支持此功能,还是必须恢复为无类型枚举?

(我知道调用者可以static_cast返回基础类型而static_cast返回,但我不希望调用者必须这样做)

5 个答案:

答案 0 :(得分:12)

当然可以将enum class es用于位图。不幸的是,这样做有点痛苦:您需要在类型上定义必要的位操作。下面是一个如何看起来的例子。如果enum class es可以从某些其他类型派生出来,这可能存在于定义必要的运算符样板代码的合适名称空间中,那将是很好的。

#include <iostream>
#include <type_traits>

enum class bitmap: unsigned char
{
    a = 0x01,
    b = 0x02,
    c = 0x04
};

bitmap operator& (bitmap x, bitmap y)
{
    typedef std::underlying_type<bitmap>::type uchar;
    return bitmap(uchar(x) & uchar(y));
}

bitmap operator| (bitmap x, bitmap y)
{
    typedef std::underlying_type<bitmap>::type uchar;
    return bitmap(uchar(x) | uchar(y));
}

bitmap operator^ (bitmap x, bitmap y)
{
    typedef std::underlying_type<bitmap>::type uchar;
    return bitmap(uchar(x) ^ uchar(y));
}

bool test(bitmap x)
{
    return std::underlying_type<bitmap>::type(x);
}

int main()
{
    bitmap v = bitmap::a | bitmap::b;
    if (test(v & bitmap::a)) {
        std::cout << "a ";
    }
    if (test(v & bitmap::b)) {
        std::cout << "b ";
    }
    if (test(v & bitmap::c)) {
        std::cout << "c ";
    }
    std::cout << '\n';
}

答案 1 :(得分:3)

模板可以与enum class一起使用,因此您可以定义处理类似枚举类型集的运算符集。关键是使用特征模板来指定每个枚举符合/订阅的接口。

首先:

enum class mood_flag {
    jumpy,
    happy,
    upset,
    count // size of enumeration
};

template<>
struct enum_traits< mood_flag > {
    static constexpr bool bit_index = true;
};

template< typename t >
struct flag_bits : std::bitset< static_cast< int >( t::count ) > {
    flag_bits( t bit ) // implicit
        { this->set( static_cast< int >( bit ) ); }

    // Should be explicit but I'm lazy to type:
    flag_bits( typename flag_bits::bitset set )
        : flag_bits::bitset( set ) {}
};

template< typename e >
typename std::enable_if< enum_traits< e >::bit_index,
    flag_bits< e > >::type
operator | ( flag_bits< e > set, e next )
    { return set | flag_bits< e >( next ); }

template< typename e >
typename std::enable_if< enum_traits< e >::bit_index,
    flag_bits< e > >::type
operator | ( e first, e next )
    { return flag_bits< e >( first ) | next; }

http://ideone.com/kJ271Z

GCC 4.9报告说,在我编译时,一些隐式成员函数是constexpr,因此模板应该也是如此。

这可能还应该有一个自由函数to_scalar或者在给定单个标志或flag_bits集的情况下返回无符号整数类型的东西。

答案 2 :(得分:1)

如何定义FindFiles以便std::initializer_list FindFilesOptions {。}}。

void FindFiles(std::wstring const& pattern, std::initializer_list<FindFilesOptions> options)
{
  auto has_option = [&](FindFilesOptions const option)
  {
    return std::find(std::begin(options), std::end(options), option) != std::end(options);
  };

  if (has_option(FindFilesOptions::LocalSearch))
  {
    // ...
  }

  if (has_option(FindFilesOptions::RecursiveSearch))
  {
    // ...
  }

  if (has_option(FindFilesOptions::IncludeDotDirectories))
  {
    // ...
  }
}

然后你可以这样称呼它:

FindFiles({}, {FindFilesOptions::RecursiveSearch, FindFilesOptions::IncludeDotDirectories});

答案 3 :(得分:0)

问题不是显式枚举类型,而是类范围。

使用C ++ 11,当需要对值进行操作(按位运算,递增等)时,作为编译时常量的枚举与一堆constexpr相比松散了很多兴趣

答案 4 :(得分:0)

如果您不关心效果,请将选项更改为set<FindFilesOptions>