Restart when the ball falls outside the ground

Signed-off-by: Manuel Quiñones <manuq@laptop.org>

1 files changed, 4 insertions, 0 deletions

diff --git a/game.py b/game.py
index 5f2289a..d239014 100755
--- a/game.py
+++ b/game.py
@@ -118,10 +118,14 @@ class TiltGame(object):
             # Move the ball
             if not self._paused:
                 if self.level.is_on_hole(self.ball.get_position()):
+                    # Success! Restart
                     self.ball.set_position(*self.level.get_ball_start())
 
                 if self.level.is_on_ground(self.ball.get_position()):
                     self.ball.update()
+                else:
+                    # Fail! Restart
+                    self.ball.set_position(*self.level.get_ball_start())
 
             # Clear Display
             self._screen.fill(BACKGROUND_COLOR)