bitmask.hxx

#include <StormByte/type_traits.hxx>
 
using namespace StormByte;
 
template <Type::UnsignedEnum E>
constexpr E operator|(E a, E b) noexcept {
    return static_cast<E>(Type::ToUnderlying(a) | Type::ToUnderlying(b));
}
 
template <Type::UnsignedEnum E>
constexpr E operator&(E a, E b) noexcept {
    return static_cast<E>(Type::ToUnderlying(a) & Type::ToUnderlying(b));
}
 
template <Type::UnsignedEnum E>
constexpr E operator^(E a, E b) noexcept {
    return static_cast<E>(Type::ToUnderlying(a) ^ Type::ToUnderlying(b));
}
 
template <Type::UnsignedEnum E>
constexpr E operator~(E a) noexcept {
    return static_cast<E>(~Type::ToUnderlying(a));
}
 
namespace StormByte {
    template<typename Derived, Type::UnsignedEnum E>
    class Bitmask {
        public:
            constexpr Bitmask() noexcept: m_value(static_cast<E>(0)) {}
 
            constexpr Bitmask(E value) noexcept: m_value(value) {}
 
            constexpr Bitmask(const Bitmask& other) noexcept = default;
 
            constexpr Bitmask(Bitmask&& other) noexcept = default;
 
            constexpr virtual ~Bitmask() noexcept = default;
 
            constexpr Bitmask& operator=(const Bitmask& other) noexcept = default;
 
            constexpr Bitmask& operator=(Bitmask&& other) noexcept = default;
 
            constexpr bool operator==(const Bitmask& other) const noexcept {
                return m_value == other.m_value;
            }
 
            constexpr bool operator!=(const Bitmask& other) const noexcept {
                return m_value != other.m_value;
            }
 
            constexpr Derived operator|(const Bitmask& other) const noexcept {
                return Derived(m_value | other.m_value);
            }
 
            constexpr Derived operator&(const Bitmask& other) const noexcept {
                return Derived(m_value & other.m_value);
            }
 
            constexpr Derived operator^(const Bitmask& other) const noexcept {
                return Derived(m_value ^ other.m_value);
            }
 
            constexpr Derived operator~() const noexcept {
                return Derived(~m_value);
            }
 
            constexpr Bitmask& operator|=(const Bitmask& other) noexcept {
                m_value = m_value | other.m_value;
                return *this;
            }
 
            constexpr Bitmask& operator&=(const Bitmask& other) noexcept {
                m_value = m_value & other.m_value;
                return *this;
            }
 
            constexpr Bitmask& operator^=(const Bitmask& other) noexcept {
                m_value = m_value ^ other.m_value;
                return *this;
            }
 
            constexpr void Add(E value) noexcept {
                m_value = m_value | value;
            }
 
            constexpr void Remove(E value) noexcept {
                m_value = m_value & ~value;
            }
 
            constexpr E Value() const noexcept {
                return m_value;
            }
 
            constexpr bool Has(E value) const noexcept {
                return (m_value & value) == value;
            }
 
            constexpr bool Has(const Bitmask& other) const noexcept {
                return Has(other.m_value);
            }
 
            constexpr bool HasAny(E value) const noexcept {
                return (m_value & value) != static_cast<E>(0);
            }
 
            constexpr bool HasAny(const Bitmask& other) const noexcept {
                return HasAny(other.m_value);
            }
 
            constexpr bool HasNone(E value) const noexcept {
                return (m_value & value) == static_cast<E>(0);
            }
 
            constexpr bool HasNone(const Bitmask& other) const noexcept {
                return HasNone(other.m_value);
            }
 
        protected:
            E m_value;
    };
}