forked from s464965/WMICraft
134 lines
4.9 KiB
Python
134 lines
4.9 KiB
Python
import sys
|
|
|
|
import pygame
|
|
|
|
from common.colors import FONT_DARK, WHITE
|
|
from common.constants import GAME_TITLE, WINDOW_WIDTH, WINDOW_HEIGHT, FPS_COUNT, TILES, COLUMNS, KNIGHTS_SPAWN_WIDTH, \
|
|
KNIGHTS_SPAWN_FIRST_ROW, KNIGHTS_SPAWN_HEIGHT, KNIGHTS_SPAWN_FIRST_COL
|
|
from common.helpers import draw_text
|
|
from models.knight import Knight
|
|
from models.monster import Monster
|
|
from ui.logs import Logs
|
|
from ui.screens.credits import Credits
|
|
from ui.screens.options import Options
|
|
from ui.stats import Stats
|
|
from .grid import Grid
|
|
from .spawner import Spawner
|
|
|
|
|
|
class Game:
|
|
def __init__(self):
|
|
pygame.init()
|
|
pygame.display.set_caption(GAME_TITLE)
|
|
pygame.display.set_icon(pygame.image.load('./resources/icons/sword.png'))
|
|
|
|
self.screen = pygame.display.set_mode((WINDOW_WIDTH, WINDOW_HEIGHT))
|
|
self.clock = pygame.time.Clock()
|
|
|
|
self.tiles = []
|
|
for tile_path in TILES:
|
|
converted_tile = pygame.image.load('resources/textures/' + tile_path).convert_alpha()
|
|
self.tiles.append((tile_path, converted_tile))
|
|
|
|
self.bg = pygame.image.load("./resources/textures/bg.jpg")
|
|
self.screens = {'credits': Credits(self.screen, self.clock), 'options': Options(self.screen, self.clock)}
|
|
|
|
click = False
|
|
|
|
def main_menu(self):
|
|
while True:
|
|
self.screen.blit(self.bg, (0, 0))
|
|
|
|
pygame.draw.rect(self.screen, (255, 255, 255), pygame.Rect(800, 100, 400, 500), 0, 5)
|
|
draw_text('MAIN MENU', FONT_DARK, self.screen, 850, 150, 30, True)
|
|
|
|
mx, my = pygame.mouse.get_pos()
|
|
|
|
button_1 = pygame.Rect(850, 250, 300, 50)
|
|
button_2 = pygame.Rect(850, 350, 300, 50)
|
|
button_3 = pygame.Rect(850, 450, 300, 50)
|
|
if button_1.collidepoint((mx, my)):
|
|
if click:
|
|
self.game()
|
|
if button_2.collidepoint((mx, my)):
|
|
if click:
|
|
self.screens['options'].display_screen()
|
|
if button_3.collidepoint((mx, my)):
|
|
if click:
|
|
self.screens['credits'].display_screen()
|
|
pygame.draw.rect(self.screen, (0, 191, 255), button_1, 0, 4)
|
|
draw_text('PLAY', WHITE, self.screen, 870, 255)
|
|
|
|
pygame.draw.rect(self.screen, (0, 191, 255), button_2, 0, 4)
|
|
draw_text('OPTIONS', WHITE, self.screen, 870, 355)
|
|
|
|
pygame.draw.rect(self.screen, (0, 191, 255), button_3, 0, 4)
|
|
draw_text('CREDITS', WHITE, self.screen, 870, 455)
|
|
|
|
click = False
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
pygame.quit()
|
|
sys.exit()
|
|
if event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_ESCAPE:
|
|
pygame.quit()
|
|
sys.exit()
|
|
if event.type == pygame.MOUSEBUTTONDOWN:
|
|
if event.button == 1:
|
|
click = True
|
|
|
|
pygame.display.update()
|
|
self.clock.tick(60)
|
|
|
|
def game(self):
|
|
running = True
|
|
grid = Grid(self.tiles)
|
|
stats = Stats()
|
|
logs = Logs()
|
|
|
|
knights_sprite_group = pygame.sprite.Group()
|
|
|
|
knights_left = self.generate_knights_team(knights_sprite_group)
|
|
knights_right = self.generate_knights_team(knights_sprite_group)
|
|
|
|
spawn_left_team = Spawner(grid, knights_left, width=KNIGHTS_SPAWN_WIDTH, height=KNIGHTS_SPAWN_HEIGHT,
|
|
pos_row=KNIGHTS_SPAWN_FIRST_ROW, pos_column=KNIGHTS_SPAWN_FIRST_COL)
|
|
spawn_right_team = Spawner(grid, knights_right, width=KNIGHTS_SPAWN_WIDTH, height=KNIGHTS_SPAWN_HEIGHT,
|
|
pos_row=KNIGHTS_SPAWN_FIRST_ROW,
|
|
pos_column=COLUMNS - KNIGHTS_SPAWN_WIDTH - 1) # - 1 to avoid list index out of range
|
|
|
|
spawn_left_team.spawn()
|
|
spawn_right_team.spawn()
|
|
|
|
while running:
|
|
self.screen.blit(self.bg, (0, 0))
|
|
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
pygame.quit()
|
|
sys.exit()
|
|
if event.type == pygame.KEYDOWN:
|
|
if event.key == pygame.K_ESCAPE:
|
|
running = False
|
|
|
|
grid.draw(self.screen)
|
|
stats.draw(self.screen)
|
|
logs.draw(self.screen)
|
|
|
|
knights_sprite_group.draw(self.screen)
|
|
monster1 = Monster(self.screen, "./resources/textures/dragon.png")
|
|
monster_list = pygame.sprite.Group()
|
|
monster_list.add(monster1)
|
|
pygame.display.update()
|
|
self.clock.tick(FPS_COUNT)
|
|
|
|
@staticmethod
|
|
def generate_knights_team(knights_sprite_group):
|
|
knights = []
|
|
for i in range(4):
|
|
knight = Knight("./resources/textures/knight.png")
|
|
knights.append(knight)
|
|
knights_sprite_group.add(knight)
|
|
return knights
|