merge.cpp

Go to the documentation of this file.

/***********************************************************************************\
* (c) Copyright 1998-2024 CERN for the benefit of the LHCb and ATLAS collaborations *
*                                                                                   *
* This software is distributed under the terms of the Apache version 2 licence,     *
* copied verbatim in the file "LICENSE".                                            *
*                                                                                   *
* In applying this licence, CERN does not waive the privileges and immunities       *
* granted to it by virtue of its status as an Intergovernmental Organization        *
* or submit itself to any jurisdiction.                                             *
\***********************************************************************************/
#include "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;
}