Included gomill framework for SGF and GTP support, and sketched out SGF game-loading code.

gomill/examples/show_sgf.py

@@ -0,0 +1,72 @@
+"""Show the position from an SGF file.
+
+This demonstrates the sgf and ascii_boards modules.
+
+"""
+
+import sys
+from optparse import OptionParser
+
+from gomill import ascii_boards
+from gomill import sgf
+from gomill import sgf_moves
+
+def show_sgf_file(pathname, move_number):
+    f = open(pathname)
+    sgf_src = f.read()
+    f.close()
+    try:
+        sgf_game = sgf.Sgf_game.from_string(sgf_src)
+    except ValueError:
+        raise StandardError("bad sgf file")
+
+    try:
+        board, plays = sgf_moves.get_setup_and_moves(sgf_game)
+    except ValueError, e:
+        raise StandardError(str(e))
+    if move_number is not None:
+        move_number = max(0, move_number-1)
+        plays = plays[:move_number]
+
+    for colour, move in plays:
+        if move is None:
+            continue
+        row, col = move
+        try:
+            board.play(row, col, colour)
+        except ValueError:
+            raise StandardError("illegal move in sgf file")
+
+    print ascii_boards.render_board(board)
+    print
+
+_description = """\
+Show the position from an SGF file. If a move number is specified, the position
+before that move is shown (this is to match the behaviour of GTP loadsgf).
+"""
+
+def main(argv):
+    parser = OptionParser(usage="%prog <filename> [move number]",
+                          description=_description)
+    opts, args = parser.parse_args(argv)
+    if not args:
+        parser.error("not enough arguments")
+    pathname = args[0]
+    if len(args) > 2:
+        parser.error("too many arguments")
+    if len(args) == 2:
+        try:
+            move_number = int(args[1])
+        except ValueError:
+            parser.error("invalid integer value: %s" % args[1])
+    else:
+        move_number = None
+    try:
+        show_sgf_file(pathname, move_number)
+    except Exception, e:
+        print >>sys.stderr, "show_sgf:", str(e)
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main(sys.argv[1:])
+