Easier debugging of MARS.
[jars.git] / koth.scm
index c6be230..45e4bb8 100644 (file)
--- a/koth.scm
+++ b/koth.scm
@@ -5,6 +5,8 @@
   (chicken pathname)
   (chicken string)
   (chicken pretty-print)
+  (chicken sort)
+  (chicken time posix)
   srfi-1
   matchable
   mars parser)
   (string->prog (with-input-from-file file read-string)))
 
 (define (score-challenger-matches spec challenger-prog other-progs)
-  (map 
-   (lambda (other-prog)
-     (score-match spec challenger-prog other-prog))
-   other-progs))
+  (foldl append '()
+         (map 
+          (lambda (other-prog)
+            (score-match spec challenger-prog other-prog))
+          other-progs)))
 
 (define (score-match spec prog1 prog2)
-  (print "Matching " (prog-name prog1) " against " (prog-name prog2))
+  (print "... Matching " (prog-name prog1) " against " (prog-name prog2))
   (let ((tally
          (foldl
           (lambda (score-a score-b)
@@ -39,8 +42,8 @@
                 results)))))
     (let ((prog1-name (prog-name prog1))
           (prog2-name (prog-name prog2)))
-      `(((,prog1-name ,prog2-name) ,(car tally))
-        ((,prog2-name ,prog1-name) ,(cadr tally))))))
+      `((,prog1-name ,prog2-name ,(car tally))
+        (,prog2-name ,prog1-name ,(cadr tally))))))
 
 (define (score-game spec prog1 prog2)
   (let* ((core (make-core (spec-core-size spec)))
           ((null? result) (error "Invalid game result."))
           ((= (length result) 1)
            (let ((winner-name (caar result)))
-             (if (eq? winner-name (prog-name prog1))
+             (if (equal? winner-name (prog-name prog1))
                  '(3 0)
                  '(0 3))))
           (else
            '(1 1)))))
-    
+
+(define (scores->rankings scores)
+  (let* ((prog-names (delete-duplicates (map car scores)))
+         (prog-scores
+          (map (lambda (prog-name)
+                 (apply +
+                        (map caddr
+                             (filter (lambda (score)
+                                       (equal? prog-name (car score)))
+                                     scores))))
+               prog-names)))
+    (sort (zip prog-names prog-scores)
+          (lambda (a b)
+            (< (cadr a) (cadr b))))))
+
 (define (challenge hill-dir challenger-file)
-  (let* ((spec (load-spec hill-dir))
-         (scores (load-scores hill-dir))
+  (let* ((spec (hill-spec hill-dir))
+         (scores (hill-scores hill-dir))
          (challenger-prog (file->prog challenger-file))
+         (challenger-name (prog-name challenger-prog))
          (hill-progs (map file->prog (hill-files hill-dir))))
-    (if (memq (prog-name challenger-prog) (map prog-name hill-progs))
-        (print "Challenger already on hill!")
-        (let ((new-scores (append (score-challenger-matches spec challenger-prog hill-progs)
-                                  scores)))
-          (save-scores new-scores hill-dir)
-          (hill-add-file challenger-file (prog-name challenger-prog) hill-dir)))))
+    (hill-news-add hill-dir "Challenger '" challenger-name "' accepted for battle.")
+    (if (member challenger-name (map prog-name hill-progs))
+        (hill-news-add hill-dir "Challenge aborted: challenger already on hill!")
+        (let* ((new-scores (append (score-challenger-matches spec challenger-prog hill-progs)
+                                   scores))
+               (rankings (scores->rankings new-scores)))
+          (if (<= (length rankings) (spec-hill-size spec))
+              (begin
+                (hill-save-scores-and-rankings hill-dir new-scores rankings)
+                (hill-add hill-dir challenger-file challenger-name)
+                (hill-news-add hill-dir "Challenger '" challenger-name "' added to the hill."))
+              (let ((loser-name (caar rankings)))
+                (hill-save-scores-and-rankings hill-dir
+                                               (filter (lambda (score)
+                                                         (not (or (equal? (car score) loser-name)
+                                                                  (equal? (cadr score) loser-name))))
+                                                       new-scores)
+                                               (cdr rankings))
+                (hill-add hill-dir challenger-file challenger-name)
+                (hill-remove hill-dir loser-name)
+                (if (equal? loser-name challenger-name)
+                    (hill-news-add hill-dir
+                                   "Challenger '" challenger-name
+                                   "' failed to best any warrior on the hill.")
+                    (begin
+                      (hill-news-add hill-dir
+                                     "Challenger '" challenger-name
+                                     "' defeated at least one warrior on the hill.")
+                      (hill-news-add hill-dir
+                                     "Warrior '" loser-name
+                                     "' has been pushed off the hill!")))))))))
   
 
 ;;; Hill initialization and specs
 ;;
 
-(define (load-scores dir)
+(define (hill-scores dir)
   (with-input-from-file (make-pathname dir "scores") read))
 
-(define (save-scores scores dir)
-  (with-output-to-file (make-pathname dir "scores")
-    (lambda () (pretty-print scores))))
+(define (hill-save-scores-and-rankings dir scores rankings)
+  (for-each
+   (lambda (p)
+     (with-output-to-file (make-pathname dir (car p))
+       (lambda () (pretty-print (cdr p)))))
+   `(("scores" . ,scores) ("rankings" . ,rankings))))
 
-(define (load-spec dir)
+(define (hill-spec dir)
   (with-input-from-file (make-pathname dir "spec") read))
 
 (define (hill-files dir)
   (glob (make-pathname dir "*.red")))
 
-(define (hill-add-file file name dir)
+(define (hill-rankings dir)
+  (with-input-from-file (make-pathname dir "rankings") read))
+
+(define (hill-news dir)
+  (with-input-from-file (make-pathname dir "news") read))
+
+(define (hill-news-add dir . args)
+  (let* ((old-news (hill-news dir))
+         (news-string (apply conc args))
+         (new-news (cons (cons (seconds->string) news-string) old-news)))
+    (print news-string)
+    (with-output-to-file (make-pathname dir "news")
+      (lambda () (pretty-print new-news)))))
+
+(define (hill-add dir file name)
   (copy-file file (make-pathname dir (->string name) ".red")))
 
+(define (hill-remove dir name)
+  (delete-file (make-pathname dir name ".red")))
+
 (define (make-spec core-size match-length games-per-match hill-size)
   (list 'spec hill-size core-size match-length games-per-match))
 
                 (print ";; Hill specifications.")
                 (print ";; ('spec hill-size core-size game-length games-per-match\n")
                 (pp (make-spec core-size game-length games-per-match hill-size))))
-            (with-output-to-file (make-pathname dir "scores")
-              (lambda ()
-                (pp '())))))))
+            (hill-save-scores-and-rankings dir '() '())
+            (with-output-to-file (make-pathname dir "news")
+              (lambda () (print '())))
+            (hill-news-add dir "Hill created.")))))
 
 ;;;; Main ;;;;
 
 
 (define default-core-size 8000)
 (define default-game-length 80000)
-(define default-games-per-match 1)
+(define default-games-per-match 3)
 (define default-hill-size 10)
 
 (define (print-usage)
   (let ((binary (pathname-file (car (argv)))))
     (print "King of the Hill Tournament Manager")
     (print "\nUsage:\t" binary " hill-directory challenger-file")
+    (print "\t" binary " [-r|--rankings] hill-directory")
     (print "\t" binary " [-h|--help]")
     (print "\t" binary " [-i|--init] hill-directory [hill-size [core-size game-length games-per-match]]")
     (print "\nDefault values are as follows:\n"
     ((or () ((or "-h" "--help")))
      (print-usage))
     (((or "-i" "--init") dir hill-size core-size game-length games-per-match)
-     (init-hill-dir dir hill-size core-size game-length games-per-match))
+     (init-hill-dir dir
+                    (string->number hill-size)
+                    (string->number core-size)
+                    (string->number game-length)
+                    (string->number games-per-match)))
     (((or "-i" "--init") dir)
      (init-hill-dir dir
                     default-hill-size
                     default-games-per-match))
     (((or "-i" "--init") dir hill-size)
      (init-hill-dir dir
-                    hill-size
+                    (string->number hill-size)
                     default-core-size
                     default-game-length
                     default-games-per-match))
+    (((or "-r" "--rankings") dir)
+     (let ((rankings (reverse (hill-rankings dir))))
+       (if (null? rankings)
+           (print "No warriors on hill!")
+           (begin
+             (print "Warrior" "\t" "Score")
+             (print "-=-=-=-" "\t" "=-=-=")
+             (for-each (lambda (r) (print (car r) "\t\t" (cadr r))) rankings)))))
     ((hill-dir challenger-file)
      (challenge hill-dir challenger-file))
     (else
      (print "Invalid arguments: " (apply conc else)))))
 
-;; (main)
+(main)