sequences.extras: Add count-head and count-tail

diff --git a/extra/sequences/extras/extras.factor b/extra/sequences/extras/extras.factor
index 334fc60b3f275f5cf73d455366fcdbb1c94b9d44..85bf1327191b2d026e55a298587944d21d0eef00 100644 (file)
--- a/extra/sequences/extras/extras.factor
+++ b/extra/sequences/extras/extras.factor
@@ -628,6 +628,13 @@ PRIVATE>
     [ '[ @ not ] find drop ] keepd swap
     [ dup length ] unless* tail-slice ; inline
 
+: count-head ( seq quot -- n )
+    [ not ] compose find drop ; inline
+
+: count-tail ( seq quot -- n )
+    [ not ] compose [ find-last drop ] 2keep drop
+    length swap [ - 1 - ] when* ; inline
+
 :: interleaved-as ( seq glue exemplar -- newseq )
     seq length dup 1 - + 0 max exemplar new-sequence :> newseq
     seq [ 2 * newseq set-nth-unsafe ] each-index