如果它是右值,则以下帮助函数返回第一个值,否则返回第二个值(可能是右值,但可能不是)。
template <class T1, class T2>
typename std::enable_if<! std::is_reference<T1>::value, T1&&>::type
get_rvalue(T1&& t1, T2&& t2) { return std::forward<T1>(t1); }
template <class T1, class T2>
typename std::enable_if<std::is_reference<T1>::value, T2&&>::type
get_rvalue(T1&& t1, T2&& t2) { return std::forward<T2>(t2); }
以下帮助函数返回上面未返回的其他值。
template <class T1, class T2>
typename std::enable_if<! std::is_reference<T1>::value, T1&&>::type
get_non_rvalue(T1&& t1, T2&& t2) { return std::forward<T2>(t2); }
template <class T1, class T2>
typename std::enable_if<std::is_reference<T1>::value, T2&&>::type
get_non_rvalue(T1&& t1, T2&& t2) { return std::forward<T1>(t1); }
这只是比较两种类型是否相同,忽略引用和 const。
template <class T1, class T2>
struct is_same_decay : public std::is_same<
typename std::decay<T1>::type,
typename std::decay<T2>::type
> {};
然后我们可以为每个函数(使用模板)只做一个重载,如下所示:
// 2ary ops
template <class M1, class M2>
friend typename std::enable_if<
is_same_decay<M1, Matrix>::value &&
is_same_decay<M2, Matrix>::value,
Matrix>::type
operator+(M1&& a, M2&& b)
{
Matrix x = get_rvalue(std::forward<M1>(a), std::forward<M2>(b));
x += get_non_rvalue(std::forward<M1>(a), std::forward<M2>(b));
return x;
}
template <class M1, class M2>
friend typename std::enable_if<
is_same_decay<M1, Matrix>::value &&
is_same_decay<M2, Matrix>::value,
Matrix>::type
operator*(M1&& a, M2&& b)
{
Matrix x = get_rvalue(std::forward<M1>(a), std::forward<M1>(b));
x *= get_non_rvalue(std::forward<M1>(a), std::forward<M1>(b));
return x;
}
请注意,如果其中一个M1
或M2
是右值,get_rvalue(a, b)
则将返回一个右值,因此在这种情况下Matrix x
将由移动而不是副本填充。命名返回值优化可能会确保不需要复制(甚至移动)到返回值中,因为x
将在返回值的位置构造。
完整代码在这里。