Add Any.tree.

rakudo · Jun 7, 2012 · d884467 · d884467
1 parent 2422204
commit d884467
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/core/Any.pm b/src/core/Any.pm
@@ -18,9 +18,9 @@ my class Any {
     method uniq() { self.list.uniq }
     method infinite() { Mu }
     method flat() { nqp::p6list(nqp::list(self), List, Bool::True) }
-    method tree(*@a) { self.list.tree(|@a) }
     method hash() { my %h = self }
     method list() { nqp::p6list(nqp::list(self), List, Mu) }
+    method lol()  { MapIter.new(self.list, { .item }, Mu).list }
     method pick($n = 1) { self.list.pick($n) }
     method roll($n = 1) { self.list.roll($n) }
     method reverse() { self.list.reverse }
@@ -118,6 +118,20 @@ my class Any {
         $self.push(@values);
     }
 
+    proto method tree(|$) { * }
+    multi method tree(Any:U:) { self }
+    multi method tree(Any:D:) { self.lol }
+    multi method tree(Any:D: Cool $count as Int) {
+        $count > 1
+          ?? MapIter.new(self.list, { .tree($count-1).item }, Mu).list
+          !! $count == 1
+             ?? self.lol
+             !! self
+    }
+    multi method tree(Any:D: &c) {
+        MapIter.new(self.list, { .&c.item }, Mu).list
+    }
+
     proto method unshift(|$) { * }
     multi method unshift(Any:U \$self: *@values) {
         &infix:<=>($self, Array.new);