aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--imago/sgfParser/astNode.py50
-rw-r--r--imago/sgfParser/sgf.py9
-rwxr-xr-ximago/sgfParser/sgfyacc.py5
-rwxr-xr-xtestSGF.py19
4 files changed, 61 insertions, 22 deletions
diff --git a/imago/sgfParser/astNode.py b/imago/sgfParser/astNode.py
index e2b74a2..f87d9c1 100644
--- a/imago/sgfParser/astNode.py
+++ b/imago/sgfParser/astNode.py
@@ -1,6 +1,6 @@
from imago.gameLogic.gameData import GameData
from imago.gameLogic.gameMove import GameMove
-from imago.gameLogic.gameState import Player
+from imago.gameLogic.gameBoard import GameBoard
class ASTNode:
"""Abstract Syntax Tree Node of SGF parser"""
@@ -12,7 +12,7 @@ class ASTNode:
if props:
self.props = props
else:
- self.props = {}
+ self.props = []
self.leaf = leaf
def addToSequence(self, move):
@@ -76,27 +76,41 @@ class ASTNode:
firstMoves = []
for child in self.children:
- firstMoves.append(child.toGameMoveTree)
+ firstMoves.append(child.toGameMoveTree(size))
return GameTree(firstMoves, gameData)
- def toGameMoveTree(self):
- player = 0
- coords = []
- for prop in self.props:
- if prop.name == "B": # White move
- player = Player.BLACK
- coords = textToCoords(prop.value)
- if prop.name == "W": # White move
- player = Player.WHITE
- coords = textToCoords(prop.value)
- gameMove = GameMove(player, coords[0], coords[1])
+ def toGameMoveTree(self, previousMove=None):
+ coords = None
+ if previousMove is None:
+ # Game root node
+ size = int(self.getPropertyValue("SZ"))
+ board = GameBoard(size, size)
+ gameMove = GameMove(board)
+ else:
+ for prop in self.props:
+ if prop.name == "B" or prop.name == "W":
+ coords = textToCoords(prop.value)
+ gameMove = previousMove.addMoveByCoords(coords)
for child in self.children:
- newMove = child.toGameMoveTree()
- gameMove.nextMoves.append(newMove)
+ newMove = child.toGameMoveTree(gameMove)
newMove.previousMove = gameMove
return gameMove
+ def hasProperty(self, propertyName):
+ """Returns True if the node contains a property with the given name."""
+ for prop in self.props:
+ if prop.name == propertyName:
+ return True
+ return False
+
+ def getPropertyValue(self, propertyName):
+ """Returns the value of the given property for this node."""
+ for prop in self.props:
+ if prop.name == propertyName:
+ return prop.value
+ raise RuntimeError("ASTNode has no property %s" % propertyName)
+
def toString(self, debug=False):
"""Returns a depth-first representation of the tree."""
out = '(' + str(self.props) + ')'
@@ -107,8 +121,8 @@ class ASTNode:
return out
def textToCoords(text): # Poner en PropertyMove, subclase de Property
- col = ord(text[1]) - ord('a')
- row = ord(text[0]) - ord('a')
+ col = ord(text[0]) - ord('a')
+ row = ord(text[1]) - ord('a')
return [row, col]
diff --git a/imago/sgfParser/sgf.py b/imago/sgfParser/sgf.py
index 56135de..4b11cfb 100644
--- a/imago/sgfParser/sgf.py
+++ b/imago/sgfParser/sgf.py
@@ -1,11 +1,14 @@
"""Module for reading and writing of SGF files."""
-from imago.gameLogic.gameMove import GameMove
from imago.sgfParser.sgfyacc import parser
def loadGameTree(filename):
# PLY?
"""Parses a GameTree instance from a source SGF file."""
- sgf = open(filename, "r")
+ file = open(filename, "r")
- return parser.parse(sgf).toGameMoveTree()
+ text = file.read()
+
+ file.close()
+
+ return parser.parse(text).toGameMoveTree()
diff --git a/imago/sgfParser/sgfyacc.py b/imago/sgfParser/sgfyacc.py
index a07d179..d18f2a8 100755
--- a/imago/sgfParser/sgfyacc.py
+++ b/imago/sgfParser/sgfyacc.py
@@ -31,7 +31,10 @@ def p_node(p):
def p_node_prop(p):
'node : node property'
- p[1].props[p[2].name] = p[2].value
+ if p[1].hasProperty(p[2].name):
+ print("Syntax error: node already contains a property named - %s" % p[2].name)
+ raise SyntaxError
+ p[1].props.append(p[2])
p[0] = p[1]
def p_property(p):
diff --git a/testSGF.py b/testSGF.py
new file mode 100755
index 0000000..6291033
--- /dev/null
+++ b/testSGF.py
@@ -0,0 +1,19 @@
+#!/usr/bin/python
+
+import sys
+from imago.sgfParser.sgf import loadGameTree
+
+"""Gets a game from an SGF file."""
+
+def main():
+ filename = sys.argv[1]
+ move = loadGameTree(filename)
+ while move is not None:
+ move.printBoard()
+ if len(move.nextMoves) > 0:
+ move = move.nextMoves[0]
+ else:
+ move = None
+
+if __name__ == '__main__':
+ main()