add GTP support

GTP/__init__.py

@@ -0,0 +1,7 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+# $File: __init__.py
+# $Date: Thu Nov 16 14:1006 2017 +0800
+# $Author: renyong15 © <mails.tsinghua.edu.cn>
+#
+

GTP/engine.py

@@ -0,0 +1,197 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+# $File: engine.py
+# $Date: Fri Nov 17 13:5624 2017 +0800
+# $Author: renyong15 © <mails.tsinghua.edu.cn>
+#
+
+from game import Game
+import utils 
+
+class GTPEngine():
+    def __init__(self, **kwargs):
+        self.size = 19
+        self.komi = 6.5
+        try:
+            self._game = kwargs['game_obj']
+            self._game.clear()
+        except:
+            self._game = None
+
+        try:
+            self._name = kwargs['name']
+        except:
+            self._name = 'gtp engine'
+        try:
+            self._version = kwargs['version']
+        except:
+            self._version = 2
+
+
+        self.disconnect = False
+
+        self.known_commands = [
+            field[4:] for field in dir(self) if field.startswith("cmd_")]
+
+    def _vertex_point2string(self, vertex):
+        if vertex == utils.PASS:
+            return "pass"
+        elif vertex == utils.RESIGN:
+            return "resign"
+        else:
+            x, y = vertex
+            return "{}{}".format("ABCDEFGHJKLMNOPQRSTYVWYZ"[x - 1], y)
+
+
+
+
+    def _vertex_string2point(self, s):
+        if s is None:
+            return False
+        elif s.lower() == "pass":
+            return utils.PASS
+        elif len(s) > 1:
+            x = "abcdefghjklmnopqrstuvwxyz".find(s[0].lower()) + 1
+            if x == 0:
+                return False
+            if s[1:].isdigit():
+                y = int(s[1:])
+            else:
+                return False
+        else:
+            return False
+        return (x, y)
+
+
+    def _parse_color(self, color):
+        if color.lower() in ["b", "black"]:
+            color = utils.BLACK
+        elif color.lower() in ["w", "white"]:
+            color = utils.WHITE
+        else:
+            color = None
+        return color
+
+
+    def _parse_move(self, move_string):
+        color, move = move_string.split(" ",1)
+        color = self._parse_color(color)
+
+        point = self._vertex_string2point(move)
+
+        if point and color:
+            return color,point
+        else:
+            return False
+        
+        
+
+    def _parse_res(self, res, id_ = None, success = True):
+        if success:
+            if id_:
+                return '={} {}\n\n'.format(id_, res)
+            else:
+                return '= {}\n\n'.format(res)
+        else:
+            if id_:
+                return '?{} {}\n\n'.format(id_, res)
+            else:
+                return '? {}\n\n'.format(res)
+
+
+    def _parse_cmd(self, message):
+        try:
+            m = message.strip().split(" ", 1)
+            if m[0].isdigit():
+                id_ = int(m[0])
+                cmd, args = (m[1].split(" ", 1) + [None])[:2]
+            else:
+                id_ = None
+                cmd, args = (m[0].split(" ", 1) + [None])[:2]
+        except:
+            return "invaild command"
+        return id_, cmd, args
+
+    def run_cmd(self, message):
+        try:
+            id_, cmd, args = self._parse_cmd(message)
+        except:
+            return self._parse_res("invaild message", id_, False)
+
+        if cmd in self.known_commands:
+            #dispatch
+            #try:
+            if True:
+                res, flag = getattr(self, "cmd_" + cmd)(args)
+                return self._parse_res(res, id_, flag)
+            #except Exception as e:
+            #    print(e)
+            #    return self._parse_res("command excution failed", id_, False)
+        else:
+            return self._parse_res("unknown command", id_, False)
+
+
+
+    def cmd_protocol_version(self, args, **kwargs):
+        return 2, True
+
+    def cmd_name(self, args, **kwargs):
+        return self._name, True
+
+    def cmd_version(self, args, **kwargs):
+        return self._version, True
+
+    def cmd_known_command(self, args, **kwargs):
+        return 'true' if (args in self.known_commands) else 'false', True
+
+    def cmd_list_commands(self, args, **kwargs):
+        return self.known_commands, True
+
+    def cmd_quit(self, args, **kwargs):
+        return None,True
+
+    def cmd_boardsize(self, args, **kwargs):
+        if args.isdigit():
+            size = int(args)
+            self.size = size
+            self._game.set_size(size)
+            return None,True
+        else:
+            return 'non digit size',False
+
+    def cmd_clear_board(self, args, **kwargs):
+        self._game.clear()
+        return None,True
+
+    def cmd_komi(self, args, **kwargs):
+        try:
+            komi = float(args)
+            self.komi = komi
+            self._game.set_komi(komi)
+            return None,True
+        except ValueError:
+            raise ValueError("syntax error")
+
+
+    def cmd_play(self, args, **kwargs):
+        move = self._parse_move(args)
+        if move:
+            color, vertex = move
+            res = self._game.do_move(color, vertex)
+            if res:
+                return None,True 
+            else:
+                return None,False
+        return None,True
+
+    def cmd_genmove(self, args, **kwargs):
+        color = self._parse_color(args)
+        if color:
+            move = self._game.gen_move(color)
+            return self._vertex_point2string(move),True
+        else:
+            return 'unknown player',False
+    
+    
+
+

