1 files changed, 6 insertions, 8 deletions

diff --git a/pacman-c++/sceneholder.h b/pacman-c++/sceneholder.h
index ccb2a42..61cff3e 100644
--- a/pacman-c++/sceneholder.h
+++ b/pacman-c++/sceneholder.h
@@ -1,30 +1,30 @@
 #ifndef SCENEHOLDER_H
 #define SCENEHOLDER_H
 
-#include <QtGui>
-
 #include "constants.h"
+#include <QtGui>
 
 class PixmapItem;
 class Actor;
 
 class SceneHolder
-  : public QWidget
+  : public QGraphicsScene
 {
   Q_OBJECT
 
 public:
-  SceneHolder(QWidget* parent = 0);
+  SceneHolder(QObject *parent = 0);
   virtual ~SceneHolder()
   {};
   unsigned int pointsLeft();
+  void updateMap(const Transmission::map_t& map);
+  void setColor(Color::Color color = Color::none);
+  Color::Color color();
 
 private slots:
   void decrementPoints();
 
 protected:
-  virtual void updateMap(const Transmission::map_t& map);
-
   /* data conversion */
   QPoint mapPositionToCoord(unsigned int x, unsigned int y);
   QPoint mapPositionToCoord(QPoint point);
@@ -42,8 +42,6 @@ protected:
 
   /* points left before round ends */
   unsigned int m_pointsLeft;
-
-  QGraphicsScene *m_scene;
 };
 
 #endif // SCENEHOLDER_H