Разрешить для Range-Based For с классами enum?

Ответ 1

Итерация перечислений с самим перечислением как итератором является плохой идеей, и я рекомендую использовать фактический итератор, как в ответе deft_code. Но если это действительно то, что вы хотите:

COLOR operator++(COLOR& x) { return x = (COLOR)(std::underlying_type<COLOR>::type(x) + 1); }
COLOR operator*(COLOR c) {return c;} 
COLOR begin(COLOR r) {return COLOR::First;}
COLOR end(COLOR r)   {COLOR l=COLOR::Last; return l++;}

int main() { 
    for(const auto& c : COLOR()) { //note I added parenthesis here to make an instance
        //do work
    }
    return 0;
}

Работа здесь: http://ideone.com/cyTGD8

extern const COLOR COLORS[(int)COLOR::Last+1];
const COLOR COLORS[] = {COLOR::Blue, COLOR::Red, COLOR::Green, COLOR::Purple};

int main() { 
    for(const auto& c : COLOR()) { //note I added parenthesis here to make an instance
        //do work
    }
    return 0;
}

Как показано здесь: http://ideone.com/9XadVt

(Отдельное объявление и дефиниция массива делает его ошибкой компилятора, если количество цветов не соответствует количеству элементов в массиве. Отличная простая проверка безопасности.)

Ответ 2

Мне лично не нравится перегружать оператор ++ для перечислений. Часто увеличение значения перечисления на самом деле не имеет смысла. Все, что действительно нужно, - это путь к итератору над перечислением.

Ниже приведен общий Enum класс, поддерживающий итерацию. Он функциональный, но неполный. Реальная реализация имела бы смысл ограничить доступ к конструктору и добавить все признаки итератора.

#include <iostream>

template< typename T >
class Enum
{
public:
   class Iterator
   {
   public:
      Iterator( int value ) :
         m_value( value )
      { }

      T operator*( void ) const
      {
         return (T)m_value;
      }

      void operator++( void )
      {
         ++m_value;
      }

      bool operator!=( Iterator rhs )
      {
         return m_value != rhs.m_value;
      }

   private:
      int m_value;
   };

};

template< typename T >
typename Enum<T>::Iterator begin( Enum<T> )
{
   return typename Enum<T>::Iterator( (int)T::First );
}

template< typename T >
typename Enum<T>::Iterator end( Enum<T> )
{
   return typename Enum<T>::Iterator( ((int)T::Last) + 1 );
}

enum class Color
{
   Red,
   Green,
   Blue,
   First = Red,
   Last = Blue
};

int main()
{
   for( auto e: Enum<Color>() )
   {
      std::cout << ((int)e) << std::endl;
   }
}

Ответ 3

enum class Color {
    blue,
    red,
    green = 5,
    purple
};
const std::array<Color,4> all_colors = {Color::blue, Color::red, Color::green, Color::purple};

Тогда:

for (Color c : all_colors) {
    //...
}

Много раз я использую его так, где я хочу значение "none":

// Color of a piece on a chess board
enum class Color {
    white,
    black,
    none
};
const std::array<Color,3> colors = {Color::white, Color::black};

template <typename CONTAINER>
bool has_item (CONTAINER const & c, typename CONTAINER::const_reference v) {
    return std::find(c.begin(), c.end(), v) != c.end();
}

bool is_valid (Color c) {
    return has_item(colors, c) || c == Color::none;
}

bool do_it (Color c) {
    assert(has_item(colors, c)); // here I want a real color, not none
    // ...
}

bool stop_it (Color c) {
    assert(is_valid(c));         // but here I just want something valid
    // ...
}

Ответ 4

Возможно, вы могли бы сделать что-то умное с boost:: mpl, грубая версия может выглядеть так:

#include <typeinfo>

// ---------------------------------------------------------------------------|
// Boost MPL
// ---------------------------------------------------------------------------|
#include <boost/mpl/for_each.hpp>
#include <boost/mpl/iterator_range.hpp>
#include <boost/mpl/range_c.hpp>

