jruby · Mar 9, 2015 · Mar 9, 2015
Showing with 43 additions and 13 deletions.

+0 −13 spec/truffle/tags/core/array/bsearch_tags.txt

+43 −0 truffle/src/main/ruby/core/rubinius/common/array.rb
diff --git a/spec/truffle/tags/core/array/bsearch_tags.txt b/spec/truffle/tags/core/array/bsearch_tags.txt
diff --git a/truffle/src/main/ruby/core/rubinius/common/array.rb b/truffle/src/main/ruby/core/rubinius/common/array.rb
@@ -348,6 +348,49 @@ def recursively_flatten(array, out, max_levels = -1)
 
   private :recursively_flatten
 
+  def bsearch
+    return to_enum :bsearch unless block_given?
+
+    m = Rubinius::Mirror::Array.reflect self
+
+    tuple = m.tuple
+
+    min = start = m.start
+    max = total = start + m.total
+
+    last_true = nil
+    i = start + m.total / 2
+
+    while max >= min and i >= start and i < total
+      x = yield tuple.at(i)
+
+      return tuple.at(i) if x == 0
+
+      case x
+        when Numeric
+          if x > 0
+            min = i + 1
+          else
+            max = i - 1
+          end
+        when true
+          last_true = i
+          max = i - 1
+        when false, nil
+          min = i + 1
+        else
+          raise TypeError, "Array#bsearch block must return Numeric or boolean"
+      end
+
+      i = min + (max - min) / 2
+    end
+
+    return tuple.at(i) if max > min
+    return tuple.at(last_true) if last_true
+
+    nil
+  end
+
   def flatten(level=-1)
     level = Rubinius::Type.coerce_to_collection_index level
     return self.dup if level == 0