diff options
Diffstat (limited to 'voerbak/main.py')
| -rw-r--r-- | voerbak/main.py | 149 |
1 files changed, 78 insertions, 71 deletions
diff --git a/voerbak/main.py b/voerbak/main.py index 433f130..d802603 100644 --- a/voerbak/main.py +++ b/voerbak/main.py @@ -6,85 +6,92 @@ DISC_A = Fore.RED + DISC_SHAPE + Fore.RESET DISC_B = Fore.BLUE + DISC_SHAPE + Fore.RESET EMPTY = Fore.LIGHTBLACK_EX + "_" + Fore.RESET + class bord: - def __init__(self, w, h): - self.width = w - self.height = h - self.board = [[EMPTY for x in range(self.width)] for u in range(self.height)] + def __init__(self, w, h): + self.width = w + self.height = h + self.board = [ + [EMPTY for x in range(self.width)] for u in range(self.height) + ] + + def print(self): + print( + "\n".join( + [ + " ".join(self.board[y]) + for y in range(len(self.board) - 1, -1, -1) + ] + ) + ) - def print(self): - print("\n".join([" ".join(self.board[y]) for y in range(len(self.board) -1, -1, -1)])) + def outside_board(self, coords): + return coords[0] < 0 or \ + coords[1] < 0 or \ + coords[0] > self.height - 1 or \ + coords[1] > self.width - 1 - def outside_board(self, coords): - return coords[0] < 0 or \ - coords[1] < 0 or \ - coords[0] > self.height - 1 or \ - coords[1] > self.width - 1 + def recursive_solve(self, coords, check_for, direction, current_length): + new_position = (coords[0] + direction[0], coords[1] + direction[1]) + if self.outside_board(new_position) or self.board[new_position[0]][ + new_position[1]] != check_for: + return current_length + else: + return self.recursive_solve( + new_position, check_for, direction, current_length + 1 + ) - def recursive_solve(self, coords, check_for, direction, current_length): - new_position = ( - coords[0] + direction[0], - coords[1] + direction[1] - ) - if self.outside_board(new_position) or self.board[new_position[0]][new_position[1]] != check_for: - return current_length - else: - return self.recursive_solve(new_position, check_for, direction, current_length + 1) + def check_win(self, coords): + directions = [ + (1, 0), (1, 1), (0, 1), (-1, 1), (-1, 0), (-1, -1), (0, -1), + (1, -1) + ] + values = list() + for direction in directions: + values.append( + self.recursive_solve( + coords, self.board[coords[0]][coords[1]], direction, 0 + ) + ) + joined_directions = [ + values[0] + values[4], values[1] + values[5], + values[2] + values[6], values[3] + values[7] + ] + won = any(i >= 3 for i in joined_directions) + if won: + for i, value in enumerate(joined_directions): + if value >= 3: + start_pos = ( + coords[0] + directions[i][0] * values[i], + coords[1] + directions[i][1] * values[i], + ) + end_pos = ( + coords[0] + directions[i + 4][0] * values[i + 4], + coords[1] + directions[i + 4][1] * values[i + 4], + ) + print(start_pos, end_pos) - def check_win(self, coords): - directions = [ - ( 1, 0), - ( 1, 1), - ( 0, 1), - (-1, 1), - (-1, 0), - (-1, -1), - ( 0, -1), - ( 1, -1) - ] - values = list() - for direction in directions: - values.append(self.recursive_solve(coords, self.board[coords[0]][coords[1]], direction, 0)) - joined_directions = [ - values[0] + values[4], - values[1] + values[5], - values[2] + values[6], - values[3] + values[7] - ] - won = any(i >= 3 for i in joined_directions) - if won: - for i, value in enumerate(joined_directions): - if value >= 3: - start_pos = ( - coords[0] + directions[i][0] * values[i], - coords[1] + directions[i][1] * values[i], - ) - end_pos = ( - coords[0] + directions[i+4][0] * values[i+4], - coords[1] + directions[i+4][1] * values[i+4], - ) - print(start_pos, end_pos) + return won - return won + def drop_fisje(self, column, disc): + for row, value in enumerate(self.board): + if self.board[row][column] == EMPTY: + self.board[row][column] = disc + won = self.check_win((row, column)) + print(won) + return - def drop_fisje(self, column, disc): - for row, value in enumerate(self.board): - if self.board[row][column] == EMPTY: - self.board[row][column] = disc - won = self.check_win((row, column)) - print(won) - return def main(): - disc_a = True - gert = bord(11, 8) - while True: - gert.print() - column = int(input("column?: ")) - 1 - os.system("clear") - gert.drop_fisje(column, DISC_A if disc_a else DISC_B) - disc_a = not disc_a + disc_a = True + gert = bord(11, 8) + while True: + gert.print() + column = int(input("column?: ")) - 1 + os.system("clear") + gert.drop_fisje(column, DISC_A if disc_a else DISC_B) + disc_a = not disc_a -if __name__ == "__main__": - main() +if __name__ == "__main__": + main() |