factor: trim using lists

[factor.git] / extra / sets / extras / extras.factor

diff --git a/extra/sets/extras/extras.factor b/extra/sets/extras/extras.factor
index 2963ad4559bb252f8b7109fe515e82f55cf6fe2a..a6ca1f974004b210670e3f1cf39939c8dc016911 100644 (file)
--- a/extra/sets/extras/extras.factor
+++ b/extra/sets/extras/extras.factor
@@ -1,6 +1,7 @@
 ! Copyright (C) 2013 Doug Coleman.
 ! See http://factorcode.org/license.txt for BSD license.
-USING: assocs kernel locals sequences sequences.extras sets ;
+USING: assocs hash-sets kernel sequences sequences.extras
+sets ;
 IN: sets.extras
 
 : setwise-xor ( seq1 seq2 -- set )
@@ -28,3 +29,11 @@ IN: sets.extras
 
 : adjoin-at* ( value key assoc -- set )
     [ [ HS{ } clone ] unless* [ adjoin ] keep dup ] change-at ;
+
+: mapped-set ( ... seq quot: ( ... elt -- ... newelt ) -- ... set )
+    over length <hash-set> [
+        '[ @ _ adjoin ] each
+    ] keep ; inline
+
+: unique-by ( seq quot: ( elt -- key ) -- seq' )
+    over length <hash-set> '[ @ _ ?adjoin ] filter ; inline