ruby 运算符重载问题

Question

为了娱乐目的，我一直在玩弄 ruby​​ 和 opengl，我决定编写一些 3d 矢量/平面/等类来完善一些数学。

简化示例：

class Vec3
    attr_accessor :x,:y,:z

    def *(a)
        if a.is_a?(Numeric) #multiply by scalar
            return Vec3.new(@x*a, @y*a, @z*a)
        elsif a.is_a?(Vec3) #dot product
            return @x*a.x + @y*a.y + @z*a.z
        end
    end
end

v1 = Vec3.new(1,1,1)
v2 = v1*5 #produces [5,5,5]

这一切都很好，花花公子，但我也希望能够写作

v2 = 5*v1

这需要向 Fixnum 或 Float 或其他任何内容添加功能，但我无法找到一种方法来重载或扩展 fixnum 的乘法而不完全替换它。这可能在红宝石中吗？有什么建议吗？

（显然，如果需要，我可以按正确的顺序编写所有乘法）

Answer 1

使用强制是比猴子修补核心类更好的方法：

class Vec3
    attr_accessor :x,:y,:z

    def *(a)
        if a.is_a?(Numeric) #multiply by scalar
            return Vec3.new(@x*a, @y*a, @z*a)
        elsif a.is_a?(Vec3) #dot product
            return @x*a.x + @y*a.y + @z*a.z
        end
    end

    def coerce(other)
        return self, other
    end
end

如果您将 v 定义为 v = Vec3.new，则以下内容将起作用：v * 5 和 5 * v coerce(self)返回的第一个元素成为操作的新接收者，第二个元素(other)成为参数，所以5 * v完全等价于v * 5

Answer 2

我相信下面会做你想要的，虽然banister's suggestion 使用coerce 而不是猴子补丁Numeric 是首选方法。仅在必要时使用此方法（例如，如果您只希望 some 二进制操作数是可传递的）。

Fixnum.class_eval do
  original_times = instance_method(:*)
  define_method(:*) do |other|
    if other.kind_of?(Vec3)
      return other * self
    else
      return original_times.bind(self).call(other)
    end
  end
end