根据一些枚举值在编译时添加额外的方法

Bob*_*ane 5 c++ templates template-specialization

想象一下我有以下课程:

class Extra final
{
public:

    void Method1() const {std::cout << "Method 1" << std::endl;}
    void Method2() const {std::cout << "Method 2" << std::endl;}
};
Run Code Online (Sandbox Code Playgroud)

我为某些类创建了T一个类模板,该模板可以Extra根据不同的场景(在枚举中列出)在编译时启用/禁用方法:

enum class Scenario
{
    None, // No extra.
    One,  // Only Method1 allowed.
    Two,  // Only Method2 allowed.
    Both  // Both Method1 and Method2 allowed.
};
Run Code Online (Sandbox Code Playgroud)

类模板如下所示:

template<typename T, Scenario R>
class ExtraAdder : public T
{
};

template<typename T>
class ExtraAdder<T, Scenario::One> : public T
{
public:

    void Method1Extra() const {m_extra.Method1();}

private:

    Extra m_extra;
};

template<typename T>
class ExtraAdder<T, Scenario::Two> : public T
{
public:

    void Method2Extra() const {m_extra.Method2();}

private:

    Extra m_extra;
};

template<typename T>
class ExtraAdder<T, Scenario::Both> : public T
{
public:

    void Method1Extra() const {m_extra.Method1();}
    void Method2Extra() const {m_extra.Method2();}

private:

    Extra m_extra;
};
Run Code Online (Sandbox Code Playgroud)

它可以如下使用:

class Base
{
public:

    virtual ~Base() = default;
    void BaseMethod() const {std::cout << "Base method" << std::endl;}

    // Some other stuff...
};

int main(int argc, char **argv)
{
    std::cout << "Scenario: None" << std::endl;
    ExtraAdder<Base, Scenario::None> none;
    none.BaseMethod();
    //none.Method1Extra();   // Does not compile.
    //none.Method2Extra();   // Does not compile.

    std::cout << std::endl << "Scenario: One" << std::endl;
    ExtraAdder<Base, Scenario::One> one;
    one.BaseMethod();
    one.Method1Extra();
    //one.Method2Extra();    // Does not compile.

    std::cout << std::endl << "Scenario: Two" << std::endl;
    ExtraAdder<Base, Scenario::Two> two;
    two.BaseMethod();
    //two.Method1Extra();    // Does not compile.
    two.Method2Extra();

    std::cout << std::endl << "Scenario: Both" << std::endl;
    ExtraAdder<Base, Scenario::Both> both;
    both.BaseMethod();
    both.Method1Extra();
    both.Method2Extra();

    return 0;
}
Run Code Online (Sandbox Code Playgroud)

关键是通过更改模板参数Scenario,我只能访问包装m_extra成员的某些方法,这正是我想要的。但是,正如您所看到的,不同的专业化非常冗长(并且有一些重复)。

问题:有没有一种方法可以生成完全相同的模板特化,但以不那么冗长的方式,重复有限或没有重复?

我发现最接近解决这个问题的这个问题(使用std::enable_if),但我无法适应我的情况。

我在 C++17 标准中使用 g++。

flo*_*tan 4

或者,您可以将 放入enable_if模板而不是返回类型声明中。我个人认为这更具可读性,并且它的优点是您可以使用返回类型推导。

template<typename T, Scenario R>
class ExtraAdder : public T {
    Extra m_extra;
public:
    template<
        Scenario S = R, 
        typename = std::enable_if_t<(S == Scenario::One || S == Scenario::Both)>
    >
    void Method1Extra() const {
        m_extra.Method1();
    }

    
    template<
        Scenario S = R, 
        typename = std::enable_if_t<(S == Scenario::Two || S == Scenario::Both)>
    >
    decltype(auto) Method2Extra() const {
       return m_extra.Method2();
    }

};
Run Code Online (Sandbox Code Playgroud)

活生生的例子在这里

使用enable_if成员函数的一个常见陷阱是,它仅在模板参数的替换格式不正确时才有效。因此,我们需要添加额外的模板参数Scenario S = R(或bool其他答案中的 ),以便将替换推迟到使用该方法的点。否则就不会有这样的替换导致硬错误。请参阅此问题进行更深入的讨论。