Go(语言)通用数字类型/接口

Question

我正在尝试在Go中编写一个使用"泛型"类型计算方程的包.具体来说,我想实现runge kutta 5近似.

这种近似计算(未知的)函数的值y在点t0 + h使用的唯一的值y在t0,开始时间t0,步宽h和微分方程dgl其为形式的dy/dt = g(t,y),其中g一些功能.

在使用标量类型时,与使用向量(甚至矩阵)时,此近似行为完全相同.更一般地说:它适用于可以添加/减少到相同类型的值的所有内容,并且可以通过标量缩放(我使用它float64)

所以我试着把它表达为Go接口:

type Numeric interface {
    Add(rhs Numeric) Numeric
    Sub(rhs Numeric) Numeric
    Mul(rhs float64) Numeric
}

但是当我尝试"实现"这个接口时,由于参数类型我遇到了麻烦:

type Vec6F struct {
    x, y, z float64
    vx, vy, vz float64
}

func (lhs *Vec6F) Add(rhs *Vec6F) rk5.Numeric {
    result := new(Vec6F)
    result.x = lhs.x + rhs.x
    result.y = lhs.y + rhs.y
    result.z = lhs.z + rhs.z
    result.vx = lhs.vx + rhs.vx
    result.vy = lhs.vy + rhs.vy
    result.vz = lhs.vz + rhs.vz
    return result
}

这给了我错误

cannot use result (type *Vec6F) as type rk5.Numeric in return argument:
        *Vec6F does not implement rk5.Numeric (wrong type for Add method
                have Add(*Vec6F) rk5.Numeric
                want Add(rk5.Numeric) rk5.Numeric

这对我来说绝对是逻辑(因为rhs可能是实现数字的另一个对象)

但另一方面:我如何在Go中表达类似的内容？在C++中,我可以使用运算符重载,但这不可能.

Answer 1

为了通用，您的Add方法必须带有一个Numeric参数。处理这个问题的正常方法是使用像这样的类型断言（在操场上）

func (lhs *Vec6F) Add(_rhs Numeric) Numeric {
    result := new(Vec6F)
    rhs := _rhs.(*Vec6F) // type assertion - will panic if wrong type passes
    result.x = lhs.x + rhs.x
    result.y = lhs.y + rhs.y
    result.z = lhs.z + rhs.z
    result.vx = lhs.vx + rhs.vx
    result.vy = lhs.vy + rhs.vy
    result.vz = lhs.vz + rhs.vz
    return result
}

如果您想要在不同的类型之间进行转换，您也可以使用类型开关。