Added new recovery mode "autoprobe" which is like "auto" but lets the user

continue with the simulation if no checkpoint file was found.
Closes PR Cactus/300.


git-svn-id: http://svn.cactuscode.org/arrangements/CactusBase/IOUtil/trunk@139 b32723a9-ab3a-4a60-88e2-2e5d99d7c17a

1 files changed, 50 insertions, 32 deletions

diff --git a/src/CheckpointRecovery.c b/src/CheckpointRecovery.c
index 6cb2582..71e98c1 100644
--- a/src/CheckpointRecovery.c
+++ b/src/CheckpointRecovery.c
@@ -42,6 +42,7 @@ void IOUtil_RecoverIDFromDatafiles (cGH *GH);
 /* Local data holding info on Recover Functions */
 static cHandledData *RecoverFunctions = NULL;
 static int num_functions = 0;
+static int checkpoint_file_exists = 0;
 
 #ifdef HAVE_SCANDIR
 /* prefix and extension of potential recovery files */
@@ -330,12 +331,19 @@ void IOUtil_RecoverGH (cGH *GH)
 
   myGH = (ioGH *) CCTK_GHExtension (GH, "IO");
 
-  myGH->recovered = IOUtil_RecoverFromFile (GH, NULL, CP_RECOVER_DATA) >= 0;
+  if (checkpoint_file_exists)
+  {
+    myGH->recovered = IOUtil_RecoverFromFile (GH, NULL, CP_RECOVER_DATA) >= 0;
 
-  /* stop if recovery failed */
-  if (! myGH->recovered)
+    /* stop if recovery failed */
+    if (! myGH->recovered)
+    {
+      CCTK_WARN (0, "Failed to restart from recovery !");
+    }
+  }
+  else
   {
-    CCTK_WARN (0, "Failed to restart from recovery !");
+    myGH->recovered = 0;
   }
 }
 
@@ -474,26 +482,34 @@ void IOUtil_RecoverIDFromDatafiles (cGH *GH)
                It is called by the IO thorns' parameter recovery routines
                scheduled at CCTK_RECOVER_PARAMETERS, and simply calls
                the given callback routine with its arguments
-               plus a recovery file name.
+               plus a checkpoint filename.
    @enddesc
 
    @var        recoverFn
    @vdesc      callback function for recovery of parameters
-               from a given recovery file
+               from a given checkpoint file
    @vtype      int (*) (cGH *, const char *, int)
    @vio        in
    @endvar
    @var        fileExtension
-   @vdesc      extension of valid recovery files for given callback
+   @vdesc      extension of valid checkpoint files for given callback
    @vtype      const char *
    @vio        in
    @endvar
    @var        fileType
-   @vdesc      string to describe the type of recovery file
+   @vdesc      string to describe the type of checkpoint file
                (used for warning/info messages)
    @vtype      const char *
    @vio        in
    @endvar
+
+   @returntype int
+   @returndesc
+                0 if in "autoprobe" mode and no cp files were found, or<BR>
+               +1 if parameter recovery was successful for some cp file,<BR>
+               -1 if in "auto" mode and no checkpoint files were found,
+                  or if parameter recovery failed for some cp file
+   @endreturndesc
 @@*/
 int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
                                                 const char *basefilename,
@@ -501,20 +517,20 @@ int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
                               const char *fileExtension,
                               const char *fileType)
 {
-  DECLARE_CCTK_PARAMETERS
-  int retval = -1;        /* the return value */
+  int retval;             /* the return value */
   cGH *dummyGH = NULL;    /* there's no GH yet but the callback routine
                              expects a GH pointer */
+#ifdef HAVE_SCANDIR
+  int i, nRecoverFiles;
+  struct dirent **recoverFileList = NULL;
+#endif
+  DECLARE_CCTK_PARAMETERS
 
 
-  if (CCTK_Equals (recover, "auto"))
+  if (CCTK_Equals (recover, "auto") || CCTK_Equals (recover, "autoprobe"))
   {
-
+    retval = CCTK_Equals (recover, "auto") ? -1 : 0;
 #ifdef HAVE_SCANDIR
-    int i, nRecoverFiles;
-    struct dirent **recoverFileList = NULL;
-
-
     if (verbose)
     {
       CCTK_VInfo (CCTK_THORNSTRING, "Searching for %s checkpoint files "
@@ -522,7 +538,7 @@ int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
                   fileType, recover_file, recovery_dir);
     }
 
-    /* set the file prefix and extension for selecting valid recovery files */
+    /* set the file prefix and extension for selecting valid checkpoint files */
     /* we have to pass it via global variables to the select() routine
        because it doesn't receive user-supplied arguments */
     recoverFilePrefix = (char *) malloc (strlen (recover_file) + 5);
@@ -536,30 +552,28 @@ int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
 
     if (nRecoverFiles <= 0)
     {
-      CCTK_VWarn (1, __LINE__, __FILE__, CCTK_THORNSTRING,
+      CCTK_VWarn (retval ? 1 : 3, __LINE__, __FILE__, CCTK_THORNSTRING,
                   "No %s checkpoint files with basefilename '%s' found in "
                   "recovery directory '%s'",
                   fileType, recover_file, recovery_dir);
     }
-
-    /* loop over all recovery files found and call the callback routine
-       skip all following files after the first successful recovery */
-    for (i = 0; i < nRecoverFiles; i++)
+    else
     {
-      if (retval < 0)
+      /* loop over all recovery files found and call the callback routine;
+         skip all following files after the first successful recovery (when
+         recoverFn() returned a positive value) */
+      for (i = 0; i < nRecoverFiles; i++)
       {
-        retval = (*recoverFn) (dummyGH, recoverFileList [i]->d_name,
-                               CP_RECOVER_PARAMETERS);
+        if (retval <= 0)
+        {
+          retval = recoverFn (dummyGH, recoverFileList[i]->d_name,
+                              CP_RECOVER_PARAMETERS);
+        }
+        free (recoverFileList[i]);
       }
-      free (recoverFileList [i]);
-    }
-    if (recoverFileList)
-    {
       free (recoverFileList);
     }
-
 #else
-
     /* no scandir(3) ? give up ! */
     CCTK_WARN (0, "You cannot use 'IO::recover = \"auto\"' on "
                "this architecture because it doesn't provide scandir(3) to "
@@ -567,7 +581,6 @@ int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
                "Please use 'IO::recover = \"manual\"' instead !");
 
 #endif
-
   }
   else
   {
@@ -582,6 +595,11 @@ int IOUtil_RecoverParameters (int (*recoverFn) (cGH *GH,
                 "with basefilename '%s' in recovery directory '%s'",
                 fileType, recover_file, recovery_dir);
   }
+
+  /* remember parameter recovery status for later evaluation in
+     IOUtil_RecoverGH() */
+  checkpoint_file_exists = retval > 0;
+
   return (retval);
 }