Вопрос о перегрузке оператора 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, не заменяя его полностью. возможно ли это в рубине? любые советы?

(очевидно, я могу просто написать все мои умножения в правильном порядке, если мне нужно)

Ответ 1

Использование coerce - это намного лучший подход, чем обезвреживание базового класса:

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), становится новым приемником для операции, а второй элемент (другой) становится параметром, поэтому 5 * v в точности эквивалентен v * 5

Ответ 2

Я полагаю, что следующее будет делать то, что вы хотите, но предложение banister использовать coerce вместо обезьян-патчей Numeric является предпочтительным методом. Используйте этот метод только в случае необходимости (например, если вы хотите, чтобы некоторые двоичные операнды были транзитивными).

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