[421] | 1 | #!/usr/bin/env python |
---|
| 2 | '''This demonstrates controlling a screen oriented application (curses). |
---|
| 3 | It starts two instances of gnuchess and then pits them against each other. |
---|
| 4 | ''' |
---|
| 5 | import pexpect |
---|
| 6 | import string |
---|
| 7 | import ANSI |
---|
| 8 | |
---|
| 9 | REGEX_MOVE = '(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)' |
---|
| 10 | REGEX_MOVE_PART = '(?:[0-9]|\x1b\[C)(?:[a-z]|\x1b\[C)(?:[0-9]|\x1b\[C)' |
---|
| 11 | |
---|
| 12 | class Chess: |
---|
| 13 | |
---|
| 14 | def __init__(self, engine = "/usr/local/bin/gnuchess -a -h 1"): |
---|
| 15 | self.child = pexpect.spawn (engine) |
---|
| 16 | self.term = ANSI.ANSI () |
---|
| 17 | |
---|
| 18 | self.child.expect ('Chess') |
---|
| 19 | if self.child.after != 'Chess': |
---|
| 20 | raise IOError, 'incompatible chess program' |
---|
| 21 | self.term.process_list (self.before) |
---|
| 22 | self.term.process_list (self.after) |
---|
| 23 | self.last_computer_move = '' |
---|
| 24 | def read_until_cursor (self, r,c) |
---|
| 25 | while 1: |
---|
| 26 | self.child.read(1, 60) |
---|
| 27 | self.term.process (c) |
---|
| 28 | if self.term.cur_r == r and self.term.cur_c == c: |
---|
| 29 | return 1 |
---|
| 30 | |
---|
| 31 | def do_first_move (self, move): |
---|
| 32 | self.child.expect ('Your move is') |
---|
| 33 | self.child.sendline (move) |
---|
| 34 | self.term.process_list (self.before) |
---|
| 35 | self.term.process_list (self.after) |
---|
| 36 | return move |
---|
| 37 | |
---|
| 38 | def do_move (self, move): |
---|
| 39 | read_until_cursor (19,60) |
---|
| 40 | #self.child.expect ('\[19;60H') |
---|
| 41 | self.child.sendline (move) |
---|
| 42 | print 'do_move' move |
---|
| 43 | return move |
---|
| 44 | |
---|
| 45 | def get_first_computer_move (self): |
---|
| 46 | self.child.expect ('My move is') |
---|
| 47 | self.child.expect (REGEX_MOVE) |
---|
| 48 | # print '', self.child.after |
---|
| 49 | return self.child.after |
---|
| 50 | |
---|
| 51 | def get_computer_move (self): |
---|
| 52 | print 'Here' |
---|
| 53 | i = self.child.expect (['\[17;59H', '\[17;58H']) |
---|
| 54 | print i |
---|
| 55 | if i == 0: |
---|
| 56 | self.child.expect (REGEX_MOVE) |
---|
| 57 | if len(self.child.after) < 4: |
---|
| 58 | self.child.after = self.child.after + self.last_computer_move[3] |
---|
| 59 | if i == 1: |
---|
| 60 | self.child.expect (REGEX_MOVE_PART) |
---|
| 61 | self.child.after = self.last_computer_move[0] + self.child.after |
---|
| 62 | print '', self.child.after |
---|
| 63 | self.last_computer_move = self.child.after |
---|
| 64 | return self.child.after |
---|
| 65 | |
---|
| 66 | def switch (self): |
---|
| 67 | self.child.sendline ('switch') |
---|
| 68 | |
---|
| 69 | def set_depth (self, depth): |
---|
| 70 | self.child.sendline ('depth') |
---|
| 71 | self.child.expect ('depth=') |
---|
| 72 | self.child.sendline ('%d' % depth) |
---|
| 73 | |
---|
| 74 | def quit(self): |
---|
| 75 | self.child.sendline ('quit') |
---|
| 76 | import sys, os |
---|
| 77 | print 'Starting...' |
---|
| 78 | white = Chess() |
---|
| 79 | white.child.echo = 1 |
---|
| 80 | white.child.expect ('Your move is') |
---|
| 81 | white.set_depth(2) |
---|
| 82 | white.switch() |
---|
| 83 | |
---|
| 84 | move_white = white.get_first_computer_move() |
---|
| 85 | print 'first move white:', move_white |
---|
| 86 | |
---|
| 87 | white.do_move ('e7e5') |
---|
| 88 | move_white = white.get_computer_move() |
---|
| 89 | print 'move white:', move_white |
---|
| 90 | white.do_move ('f8c5') |
---|
| 91 | move_white = white.get_computer_move() |
---|
| 92 | print 'move white:', move_white |
---|
| 93 | white.do_move ('b8a6') |
---|
| 94 | move_white = white.get_computer_move() |
---|
| 95 | print 'move white:', move_white |
---|
| 96 | |
---|
| 97 | sys.exit(1) |
---|
| 98 | |
---|
| 99 | |
---|
| 100 | |
---|
| 101 | black = Chess() |
---|
| 102 | white = Chess() |
---|
| 103 | white.child.expect ('Your move is') |
---|
| 104 | white.switch() |
---|
| 105 | |
---|
| 106 | move_white = white.get_first_computer_move() |
---|
| 107 | print 'first move white:', move_white |
---|
| 108 | |
---|
| 109 | black.do_first_move (move_white) |
---|
| 110 | move_black = black.get_first_computer_move() |
---|
| 111 | print 'first move black:', move_black |
---|
| 112 | |
---|
| 113 | white.do_move (move_black) |
---|
| 114 | |
---|
| 115 | done = 0 |
---|
| 116 | while not done: |
---|
| 117 | move_white = white.get_computer_move() |
---|
| 118 | print 'move white:', move_white |
---|
| 119 | |
---|
| 120 | black.do_move (move_white) |
---|
| 121 | move_black = black.get_computer_move() |
---|
| 122 | print 'move black:', move_black |
---|
| 123 | |
---|
| 124 | white.do_move (move_black) |
---|
| 125 | print 'tail of loop' |
---|
| 126 | |
---|
| 127 | g.quit() |
---|
| 128 | |
---|
| 129 | |
---|