带范围枚举的QFlags的相等运算符

时间:2019-06-19 13:39:55

标签: qt c++11 enums flags

我使用范围限定的枚举定义了一些标志,如下所示:

enum class Capability : int
{
  NoCapabilities  = 0,
  SomethingCool = 1,
  UberCool = 1 << 1,
  EvenCooler = 1 << 2,
};
Q_DECLARE_FLAGS( Capabilities, Capability )

现在,我正在尝试使用相等运算符:

Capabilities(DataCapability::NoCapabilities) == Capability::NoCapabilities

在当前示例中,由于testFlag,我无法使用NoCapabilities = 0。如果枚举没有作用域(删除class关键字),则此方法有效。

显然,强制转换为int(static_cast<int>(NoCapabilities))是有效的,但这确实很难看。

解决此问题的方法是什么? 设置0零值并对其进行测试是一种不好的做法吗?

1 个答案:

答案 0 :(得分:3)

您可以为数据类型组合重载等于运算符。但是强制转换仍然是必需的,但被操作员隐藏。

#include <QCoreApplication>
#include <QFlags>
#include <iostream>

enum class Capability : int
{
    NoCapabilities = 0,
    SomethingCool  = 1,
    UberCool       = (1 << 1),
    EvenCooler     = (1 << 2),
};
Q_DECLARE_FLAGS(Capabilities, Capability)

template <typename T>
bool operator==(const QFlags<T> lhs, const Capability rhs)
{
    return (QFlags<T>::Int(lhs) == static_cast<int>(rhs));
}

int main(int argc, char *argv[])
{
    Q_UNUSED(argc);
    Q_UNUSED(argv);

    if (Capabilities(Capability::NoCapabilities) == Capability::NoCapabilities)
    {
        std::cout << "true" << std::endl;
    }
    else
    {
        std::cout << "false" << std::endl;
    }

    if (Capabilities(Capability::EvenCooler) == Capability::NoCapabilities)
    {
        std::cout << "true" << std::endl;
    }
    else
    {
        std::cout << "false" << std::endl;
    }

    return 0;
}