summaryrefslogtreecommitdiffhomepage
path: root/mrbgems/mruby-array-ext/mrblib/array.rb
diff options
context:
space:
mode:
Diffstat (limited to 'mrbgems/mruby-array-ext/mrblib/array.rb')
-rw-r--r--mrbgems/mruby-array-ext/mrblib/array.rb32
1 files changed, 32 insertions, 0 deletions
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