diff options
| -rw-r--r-- | mrbgems/mruby-array-ext/mrblib/array.rb | 32 | ||||
| -rw-r--r-- | mrbgems/mruby-array-ext/test/array.rb | 7 |
2 files changed, 39 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 diff --git a/mrbgems/mruby-array-ext/test/array.rb b/mrbgems/mruby-array-ext/test/array.rb index 1c441cec4..8a6f50fe4 100644 --- a/mrbgems/mruby-array-ext/test/array.rb +++ b/mrbgems/mruby-array-ext/test/array.rb @@ -107,3 +107,10 @@ assert("Array#compact!") do a.compact! a == [1, "2", :t, false] end + +assert("Array#[]") do + a = [ "a", "b", "c", "d", "e" ] + a[1.1] == "b" and + a[1,2] == ["b", "c"] and + a[1..-2] == ["b", "c", "d"] +end |
