GraphicalGoBoards: sgf2ascii.py

#!/usr/bin/env python

try:
        from sgfpars import Node, Cursor, SGFError, SGFescape
except:
        from sgfparser import Node, Cursor, SGFError, SGFescape

import os, sys, getopt, re
from sys import stdout



def writeheader(rootnode, filename):
        global boardsize
        print 'File', filename
        if (rootnode.has_key('PW')):
                print "White: " + rootnode['PW'][0]
        if (rootnode.has_key('PB')):
                print "Black: " + rootnode['PB'][0]
        if (rootnode.has_key('RE')):
                print "# Result: " + rootnode['RE'][0]
        for el in ['DT', 'EV', 'RO']:
                if (rootnode.has_key(el)):
                        print rootnode[el][0]
        print

        if (rootnode.has_key('SZ')):
                try:
                        boardsize = int(rootnode['SZ'][0])
                except TypeError:
                        print "SGF trouble: Don't understand board size."
                        sys.exit(2)

def sgfpos2coord(pos):
        if len(pos) == 0 or pos == 'tt':
                return "PASS"
        elif len(pos) != 2:
                raise SGFError("Couldn't parse coordinate.")
        x, y = ord(pos[0])-ord('a'), ord(pos[1])-ord('a')
        return x, y


def writeposition(cursor, color):
        board = [['.' for x in range(0, boardsize)] for y in range(0, boardsize)]
        nodenum = 0
        while nodenum < movenum and not cursor.atEnd:
                try:
                        thisnode = cursor.next()
                except SGFError:
                        break
                if thisnode.has_key('AB'):
                        for pos in thisnode['AB']:
                                x, y = sgfpos2coord(pos)
                                board[x][y] = 'X'
                if thisnode.has_key('AW'):
                        for pos in thisnode['AW']:
                                x, y = sgfpos2coord(pos)
                                board[sgfpos2coord(pos)] = 'O'
                if thisnode.has_key('B'):
                        x, y = sgfpos2coord(thisnode['B'][0])
                        board[x][y] = 'X'
                elif thisnode.has_key('W'):
                        x, y = sgfpos2coord(thisnode['W'][0])
                        board[x][y] = 'O'
                else:
                        continue
                nodenum = nodenum + 1

        stdout.write('{{{#!goban\n+')
        for x in range(0, boardsize):
                stdout.write('-')
        print '+'
        for y in range(boardsize-1, -1, -1):
                stdout.write('|')
                for x in range(0, boardsize):
                        stdout.write(board[x][-y-1])
                print '|'
        stdout.write('+')
        for x in range(0, boardsize):
                stdout.write('-')
        print '+\n}}}'


# Main Program.
###############

help_string = """
sgf2gtp.py converts .sgf-files to an ASCII output that is usable for
GNU Go's trac Wiki.
Usage:
sgf2ascii.py --move=<movenum> <file>
"""

color = []
start = 0
to = 100000
boardsize = 19
movenum = 0
maxauto = 300

try:
        opts, args = getopt.getopt(sys.argv[1:], "",
                                   ["help", "move="])
except getopt.GetoptError: 
        print "Unrecognized option. Try sgf2gtp.py --help"
        sys.exit(2)


for o, a in opts:
        if (o == "--help"):
                print help_string
                sys.exit()
        if (o == "--move"):
                try:
                        movenum = int(a)
                except TypeError:
                        print "Cannot convert", a, "into a number."
                        sys.exit(2)

if not len(args) == 1:
        print "Error: Need exactly one file as argument."
        sys.exit(2)

filename = args[0]
try:
        file = open(filename)
        sgf = file.read()
        file.close()
        cursor = Cursor(sgf, 1)
        rootnode = cursor.getRootNode(0)
except IOError:
        print "Couldn't read file %s." % filename
        sys.exit(2)
except SGFError, message:
        print "Couldn't parse SGF in file %s: %s" % (filename, message)
        sys.exit(2)


writeheader(rootnode, filename)
writeposition(cursor, color)