混合模板化参数类型

时间:2016-03-10 00:55:58

标签: c++ templates operator-overloading std generic-programming

如何在C ++模板/泛型编程和重载运算符中处理混合数据类型?

例如,假设我们正在创建一个带有x和y参数的二维坐标类,我们想要添加它们:

template <class T>
class Cartesian {
public:
    Cartesian();
    Cartesian(T _x, T _y);
    Cartesian<T> operator + (const Cartesian<T> & rhs);
    // setters, getters
private:
    T x, y;
};

+运算符重载以添加两个坐标:

template <class T>
Cartesian<T> Cartesian<T>::operator + (const Cartesian<T> & rhs) {return Cartesian(x+rhs.x,y+rhs.y);}

现在,我们实例化四个点:两个点int系数;其他两个float

int main() {
    Cartesian<int> i1(1,2), i2(3,4);
    Cartesian<float> f1(5.7, 2.3), f2(9.8, 7.43);

添加两个整数是没有问题的,就像添加两个浮点数一样。但是如果我们想在浮点数中添加一个int呢?即使在四年级的教室,这也不会出现问题,但在这里......

(i1 + i2); // ok
(f1 + f2); // ok
(i1 + f2); // uh oh!!!

有没有一种简单的方法来处理这种情况?谢谢! :)

2 个答案:

答案 0 :(得分:4)

您可以使用免费operator+重载。

template <class T, class U>
typename std::enable_if<std::is_arithmetic<T>::value && 
  std::is_arithmetic<U>::value, Cartesian<std::common_type_t<T, U>>>::type 
operator + (Cartesian<T> const & lhs, Cartesian<U> const & rhs) 
{
    return Cartesian<std::common_type_t<T, U>>(lhs.x+rhs.x,lhs.y+rhs.y);
}

如果预先C ++ 14,请将std::common_type_t<T, U>替换为typename std::common_type<T,U>::type

现在您可以执行上述操作:

Cartesian<int> i1(1, 2), i2(3, 4);
Cartesian<float> f1(5.7, 2.3), f2(9.8, 7.43);
auto a = i1 + i2; // a === Cartesian<int>
auto b = f1 + f2; // b === Cartesian<float>
auto c = i1 + f2; // c === Cartesian<float>

答案 1 :(得分:3)

我只想定义一个提供参数类型推导的工厂函数,

template< class T >
auto cartesian( T const x, T const y )
    -> Cartesian<T>
{ return {x, y}; }

然后是一个独立的operator+喜欢

template< class U, class V >
auto operator+( Cartesian<U> const& a, Cartesian<V> const& b )
{ return cartesian( a.x + b.x, a.y + b.y ); }
嗯,这很容易。