runner-and-chasers-game/game.py at main · bersoy12/runner-and-chasers-game · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
from agent import *
import pygame
from constants import *
from board import Board
import random as rnd


class Game:
    def __init__(self, window):
        self._init()
        self.window = window
        self.moves_list = {}
        self.round = 0
        self.win = None


    def update(self):
        self.board.draw(self.window)
        self.draw_valid_moves(self.valid_moves)
        pygame.display.update()


    def _init(self):
        self.selected = None
        self.board = Board()
        self.turn = self.board.agent_list[0]
        self.valid_moves = {}


    def reset(self):
        self._init()

    def select(self, row, col):
        if self.selected:
            result = self._move(row, col)
            if not result:
                self.selected = None
                self.select(row, col)

        agent = self.board.get_agent(row, col)

        if agent != 0 and agent == self.turn:
            self.selected = agent
            self.valid_moves = self.board.get_valid_moves(agent)
            return True

        return False


    def winner(self):
        return self.board.winner()


    def collision(self, row, col):
        agent = self.board.get_agent(row, col)
        print("Runner is caught by {}".format(agent.name))


    def _move(self, row, col):
        agent = self.board.get_agent(row, col)
        if self.selected and agent == 0 and (row, col) in self.valid_moves.values():
            self.board.move(self.selected, row, col)
            self.change_turn()
        elif self.selected and agent != 0 and (row, col) in self.valid_moves.values():
            self.collision(row, col)
            return self.winner()
        else:
            return False
        return True


    def draw_valid_moves(self, moves):
        for r, c in moves.values():
            row, col = r, c
            pygame.draw.circle(self.window, BLACK, (col*SQUARE_SIZE + SQUARE_SIZE//2, row*SQUARE_SIZE + SQUARE_SIZE//2), 5)


    def change_turn(self):
        self.valid_moves = {}
        self.board.agent_list = []
        for agent in self.board.get_all_agents():
            self.board.agent_list.append(agent)

        if self.turn.name == "C2":
            self.turn = self.board.agent_list[0]
        elif self.turn.name == "R":
            self.turn = self.board.agent_list[1]
        else:
            self.turn = self.board.agent_list[2]


    def get_board(self):
        return self.board

    def ai_move(self, new_board):
        self.board = new_board
        """
        moves = self.board.get_valid_moves(self.turn)
        row, col = rnd.choice(list(moves.values()))
        self.board.move(self.turn, row, col)
        """
        self.change_turn()