Rakefile test/peach_test.rb test/test_helper.rb @@ -1,32 +1,37 @@ -class Array +module Peach def peach(n = nil, &b) - peachrun(:each, b, n) + peach_run(:each, b, n) end + def pmap(n = nil, &b) - peachrun(:map, b, n) + peach_run(:map, b, n) end - def pdelete_if(n = nil, &b) - peachrun(:delete_if, b, n) + + def pselect(n = nil, &b) + peach_run(:select, b, n) end + + protected - def peachrun(meth, b, n = nil) + def peach_run(meth, b, n = nil) threads, results, result = [],[],[] - divvy(n).each_with_index do |x,i| - if x.size > 0 - threads << Thread.new { results[i] = x.send(meth, &b)} - else - results[i] = [] - end + peach_divvy(n).each_with_index do |x,i| + threads << Thread.new { results[i] = x.send(meth, &b)} end threads.each {|t| t.join } results.each {|x| result += x if x} result end - def divvy(n = nil) + def peach_divvy(n = nil) + return [] if size == 0 + n ||= $peach_default_threads || size + n = size if n > size + lists = [] + div = (size/n).floor offset = 0 for i in (0...n-1) @@ -37,3 +42,5 @@ class Array lists end end + +Array.send(:include, Peach) lib/peach.rb 0083989fc951b3fbccd6cc50c19199089ea1a64d Ben Hughes ben@pixelmachine.org http://github.com/schleyfox/peach/commit/7b43f10a4c37be9e8dbc108250f21cc0eeb17998 7b43f10a4c37be9e8dbc108250f21cc0eeb17998 2009-04-02T23:32:34-07:00 2009-04-02T23:32:34-07:00 Added tests, fixed a few potential bugs, cleaned up code dc7806ed501a9996077eca9843d0481efd8aec2a Ben Hughes ben@pixelmachine.org