Support actor in map

1 files changed, 8 insertions, 1 deletions

diff --git a/pacman-c++/mainwidget.h b/pacman-c++/mainwidget.h
index 4cdb3b8..93af229 100644
--- a/pacman-c++/mainwidget.h
+++ b/pacman-c++/mainwidget.h
@@ -2,9 +2,12 @@
 #define MAINWIDGET_H
 
 #include <QtGui>
+#include <map>
 
 #include "constants.h"
 
+class Actor;
+
 class MainWidget
   : public QWidget
 {
@@ -19,7 +22,11 @@ private:
   // data conversion
   QPoint mapPositionToCoord(unsigned int x, unsigned int y);
 
-  QGraphicsScene *scene;
+  QGraphicsScene *m_scene;
+
+  Actor *getActor(Color);
+  typedef std::map<Color, Actor*> ActorMap;
+  ActorMap m_actors;
 };
 
 #endif // MAINWIDGET_H
\ No newline at end of file