fink/gtp.lisp

97 lines
3.8 KiB
Common Lisp

(in-package :gtp-handler)
(defparameter *quit?* nil)
;(defparameter *cputime* 0)
(defmacro inc-cpu-timer (body)
`(let ((start (get-internal-run-time))
(val ,body)
(end (get-internal-run-time)))
(setf go-bot:*cputime* (+ go-bot:*cputime* (float (/ (- end start) 1000))))
val))
(defun gtp-net-client (server port)
(go-bot:init)
(setf *quit?* nil)
(let ((socket (netpipe:tcp-connect server port)))
(if (eql socket nil)
()
(progn
; (format t "Connection establish, playing...~%")
(do ()
((or (eql socket nil) (eql *quit?* t)))
(let ((cmd (netpipe:tcp-read socket)))
;(format t "cmd: '~a'~%" cmd)
(let ((resp (inc-cpu-timer (dispatch-gtp-command cmd))))
;(format t "resp: '~a'~%" resp)
(netpipe:tcp-print socket (concatenate 'string "= " resp (string #\newline) (string #\newline))))))))))
(defun gtp-client ()
(go-bot:init)
(setf *quit?* nil)
(do ()
((eql *quit?* t))
(format t "= ~a~%~%" (inc-cpu-timer (dispatch-gtp-command (read-line t))))))
(defun split-string (string pivot-str)
(do ((pivot (char pivot-str 0))
(i 0 (+ i 1))
(beg 0)
(strings '()))
((> i (length string)) (reverse strings))
(if (or (eql (length string) i) (eql (aref string i) pivot))
(progn (push (subseq string beg i) strings) (setf beg (+ i 1))))))
(defparameter *supported_commands* '("name" "version" "protocol_version" "komi" "boardsize" "clear_board" "play" "genmove" "cputime" "quit" "game_score" "list_commands" "known_command" "gogui-analyze_commands" ))
(defparameter *analyze_commands* '("gfx/Liberties/liberties" "gfx/Shapes/shapes" "gfx/Shape-Liberties/shape-liberties" "gfx/Shape-Stone-Liberties/shape-stone-liberties"))
(defun match-string (str)
(lambda (elem) (string-equal str elem)))
(defun dispatch-gtp-command (command-string)
(let* ((commands (split-string (string-trim #(#\newline #\space) (string-upcase command-string)) " "))
;(cl-ppcre:split "[\\s\\n]+" (string-upcase command-string)))
(command (intern (first commands) :gtp-handler)))
;(format t "~a~%" commands)
(case command
(name go-bot:*name*)
(version go-bot:*version*)
(protocol_version "gtp2ip-0.1")
(boardsize (go-bot:set-boardsize (parse-integer (second commands)))
(go-bot:init-board)
"")
; warning: read-from-string pulls full reader. not safe
(komi (go-bot:set-komi (read-from-string (second commands)))
"")
(clear_board (go-bot:init) "")
(play (go-bot:do-play (char (second commands) 0) (third commands)) "")
(genmove (go-bot:do-genmove (char (second commands) 0)))
(genmove_black (go-bot:do-genmove #\b))
(genmove_white (go-bot:do-genmove #\w))
(cputime (write-to-string go-bot:*cputime*))
;(get_random_seed "0")
(known_command (write-to-string (count-if (match-string (second commands)) *supported_commands*)))
(list_commands (let ((str ""))
(loop for command in *supported_commands* do (setf str (concatenate 'string str command " ")))
str))
(gogui-analyze_commands (let ((str ""))
(loop for command in *analyze_commands* do (setf str (concatenate 'string str command (string #\newline))))
(string-trim #(#\newline) str)))
(game_score (format t "Score for ~c: ~s~%" go-bot:*player* (string-trim (string #\newline) (second commands))) "")
(liberties (string-trim #(#\newline) (analyze-liberty)))
(shapes (string-trim #(#\newline) (analyze-shapes)))
(shape-liberties (string-trim #(#\newline) (analyze-shape-liberties)))
(shape-stone-liberties (string-trim #(#\newline) (analyze-shape-stone-liberties)))
;(scores (string-trim #(#\newline)(analyze-score)))
(quit (setf *quit?* t) "")
(otherwise (concatenate 'string "? unknown command: " (string-downcase (first commands)))))))