diff --git a/state.py b/state.py index 93601e2..c980df3 100644 --- a/state.py +++ b/state.py @@ -20,4 +20,67 @@ def __str__(self): string += str(self.board[row]) + "\n" return string - + def first_moves_set(self): + firstMoves = [] + firstMoves.append((1, 1)) + firstMoves.append((1, 18)) + firstMoves.append((18, 1)) + firstMoves.append((18, 18)) + firstMoves.append((9, 9)) + firstMoves.append((9, 10)) + firstMoves.append((10, 9)) + firstMoves.append((10, 10)) + return firstMoves + + def second_moves_set(self): + secondMoves = [] + #If bottom left corner of the board is removed + if (self.board[1][1] == ' '): + secondMoves.append((1, 2)) + secondMoves.append((2, 1)) + return secondMoves + #Top left corner is removed + elif (self.board[1][18] == ' '): + secondMoves.append((1, 17)) + secondMoves.append((2, 18)) + return secondMoves + #Top right + elif (self.board[18][18] == ' '): + secondMoves.append((17, 18)) + secondMoves.append((18, 17)) + return secondMoves + #Bottom right + elif (self.board[18][1] == ' '): + secondMoves.append((17, 1)) + secondMoves.append((18, 2)) + return secondMoves + + #Middle game states, bottom left + elif (self.board[9][9] == ' '): + secondMoves.append((9, 10)) + secondMoves.append((9, 8)) + secondMoves.append((8, 9)) + secondMoves.append((10, 9)) + return secondMoves + #Mid top left + elif (self.board[9][10] == ' '): + secondMoves.append((9, 11)) + secondMoves.append((9, 9)) + secondMoves.append((10, 10)) + secondMoves.append((8, 10)) + return secondMoves + #Mid top right + elif (self.board[10][10] == ' '): + secondMoves.append((9, 10)) + secondMoves.append((11, 10)) + secondMoves.append((10, 9)) + secondMoves.append((10, 11)) + return secondMoves + #Mid bot right + elif (self.board[10][9] == ' '): + secondMoves.append((11, 9)) + secondMoves.append((9, 9)) + secondMoves.append((10, 10)) + secondMoves.append((10, 8)) + return secondMoves +