Fail more loudly if a diagram file list is missing.

1 files changed, 8 insertions, 0 deletions

diff --git a/geom_matching/wasserstein/mpi/main.cpp b/geom_matching/wasserstein/mpi/main.cpp
index a1d980a..707cd38 100644
--- a/geom_matching/wasserstein/mpi/main.cpp
+++ b/geom_matching/wasserstein/mpi/main.cpp
@@ -245,6 +245,10 @@ int main(int argc, char ** argv)
   std::vector<std::string> files_2;
 
   std::ifstream list_file(list_1_file_name, std::ios::in);
+  if (!list_file.is_open())
+  {
+    fail(world_rank, std::string("Failed to load diagram list file ") + list_1_file_name + std::string("."));
+  }
   std::string line;
   while (std::getline(list_file, line))
   {
@@ -259,6 +263,10 @@ int main(int argc, char ** argv)
   else
   {
     list_file.open(list_2_file_name, std::ios::in);
+    if (!list_file.is_open())
+    {
+      fail(world_rank, std::string("Failed to load diagram list file ") + list_2_file_name + std::string("."));
+    }
     while (std::getline(list_file, line))
     {
       files_2.push_back(line);