続:Rubyで内包表現
配列リテラル([1,2,3]のような配列のことをこう呼ぶらしい)のinitialize上書きは無理っぽいので諦めて、コードを整理してみました(ついでにバグ取りも)。
用語はこちらを参考にさせて頂きました。
書式:
I['value_expression' | 'var <- source', 'predicate_1', 'predicate_2', ・・・, 'predicate_n']
この式は以下のような式に展開されます。
source.select do |var|
predicate_1(var) && predicate_2(var) && ・・・ && predicate_n(var)
end.map do |var|
value_expression(var)
end
例1:
xs = [10,30,3,-5,1,10,100,-60]
p I['(x*2).abs' | 'x<-xs', 'x < 50']
=> [20, 60, 6, 10, 2, 20, 120]
例2:
def quicksort(x = nil, *xs)
return [] if x == nil
quicksort(*I['y' | 'y <- xs', 'y < x']) + [x] + quicksort(*I['y' | 'y <- xs', 'y >= x'])
end
p quicksort(*xs)
=> [-60, -5, 1, 3, 10, 10, 30, 100]
コードはこちら:
require 'active_support/binding_of_caller'class Intention
def initialize(value_expression, var, source)
@value_expression = value_expression
@var = var
@source = source
end
def to_a(predicates, binding)
predicates << 'true' if predicates == []
eval(<<-EOF, binding)
#@source.select {|#@var| #{predicates.join('&&')} }.map do |#@var|
#@value_expression
end
EOF
end
endclass String
def |(other)
raise unless other =~ /^(.+)<-(.+)$/
Intention.new(self, $1, $2)
end
endclass I
def self.[](intention, *predicates)
raise unless intention.instance_of?(Intention)
Binding.of_caller do |binding|
intention.to_a(predicates, binding)
end
end
end