c07_aliengame.py

#!/usr/bin/env python3
"""
@author  Michele Tomaiuolo - https://tomamic.github.io/
@license This software is free - https://opensource.org/license/mit
"""

import random
from actor import Actor, Arena, Point

class Alien(Actor):
    def __init__(self, pos: Point):
        self._x, self._y = pos
        self._w, self._h = 24, 16
        self._dx, self._dy = 4, 8
        # each alien has its own moving space, e.g. 150px
        self._xmin, self._xmax = self._x, self._x + 150
        self._pose = 0  # which sprite to use?

    def move(self, arena: Arena):
        for other in arena.collisions():
            if isinstance(other, Missile):
                arena.kill(self)
        if self._xmin <= self._x + self._dx <= self._xmax:
            self._x += self._dx
        else:
            self._dx = -self._dx
            self._y += self._dy

        chances = 25000 // (1 + arena.count())
        if random.randrange(chances) == 0:
            pos = self._x + self._w / 2, self._y + self._h
            arena.spawn(Bomb(pos))
        self._pose = arena.count() // 8 % 2

    def pos(self) -> Point:
        return self._x, self._y

    def size(self) -> Point:
        return self._w, self._h

    def sprite(self) -> Point:
        if self._pose:
            return 24, 354
        return 0, 354


class Missile(Actor):
    def __init__(self, pos):
        self._w, self._h = 4, 8
        self._x, self._y = pos
        self._x -= self._w / 2
        self._y -= self._h

    def move(self, arena):
        for other in arena.collisions():
            if isinstance(other, Alien):
                arena.kill(self)

        self._y -= 4
        if self._y < 0:
            arena.kill(self)

    def pos(self):
        return self._x, self._y

    def size(self):
        return self._w, self._h

    def sprite(self):
        return 165, 354


class Bomb(Actor):
    def __init__(self, pos):
        self._w, self._h = 8, 16
        self._x, self._y = pos
        self._x -= self._w / 2

    def move(self, arena):
        for other in arena.collisions():
            if isinstance(other, Cannon):
                arena.kill(self)

        aw, ah = arena.size()
        self._y += 4
        if self._y > ah:
            arena.kill(self)

    def pos(self):
        return self._x, self._y

    def size(self):
        return self._w, self._h

    def sprite(self):
        return 53, 338


class Cannon(Actor):
    def __init__(self, pos):
        self._x, self._y = pos
        self._w, self._h = 28, 16

    def move(self, arena):
        for other in arena.collisions():
            if isinstance(other, Bomb):
                arena.kill(self)

        aw, ah = arena.size()
        keys = arena.current_keys()
        prev = arena.previous_keys()
        if "ArrowUp" in keys and "ArrowUp" not in prev:
            pos = self._x + self._w / 2, self._y
            arena.spawn(Missile(pos))

        if "ArrowLeft" in keys:
            self._x -= 4
        elif "ArrowRight" in keys:
            self._x += 4
        self._x = max(self._x, 0)  # clamp
        self._x = min(self._x, aw - self._w)  # clamp

    def pos(self):
        return self._x, self._y

    def size(self):
        return self._w, self._h

    def sprite(self):
        return 153, 364

class AlienGame(Arena):
    def __init__(self, size=(480, 360)):
        super().__init__(size)
        for y in range(4):
            for x in range(8):
                self.spawn(Alien((x * 42, y * 24)))
        self.spawn(Cannon((240, 340)))

    def game_over(self) -> bool:
        cannon = False
        for a in self.actors():
            x, y = a.pos()
            if isinstance(a, Alien) and y > 320:
                return True
            if isinstance(a, Cannon):
                cannon = True
        return not cannon

    def game_won(self) -> bool:
        for a in self.actors():
            if isinstance(a, Alien):
                return False
        return True

class AlienGui:
    def __init__(self):
        self._game = AlienGame()
        g2d.init_canvas(self._game.size())
        g2d.main_loop(self.tick)

    def tick(self):
        sprites = "https://fondinfo.github.io/sprites/invaders.png"
        g2d.set_color((0, 0, 0))
        g2d.draw_rect((0, 0), self._game.size())
        for a in self._game.actors():
            g2d.draw_image(sprites, a.pos(), a.sprite(), a.size())

        if self._game.game_over():
            g2d.alert("Game over")
            g2d.close_canvas()
        elif self._game.game_won():
            g2d.alert("Game won")
            g2d.close_canvas()
        else:
            self._game.tick(g2d.current_keys())  # Game logic

if __name__ == "__main__":
    import g2d
    gui = AlienGui()