Introduce :as parameter to classify

Use case: my %h = (a => 1, a => 2, b=> 3).classify(*.key, :what(*.value)); say %h.perl {:a([1, 2]), :b([3])}<> Inspired by smls++'s comments at: http://irclog.perlgeek.de/perl6/2015-05-24#i_10648933
rakudo · May 24, 2015 · 922bcf0 · 922bcf0
1 parent 800f703
commit 922bcf0
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 11 deletions.
diff --git a/src/core/Any.pm b/src/core/Any.pm
@@ -155,11 +155,11 @@ my class Any { # declared in BOOTSTRAP
     multi method roll($n) { self.list.roll($n) }
 
     proto method classify(|) is nodal { * }
-    multi method classify($test)   {
-        Hash.^parameterize(Any,Any).new.classify-list( $test, self.list );
+    multi method classify($test, |c)   {
+        Hash.^parameterize(Any,Any).new.classify-list( $test, self.list, |c );
     }
-    multi method classify($test, :$into!)   {
-        ( $into // $into.new ).classify-list( $test, self.list );
+    multi method classify($test, :$into!, |c)   {
+        ( $into // $into.new ).classify-list( $test, self.list, |c );
     }
 
     proto method categorize(|) is nodal { * }

diff --git a/src/core/Hash.pm b/src/core/Hash.pm
@@ -129,7 +129,7 @@ my class Hash { # declared in BOOTSTRAP
     }
 
     proto method classify-list(|) { * }
-    multi method classify-list( &test, @list ) {
+    multi method classify-list( &test, @list, :&as ) {
         fail X::Cannot::Infinite.new(:action<.classify>) if @list.infinite;
         if @list {
 
@@ -141,22 +141,39 @@ my class Hash { # declared in BOOTSTRAP
                     my $hash  = self;
                     $hash = $hash{$_} //= self.new for @keys;
                     nqp::push(
-                      nqp::p6listitems(nqp::decont($hash{$last} //= [])), $l );
+                      nqp::p6listitems(nqp::decont($hash{$last} //= [])),
+                      &as ?? as($l) !! $l
+                    );
+                }
+            }
+
+            # simple classify with to store a specific value
+            elsif &as {
+                @list.map: {
+                    nqp::push(
+                      nqp::p6listitems(nqp::decont(self{test $_} //= [])),
+                      as($_)
+                    )
                 }
             }
 
             # just a simple classify
             else {
-                @list.map: { nqp::push( nqp::p6listitems(nqp::decont(self{test $_} //= [])), $_ ) }
+                @list.map: {
+                    nqp::push(
+                      nqp::p6listitems(nqp::decont(self{test $_} //= [])),
+                      $_
+                    )
+                }
             }
         }
         self;
     }
-    multi method classify-list( %test, $list ) {
-        samewith( { %test{$^a} }, $list );
+    multi method classify-list( %test, $list, |c ) {
+        samewith( { %test{$^a} }, $list, |c );
     }
-    multi method classify-list( @test, $list ) {
-        samewith( { @test[$^a] }, $list );
+    multi method classify-list( @test, $list, |c ) {
+        samewith( { @test[$^a] }, $list, |c );
     }
 
     proto method categorize-list(|) { * }