Handle SIGINT

1 files changed, 12 insertions, 3 deletions

diff --git a/src/navier-stokes.c b/src/navier-stokes.c
index 5d71d73..0fe4407 100644
--- a/src/navier-stokes.c
+++ b/src/navier-stokes.c
@@ -230,8 +230,20 @@ int eea(
       fprintf(stderr,"%d % .8e % .8e % .8e % .8e % .8e % .8e % .8e\n",t,t*delta, avg_a, avg_e, avg_en, alpha, energy, enstrophy);
       printf("%8d % .15e % .15e % .15e % .15e % .15e % .15e % .15e\n",t,t*delta, avg_a, avg_e, avg_en, alpha, energy, enstrophy);
     }
+
+    // catch abort signal
+    if (g_abort){
+      // print u to stderr if no savefile
+      if (savefile==NULL){
+	savefile=stderr;
+      }
+      break;
+    }
   }
 
+  // save final entry to savefile
+  write_u(u, K1, K2, savefile);
+
   if(running_avg_window!=0){
     free(save_print_e);
     free(save_print_a);
@@ -241,9 +253,6 @@ int eea(
     free(save_print_short_en);
   }
 
-  // save final entry to savefile
-  write_u(u, K1, K2, savefile);
-
   ns_free_tmps(u, tmp1, tmp2, tmp3, fft1, fft2, ifft);
   return(0);
 }