lib/open_classes/array.rb in tbpgr_utils-0.0.15 vs lib/open_classes/array.rb in tbpgr_utils-0.0.16

@@ -1,7 +1,8 @@
 # encoding: utf-8
 require 'open_classes/object'
+require 'open_classes/module'
 
 # Array
 class Array
   # Arrays loop together.
   #
@@ -54,19 +55,42 @@
     if_not_contain_array_rails_type_error
     ret = []
     first.each_with_index do |i_v, i|
       eval_each_str = get_args_str_for_together i
       each_ret = instance_eval "yield(#{eval_each_str})"
-      ret = set_together_each_return(ret, each_ret, i)
+      ret = set_together_each_return_map(ret, each_ret, i)
     end
     ret
   end
 
-  alias_method :together_collect, :together_map
-  alias_method :tmap, :together_map
-  alias_method :tcollect, :together_map
+  # Arrays loop together select.
+  #
+  # if you want to single Array return
+  #   firsts = [1, 2, 3, 4]
+  #   seconds =  [4, 2, 3, 1]
+  #   ret = [firsts, seconds].together_select{|first, second|first == second}
+  #   print ret # => output  [[2, 3], [2, 3]]
+  #
+  # if you want to multi Array return
+  #   firsts = [1, 2, 3, 4]
+  #   seconds =  [4, 2, 3, 1]
+  #   ret = [firsts, seconds].together_select{|first, second|[first.odd?, second.even?}
+  #   print ret # => output  [[1, 3], [4, 2]]
+  def together_select
+    if_not_contain_array_rails_type_error
+    ret = []
+    first.each_with_index do |i_v, i|
+      eval_each_str = get_args_str_for_together i
+      each_ret = instance_eval "yield(#{eval_each_str})"
+      ret = set_together_each_return_select(ret, each_ret, i)
+    end
+    ret
+  end
 
+  alias_methods [:together_collect, :tmap, :tcollect], :together_map
+  alias_methods [:together_find_all, :tselect, :tfindall], :together_select
+
   private
 
   def if_not_contain_array_rails_type_error
     each { |f|fail TypeError, "you have to use [Array1, Array2, ...] | #{f.class} is invalid" unless f.class == Array }
   end
@@ -80,16 +104,27 @@
   def get_args_str_for_together(i, with_index = false)
     each_eval = with_index ? get_args_for_together(i) << i : get_args_for_together(i)
     each_eval.join(',')
   end
 
-  def set_together_each_return(ret, each_ret, index)
+  def set_together_each_return_map(ret, each_ret, index)
     if together_return_multi?(each_ret)
       size.times { |i|ret << [] } if index == 0
       (0..(size - 1)).each { |i|ret[i] << each_ret[i] }
     else
       ret << each_ret
     end
+    ret
+  end
+
+  def set_together_each_return_select(ret, each_ret, index)
+    unless together_return_multi?(each_ret)
+      tmp_each_ret = []
+      size.times { tmp_each_ret << each_ret }
+      each_ret = tmp_each_ret
+    end
+    size.times { |i|ret << [] } if index == 0
+    (0..(size - 1)).each { |i|ret[i] << self[i][index] if each_ret[i] }
     ret
   end
 
   def together_return_multi?(list)
     (list.class == Array && list.size == size).to_bool