From 921e6e24f4e87cf75827b6deba18fc7bbea48d5c Mon Sep 17 00:00:00 2001
From: Paolo Bosetti <paolo.bosetti@unitn.it>
Date: Fri, 10 Jan 2014 11:13:47 +0100
Subject: Added rewrite of Array#[] to mruby-array-ext gem, so that arrays can
 be sliced with Ranges (as a[1..-2])

---
 mrbgems/mruby-array-ext/mrblib/array.rb | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

(limited to 'mrbgems/mruby-array-ext/mrblib/array.rb')

diff --git a/mrbgems/mruby-array-ext/mrblib/array.rb b/mrbgems/mruby-array-ext/mrblib/array.rb
index 337cef632..f8d89dc7b 100644
--- a/mrbgems/mruby-array-ext/mrblib/array.rb
+++ b/mrbgems/mruby-array-ext/mrblib/array.rb
@@ -201,4 +201,36 @@ class Array
       self.replace(result)
     end
   end
+  
+  ##
+  # call-seq:
+  #    ary[rng]    -> ary slice
+  #
+  # Remeturns a slice of +ary+ according to the Range instance +rng+.
+  #
+  #    a = [ "a", "b", "c", "d", "e" ]
+  #    a[1]     => "b"
+  #    a[1,2]   => ["b", "c"]
+  #    a[1..-2] => ["b", "c", "d"]
+  #
+  def [](idx, len=nil)
+    case idx
+    when Range
+      if idx.last < 0 then
+        len = self.length - idx.first + idx.last + 1
+      else
+        len = idx.last - idx.first + 1
+      end
+      return self.slice(idx.first, len)
+    when Numeric
+      if len then
+        return self.slice(idx.to_i, len.to_i)
+      else
+        return self.slice(idx.to_i)
+      end
+    else
+      self.slice(idx)
+    end
+  end
+  
 end
-- 
cgit v1.2.3