wip: Listen to signal interuption and do early exit

Author: danieledler

src/core/InfomapBase.cpp

@@ -30,13 +30,16 @@
 #include <cstdlib>
 #include <algorithm>
 #include <cmath>
+#include <csignal>
 
 #ifdef _OPENMP
 #include <omp.h>
 #endif
 
 namespace infomap {
 
+bool InfomapBase::shouldExit = false;
+
 std::map<unsigned int, std::vector<unsigned int>> InfomapBase::getMultilevelModules(bool states)
 {
   if (haveMemory() && !states) {
@@ -95,6 +98,12 @@ unsigned int InfomapBase::maxTreeDepth() const
   return maxDepth;
 }
 
+void InfomapBase::signalHandler(int signal)
+{
+  // shouldExit.store(true);
+  InfomapBase::shouldExit = true;
+}
+
 // ===================================================
 // Run
 // ===================================================
@@ -106,6 +115,12 @@ void InfomapBase::run(const std::string& parameters)
     return;
   }
 
+  // Register signal handlers
+  std::signal(SIGINT, InfomapBase::signalHandler);
+  std::signal(SIGTERM, InfomapBase::signalHandler);
+  // shouldExit.store(false);
+  shouldExit = false;
+
   m_elapsedTime = Stopwatch(true);
   m_startDate = Date();
 

src/core/InfomapBase.h

@@ -29,6 +29,7 @@
 #include <string>
 #include <iostream>
 #include <sstream>
+#include <atomic>
 
 namespace infomap {
 
@@ -171,6 +172,7 @@ class InfomapBase : public InfomapConfig<InfomapBase> {
   void run(Network& network);
 
 private:
+  static void signalHandler(int signal);
   bool isFullNetwork() const { return m_isMain && m_aggregationLevel == 0; }
   bool isFirstLoop() const { return m_tuneIterationIndex == 0 && isFullNetwork(); }
 
@@ -516,6 +518,10 @@ class InfomapBase : public InfomapConfig<InfomapBase> {
   std::string m_initialParameters;
   std::string m_currentParameters;
 
+  // Atomic flag to indicate the program should exit
+  // static std::atomic<bool> shouldExit;
+  static bool shouldExit;
+
   std::unique_ptr<InfomapOptimizerBase> m_optimizer;
 };
 

src/core/InfomapOptimizer.h

@@ -20,6 +20,9 @@
 #include <set>
 #include <utility>
 
+#include <chrono>
+#include <thread>
+
 namespace infomap {
 
 template <typename Objective>
@@ -272,6 +275,12 @@ inline unsigned int InfomapOptimizer<Objective>::optimizeActiveNetwork()
   }
 
   do {
+    std::this_thread::sleep_for(std::chrono::nanoseconds(1));
+    if (m_infomap->shouldExit) {
+      Log() << "\nGot interrupt signal, early exiting.\n";
+      return numEffectiveLoops;
+    }
+
     ++coreLoopCount;
     unsigned int numNodesMoved = m_infomap->innerParallelization
         ? tryMoveEachNodeIntoBestModuleInParallel()