YearProject/single_mode.py at master · Sanya239/YearProject · 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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
import pygame
import os
import random
import buttons.button_class
import disks.disk_class


class single_game():
    def __init__(self, screen):
        self.screen = screen

        self.desk = []
        self.possibilities = []
        for i in range(8):
            self.desk.append([0] * 8)
        self.disks = pygame.sprite.Group()
        self.player = 1

    def start_a_game(self):
        self.screen.fill((0, 255, 0))
        for i in range(0, 9):
            pygame.draw.line(self.screen, (0, 0, 0), [i * 80 + 80, 80], [i * 80 + 80, 720], 2)
            pygame.draw.line(self.screen, (0, 0, 0), [80, i * 80 + 80], [720, i * 80 + 80], 2)
        disk1 = disks.disk_class.disk(1, (3,3))
        self.disks.add(disk1)
        self.desk[3][3] = disk1

        disk2 = disks.disk_class.disk(2, (4, 3))
        self.disks.add(disk2)
        self.desk[4][3] = disk2

        disk3 = disks.disk_class.disk(2, (3, 4))
        self.disks.add(disk3)
        self.desk[3][4] = disk3

        disk4 = disks.disk_class.disk(1, (4, 4))
        self.disks.add(disk4)
        self.desk[4][4] = disk4
        self.disks.draw(self.screen)
        self.possibilities = [(2, 4), (3, 5), (4, 2), (5, 3)]

    def parcing(self, event):
        pos = event.pos
        if (pos[0] >= 80 and pos[0] <= 720) and (pos[1] >= 80 and pos[1] <= 720):
            pos_x = pos[0] // 80 - 1
            pos_y = pos[1] // 80 - 1
            return (pos_x, pos_y)
        else:
            return 'not desk'

    def update(self, event: pygame.MOUSEBUTTONDOWN):

        if self.player == 2:
            max_points = 0
            for move in self.possibilities:
                points = self.all_directions(move,True,2)
                if points > max_points:
                    best_move = move
            self.turn(best_move)
            self.player = 1
            self.disks.draw(self.screen)

        elif self.parcing(event) != 'not desk':
            cell_X, cell_Y = self.parcing(event)

            if (cell_X, cell_Y) in self.possibilities:
                self.turn((cell_X, cell_Y))

                self.player = 2

                self.disks.draw(self.screen)

    def possibilities_update(self):
        self.possibilities = []
        for x in range(8):
            for y in range(8):
                if self.desk[x][y] == 0:
                    if self.all_directions((x, y), True, 3 - self.player) != 0:
                        self.possibilities.append((x, y))
        print(self.possibilities)

    def all_directions(self, position, checking, player):
        summ = 0
        for dir_x in range(-1, 2):
            for dir_y in range(-1, 2):
                if (dir_x, dir_y) == (0, 0):
                    continue
                if checking == False:
                    self.chain(position, (dir_x, dir_y), checking, player)
                else:
                    summ += self.chain(position, (dir_x, dir_y), checking, player)
        return summ

    def turn(self, position):
        new_disk = disks.disk_class.disk(self.player, position)
        self.disks.add(new_disk)
        self.desk[position[0]][position[1]] = new_disk
        self.all_directions(position, False, self.player)
        self.possibilities_update()

    def chain(self, start_cell, dir, checking, player):
        curr_x = start_cell[0]
        curr_y = start_cell[1]
        sttrclr = player
        line = []
        while ((0 <= curr_x + dir[0] <= 7) and (0 <= curr_y + dir[1] <= 7)):
            curr_x += dir[0]
            curr_y += dir[1]
            if self.desk[curr_x][curr_y] != 0:
                if self.desk[curr_x][curr_y].colour == sttrclr:
                    if checking == False:
                        for disk in line:
                            disk.flip()
                    if checking == True:
                        return len(line)
                    break
                else:
                    line.append(self.desk[curr_x][curr_y])
            else:
                '''if not (curr_x - dir[0] == start_cell[0] and curr_y - dir[1] == start_cell[1]):
                    print('aaaa')'''
                break
        return 0