fixed object removal via setting state to delete

apps/opencs/view/render/cell.cpp

@@ -22,11 +22,18 @@ bool CSVRender::Cell::removeObject (const std::string& id)
     if (iter==mObjects.end())
         return false;
 
-    delete iter->second;
+    removeObject (iter);
-    mObjects.erase (iter);
     return true;
 }
 
+std::map<std::string, CSVRender::Object *>::iterator CSVRender::Cell::removeObject (
+    std::map<std::string, Object *>::iterator iter)
+{
+    delete iter->second;
+    mObjects.erase (iter++);
+    return iter;
+}
+
 bool CSVRender::Cell::addObjects (int start, int end)
 {
     bool modified = false;
@@ -161,8 +168,8 @@ bool CSVRender::Cell::referenceDataChanged (const QModelIndex& topLeft,
     // perform update and remove where needed
     bool modified = false;
 
-    for (std::map<std::string, Object *>::iterator iter (mObjects.begin());
+    std::map<std::string, Object *>::iterator iter = mObjects.begin();
-        iter!=mObjects.end(); ++iter)
+    while (iter!=mObjects.end())
     {
         if (iter->second->referenceDataChanged (topLeft, bottomRight))
             modified = true;
@@ -171,24 +178,31 @@ bool CSVRender::Cell::referenceDataChanged (const QModelIndex& topLeft,
 
         if (iter2!=ids.end())
         {
-            if (iter2->second)
+            bool deleted = iter2->second;
+            ids.erase (iter2);
+
+            if (deleted)
             {
-                removeObject (iter->first);
+                iter = removeObject (iter);
                 modified = true;
+                continue;
             }
-
-            ids.erase (iter2);
         }
+
+        ++iter;
     }
 
     // add new objects
     for (std::map<std::string, bool>::iterator iter (ids.begin()); iter!=ids.end(); ++iter)
+    {
+        if (!iter->second)
         {
             mObjects.insert (std::make_pair (
                 iter->first, new Object (mData, mCellNode, iter->first, false)));
 
             modified = true;
         }
+    }
 
     return modified;
 }

apps/opencs/view/render/cell.hpp

@@ -49,6 +49,10 @@ namespace CSVRender
             /// \return Was the object deleted?
             bool removeObject (const std::string& id);
 
+            // Remove object and return iterator to next object.
+            std::map<std::string, Object *>::iterator removeObject (
+                std::map<std::string, Object *>::iterator iter);
+
             /// Add objects from reference table that are within this cell.
             ///
             /// \return Have any objects been added?