enum_type.hpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD 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, version 2.
 * OpenTTD 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 OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef ENUM_TYPE_HPP
#define ENUM_TYPE_HPP
 
template <typename enum_type>
constexpr std::underlying_type_t<enum_type> to_underlying(enum_type e) { return static_cast<std::underlying_type_t<enum_type>>(e); }
 
template <typename enum_type>
struct is_enum_incrementable {
  static constexpr bool value = false;
};
 
template <typename enum_type>
constexpr bool is_enum_incrementable_v = is_enum_incrementable<enum_type>::value;
 
template <typename enum_type, std::enable_if_t<is_enum_incrementable_v<enum_type>, bool> = true>
inline constexpr enum_type &operator ++(enum_type &e)
{
  e = static_cast<enum_type>(to_underlying(e) + 1);
  return e;
}
 
template <typename enum_type, std::enable_if_t<is_enum_incrementable_v<enum_type>, bool> = true>
inline constexpr enum_type operator ++(enum_type &e, int)
{
  enum_type e_org = e;
  ++e;
  return e_org;
}
 
template <typename enum_type, std::enable_if_t<is_enum_incrementable_v<enum_type>, bool> = true>
inline constexpr enum_type &operator --(enum_type &e)
{
  e = static_cast<enum_type>(to_underlying(e) - 1);
  return e;
}
 
template <typename enum_type, std::enable_if_t<is_enum_incrementable_v<enum_type>, bool> = true>
inline constexpr enum_type operator --(enum_type &e, int)
{
  enum_type e_org = e;
  --e;
  return e_org;
}
 
#define DECLARE_INCREMENT_DECREMENT_OPERATORS(enum_type) \
  template <> struct is_enum_incrementable<enum_type> { \
    static const bool value = true; \
  };
 
 
#define DECLARE_ENUM_AS_BIT_SET(enum_type) \
  inline constexpr enum_type operator | (enum_type m1, enum_type m2) { return static_cast<enum_type>(to_underlying(m1) | to_underlying(m2)); } \
  inline constexpr enum_type operator & (enum_type m1, enum_type m2) { return static_cast<enum_type>(to_underlying(m1) & to_underlying(m2)); } \
  inline constexpr enum_type operator ^ (enum_type m1, enum_type m2) { return static_cast<enum_type>(to_underlying(m1) ^ to_underlying(m2)); } \
  inline constexpr enum_type& operator |= (enum_type& m1, enum_type m2) { m1 = m1 | m2; return m1; } \
  inline constexpr enum_type& operator &= (enum_type& m1, enum_type m2) { m1 = m1 & m2; return m1; } \
  inline constexpr enum_type& operator ^= (enum_type& m1, enum_type m2) { m1 = m1 ^ m2; return m1; } \
  inline constexpr enum_type operator ~(enum_type m) { return static_cast<enum_type>(~to_underlying(m)); }
 
#define DECLARE_ENUM_AS_ADDABLE(EnumType) \
  template <typename OtherEnumType, typename = typename std::enable_if<std::is_enum_v<OtherEnumType>, OtherEnumType>::type> \
  constexpr OtherEnumType operator + (OtherEnumType m1, EnumType m2) { \
    return static_cast<OtherEnumType>(to_underlying(m1) + to_underlying(m2)); \
  }
 
template <typename T, class = typename std::enable_if_t<std::is_enum_v<T>>>
debug_inline constexpr bool HasFlag(const T x, const T y)
{
  return (x & y) == y;
}
 
template <typename T, class = typename std::enable_if_t<std::is_enum_v<T>>>
debug_inline constexpr void ToggleFlag(T &x, const T y)
{
  if (HasFlag(x, y)) {
    x &= ~y;
  } else {
    x |= y;
  }
}
 
template <typename Tenum, typename Tstorage>
class EnumBitSet {
public:
  using EnumType = Tenum; 
  using BaseType = Tstorage; 
 
  constexpr EnumBitSet() : data(0) {}
  constexpr EnumBitSet(Tenum value) : data(0) { this->Set(value); }
  explicit constexpr EnumBitSet(Tstorage data) : data(data) {}
 
  constexpr EnumBitSet(std::initializer_list<const Tenum> values) : data(0)
  {
    for (const Tenum &value : values) {
      this->Set(value);
    }
  }
 
  constexpr auto operator <=>(const EnumBitSet &) const noexcept = default;
 
  inline constexpr EnumBitSet &Set(Tenum value)
  {
    this->data |= (1U << to_underlying(value));
    return *this;
  }
 
  inline constexpr EnumBitSet &Reset(Tenum value)
  {
    this->data &= ~(1U << to_underlying(value));
    return *this;
  }
 
  inline constexpr EnumBitSet &Flip(Tenum value)
  {
    if (this->Test(value)) {
      return this->Reset(value);
    } else {
      return this->Set(value);
    }
  }
 
  inline constexpr bool Test(Tenum value) const
  {
    return (this->data & (1U << to_underlying(value))) != 0;
  }
 
  inline constexpr bool All(const EnumBitSet &other) const
  {
    return (this->data & other.data) == other.data;
  }
 
  inline constexpr bool Any(const EnumBitSet &other) const
  {
    return (this->data & other.data) != 0;
  }
 
  inline constexpr EnumBitSet operator |(const EnumBitSet &other) const
  {
    return EnumBitSet{static_cast<Tstorage>(this->data | other.data)};
  }
 
  inline constexpr EnumBitSet operator &(const EnumBitSet &other) const
  {
    return EnumBitSet{static_cast<Tstorage>(this->data & other.data)};
  }
 
  inline constexpr Tstorage base() const noexcept
  {
    return this->data;
  }
 
private:
  Tstorage data; 
};
 
#endif /* ENUM_TYPE_HPP */