GTP/game.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+# $File: game.py
+# $Date: Fri Nov 17 15:0745 2017 +0800
+# $Author: renyong15 © <mails.tsinghua.edu.cn>
+#
+
+import utils
+
+
+class Game:
+    def __init__(self, size=19, komi=6.5):
+        self.size = size
+        self.komi = 6.5
+        self.board = [utils.EMPTY] * (self.size * self.size)
+        self.strategy = None
+
+    def _flatten(self, vertex):
+        x,y = vertex
+        return (x-1) * self.size + (y-1)
+        
+
+    def clear(self):
+        self.board = [utils.EMPTY] * (self.size * self.size)
+
+    def set_size(self, n):
+        self.size = n
+        self.clear()
+
+    def set_komi(self, k):
+        self.komi = k
+
+    def do_move(self, color, vertex):
+        if vertex == utils.PASS:
+            return True
+
+        id_ = self._flatten(vertex)
+        if self.board[id_] == utils.EMPTY:
+            self.board[id_] = color
+            return True
+        else:
+            return False
+
+    def gen_move(self, color):
+        move = self.strategy.gen_move(color)
+        return move
+        #return utils.PASS
+
+
+    

GTP/test.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+# $File: test.py
+# $Date: Fri Nov 17 13:5600 2017 +0800
+# $Author: renyong15 © <mails.tsinghua.edu.cn>
+#
+
+from game import Game
+from engine import GTPEngine
+
+
+
+g = Game()
+e = GTPEngine(game_obj = g)
+res = e.run_cmd('1 protocol_version')
+print(e.known_commands)
+print(res)
+
+res = e.run_cmd('2 name')
+print(res)
+
+res = e.run_cmd('3 known_command quit')
+print(res)
+
+res = e.run_cmd('4 unknown_command quitagain')
+print(res)
+
+res = e.run_cmd('5 list_commands')
+print(res)
+
+res = e.run_cmd('6 komi 6')
+print(res)
+
+res = e.run_cmd('7 play BLACK C3')
+print(res)
+
+res = e.run_cmd('8 genmove BLACK')
+print(res)
+
+

GTP/utils.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# vim:fenc=utf-8
+# $File: utils.py
+# $Date: Fri Nov 17 10:2407 2017 +0800
+# $Author: renyong15 © <mails.tsinghua.edu.cn>
+#
+
+WHITE = -1
+BLACK = +1
+EMPTY = 0
+
+PASS = (0,0)
+RESIGN = "resign"
+
+
+