如何在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!!!
有没有一种简单的方法来处理这种情况?谢谢! :)
答案 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 ); }
嗯,这很容易。