X-Git-Url: https://git.ucc.asn.au/?p=progcomp10.git;a=blobdiff_plain;f=src%2Flink%2Fpexpect%2Fexamples%2Fchess3.py;fp=src%2Flink%2Fpexpect%2Fexamples%2Fchess3.py;h=44044420a1c9879bf1af353cee2df8882fd8c738;hp=0000000000000000000000000000000000000000;hb=cd42b53c196672694396e695ae17fd94ba7d58b4;hpb=e9a8105a8f22404f4ac550d79954eaa6b7f5d8ff

diff --git a/src/link/pexpect/examples/chess3.py b/src/link/pexpect/examples/chess3.py
new file mode 100755
index 0000000..4404442
--- /dev/null
+++ b/src/link/pexpect/examples/chess3.py
@@ -0,0 +1,138 @@
+#!/usr/bin/env python
+
+'''This demonstrates controlling a screen oriented application (curses).
+It starts two instances of gnuchess and then pits them against each other.
+'''
+
+import pexpect
+import string
+import ANSI
+
+REGEX_MOVE = '(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)'
+REGEX_MOVE_PART = '(?:[0-9]|\x1b\[C)(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)'
+
+class Chess:
+
+	def __init__(self, engine = "/usr/local/bin/gnuchess -a -h 1"):
+		self.child = pexpect.spawn (engine)
+                self.term = ANSI.ANSI ()
+             
+#		self.child.expect ('Chess')
+	#	if self.child.after != 'Chess':
+	#		raise IOError, 'incompatible chess program'
+        #        self.term.process_list (self.before)
+        #        self.term.process_list (self.after)
+		self.last_computer_move = ''
+        def read_until_cursor (self, r,c):
+            fout = open ('log','a')
+            while 1:
+                k = self.child.read(1, 10)
+                self.term.process (k)
+                fout.write ('(r,c):(%d,%d)\n' %(self.term.cur_r, self.term.cur_c))
+                fout.flush()
+                if self.term.cur_r == r and self.term.cur_c == c:
+                    fout.close()
+                    return 1
+                sys.stdout.write (k)
+                sys.stdout.flush()
+
+	def do_scan (self):
+            fout = open ('log','a')
+            while 1:
+                c = self.child.read(1,10)
+                self.term.process (c)
+                fout.write ('(r,c):(%d,%d)\n' %(self.term.cur_r, self.term.cur_c))
+                fout.flush()
+                sys.stdout.write (c)
+                sys.stdout.flush()
+
+	def do_move (self, move):
+                self.read_until_cursor (19,60)
+		self.child.sendline (move)
+		return move
+	
+	def get_computer_move (self):
+		print 'Here'
+		i = self.child.expect (['\[17;59H', '\[17;58H'])
+		print i
+		if i == 0:
+			self.child.expect (REGEX_MOVE)
+			if len(self.child.after) < 4:
+				self.child.after = self.child.after + self.last_computer_move[3]
+		if i == 1:
+			self.child.expect (REGEX_MOVE_PART)
+			self.child.after = self.last_computer_move[0] + self.child.after
+		print '', self.child.after
+		self.last_computer_move = self.child.after
+		return self.child.after
+
+	def switch (self):
+		self.child.sendline ('switch')
+
+	def set_depth (self, depth):
+		self.child.sendline ('depth')
+		self.child.expect ('depth=')
+		self.child.sendline ('%d' % depth)
+
+	def quit(self):
+		self.child.sendline ('quit')
+import sys, os
+print 'Starting...'
+white = Chess()
+white.do_move('b2b4')
+white.read_until_cursor (19,60)
+c1 = white.term.get_abs(17,58)
+c2 = white.term.get_abs(17,59)
+c3 = white.term.get_abs(17,60)
+c4 = white.term.get_abs(17,61)
+fout = open ('log','a')
+fout.write ('Computer:%s%s%s%s\n' %(c1,c2,c3,c4))
+fout.close()
+white.do_move('c2c4')
+white.read_until_cursor (19,60)
+c1 = white.term.get_abs(17,58)
+c2 = white.term.get_abs(17,59)
+c3 = white.term.get_abs(17,60)
+c4 = white.term.get_abs(17,61)
+fout = open ('log','a')
+fout.write ('Computer:%s%s%s%s\n' %(c1,c2,c3,c4))
+fout.close()
+white.do_scan ()
+
+#white.do_move ('b8a6')
+#move_white = white.get_computer_move()
+#print 'move white:', move_white
+
+sys.exit(1)
+
+
+
+black = Chess()
+white = Chess()
+white.child.expect ('Your move is')
+white.switch()
+
+move_white = white.get_first_computer_move()
+print 'first move white:', move_white
+
+black.do_first_move (move_white)
+move_black = black.get_first_computer_move()
+print 'first move black:', move_black
+
+white.do_move (move_black)
+
+done = 0
+while not done:
+    move_white = white.get_computer_move()
+    print 'move white:', move_white
+
+    black.do_move (move_white)
+    move_black = black.get_computer_move()
+    print 'move black:', move_black
+   
+    white.do_move (move_black)
+    print 'tail of loop'
+
+g.quit()
+
+