Refactored a wee bit of code in the GameStack... stack, and started building the title screen code

Makefile

@@ -4,7 +4,7 @@ PROJECT=treewars
 CXX=g++
 CXXFLAGS=-DDEBUG -g `sdl-config --cflags`
 LDFLAGS=`sdl-config --libs`
-OBJECTS=drawutils.o game.o graph.o main.o mainevent.o mathutils.o gamedata.o
+OBJECTS=drawutils.o game.o graph.o main.o mainevent.o mathutils.o gamedata.o gamestate.o
 
 all: $(PROJECT)
 

game.cpp

@@ -20,15 +20,6 @@ Game::~Game()
 }
 
 
-void Game::execute(stack<GameState*> &state_stack) throw(StateExit)
-{
-    SDL_Event event;
-    while(SDL_PollEvent(&event))
-	handle_event(&event);
-    render();
-}
-
-
 bool Game::init()
 {
     background = DrawUtils::load("background.bmp");

game.h

@@ -21,9 +21,11 @@ class Game : public GameState
     ~Game();
 
     bool init();
-    void execute(stack<GameState*> &state_stack) throw(StateExit);
 
  protected:
+    void render();
+    void iterate() {}
+
     // event handlers
     void on_exit();
     void on_lbutton_down(int x, int y);
@@ -31,7 +33,6 @@ class Game : public GameState
     void on_key_down(SDLKey sym, SDLMod mod, Uint16 unicode);
 
  private:
-    void render();
 
     // data
     GameData data;

gamestate.cpp

@@ -0,0 +1,10 @@
+#include "gamestate.h"
+
+void GameState::execute(stack<GameState*> &state_stack) throw(StateExit)
+{
+    SDL_Event event;
+    while(SDL_PollEvent(&event))
+	handle_event(&event);
+    iterate();
+    render();
+}

gamestate.h

@@ -20,12 +20,15 @@ class GameState : public MainEvent
 {
  public:
     GameState(SDL_Surface* display) { this->display = display; }
-    ~GameState() {}
+    virtual ~GameState() {}
 
     virtual bool init() = 0;
-    virtual void execute(stack<GameState*> &game_state) throw(StateExit) = 0;
+    void execute(stack<GameState*> &game_state) throw(StateExit);
 
   protected:
+    virtual void iterate() = 0;
+    virtual void render() = 0;
+    
     SDL_Surface* display;
 };
 

titlescreen.cpp

@@ -0,0 +1,30 @@
+#include "titlescreen.h"
+
+TitleScreen::TitleScreen(SDL_Surface* display)
+    : GameState(display)
+{
+    background = NULL;
+    title_banner = NULL;
+}
+
+
+GameState::~GameState()
+{
+    if (background != NULL)
+	SDL_FreeSurface(background);
+
+    if (background != NULL)
+	SDL_FreeSurface(title_banner);
+}
+
+
+bool TitleScreen::init()
+{
+
+}
+
+
+void TitleScreen::execute(stack<TitleScreen*> &game_state) throw(StateExit)
+{
+
+}

titlescreen.h

@@ -0,0 +1,27 @@
+/*
+  This renders and handles the title screen / main menu
+ */
+
+#ifndef _TITLE_SCREEN_H_
+#define _TITLE_SCREEN_H_
+
+#include "gamestate.h"
+
+class TitleScreen : public GameState
+{
+ public:
+    GameState(SDL_Surface* display);
+    ~GameState();
+
+    bool init();
+    void execute(stack<GameState*> &game_state) throw(StateExit);
+
+  protected:
+    SDL_Surface* display;
+
+    SDL_Surface* background;
+    SDL_Surface* title_banner;
+};
+
+
+#endif