1 files changed, 10 insertions, 0 deletions
diff --git a/src/game/GameModel.h b/src/game/GameModel.h
index c709535..753e19d 100644
--- a/src/game/GameModel.h
+++ b/src/game/GameModel.h
@@ -2,9 +2,11 @@
 #define GAMEMODEL_H
 
 #include <vector>
+#include "search/Save.h"
 #include "simulation/Simulation.h"
 #include "Renderer.h"
 #include "GameView.h"
+#include "Brush.h"
 
 using namespace std;
 
@@ -16,20 +18,28 @@ class GameModel
 {
 private:
 	vector<GameView*> observers;
+	Brush * currentBrush;
+	Save * currentSave;
 	Simulation * sim;
 	Renderer * ren;
 	int activeElement;
 	void notifyRendererChanged();
 	void notifySimulationChanged();
 	void notifyPausedChanged();
+	void notifySaveChanged();
+	void notifyBrushChanged();
 public:
 	GameModel();
 	~GameModel();
+	Save * GetSave();
+	Brush * GetBrush();
+	void SetSave(Save * newSave);
 	void AddObserver(GameView * observer);
 	int GetActiveElement();
 	void SetActiveElement(int element);
 	bool GetPaused();
 	void SetPaused(bool pauseState);
+	void ClearSimulation();
 
 	Simulation * GetSimulation();
 	Renderer * GetRenderer();