From 11cf6c6a268b17c02df7673c846ba991a04842fa Mon Sep 17 00:00:00 2001
From: Andrés Ambrois <andresambrois@gmail.com>
Date: Sat, 16 Aug 2008 04:40:33 +0000
Subject:  First sketch of a GnuGo wrapper class gtp.py

---
(limited to 'gtp.py')

diff --git a/gtp.py b/gtp.py
new file mode 100644
index 0000000..5e7164a
--- /dev/null
+++ b/gtp.py
@@ -0,0 +1,66 @@
+# -*- coding: UTF-8 -*-
+# Copyright 2007-2008 One Laptop Per Child
+# Copyright 2008 Andrés Ambrois <andresambrois@gmail.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+from subprocess import Popen, PIPE
+import logging
+
+logger = logging.getLogger('PlayGo')
+
+class gnugo:
+    ''' A wrapper for talking to gnugo over GTP '''
+    def __init__(self, boardsize=19, color='black', handicap=0, komi=5.5, level=3):
+        ''' Start the gnugo subprocess '''
+        self.color = color
+        try: 
+            self.gnugo = Popen(['gnugo', '--mode', 'gtp', '--boardsize', str(boardsize),
+                                '--handicap', str(handicap), '--komi', str(komi), '--level', str(level) ], 
+                                stdout=PIPE, stdin=PIPE)
+        except:
+            logger.error('Could not start gnugo subprocess')
+        else:
+            logger.debug('Successfuly loaded gnugo!')
+            self.stdin = self.gnugo.stdin
+            self.stdout = self.gnugo.stdout
+    
+    def __del__(self):
+        self.stdin.write('quit \n')
+    
+    def _xy_to_coords(self, x, y):
+        return dict(zip(range(0, 26), 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'))[x] + str(y)
+        
+    def _coords_to_xy(self, coords):
+        return int(dict(zip('ABCDEFGHIJKLMNOPQRSTUVWXYZ', range(0, 26)))[coords[0]]), int(coords[1:])
+    
+    def make_play(self, x, y):
+        self.stdin.write('play %s %s\n' % (self.color, self._xy_to_coords(x, y)))
+        self.stdin.flush()
+        output = self.stdout.readline()
+        self.stdout.readline()
+        if output[0] == '?':
+            return False
+        return True
+    
+    def get_move(self, color):
+        self.stdin.write('genmove %s\n' % color)
+        self.stdin.flush()
+        output = self.stdout.readline()
+        self.stdout.readline()
+        if output[0] == '?':
+            # FIXME: Handle error
+            return False
+        return self._coords_to_xy(output[2:])
--
cgit v0.9.1