2 files changed, 13 insertions, 2 deletions

diff --git a/pacman-c++/mainwidget.cpp b/pacman-c++/mainwidget.cpp
index bf049f2..a3acc45 100644
--- a/pacman-c++/mainwidget.cpp
+++ b/pacman-c++/mainwidget.cpp
@@ -56,6 +56,11 @@ Transmission::map_t createDummyMap()
 MainWidget::MainWidget()
   : currentKey(0)
 {
+  visualMap.resize(Constants::map_size.width);
+  for (int i=0; i<visualMap.size(); ++i) {
+    visualMap[i].resize(Constants::map_size.height);
+  }
+
   createGui();
   updateMap(createDummyMap());
 }
@@ -133,7 +138,7 @@ void MainWidget::updateMap(const Transmission::map_t& map)
       Color::Color color = static_cast<Color::Color>(cur & Transmission::color_mask);
       qDebug() << "col=" << color;
 
-      PixmapItem *item = NULL;
+      PixmapItem* item = NULL;
       if (cur & Transmission::block)
       {
         unsigned int neighbours = Block::None;
@@ -194,6 +199,12 @@ void MainWidget::updateMap(const Transmission::map_t& map)
       {
         m_scene->addItem(item);
         item->setPos(mapPositionToCoord(x, y));
+        PixmapItem* oldItem = visualMap[x][y];
+        visualMap[x][y] = item;
+        if (oldItem != NULL) {
+          m_scene->removeItem(item);
+          delete oldItem;
+        }
       }
     }
   }
diff --git a/pacman-c++/mainwidget.h b/pacman-c++/mainwidget.h
index 36e5624..951b6c6 100644
--- a/pacman-c++/mainwidget.h
+++ b/pacman-c++/mainwidget.h
@@ -29,7 +29,7 @@ private:
 
   void updateScore();
 
-  PixmapItem ** visualMap;
+  QVector< QVector< PixmapItem* > > visualMap;
 
   // data conversion
   QPoint mapPositionToCoord(unsigned int x, unsigned int y);