Added lives to the HUD

2 files changed, 6 insertions, 2 deletions

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
   }