enum_vector.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2017 CERN
 * @author Maciej Suminski <[email protected]>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 3
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef ENUM_VECTOR_H
#define ENUM_VECTOR_H
 
#include <iterator>
#include <type_traits>
 
#define DEFINE_ENUM_VECTOR( enumName, ... ) \
    enum enumName __VA_ARGS__; \
    static constexpr enumName enumName##_vector[] = __VA_ARGS__;
 
#define DECLARE_ENUM_VECTOR( className, enumName ) \
    constexpr className::enumName className::enumName##_vector[];
 
 
#define DEFINE_ENUM_CLASS_WITH_ITERATOR( enumName, beginVal, ... ) \
    enum class enumName : int { beginVal, __VA_ARGS__, _ENUM_END }; \
    typedef ENUM_ITERATOR<enumName, enumName::beginVal, enumName::_ENUM_END> enumName##_ITERATOR;
 
template <typename T, T beginVal, T endVal>
class ENUM_ITERATOR
{
    typedef typename std::underlying_type<T>::type val_t;
    int val;
 
public:
    using value_type = T;
    using difference_type = std::ptrdiff_t;
    using pointer = T*;
    using reference = T&;
    using iterator_category = std::input_iterator_tag;
 
    ENUM_ITERATOR( const T& f ) : val( static_cast<val_t>( f ) ) {}
    ENUM_ITERATOR() : val( static_cast<val_t>( beginVal ) ) {}
    ENUM_ITERATOR operator++()
    {
        val++;
        return *this;
    }
 
    T operator*() { return static_cast<T>( val ); }
    ENUM_ITERATOR begin() { return *this; }
    ENUM_ITERATOR end() { return ENUM_ITERATOR( endVal ); }
    bool operator==( const ENUM_ITERATOR& aIt ) const { return val == aIt.val; }
    bool operator!=( const ENUM_ITERATOR& aIt ) const { return val != aIt.val; }
};
 
#endif /* ENUM_VECTOR_H */