merge.cpp

Go to the documentation of this file.

#include "merge/merge.C"
#include "TError.h"
#include <cstdlib>

static int usage() {
  ::printf("Gaudi merge facility for ROOT tree based files.\n"
       " Usage: \n"
       "gaudi_merge -o <output-file> -i <input-file 1> [ -i <input-file 2> ...]\n\n"
       "input- and output files may specify any legal (ROOT) file name.\n" 
       "   -output      Specify output file name.\n"
       "   -input       Specify input file name.\n"
       "   -debug       Switch debug flag on.\n"
       );
  return 1;
}

static ErrorHandlerFunc_t s_err = nullptr;
static void err_handler(Int_t level, Bool_t abort_bool, const char *location, const char *msg)
{
  if ( msg && strstr(msg,"no dictionary for class") ) return;
  s_err(level,abort_bool, location, msg);
}

int main(int argc, char** argv) {
  bool dbg = false, fixup=true;
  vector<string> input;
  string output;
  for(int i=1; i < argc; ++i) {
    if ( *argv[i] == '-' ) {
      switch(::toupper(*(argv[i]+1))) {
      case 'N':
    fixup = false;
    break;
      case 'D':
    dbg = true;
    break;
      case 'O':
    if ( i+1 < argc ) output = argv[i+1];
    ++i;
    break;
      case 'I':
    if ( i+1 < argc ) input.push_back(argv[i+1]);
    ++i;
    break;
      default:
    return usage();
      }
    }
  }
  if ( input.empty() ) {
    ::printf("\nERROR: No input file(s) supplied\n\n");
    return usage();
  }
  else if ( output.empty() ) {
    ::printf("\nERROR: No output file supplied.\n\n");
    return usage();
  }
  gROOT->SetBatch(kTRUE);
  s_err = SetErrorHandler(err_handler);
  for(size_t i=0; i<input.size();++i)  {
    const string& in = input[i];
    bool do_fixup = fixup && ((i+1)==input.size());
    //::printf("+++ Target:%s\n+++ Source file:%s Fixup:%s Dbg:%s %d %d\n",
    //         output.c_str(),in.c_str(),do_fixup ? "YES" : "NO",dbg ? "YES" : "NO",i,input.size());
    int result = merge(output.c_str(),in.c_str(),do_fixup,dbg);
    if ( result == MERGE_ERROR ) {
      printf("\nERROR: File merge failed after %ld files.\n\n",long(i+1));
      return 1;
    }
  }
  return 0;
}