added flatten_deep with tests

[gitmo/Moose-Autobox.git] / lib / Moose / Autobox / Array.pm

diff --git a/lib/Moose/Autobox/Array.pm b/lib/Moose/Autobox/Array.pm
index 6234435..92b919b 100644 (file)
--- a/lib/Moose/Autobox/Array.pm
+++ b/lib/Moose/Autobox/Array.pm
@@ -121,6 +121,24 @@ sub flatten {
     @{$_[0]}
 }
 
+sub _flatten_deep { 
+       my @array = @_;
+       my $depth = CORE::pop @array;
+       --$depth if (defined($depth));
+       
+       CORE::map {
+               (ref eq 'ARRAY')
+                       ? (defined($depth) && $depth == -1) ? $_ : _flatten_deep( @$_, $depth )
+                       : $_
+       } @array;
+
+}
+
+sub flatten_deep { 
+       my ($array, $depth) = @_;       
+       [ _flatten_deep(@$array, $depth) ];
+}
+
 ## Junctions
 
 sub all {
@@ -196,6 +214,8 @@ This is a role to describe operations on the Array type.
 
 =item B<flatten>
 
+=item B<flatten_deep ($depth)>
+
 =back
 
 =head2 Indexed implementation