Changes in src/Actions/AtomAction/RemoveAction.cpp [112b09:d55743e]

File:

• src/Actions/AtomAction/RemoveAction.cpp (modified) (2 diffs)

src/Actions/AtomAction/RemoveAction.cpp

@@ -6 +6 @@
  */
 
-#include "Helpers/MemDebug.hpp"
-
 #include "Actions/AtomAction/RemoveAction.hpp"
-#include "CommandLineParser.hpp"
 #include "atom.hpp"
 #include "Descriptors/AtomDescriptor.hpp"
 #include "log.hpp"
+#include "molecule.hpp"
 #include "verbose.hpp"
 #include "World.hpp"
@@ -43 +41 @@
     delete dialog;
     DoLog(1) && (Log() << Verbose(1) << "Removing atom " << first->getId() << "." << endl);
+    // TODO: this is not necessary when atoms and their storing to file are handled by the World
+    // simply try to erase in every molecule found
+    std::vector<molecule *> molecules = World::getInstance().getAllMolecules();
+    for (std::vector<molecule *>::iterator iter = molecules.begin();iter != molecules.end(); ++iter) {
+      (*iter)->erase(first);
+    }
     World::getInstance().destroyAtom(first);
     return Action::success;