有条件地重载运算符

Conditionally overload an operator

我目前正在努力实现一些数学基础操作,并尽量避免使用第三方库。我一直在为 Scalar*VectorVector*Scalar 的乘法重载 operator*Scalar*Vector的点积的当前代码:

#include <vector>
#include <type_traits>

template<class Vector, class Scalar>
typename std::enable_if<std::is_floating_point<Scalar>::value, Vector>::type operator*
(
    const Scalar &a,
    const Vector &b
)
{
    return Vector
            (
                a*b[0],
                a*b[1],
                a*b[2]
            );
}

int main()
{
    const std::vector<double> v1({1,2,3});
    const double s1(2);
    const auto result(s1*v1);
    std::cout<< result << std::endl;
}

编译器错误信息为:

error: invalid operands to binary expression ('double' and 'const std::vector')

任何关于如何重载 * 运算符的建议,以便两个点积都是可能的?我不打算在自定义向量 class 中将这两个运算符实现为重载运算符。而不是那样,我的目标是模板化运算符。

我找到了很好的解释 here 并用它来调整矢量运算。在向量 class 头文件中,您基本上通过 struct is_vector 定义任何类型 T 默认情况下都不是向量。在下文中,所有 可以 充当向量的类型都必须明确列出,例如 std::vector.

#include <vector>
#include <type_traits>

template <typename T>
struct is_vector
{
    static const bool value = false;
};

template <>
struct is_vector< std::vector >
{
    static const bool value = true;
};

template <class Vector>
typename std::enable_if<std::is_vector<Vector>::value, double>::type
operator*(const Vector &a, const Vector &b);

在可执行文件中,代码看起来是一样的。

int main()
{
    const std::vector<double> v1({1,2,3});
    const double s1(2);
    const auto result(s1*v1);
    std::cout<< result << std::endl;
}