Add push-exactly methods to grep

Should make some greps in some contexts faster
rakudo · Oct 11, 2015 · 9b86933 · 9b86933
1 parent be66ad4
commit 9b86933
Showing 1 changed file with 40 additions and 1 deletion.
diff --git a/src/core/Any-iterable-methods.pm b/src/core/Any-iterable-methods.pm
@@ -328,6 +328,17 @@ augment class Any {
                   until ($value := $!iter.pull-one) =:= IterationEnd;
                 IterationEnd
             }
+            method push-exactly($target, int $n) {
+                my Mu $value;
+                my int $done;
+                until IterationEnd =:= ($value := $!iter.pull-one) {
+                    if $value.match($!test) {
+                        $target.push($value);
+                        return $done unless ($done = $done + 1) < $n;
+                    }
+                }
+                IterationEnd
+            }
             method push-all($target) {
                 my Mu $value;
                 $target.push($value) if $value.match($!test)
@@ -347,11 +358,28 @@ augment class Any {
                            until ($value := $!iter.pull-one) =:= IterationEnd;
                          IterationEnd   # in case of last
                      }
+                     method push-exactly($target, int $n) {
+                         my Mu $value;
+                         my int $done;
+                         until IterationEnd =:= ($value := $!iter.pull-one) {
+                             if $!test($value) {
+                                 $target.push($value);
+                                 return $done unless ($done = $done + 1) < $n;
+                             }
+                         }
+                         IterationEnd
+                     }
                      method push-all($target) {
                          my Mu $value;
                          $target.push($value) if $!test($value)
                            until ($value := $!iter.pull-one) =:= IterationEnd;
-                         IterationEnd   # in case of last
+                         IterationEnd
+                     }
+                     method sink-all() {
+                         my Mu $value;
+                         $!test($value)
+                           until ($value := $!iter.pull-one) =:= IterationEnd;
+                         IterationEnd
                      }
                  }.new(self, $test))
         } else {
@@ -387,6 +415,17 @@ augment class Any {
                   until ($value := $!iter.pull-one) =:= IterationEnd;
                 IterationEnd
             }
+            method push-exactly($target, int $n) {
+                my Mu $value;
+                my int $done;
+                until IterationEnd =:= ($value := $!iter.pull-one) {
+                    if $!test.ACCEPTS($value) {
+                        $target.push($value);
+                        return $done unless ($done = $done + 1) < $n;
+                    }
+                }
+                IterationEnd
+            }
             method push-all($target) {
                 my Mu $value;
                 $target.push($value) if $!test.ACCEPTS($value)