1 files changed, 9 insertions, 1 deletions
diff --git a/include/graphics.hpp b/include/graphics.hpp
index 69ac808..0990247 100644
--- a/include/graphics.hpp
+++ b/include/graphics.hpp
@@ -2,15 +2,20 @@
 # define GRAPHICS_HPP
 
 #include <string>
+#include <iostream>
 #include <SDL2/SDL.h>
+#include <SDL2/SDL_ttf.h>
+#include "game.hpp"
 
 class Graphics
 {
     public:
-    Graphics(std::string t, int w, int h);
+    Graphics(Game *g, std::string t, int w, int h);
     ~Graphics();
 
     void update();
+    void drawGame();
+    void drawCell(int x, int y);
     bool isRunning();
 
     private:
@@ -18,10 +23,13 @@ class Graphics
     std::string title;
     int width;
     int height;
+    Game *game;
     SDL_Renderer *renderer;
     SDL_Window *window;
+    TTF_Font *font;
 
     void handleEvent();
+    void error();
 };
 
 #endif