Added lives to the HUD

3 files changed, 12 insertions, 3 deletions

diff --git a/src/Game.hs b/src/Game.hs
index a7454b8..bc604ad 100644
--- a/src/Game.hs
+++ b/src/Game.hs
@@ -26,6 +26,9 @@ gameWidth, gameHeight :: CInt
 gameScale :: CInt
 gameScale = 3
 
+maxLives :: Int
+maxLives = 4
+
 windowWidth, windowHeight :: CInt
 (windowWidth, windowHeight) = (gameWidth * gameScale, gameHeight * gameScale)
 
@@ -77,7 +80,9 @@ main = do
     newIORef
       GS.State
         { batteries = 0,
-          totalBatteries = M.totalBatteries map'
+          totalBatteries = M.totalBatteries map',
+          lives = maxLives,
+          totalLives = maxLives
         }
   hud <- H.mkHud sprites state
   entities <- newIORef =<< E.mkEntities sprites map' controls state
diff --git a/src/Game/Hud.hs b/src/Game/Hud.hs
index 3b59558..dd72bb3 100644
--- a/src/Game/Hud.hs
+++ b/src/Game/Hud.hs
@@ -19,4 +19,6 @@ render :: SDL.Renderer -> Hud -> IO ()
 render renderer hud = do
   state <- readIORef hud.stateRef
   let xs = [0 .. state.totalBatteries - 1]
-  mapM_ (\x -> S.render renderer hud.sprite (4 + x * 8) 178 0 (if state.batteries <= x then 0 else 1)) xs
+   in mapM_ (\x -> S.render renderer hud.sprite (4 + x * 8) 178 0 (if state.batteries <= x then 0 else 1)) xs
+  let xs = [0 .. state.totalLives - 1]
+   in mapM_ (\x -> S.render renderer hud.sprite (320 - 4 - state.totalLives * 8 + x * 8) 178 0 (if state.lives <= x then 2 else 3)) xs
diff --git a/src/Game/State.hs b/src/Game/State.hs
index 9988936..2223434 100644
--- a/src/Game/State.hs
+++ b/src/Game/State.hs
@@ -2,5 +2,7 @@ module Game.State (State (..)) where
 
 data State = State
   { batteries :: Int,
-    totalBatteries :: Int
+    totalBatteries :: Int,
+    lives :: Int,
+    totalLives :: Int
   }