DefenseGame/enemy.py at master · InkasRas/DefenseGame · 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
import pygame
from need_fncts import load_image
from all_variables import *
from a_star import find_path


class Enemy(pygame.sprite.Sprite):
    def __init__(self, x, y, lvl, castle, game):
        super().__init__()
        self.x = x
        self.y = y
        self.board_x = self.x // CELL_SIZE
        self.board_y = self.y // CELL_SIZE
        self.lvl = lvl
        self.health = enemy_lvls[lvl][0]
        self.max_health = enemy_lvls[lvl][0]
        self.force = enemy_lvls[lvl][1]
        self.run_images = []
        for i in range(7):
            self.run_images.append(pygame.transform.scale(load_image(f'enemy/_RUN/_RUN_00{i}.png', -1), (40, 40)))
        self.attack_images = []
        for i in range(7):
            self.attack_images.append(
                pygame.transform.scale(load_image(f'enemy/_ATTACK/_ATTACK_00{i}.png', -1), (40, 40)))
        self.image = self.run_images[0]
        self.img_id = 0
        self.castle = castle
        self.on_field = False
        self.radius = 10
        self.active = True
        self.game = game
        self.is_attack = False
        self.on_wall_archer = False
        self.n_x, self.n_y = None, None
        self.rect = pygame.Rect((x, y), (40, 40))
        if self.board_x > self.castle.board_x:
            self.img_flip = True
        else:
            self.img_flip = False
        print('ENENMR', self.board_x, self.board_y)

    def get_board_pos(self):
        return self.board_x, self.board_y

    def hurt(self, f):
        self.health -= f
        if self.health <= 0:
            self.active = False

    def break_castle(self):
        return self.castle.hurt(self.force)

    def update_pos(self):
        print('update_pos', self.castle.board_x, self.castle.board_y, self.board_x, self.board_y)
        if self.board_x > self.castle.board_x:
            self.img_flip = True
        else:
            self.img_flip = False
        if self.is_attack is True:
            a = self.break_castle()
            print(a)
        if self.path is not None:
            if len(self.path) == 0:
                self.is_attack = True
            else:
                new_pos = self.path.pop(0)
                self.board_x = new_pos[1]
                self.board_y = new_pos[0] - 1
                self.x = self.board_x * CELL_SIZE
                self.y = self.board_y * CELL_SIZE
                self.rect.x = self.x
                self.rect.y = self.y
            return False
        else:
            if not self.on_wall_archer:
                if self.castle.board_x <= self.board_x <= self.castle.board_x + 10:
                    if pygame.sprite.collide_rect(self, self.castle):
                        self.is_attack = True
                    if self.board_y <= self.castle.board_y:
                        self.board_y += 1
                        self.n_x, self.n_y = 0, 1
                        self.y = self.board_y * CELL_SIZE
                        self.rect.y = self.y
                    elif self.board_y >= self.castle.board_y + 10:
                        self.board_y -= 1
                        self.n_x, self.n_y = 0, -1
                        self.y = self.board_y * CELL_SIZE
                        self.rect.y = self.y
                if self.board_x <= self.castle.board_x:
                    self.board_x += 1
                    self.n_x, self.n_y = 1, 0
                    self.x = self.board_x * CELL_SIZE
                    self.rect.x = self.x
                elif self.board_x >= self.castle.board_x + 10:
                    self.board_x -= 1
                    self.n_x, self.n_y = -1, 0
                    self.x = self.board_x * CELL_SIZE
                    self.rect.x = self.x
                print('TY', self.board_x, self.board_y, self.n_x, self.n_y)
                if self.game.board[self.board_y + self.n_y][self.board_x + self.n_x] == OBSTACLE:
                    self.on_wall_archer = True
                    self.attacking_bjs = pygame.sprite.spritecollide(self, self.game.walls, dokill=False)
                    self.attacking_bjs.extend(pygame.sprite.spritecollide(self, self.game.archers, dokill=False))
                    print('TYEEEEEEEEE', self.attacking_bjs)
                    print('new', self.game.board[self.board_y][self.board_x])
            else:
                for el in self.attacking_bjs:
                    el.hurt(self.force)
                    if el.health <= 0:
                        for i in range(el.rect.w // CELL_SIZE):
                            for j in range(el.rect.h // CELL_SIZE):
                                x = el.board_x + i
                                y = el.board_y + j
                                self.game.board[y][x] = CLEAR
                        el.kill()
                print('on wall ardwe', self.n_x, self.n_y)
                if self.game.board[self.board_y + self.n_y][self.board_x + self.n_x] != OBSTACLE:
                    self.on_wall_archer = False

    def change_pos(self, x, y):
        self.x = x
        self.y = y
        self.rect.x = x
        self.rect.y = y

    def draw(self, srfc):
        if self.img_flip:
            srfc.blit(pygame.transform.flip(self.image, True, False), (self.x + 4, self.y + 4))
        else:
            srfc.blit(self.image, (self.x + 4, self.y + 4))
        rx = round(40 * (self.max_health - self.health) // self.max_health)
        pygame.draw.line(srfc, (255, 0, 0), (self.x, self.y - 2), (self.x + rx, self.y - 2), 2)
        pygame.draw.line(srfc, (0, 255, 0), (self.x + rx, self.y - 2), (self.x + 40, self.y - 2), 2)

    def update_img(self):
        self.img_id += 1 if self.img_id < 6 else -6
        if not self.is_attack:
            self.image = self.run_images[self.img_id]
        else:
            self.image = self.attack_images[self.img_id]

    def get_pos(self):
        return self.x, self.y