Changeset 56c55f3 for src/Actions/FragmentationAction/FragmentationAction.cpp

Timestamp:

Aug 31, 2016, 10:54:47 AM (9 years ago)

Author:

Frederik Heber <heber@…>

Branches:

FragmentMolecule_checks_bonddegrees, Gui_Fixes

Children:

e3d0fe

Parents:

6e5ca9

git-author:

Frederik Heber <heber@…> (08/31/16 10:45:35)

git-committer:

Frederik Heber <heber@…> (08/31/16 10:54:47)

Message:

Added "parse-state-files" option to fragment-molecule which is off by default.

• this prevents accidental parsing of stale keysets, adjacency, or orderatsite files. These hard to find errors ever and again cause failing fragmentation because e.g. some hydrogen is suddenly in the keysets although hydrogens are excluded.

File:

• src/Actions/FragmentationAction/FragmentationAction.cpp (modified) (4 diffs)

src/Actions/FragmentationAction/FragmentationAction.cpp

@@ -41 +41 @@
 #include "Descriptors/AtomSelectionDescriptor.hpp"
 #include "Descriptors/MoleculeIdDescriptor.hpp"
+#include "Fragmentation/AdaptivityMap.hpp"
 #include "Fragmentation/Exporters/ExportGraph_ToAtomFragments.hpp"
 #include "Fragmentation/Exporters/ExportGraph_ToFiles.hpp"
@@ -117 +118 @@
   boost::shared_ptr<AdjacencyList> FileChecker;
   boost::filesystem::path filename(params.prefix.get() + std::string(ADJACENCYFILE));
-  if (boost::filesystem::exists(filename) && boost::filesystem::is_regular_file(filename)) {
+  if ((params.ParseStateFiles.get())
+    && boost::filesystem::exists(filename)
+    && boost::filesystem::is_regular_file(filename)) {
     std::ifstream File;
     File.open(filename.string().c_str(), ios::out);
@@ -144 +147 @@
   // we parse in the keysets from last time if present
   Graph StoredGraph;
-  StoredGraph.ParseKeySetFile(params.prefix.get());
+  if (params.ParseStateFiles.get()) {
+    StoredGraph.ParseKeySetFile(params.prefix.get());
+    // check parsed graph against the set of atoms
+    {
+      AdaptivityMap *amap = StoredGraph.GraphToAdaptivityMap();
+      bool status = true;
+      for (World::AtomSelectionConstIterator iter = world.beginAtomSelection();
+          iter != world.endAtomSelection(); ++iter) {
+        const atomId_t atomid = iter->second->getId();
+        // skip hydrogens in check if saturation is turned on
+        if ((iter->second->getType()->getAtomicNumber() != 1)
+            || (!params.DoSaturation.get())) {
+          if (amap->count(atomid) == 0) {
+            ELOG(1, "Atom #" << atomid << " not contained in KeySet file. ");
+            status = false;
+            break;
+          }
+        } else if (amap->count(atomid) != 0) {
+          ELOG(1, "Atom #" << atomid << " in KeySet file is a hydrogen, but is now excluded. ");
+          status = false;
+          break;
+        }
+      }
+      delete amap;
+
+      if (!status) {
+        ELOG(1, "KeySetsFile seems to contain leftover from an old fragmentation, hence not using file.");
+        StoredGraph.clear();
+      }
+    }
+  }
 
   start = clock();
@@ -174 +207 @@
     {
       Graph StoredLocalGraph(StoredGraph.getLocalGraph(mol));
-      const int tempFlag = Fragmenter.FragmentMolecule(mols_atomids, params.order.get(), params.prefix.get(), StoredLocalGraph);
+      const int tempFlag = Fragmenter.FragmentMolecule(mols_atomids, params.order.get(), params.prefix.get(), StoredLocalGraph, params.ParseStateFiles.get());
       if ((ExitFlag == 2) && (tempFlag != 2))
         ExitFlag = tempFlag; // if there is one molecule that needs further fragmentation, it overrides others