import arcade import json import math from fill_level import filler SCREEN_WIDTH = 800 SCREEN_HEIGHT = 600 class body(): def __init__(self, x, y, rotate, type): self.x = x self.y = y self.rotate = rotate self.type = type class SNAKE(): def __init__(self, star_point_x, star_point_y, start_rotate = 90): self.body = [ body(star_point_x, star_point_y, start_rotate, "head"), body(star_point_x - 1, star_point_y, start_rotate, "tail") ] def move(self, vector): self.body[0].rotate = vector * 90 i = 0 while i < len(self.body): if self.body[i].type == "body" or self.body[i].type == "tail": self.body[i].x = self.body[i-1].x self.body[i].y = self.body[i-1].y self.body[i].rotate = self.body[i-1].rotate i = i + 1 a = int(math.cos(-self.body[0].rotate * math.pi / 180)) self.body[0].x = int(-math.sin(-self.body[0].rotate * math.pi / 180)) + self.body[0].x self.body[0].y = int(math.cos(-self.body[0].rotate * math.pi / 180)) + self.body[0].y # print(f"Move {vector}") class Level_data(): def __init__(self, level_data_json = []): if level_data_json: self.fill_grid(level_data_json) def fill_grid(self, level_data_json): self.name = level_data_json["name"] self.floor = level_data_json["floor"] self.size = level_data_json["size"] self.enemy_types = level_data_json["enemy_types"] self.enemy_spawns = level_data_json["enemy_spawns"] self.start_point = level_data_json["start_point"] class MYGAME(arcade.Window): Sprites = [] def __init__(self, width, height): super().__init__(width, height) arcade.set_background_color(arcade.color.AMAZON) def setup(self, level_name): print(f"Upload level: {level_name}") with open(f'./DATA/Levels/{level_name}/data.json', 'r') as file: self.Lvl_data = Level_data(json.load(file)) if self.Lvl_data.name : print(f"{self.Lvl_data.name} loaded") else: print("Error to load level") self.player_list = arcade.SpriteList() self.floor_list = arcade.SpriteList() self.eat_list = arcade.SpriteList() self.enemies_list = arcade.SpriteList() self.enemy_spanw = arcade.SpriteList() grid = fill_empty_grid() self.floor_grid = self.arr2grid(self.Lvl_data.floor, grid) self.print_grid(self.floor_grid) self.snake = SNAKE(self.Lvl_data.start_point["x"], self.Lvl_data.start_point["y"]) # Счет self.score = 0 self.scale_ = 1.3 screen_center = [SCREEN_WIDTH / 2, SCREEN_HEIGHT / 2] self.level_start = [screen_center[0] - self.Lvl_data.size[0] * 16 * self.scale_, screen_center[1] - self.Lvl_data.size[1] * 16 * self.scale_] # Загрузка змеи for body in self.snake.body: snake_sprite = arcade.Sprite(f"./DATA/Sprites/Snake/{body.type}.png", self.scale_) snake_sprite.angle = 90 snake_sprite.center_x = self.level_start[0] - 16 + ( body.x + 1 ) * 32 * self.scale_ snake_sprite.center_y = self.level_start[1] - 16 + ( body.y + 1 ) * 32 * self.scale_ self.player_list.append(snake_sprite) # Загрузка пола for floor in self.Lvl_data.floor: floor_sprite = arcade.Sprite(floor["sprite"], self.scale_) floor_sprite.center_x = self.level_start[0] - 16 + ( floor["x"] + 1 ) * 32 * self.scale_ floor_sprite.center_y = self.level_start[1] - 16 + ( floor["y"] + 1 ) * 32 * self.scale_ self.floor_list.append(floor_sprite) # Загрузка врагов # if "enemies" in self.level_data: # for enemy in self.level_data["enemies"]: # enemy_sprite = arcade.Sprite(enemy["sprite"], 1) # enemy_sprite.center_x = enemy["x"] * 32 # enemy_sprite.center_y = enemy["y"] * 32 # self.enemies_list.append(enemy_sprite) def update(self): if len(self.snake.body) > 0: i = 0 while i < len(self.snake.body): self.player_list[i] = arcade.Sprite(f"./DATA/Sprites/Snake/{self.snake.body[i].type}.png", self.scale_) self.player_list[i].angle = self.snake.body[i].rotate self.player_list[i].center_x = self.level_start[0] - 16 + ( self.snake.body[i].x + 1 ) * 32 * self.scale_ self.player_list[i].center_y = self.level_start[1] - 16 + ( self.snake.body[i].y + 1 ) * 32 * self.scale_ i = i + 1 def print_grid(self, grid): x = 0 y = 15 print_ = '' while y > 0: while x < 15: print_ += " " if grid[y][x] == 0 else "1 " x += 1 print(print_) print_ = '' y -= 1 x = 0 def arr2grid(self, arr, grid): for point in arr: grid[point["y"]][point["x"]] = 1 return grid # Обработка нажатий MOVEMENT_SPEED = 1 def on_key_press(self, key, modifiers): """Вызывается при нажатии пользователем клавиши""" # Get the first sprite (head) from the player list if len(self.player_list) > 0: if key in (arcade.key.UP, arcade.key.W): if self.floor_grid[self.snake.body[0].y + 1][self.snake.body[0].x] == 1 and self.snake.body[0].rotate != 180: self.snake.move(0) elif key in (arcade.key.DOWN, arcade.key.S): if self.floor_grid[self.snake.body[0].y - 1][self.snake.body[0].x] == 1 and self.snake.body[0].rotate != 0: self.snake.move(2) elif key in (arcade.key.LEFT, arcade.key.A): if self.floor_grid[self.snake.body[0].y][self.snake.body[0].x - 1] == 1 and self.snake.body[0].rotate != 90: self.snake.move(3) elif key in (arcade.key.RIGHT, arcade.key.D): if self.floor_grid[self.snake.body[0].y][self.snake.body[0].x + 1] == 1 and self.snake.body[0].rotate != 270: self.snake.move(1) def on_key_release(self, key, modifiers): """Вызывается, когда пользователь отпускает клавишу""" if len(self.player_list) > 0: self.player_sprite = self.player_list[0] if key in (arcade.key.UP, arcade.key.DOWN, arcade.key.W, arcade.key.S): self.player_sprite.change_y = 0 elif key in (arcade.key.LEFT, arcade.key.RIGHT, arcade.key.A, arcade.key.D): self.player_sprite.change_x = 0 def on_draw(self): self.update() # Очищаем экран перед каждой отрисовкой self.clear() # Отрисовываем все спрайты self.floor_list.draw() self.enemies_list.draw() self.eat_list.draw() self.player_list.draw() def fill_empty_grid(): grid = [] y = 0 x = 0 while y < 15: grid.append([]) while x < 15: grid[y].append([x]) grid[y][x] = 0 x = x + 1 y = y + 1 x = 0 return grid def main(): filler_ = filler("Default") filler_.fill_level(filler_.floor_code) filler_.export2json() game = MYGAME(SCREEN_WIDTH, SCREEN_HEIGHT) game.setup( "Default" ) arcade.run() if __name__ == "__main__": main()