game.py

#!/usr/bin/python
# -*- coding: <utf-8> -*-
import itertools
import sys

from pygame import Surface
from pygame.locals import MOUSEMOTION, KEYDOWN, KEYUP, QUIT

from entities import *


class Game:
    def __init__(self):
        pygame.init()
        self.screen = self.__init_screen()
        resourceloader.init_resource_loader()
        self.player = Player(np.array((250.0, 250.0)))
        self.players = pygame.sprite.RenderPlain(self.player)
        self.destroits = pygame.sprite.Group()
        self.bullets = pygame.sprite.Group()
        self.clock = pygame.time.Clock()
        self.__pressed_keys = set()
        self.player_score = 0

    def __init_screen(self):
        screen = pygame.display.set_mode(GAMESIZE)
        pygame.display.set_caption("DESTROITS")

        # Fill background
        background = pygame.Surface(screen.get_size())
        background = background.convert()
        background.fill(BLACK)
        self.background = background

        # Blit everything to the screen
        screen.blit(background, (0, 0))
        pygame.display.flip()

        return screen

    def start(self):
        # Event loop
        while True:
            self.clock.tick(FPS)
            self.process_input()
            self.update()
            self.render()

    def process_input(self):
        pygame.event.pump()
        for event in pygame.event.get():
            if event.type == QUIT:
                sys.exit(0)
            elif event.type == KEYUP or event.type == KEYDOWN:
                if event.type == KEYDOWN:
                    self.__pressed_keys.add(event.key)
                else:
                    self.__pressed_keys.remove(event.key)


            elif event.type == MOUSEMOTION:
                self.player.mouse_position = np.array(event.pos)

    def render(self):
        self.screen.blit(self.background, (0, 0))
        self.players.draw(self.screen)
        self.destroits.draw(self.screen)
        self.bullets.draw(self.screen)
        self.blit_playerscore()

        if DEBUG:
            for sprite in itertools.chain(self.players.sprites(), self.destroits.sprites(), self.bullets.sprites()):
                textsurface = DBG_FONT.render(np.array2string(sprite.pos, precision=1), False, WHITE)
                self.screen.blit(textsurface, (sprite.pos))
        pygame.display.flip()

    def update(self):
        self.players.update()
        self.destroits.update()
        self.bullets.update()

        for key in self.__pressed_keys:
            if key in self.player.KEY_MAPPING.keys():
                getattr(self.player, self.player.KEY_MAPPING[key])()

        collisions = pygame.sprite.groupcollide(self.bullets, self.destroits, True, True)
        for asteroids in collisions.values():
            for asteroid in asteroids:
                self.player_score += asteroid.points
        lost = pygame.sprite.groupcollide(self.destroits, self.players, False, True)
        if lost:
            self.handle_loose()
        self.spawn_destroits()

        if self.player.can_shoot():
            if CAN_SHOOT_ANYWHERE:
                rel_pos = (self.player.mouse_position - self.player.pos)
                direction = (rel_pos / np.linalg.norm(rel_pos))
            else:
                direction = self.player.direction
            bullet = Bullet(self.player.pos, direction)
            self.bullets.add(bullet)

    def spawn_destroits(self):
        if random() < Asteroid.SPAWN_CHANCE:
            self.destroits.add(Asteroid(player_pos=self.player.pos))

    def blit_playerscore(self):
        textsurface = GAME_FONT.render("SCORE: {}".format(self.player_score), False, WHITE)
        self.screen.blit(textsurface, (0, 0))

    def handle_loose(self):
        pass