Add replay mode

1 files changed, 53 insertions, 8 deletions

diff --git a/turtetris_master/state_machine.py b/turtetris_master/state_machine.py
index 7cbfcaa..e0cc6ee 100644
--- a/turtetris_master/state_machine.py
+++ b/turtetris_master/state_machine.py
@@ -1,5 +1,10 @@
 from .screen_checker import ScreenChecker
 from .game import Game
+from .recorder import Recorder
+from .recorder import Replayer
+
+# We run 1/30 loops per second so this gives us one minute delay
+SWITCH_TIME_DELAY = 1800
 
 
 class StateMachine:
@@ -10,21 +15,43 @@ class StateMachine:
         self.state = "initializing"
         self.matrix = matrix
         self.input = input
-        self.__update_state__('screen_checker')
+        self.game = None
+        self.recorder = None
+        self.replayer = None
+        self.timeout = SWITCH_TIME_DELAY
+        self.__update_state__('screen-checker')
+
+    def __new_game__(self):
+        "Initialize new game"
+        self.game = Game(self.matrix)
+        self.recorder = Recorder(self.matrix)
+
+    def __reset_delay__(self):
+        self.timeout = SWITCH_TIME_DELAY
+
+    def __delay__(self):
+        if self.timeout > 0:
+            self.timeout -= 1
+            return False
+        else:
+            self.__reset_delay__()
+            return True
 
     def __update_state__(self, state):
         "Applies given state"
         def __exception__():
-            raise Exception('Can\'t transfer from ' + self.state + ' to '
-                            + state)
-        if state == "screen_checker":
+            raise Exception('Can\'t transfer from ' + self.state + ' to ' +
+                            state)
+        if state == "screen-checker":
             if self.state == "initializing":
                 self.screen_checker = ScreenChecker(self.matrix)
             else:
                 __exception__()
         elif state == "game":
-            if self.state == "screen_checker" or self.state == "game-over":
-                self.game = Game(self.matrix)
+            if self.state == "screen-checker" or \
+                    self.state == "game-over" or \
+                    self.state == "replay":
+                self.__new_game__()
             elif self.state == "game-pause":
                 pass
             else:
@@ -35,6 +62,8 @@ class StateMachine:
         elif state == "game-over":
             if self.state != "game":
                 __exception__()
+            self.__reset_delay__()
+            self.recorder.store(self.game.score)
             self.matrix.fill('red')
             i = self.matrix.height - 2
             # Show score
@@ -45,6 +74,10 @@ class StateMachine:
             for s in range(self.game.score % 5):
                 self.matrix.pixel(s + 3, i, 'green')
             self.matrix.display()
+        elif state == "replay":
+            if self.state != "game-over" and self.state != "screen-checker":
+                __exception__()
+            self.replayer = Replayer(self.matrix)
         else:
             __exception__()
         self.state = state
@@ -52,12 +85,17 @@ class StateMachine:
     def tick(self):
         "Do tick"
         inpt = self.input.check()
-        if self.state == "screen_checker":
+        if self.state == "screen-checker":
             if inpt['start']:
                 self.__update_state__('game')
+            if inpt['select']:
+                self.__update_state__('replay')
             else:
                 self.screen_checker.tick()
         elif self.state == "game":
+            # Note: This records previous tick output but that doens't matter
+            # as loosing latest frame is harmless
+            self.recorder.tick()
             if inpt['start']:
                 self.__update_state__('game-pause')
             elif not self.game.tick(inpt):
@@ -66,10 +104,17 @@ class StateMachine:
             if inpt['start']:
                 self.__update_state__('game')
             elif inpt['select']:
-                self.game = Game(self.matrix)
+                self.__new_game__()
                 self.__update_state__('game')
         elif self.state == "game-over":
             if inpt['start']:
                 self.__update_state__('game')
+            elif self.__delay__():
+                self.__update_state__('replay')
+        elif self.state == "replay":
+            if inpt['start']:
+                self.__update_state__('game')
+            elif not self.replayer.tick():
+                self.replayer = Replayer(self.matrix)
         else:
             raise Exception('Invalid state ' + self.state)