[hadd] add option to specify which objects should (not) be merged

main/src/hadd-argparse.py

@@ -23,6 +23,8 @@ def get_argparse():
 	parser.add_argument("-dbg", help="Parallelize the execution in multiple processes in debug mode (Does not delete partial files stored inside working directory)")
 	parser.add_argument("-d", help="Carry out the partial multiprocess execution in the specified directory")
 	parser.add_argument("-n", help="Open at most 'maxopenedfiles' at once (use 0 to request to use the system maximum)")
+        parser.add_argument("-l", help="Comma-separated list of objects include (or exclude) from the merge.")
+        parser.add_argument("-e", help="Use objects passed to -l as an exclusion list.")
 	parser.add_argument("-cachesize", help="Resize the prefetching cache use to speed up I/O operations(use 0 to disable)")
 	parser.add_argument("-experimental-io-features", help="Used with an argument provided, enables the corresponding experimental feature for output trees")
 	parser.add_argument("-f", help="Gives the ability to specify the compression level of the target file(by default 4) ")

main/src/hadd.cxx

@@ -116,6 +116,8 @@ int main( int argc, char **argv )
    Int_t maxopenedfiles = 0;
    Int_t verbosity = 99;
    TString cacheSize;
+   std::vector<std::string> AddObjectNamesList;
+   Bool_t exclusionlist = kFALSE;
    SysInfo_t s;
    gSystem->GetSysInfo(&s);
    auto nProcesses = s.fCpus;
@@ -143,7 +145,10 @@ int main( int argc, char **argv )
          debug = kTRUE;
          verbosity = kTRUE;
          ++ffirst;
-      } else if (strcmp(argv[a], "-d") == 0) {
+      } else if ( strcmp(argv[a],"-e") == 0 ) {
+         exclusionlist = kTRUE;
+         ++ffirst;
+      }  else if (strcmp(argv[a], "-d") == 0) {
          if (a + 1 != argc && argv[a + 1][0] != '-') {
             if (gSystem->AccessPathName(argv[a + 1])) {
                std::cerr << "Error: could not access the directory specified: " << argv[a + 1]
@@ -288,6 +293,18 @@ int main( int argc, char **argv )
             }
          }
          ++ffirst;
+      } else if (!strcmp(argv[a], "-l")) {
+         if (a+1 >= argc) {
+            std::cerr << "Error: no AddObjectNamesList items were specified after -l; ignoring\n";
+         } else {
+            std::stringstream ss;
+            ss.str(argv[++a]);
+            ++ffirst;
+            std::string item;
+            while (std::getline(ss, item, ','))
+              AddObjectNamesList.push_back(item);
+         }
+         ++ffirst;
       } else if ( argv[a][0] == '-' ) {
          bool farg = false;
          if (force && argv[a][1] == 'f') {
@@ -350,6 +367,8 @@ int main( int argc, char **argv )
    TFileMerger fileMerger(kFALSE, kFALSE);
    fileMerger.SetMsgPrefix("hadd");
    fileMerger.SetPrintLevel(verbosity - 1);
+   for(const auto& objn : AddObjectNamesList)
+     fileMerger.AddObjectNames(objn.data());
    if (maxopenedfiles > 0) {
       fileMerger.SetMaxOpenedFiles(maxopenedfiles);
    }
@@ -437,8 +456,12 @@ int main( int argc, char **argv )
       Bool_t status;
       if (append)
          status = merger.PartialMerge(TFileMerger::kIncremental | TFileMerger::kAll);
-      else
-         status = merger.Merge();
+      else {
+         Int_t merge_type = TFileMerger::kAll | TFileMerger::kRegular;
+         if (!AddObjectNamesList.empty())
+            exclusionlist ? merge_type |= TFileMerger::kSkipListed : merge_type |= TFileMerger::kOnlyListed;
+         status = merger.PartialMerge(merge_type);
+      }
       return status;
    };