Signal handling - SIGTERM will stop calculation and force file saving

SU2_CFD/src/drivers/CSinglezoneDriver.cpp

@@ -30,6 +30,8 @@
 #include "../../include/output/COutput.hpp"
 #include "../../include/iteration/CIteration.hpp"
 
+
+
 CSinglezoneDriver::CSinglezoneDriver(char* confFile,
                        unsigned short val_nZone,
                        SU2_Comm MPICommunicator) : CDriver(confFile,
@@ -45,6 +47,7 @@ CSinglezoneDriver::~CSinglezoneDriver() = default;
 
 void CSinglezoneDriver::StartSolver() {
 
+
   StartTime = SU2_MPI::Wtime();
 
   config_container[ZONE_0]->Set_StartTime(StartTime);
@@ -66,8 +69,8 @@ void CSinglezoneDriver::StartSolver() {
     TimeIter = config_container[ZONE_0]->GetRestart_Iter();
 
   /*--- Run the problem until the number of time iterations required is reached. ---*/
-  while ( TimeIter < config_container[ZONE_0]->GetnTime_Iter() ) {
-
+  /*--- or until a SIGTERM signal stops the loop. We catch SIGTERM and exit gracefully ---*/
+  while ( TimeIter < config_container[ZONE_0]->GetnTime_Iter()) {
     /*--- Perform some preprocessing before starting the time-step simulation. ---*/
 
     Preprocess(TimeIter);

SU2_CFD/src/iteration/CFluidIteration.cpp

@@ -25,9 +25,21 @@
  * License along with SU2. If not, see <http://www.gnu.org/licenses/>.
  */
 
+#include <iostream>
+#include <csignal>
+
 #include "../../include/iteration/CFluidIteration.hpp"
 #include "../../include/output/COutput.hpp"
 
+volatile sig_atomic_t stop;
+
+void signalHandler( int signum ) {
+   cout << "NIJSO: Interrupt signal (" << signum << ") received.\n";
+   // cleanup and close up stuff here
+   stop = 1;
+   // terminate program  
+   //exit(signum);
+}
 void CFluidIteration::Preprocess(COutput* output, CIntegration**** integration, CGeometry**** geometry,
                                  CSolver***** solver, CNumerics****** numerics, CConfig** config,
                                  CSurfaceMovement** surface_movement, CVolumetricMovement*** grid_movement,
@@ -254,6 +266,7 @@ void CFluidIteration::Solve(COutput* output, CIntegration**** integration, CGeom
   unsigned long Inner_Iter, nInner_Iter = config[val_iZone]->GetnInner_Iter();
   bool StopCalc = false;
 
+  signal(SIGTERM, signalHandler);
   /*--- Synchronization point before a single solver iteration.
         Compute the wall clock time required. ---*/
 
@@ -283,6 +296,9 @@ void CFluidIteration::Solve(COutput* output, CIntegration**** integration, CGeom
       Output(output, geometry, solver, config, Inner_Iter, StopCalc, val_iZone, val_iInst);
     }
 
+    /*--- If signal was sent, we set stopcalc to force writing the files ---*/
+    if (stop == 1) StopCalc = true;
+
     /*--- If the iteration has converged, break the loop ---*/
     if (StopCalc) break;
   }