Keep track of PixmapItems in client

1 files changed, 12 insertions, 1 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;
+        }
       }
     }
   }