factor: trim using lists

[factor.git] / extra / rosetta-code / bulls-and-cows / bulls-and-cows.factor

diff --git a/extra/rosetta-code/bulls-and-cows/bulls-and-cows.factor b/extra/rosetta-code/bulls-and-cows/bulls-and-cows.factor
index 061fab3c04efb7b0ec91f65cc795553e90fab3e4..3e84fa2e9e2d28a1be3a1da81d84947b54980d71 100644 (file)
--- a/extra/rosetta-code/bulls-and-cows/bulls-and-cows.factor
+++ b/extra/rosetta-code/bulls-and-cows/bulls-and-cows.factor
@@ -1,8 +1,7 @@
 ! Copyright (c) 2012 Anonymous
 ! See http://factorcode.org/license.txt for BSD license.
-USING: accessors assocs combinators fry grouping hashtables
-kernel locals math math.parser math.ranges random sequences
-strings io ascii ;
+USING: accessors ascii assocs combinators io kernel math
+math.parser random ranges sequences ;
 IN: rosetta-code.bulls-and-cows
 
 ! http://rosettacode.org/wiki/Bulls_and_cows
@@ -37,10 +36,10 @@ TUPLE: cow ;
 TUPLE: bull ;
 : <bull> ( -- bull ) bull new ;
 
-: inc-bulls ( score -- score ) dup bulls>> 1 + >>bulls ;
-: inc-cows ( score -- score ) dup cows>> 1 + >>cows ;
+: inc-bulls ( score -- score ) [ 1 + ] change-bulls ;
+: inc-cows ( score -- score ) [ 1 + ] change-cows ;
 
-: random-nums ( -- seq ) 9 [1,b] 4 sample ;
+: random-nums ( -- seq ) 9 [1..b] 4 sample ;
 
 : add-digits ( seq -- n ) 0 [ swap 10 * + ] reduce number>string ;
 
@@ -49,7 +48,7 @@ TUPLE: bull ;
 : narr>nhash ( narr -- nhash ) { 1 2 3 4 } swap zip ;
 
 : num>hash ( n -- hash )
-    [ 1string string>number ] { } map-as narr>nhash ;
+    [ digit> ] { } map-as narr>nhash ;
 
 :: cow-or-bull ( n g -- arr )
     {
@@ -66,23 +65,26 @@ TUPLE: bull ;
 : sum-score ( n g -- score ? )
     '[ _ cow-or-bull ] map sift add-to-score dup check-win ;
 
-: print-sum ( score -- str )
-    dup bulls>> number>string "Bulls: " swap append swap cows>> number>string
-    " Cows: " swap 3append "\n" append ;
+: score-to-answer ( score -- str )
+    [ bulls>> number>string "Bulls: " prepend ]
+    [ cows>> number>string " Cows: " prepend ] bi "\n" glue ;
 
-: (validate-readln) ( str -- ? ) dup length 4 = not swap [ letter? ] all? or ;
+: (validate-readln) ( str -- ? )
+    [ length 4 = not ]
+    [ [ letter? ] all? ] bi or ;
 
 : validate-readln ( -- str )
     readln dup (validate-readln)
-    [ "Invalid input.\nPlease enter a valid 4 digit number: "
-      write flush drop validate-readln ]
-    when ;
+    [
+        "Invalid input.\nPlease enter a valid 4 digit number: "
+        write flush drop validate-readln
+    ] when ;
 
-: win ( -- ) "\nYou've won! Good job. You're so smart." print flush ; 
+: win ( -- ) "You've won! Good job. You're so smart." print flush ;
 
 : main-loop ( x -- )
     "Enter a 4 digit number: " write flush validate-readln num>hash swap
-    [ sum-score swap print-sum print flush ] keep swap not
+    [ sum-score swap score-to-answer print flush ] keep swap not
     [ main-loop ] [ drop win ] if ;
 
 : bulls-and-cows-main ( -- ) new-number drop narr>nhash main-loop ;