对不同的特性专门化同一个操作符

Specialize same operator for different traits

本文关键字：专门化 同一个 操作符      更新时间：2023-10-16

我想按特征进行专门化。

  1. Array Aa = Scalar in_a将使用overload I

  2. Array Aa = Array Bb将使用overload II

在下面的代码中，没有使用overload II。

有人提到T1不能从overload II中推导出来。

如何解决这个问题?

我使用c++ shell用c++ 14编译代码。

#include <iostream>
#include <type_traits>
using namespace std;
class A; // forward declaration.
template <typename T>
struct is_A : false_type {};
template <> struct is_A<A> : true_type {};
template <typename T>
struct is_int : false_type {};
template <> struct is_int<int> : true_type {};
template <> struct is_int<long> : true_type {};
class A{
    public:
        int val;
        void print(void){
            std::cout << val << std::endl;
        }
        template <typename T1>
        enable_if_t<is_int<T1>::value,void>
        operator=(const T1 & input){
            val = 2*input; //Overload I
        }
        template <typename T1>
        enable_if_t<is_A<T1>::value,void>
        operator=(const T1 & Bb){
            val = 5*Bb.val; //Overload II
        }
};
int main(void){
    A Aa;
    A Bb;
    int in_a = 3;
    Aa = in_a; //This uses overload I as intended.
    Bb = Aa; //I want this to use overload II, but
             //actually overload I is used.
             //This leads to an error during compilation.
    Aa.print(); //This should give 6. (3x2)
    Bb.print(); //This should give 30. (6x5)
}

这是您的代码简化和工作的预期:

#include <iostream>
#include <type_traits>
#include<utility>
class A;
template <typename T>
struct is_A : std::false_type {};
template <> struct is_A<A> : std::true_type {};
template <typename T>
struct is_int : std::false_type {};
template <> struct is_int<int> : std::true_type {};
template <> struct is_int<long> : std::true_type {};
class A{
public:
    int val;
    void print(void){
        std::cout << val << std::endl;
    }
    template <typename T1>
    std::enable_if_t<is_int<std::decay_t<T1>>::value, void>
    operator=(T1 && input){
        val = 2*std::forward<T1>(input);
    }
    template <typename T1>
    std::enable_if_t<is_A<std::decay_t<T1>>::value,void>
    operator=(T1 && Bb){
        val = 5*std::forward<T1>(Bb).val;
    }
};
int main(void){
    A Aa;
    A Bb;
    int in_a = 3;
    Aa = in_a;
    Bb = Aa;
    Aa.print(); //This should give 6. (3x2)
    Bb.print(); //This should give 30. (6x5)
}

你的代码应该是

template <typename T>
std::enable_if_t<is_int<T>::value, A&>
operator=(const T& input){
    val = 2 * input; //Overload I
    return *this;
}
template <typename T>
std::enable_if_t<is_A<T>::value, A&>
operator=(T& rhs){
    val = 5 * rhs.val; //Overload II
    return *this;
}

演示

但你的情况更简单

A& operator=(int input){
    val = 2 * input; //Overload I
    return *this;
}
A& operator=(const A& rhs){
    val = 5 * rhs.val; //Overload II
    return *this;
}

您真的需要为您的简单案例使用所有的模板魔法吗?

#include <iostream>
class A;
class A{
public:
    int val;
    void print(void){
        std::cout << val << std::endl;
    }
    void operator =(const A& in){ val = in.val*5; }
    void operator =(int in) { val = in*2; }
};

int main(void){
    A Aa;
    A Bb;
    Aa = 3;
    Bb = Aa;
    Aa.print(); //This should give 6. (3x2)
    Bb.print(); //This should give 30. (6x5)
    return 0;
}
相关文章：