namespace mpl = boost::mpl;

using namespace std;

enum class COLOR 
{ 
   Blue,
   Red,
   Green,
   Purple,
   Last
};

struct enumValPrinter
{
    template< typename T >
    void operator() (const T&)
    {
        cout << "enumValPrinter with: " << typeid( T ).name() << " : " 
             << T::value << "\n";
    }
};

int main(int, char**)
{
    typedef mpl::range_c< int, static_cast<int>( COLOR::Blue ), 
                            static_cast<int>( COLOR::Last ) > Colors;
    mpl::for_each< Colors >( enumValPrinter() );
    return 0;
}

Ответ 5

Здесь приведен пример (GCC 4.6.1):

enum class COLOR
{
    Blue,
    Red,
    Green,
    Purple,
    First=Blue,
    Last=Purple
};

COLOR operator++( COLOR& x ) { return x = (COLOR)(((int)(x) + 1)); }

COLOR operator*(COLOR c) {return c;}

COLOR begin(COLOR r) {return COLOR::First;}
// end iterator needs to return one past the end!
COLOR end(COLOR r)   {return COLOR(int(COLOR::Last) + 1);}


int main()
{
    for (const auto& color : COLOR()) std::cout << int(color); //0123
    return 0;
}

Ответ 6

Мне очень нравится идея и часто ее желаю.

Проблема, которую я вижу, это то, что происходит, когда есть повторяющееся числовое значение для элемента перечисления. Все реализации, которые я вижу выше, требуют отбрасывания к интегральному типу и ++. В конечном счете, я думаю, что языковая поддержка может потребоваться для поистине перебора каждого элемента во всех случаях. Это устранит необходимость иметь First, Last или Begin, End, хотя я не возражаю против этого слишком много. Это похоже на поиск begin() end() для контейнеров.

enum class COLOR 
{
   Blue,
   Red,
   Green,
   Mauve = 0,
   Purple,
   Last
};

Нумерация начинается с Mauve.

Ответ 7

Если вы ужасный человек, вы можете получить это поведение с препроцессором, что-то вроде:

#include <vector>
#include <cstdio>

#define ENUM_NAME COLOR
#define ENUM_VALUES \
    ENUM_VALUE(Blue) \
    ENUM_VALUE(Red) \
    ENUM_VALUE(Green) \
    ENUM_VALUE(Purple)

// This block would be a #include "make_iterable_enum.h"
#define ENUM_VALUE(v) v,
enum class ENUM_NAME {ENUM_VALUES};
#undef ENUM_VALUE
#define ENUM_VALUE(v) ENUM_NAME::v,
#define VECTOR_NAME(v) values_ ## v
#define EXPAND_TO_VECTOR_NAME(v) VECTOR_NAME(v)
const std::vector<ENUM_NAME> EXPAND_TO_VECTOR_NAME(ENUM_NAME){ENUM_VALUES};
#undef ENUM_VALUE
#undef ENUM_NAME
#undef ENUM_VALUES
#undef VECTOR_NAME
#undef EXPAND_TO_VECTOR_NAME
// end #included block

int main() {
    for (auto v : COLOR_values) {
        printf("%d\n", (int)v);
    }
}

С небольшими изменениями это может также поддерживать, например. ENUM_SETVALUE (синий, 4) и создание карты констант, например. ЦВЕТ:: Синий - синий. И наоборот.

Я бы хотел, чтобы стандарт только что построил эти функции в качестве опций для перечисления класса. Ни один из обходных решений не является хорошим.

Ответ 8

Я уверен, что вы можете перебирать элементы списка инициализаторов С++, поэтому я считаю, что я сделал это в прошлом:

enum class Color {Red, Green, Blue};

for (const Color c : {Color::Red, Color::Green, Color::Blue})
{
}

Есть ли проблемы с этим, я не знаю, но я думал, что предлагаю его, поскольку он краток, но не идеален, если есть много цветов.