ruby - 如何将任何方法转换为ruby中的中缀运算符

Question

在某些语言中，例如 Haskell，it is possible使用任何带有两个参数的函数作为中缀运算符。

我觉得这个符号很有趣，并希望在 ruby​​ 中实现同样的功能。

给定一个虚构的方法or_if_familiar我希望能够写出类似 "omg"or_if_familiar "oh!" 而不是 or_if_familiar("omg", "oh!")

如何在 ruby​​ 中创建这样的符号(不修改 ruby​​ 本身)？

Answer 1

聚会有点晚了，但我一直在玩弄它，你可以使用运算符重载来创建中缀运算符，就像在 python 中一样(但需要更多工作)，语法变为 a |op| b ，方法如下:

首先是一个快速而肮脏的复制粘贴来玩中缀:

class Infix def initialize*a,&b;raise'arguments size mismatch'if a.length<0||a.length>3;raise'both method and b passed'if a.length!=0&&b;raise'no arguments passed'if a.length==0&&!b;@m=a.length>0? a[0].class==Symbol ? method(a[0]):a[0]:b;if a.length==3;@c=a[1];@s=a[2]end end;def|o;if@c;o.class==Infix ? self:@m.(@s,o)else;raise'missing first operand'end end;def coerce o;[Infix.new(@m,true,o),self]end;def v o;Infix.new(@m,true,o)end end;[NilClass,FalseClass,TrueClass,Object,Array].each{|c|c.prepend Module.new{def|o;o.class==Infix ? o.v(self):super end}};def Infix*a,&b;Infix.new *a,&b end
#

好的

第 1 步:创建 Infix类

class Infix
  def initialize *args, &block
    raise 'error: arguments size mismatch' if args.length < 0 or args.length > 3
    raise 'error: both method and block passed' if args.length != 0 and block
    raise 'error: no arguments passed' if args.length == 0 and not block
    @method = args.length > 0 ? args[0].class == Symbol ? method(args[0]) : args[0] : block
    if args.length == 3; @coerced = args[1]; @stored_operand = args[2] end
  end
  def | other
    if @coerced
      other.class == Infix ? self : @method.call(@stored_operand, other)
    else
      raise 'error: missing first operand'
    end
  end
  def coerce other
    [Infix.new(@method, true, other), self]
  end
  def convert other
    Infix.new(@method, true, other)
  end
end

第 2 步:修复所有没有 | 的类方法和三个特殊情况(true、false 和 nil)(注意:您可以在此处添加任何类，它可能会正常工作)

[ NilClass, FalseClass, TrueClass,
  Float, Symbol, String, Rational,
  Complex, Hash, Array, Range, Regexp
].each {|c| c.prepend Module.new {
  def | other
    other.class == Infix ? other.convert(self) : super
  end}}

第 3 步:以 5 种方式之一定义您的运算符

# Lambda
pow = Infix.new -> (x, y) {x ** y}
# Block
mod = Infix.new {|x, y| x % y}
# Proc
avg = Infix.new Proc.new {|x, y| (x + y) / 2.0}
# Defining a method on the spot (the method stays)
pick = Infix.new def pick_method x, y
  [x, y][rand 2]
end
# Based on an existing method
def diff_method x, y
  (x - y).abs
end
diff = Infix.new :diff_method

第 4 步:使用它们(间距无关紧要):

2 |pow| 3      # => 8
9|mod|4        # => 1
3| avg |6      # => 4.5
0 | pick | 1   # => 0 or 1 (randomly chosen)

你甚至可以有点 curry :(这只适用于第一个操作数)

diff_from_3 = 3 |diff

diff_from_3| 2    # => 1
diff_from_3| 4    # => 1
diff_from_3| -3   # => 6

作为奖励，这个小方法允许您在不使用 .new 的情况下定义中缀(或任何对象)。 :

def Infix *args, &block
  Infix.new *args, &block
end

pow = Infix -> (x, y) {x ** y} # and so on

剩下要做的就是将它包装在一个模块中

希望对您有所帮助

附言你可以和运营商一起搞事情，得到类似 a <<op>> b 的东西, a -op- b , a >op> b和 a <op<b对于方向性，a **op** b用于优先级和您想要的任何其他组合，但在使用 true 时要小心, false和 nil作为带有逻辑运算符( | 、 && 、 not 等)的第一个操作数，因为它们倾向于在调用中缀运算符之前返回。

例如:false |equivalent_of_or| 5 # => true如果您不更正。

最后，运行它来检查所有内置类的一堆情况作为第一个和第二个操作数:

# pp prints both inputs
pp = Infix -> (x, y) {"x: #{x}\ny: #{y}\n\n"}

[ true, false, nil, 0, 3, -5, 1.5, -3.7, :e, :'3%4s', 'to',
  /no/, /(?: [^A-g7-9]\s)(\w{2,3})*?/,
  Rational(3), Rational(-9.5), Complex(1), Complex(0.2, -4.6),
  {}, {e: 4, :u => 'h', 12 => [2, 3]},
  [], [5, 't', :o, 2.2, -Rational(3)], (1..2), (7...9)
].each {|i| puts i.class; puts i |pp| i}