aod-merger: add option to skip non-existing files

Author: mario-krueger

Common/Core/aodMerger.cxx

@@ -28,13 +28,15 @@ int main(int argc, char* argv[])
   std::string inputCollection("input.txt");
   std::string outputFileName("AO2D.root");
   long maxDirSize = 100000000;
+  bool skipNonExistingFiles = false;
 
   int option_index = 0;
   static struct option long_options[] = {
     {"input", required_argument, nullptr, 0},
     {"output", required_argument, nullptr, 1},
     {"max-size", required_argument, nullptr, 2},
-    {"help", no_argument, nullptr, 3},
+    {"skip-non-existing-files", required_argument, nullptr, 3},
+    {"help", no_argument, nullptr, 4},
     {nullptr, 0, nullptr, 0}};
 
   while (true) {
@@ -48,10 +50,13 @@ int main(int argc, char* argv[])
     } else if (c == 2) {
       maxDirSize = atol(optarg);
     } else if (c == 3) {
+      skipNonExistingFiles = (strcmp(optarg, "true")) ? false : true;
+    } else if (c == 4) {
       printf("AOD merging tool. Options: \n");
-      printf("  --input <inputfile.txt>      Contains path to files to be merged. Default: %s\n", inputCollection.c_str());
-      printf("  --output <outputfile.root>   Target output ROOT file. Default: %s\n", outputFileName.c_str());
-      printf("  --max-size <size in Bytes>   Target directory size: %ld \n", maxDirSize);
+      printf("  --input <inputfile.txt>                  Contains path to files to be merged. Default: %s\n", inputCollection.c_str());
+      printf("  --output <outputfile.root>               Target output ROOT file. Default: %s\n", outputFileName.c_str());
+      printf("  --max-size <size in Bytes>               Target directory size. Default: %ld\n", maxDirSize);
+      printf("  --skip-non-existing-files <true|false>   Non-existing files are skipped. Default: %s\n", "false");
       return -1;
     } else {
       return -2;
@@ -62,6 +67,7 @@ int main(int argc, char* argv[])
   printf("  Input file: %s\n", inputCollection.c_str());
   printf("  Ouput file name: %s\n", outputFileName.c_str());
   printf("  Maximal folder size (uncompressed): %ld\n", maxDirSize);
+  printf("  Skip non-existing files: %s\n", (skipNonExistingFiles) ? "true" : "false");
 
   std::map<std::string, TTree*> trees;
   std::map<std::string, int> offsets;
@@ -93,6 +99,16 @@ int main(int argc, char* argv[])
     printf("Processing input file: %s\n", line.Data());
 
     auto inputFile = TFile::Open(line);
+    if (!inputFile) {
+      printf("Could not open input file %s.\n", line.Data());
+      if (skipNonExistingFiles) {
+        continue;
+      } else {
+        printf("Aborting merge!\n");
+        return 1;
+      }
+    }
+
     TList* keyList = inputFile->GetListOfKeys();
     keyList->Sort();
 
@@ -234,9 +250,9 @@ int main(int argc, char* argv[])
       }
 
       if (currentDirSize > maxDirSize) {
-        printf("Maximum size reached: %ld. Closing folder.\n", currentDirSize);
+        printf("Maximum size reached: %ld. Closing folder %s.\n", currentDirSize, dfName);
         for (auto const& tree : trees) {
-          //printf("Writing %s\n", tree.first.c_str());
+          // printf("Writing %s\n", tree.first.c_str());
           outputDir->cd();
           tree.second->Write();
           delete tree.second;
@@ -249,6 +265,7 @@ int main(int argc, char* argv[])
     }
     inputFile->Close();
   }
+
   outputFile->Write();
   outputFile->Close();