warcraft4/Game.py at master · lkmaks/warcraft4 · 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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
import pygame
import sys
from Board import Board
from CPanel import CPanel
from GameState import GameState
from Units import UnitFactory, Unit
from Rules import Rules
from Player import Player


class Game:
    def __init__(self, player1, player2):
        pygame.init()
        pygame.font.init()
        self.player1 = player1
        self.player2 = player2
        self.neutralPlayer = Player('neutral', 10000000000000, 3, 'Neutrals')

        self.factory1 = UnitFactory(player1, self)
        self.factory2 = UnitFactory(player2, self)
        self.factoryNeutral = UnitFactory(self.neutralPlayer, self)

        self.player1.factory = self.factory1
        self.player2.factory = self.factory2
        self.neutralPlayer.factory = self.factoryNeutral

        self.board = Board(self)

        bw, bh = self.board.real_size_with_offsets()

        self.cpanel = CPanel(self, bw, self.board.top_offset, 400, self.board.real_size_with_offsets()[1])
        self.screen = pygame.display.set_mode((bw + self.cpanel.width, bh))
        self.clock = pygame.time.Clock()

        self.gamestate = GameState(player1, self)

        self.rules = Rules(self)

        self.render_mode = 'normal'

        self.board.generate_neutrals(Rules.NEUTRAL_GENERATING_STRATEGY())

    def render(self):
        self.board.render()
        self.cpanel.render()

    def endmove(self):
        if isinstance(self.gamestate.chosen_unit, Unit):
            self.gamestate.chosen_unit.chosen = False
        self.gamestate.chosen_unit = None
        self.gamestate.state = 0
        self.board.endmove()
        self.gamestate.player = self.player1 if self.gamestate.player == self.player2 else self.player2
        self.cpanel.endmove()
        self.player1.gold += self.rules.gold_add(self.player1)
        self.player2.gold += self.rules.gold_add(self.player2)
        self.neutralPlayer.act()

        res = self.rules.game_result()
        if res != 0:
            print('Player {} win!!!'.format(res))
            exit(0)

    def handle_keyup(self, key):
        zero = self.gamestate.state == 0
        if key == ord('e'):
            self.endmove()
        elif key == 308 or key == ord('q'):
            self.gamestate.default()
        elif key == ord('f') and zero:
            self.try_choose_unit_to_train('footman')
        elif key == ord('g') and zero:
            self.try_choose_unit_to_train('grunt')
        elif key == ord('r') and zero:
            self.try_choose_unit_to_train('rifleman')
        elif key == ord('h') and zero:
            self.try_choose_unit_to_train('headhunter')
        elif key == ord('p') and zero:
            self.try_choose_unit_to_train('priest')
        elif key == ord('b') and zero:
            self.try_choose_unit_to_train('bomber')
        elif key == ord('c') and zero:
            self.gamestate.state = 3
        elif key == ord('s') and zero:
            if self.render_mode == 'normal':
                self.render_mode = 'subordination'
            else:
                self.render_mode = 'normal'

    def try_choose_unit_to_train(self, name):
        if self.gamestate.player.factory.affordable(name):
            self.gamestate.state = 2
            # gamestate 2: ready to drop unit
            self.gamestate.chosen_unit = name

    def handle_mouseup(self, pos, mouse_button):
        cell = self.board.get_cell(pos)
        button = self.cpanel.get_button(pos)

        if self.gamestate.state == 0:
            if not cell is None and mouse_button == 1:
                unit = self.board.units_array[cell[0]][cell[1]]
                if not unit is None and unit.owner == self.gamestate.player:
                    self.gamestate.state = 1
                    self.gamestate.chosen_unit = unit
                    unit.chosen = True
            elif not button is None:
                if button == "endmove":
                    self.endmove()
                elif button.startswith('train_'):
                    name = button[6:]
                    self.try_choose_unit_to_train(name)
            else:
                self.gamestate.default()
        elif self.gamestate.state == 1:
            if not cell is None and mouse_button == 3:
                unit = self.gamestate.chosen_unit
                if unit.able(cell):
                    unit.take_action(cell)
            elif not cell is None and mouse_button == 1:
                unit = self.board.units_array[cell[0]][cell[1]]
                if not unit is None and unit.owner == self.gamestate.player:
                    self.gamestate.chosen_unit.chosen = False
                    self.gamestate.chosen_unit = unit
                    unit.chosen = True
                else:
                    self.gamestate.default()
            elif not button is None and mouse_button == 1:
                if button == 'endmove':
                    self.endmove()
                elif button.startswith('train_'):
                    name = button[6:]
                    self.try_choose_unit_to_train(name)
            else:
                self.gamestate.default()
        elif self.gamestate.state == 2:
            if not cell is None and mouse_button == 1:
                unit = self.board.units_array[cell[0]][cell[1]]
                if self.gamestate.player.factory.creatable(self.gamestate.chosen_unit, cell):
                    self.gamestate.player.factory.create_unit(self.gamestate.chosen_unit, cell)
                elif not unit is None and unit.owner == self.gamestate.player:
                    self.gamestate.chosen_unit = unit
                    unit.chosen = True
                    self.gamestate.state = 1
                else:
                    self.gamestate.default()
            elif not button is None and mouse_button == 1:
                if button == 'endmove':
                    self.endmove()
                elif button.startswith('train_'):
                    name = button[6:]
                    self.try_choose_unit_to_train(name)
            else:
                self.gamestate.default()
        elif self.gamestate.state == 3:
            # state of choosing first (child) unit for bounding
            if not cell is None and mouse_button == 1:
                unit = self.board.units_array[cell[0]][cell[1]]
                if not unit is None and unit.owner == self.gamestate.player:
                    self.gamestate.state = 4
                    self.gamestate.chosen_unit = unit
                    unit.chosen_to_bound = True
                else:
                    self.gamestate.default()
            else:
                self.gamestate.default()
        elif self.gamestate.state == 4:
            if not cell is None and mouse_button == 1:
                unit = self.board.units_array[cell[0]][cell[1]]
                if not unit is None and unit.owner == self.gamestate.player and\
                        self.gamestate.chosen_unit.able_to_set_parent(unit):
                    self.gamestate.chosen_unit.set_parent(unit)
                self.gamestate.default()
            else:
                self.gamestate.default()

    def run(self):
        while True:
            self.clock.tick(50)
            for event in pygame.event.get():
                if event.type == pygame.QUIT:
                    sys.exit()
                elif event.type == pygame.MOUSEBUTTONUP:
                    try:
                        self.handle_mouseup(event.pos, event.button)
                    except Exception:
                        print('Error =(')
                elif event.type == pygame.KEYUP:
                    try:
                        self.handle_keyup(event.key)
                    except Exception:
                        print("Error =(")
            self.render()
            pygame.display.flip()