add unfinished breakout clone

breakout.py

@@ -0,0 +1,189 @@
+#!/usr/bin/env python3
+
+# ----------------------------------------------------------------------------
+# "THE BEER-WARE LICENSE" (Revision 42):
+# <xythobuz@xythobuz.de> wrote this file.  As long as you retain this notice
+# you can do whatever you want with this stuff. If we meet some day, and you
+# think this stuff is worth it, you can buy me a beer in return.   Thomas Buck
+# ----------------------------------------------------------------------------
+
+from scroll import ScrollText
+import time
+import random
+import math
+
+class Breakout:
+    def __init__(self, g, i, ts = 0.1, to = 60.0):
+        self.gui = g
+        self.input = i
+        self.timestep = ts
+        self.timeout = to
+
+        self.paddle_width = 9
+
+        self.winText = ScrollText(self.gui, "You Won!", "uushi",
+                                  2, 50, (0, 255, 0))
+        self.loseText = ScrollText(self.gui, "Game Over!", "uushi",
+                                   2, 50, (255, 0, 0))
+        self.scoreText = ScrollText(self.gui, "Score:", "uushi",
+                                    2, 50, (255, 255, 255))
+
+        random.seed()
+        self.restart()
+
+    def place(self):
+        self.player = int(self.gui.width / 2)
+        self.ball = [
+            self.player, # x
+            self.gui.height - 2, # y
+            1, # v x
+            -1, # v y
+        ]
+
+    def restart(self):
+        self.start = time.time()
+        self.last = time.time()
+        self.lives = 3
+        self.score = 0
+        self.direction = ""
+
+        self.data = [[(0, 0, 0) for y in range(self.gui.height)] for x in range(self.gui.width)]
+
+        for x in range(self.gui.width - 2):
+            for y in range(5):
+                self.data[x + 1][y] = (0, 255, 0)
+
+        self.place()
+
+        self.old_keys = {
+            "left": False,
+            "right": False,
+            "up": False,
+            "down": False,
+            "a": False,
+            "b": False,
+            "x": False,
+            "y": False,
+            "l": False,
+            "r": False,
+            "start": False,
+            "select": False,
+        }
+
+    def finished(self):
+        if self.input == None:
+            # backup timeout for "AI"
+            if (time.time() - self.start) >= self.timeout:
+                return True
+
+        if self.lives < 0:
+            # game over screen
+            return self.scoreText.finished()
+
+        return False
+
+    def buttons(self):
+        keys = self.input.get()
+
+        if keys["left"] and (not self.old_keys["left"]) and (not self.old_keys["select"]):
+            self.direction = "l"
+        elif keys["right"] and (not self.old_keys["right"]) and (not self.old_keys["select"]):
+            self.direction = "r"
+        elif (keys["select"] and keys["start"] and (not self.old_keys["start"])) or (keys["start"] and keys["select"] and (not self.old_keys["select"])):
+            self.restart()
+
+        self.old_keys = keys.copy()
+
+    def step(self):
+        # TODO check for collisions with pieces
+
+        # move ball
+        self.ball[0] += self.ball[2]
+        self.ball[1] += self.ball[3]
+
+        # check for collision with left wall
+        if self.ball[0] <= 0:
+            self.ball[2] = -self.ball[2]
+
+        # check for collision with right wall
+        if self.ball[0] >= self.gui.width - 1:
+            self.ball[2] = -self.ball[2]
+
+        # check for collision with ceiling
+        if self.ball[1] <= 0:
+            self.ball[3] = -self.ball[3]
+
+        # check for collision with paddle
+        if (self.ball[1] == self.gui.height - 2) and (self.ball[0] >= (self.player - int(self.paddle_width / 2))) and (self.ball[0] <= (self.player + int(self.paddle_width / 2))):
+            # TODO angle
+            self.ball[3] = -self.ball[3]
+
+        # check for collision with floor
+        if self.ball[1] >= self.gui.height - 1:
+            self.place()
+            self.lives -= 1
+
+    def finishedEndScreen(self):
+        if self.score >= self.gui.width * self.gui.height:
+            return self.winText.finished()
+        else:
+            return self.loseText.finished()
+
+    def drawEndScreen(self):
+        if self.score >= self.gui.width * self.gui.height:
+            self.winText.draw()
+        else:
+            self.loseText.draw()
+
+    def drawScoreScreen(self):
+        self.scoreText.draw()
+
+    def draw(self):
+        if self.input != None:
+            self.buttons()
+        else:
+            # TODO "AI"
+            pass
+
+        if self.lives < 0:
+            if self.finishedEndScreen():
+                self.drawScoreScreen()
+            else:
+                self.drawEndScreen()
+                self.scoreText.restart()
+            return
+
+        if self.direction == "l":
+            self.player = max(self.player - 1, 0)
+        elif self.direction == "r":
+            self.player = min(self.player + 1, self.gui.width - 1)
+        self.direction = ""
+
+        now = time.time()
+        if (now - self.last) >= self.timestep:
+            self.last = now
+            self.step()
+
+            if self.lives < 0:
+                self.scoreText.setText("Score: " + str(self.score), "uushi")
+                self.winText.restart()
+                self.loseText.restart()
+                self.scoreText.restart()
+
+        for x in range(0, self.gui.width):
+            for y in range(0, self.gui.height):
+                self.gui.set_pixel(x, y, self.data[x][y])
+
+        for x in range(0, self.paddle_width):
+            self.gui.set_pixel(x + self.player - int(self.paddle_width / 2), self.gui.height - 1, (255, 255, 255))
+
+        self.gui.set_pixel(self.ball[0], self.ball[1], (255, 0, 0))
+
+if __name__ == "__main__":
+    import util
+    # Need to import InputWrapper before initializing RGB Matrix on Pi
+    i = util.getInput()
+    t = util.getTarget(i)
+
+    d = Breakout(t, i)
+    util.loop(t, d.draw)