/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/xAODRootAccess/xAODRootAccess/tools/TFileMerger.h

// Dear emacs, this is -*- c++ -*-
// $Id: TFileMerger.h 676314 2015-06-18 12:07:05Z krasznaa $
#ifndef XAODROOTACCESS_TOOLS_TFILEMERGER_H
#define XAODROOTACCESS_TOOLS_TFILEMERGER_H

// System include(s):
#include <string>
#include <vector>
#include <map>
#include <set>

// ROOT include(s):
#include <TObject.h>

// Local include(s):
#include "xAODRootAccess/TEvent.h"
#include "xAODRootAccess/tools/TReturnCode.h"
#include "xAODRootAccess/tools/THolder.h"

// Forward declaration(s):
class TFile;
class TDirectory;
class TTree;
class TBranch;

namespace xAOD {

   class TFileMerger : public ::TObject {

   public:
      TFileMerger();
      ~TFileMerger();

      TReturnCode setOutputFileName( const std::string& name,
                                     const std::string& mode = "RECREATE" );

      TReturnCode addFile( const std::string& name,
                           bool copyLocally = false );

      TReturnCode addMetaDataTool( const std::string& typeName );

      enum EMergeMode {
         kSlowMerge = 0, 
         kFastMerge = 1  
      };

      static const ::Long64_t kBigNumber = 1234567890;

      TReturnCode merge( EMergeMode mode = kFastMerge,
                         ::Long64_t entries = kBigNumber );

      TEvent::EAuxMode accessMode() const;
      void setAccessMode( TEvent::EAuxMode mode );

      int verbosity() const;
      void setVerbosity( int value );

      const std::set< std::string >& treesSkipped() const;
      void addTreeToSkip( const std::string& name );

   private:
      TReturnCode closeFiles();
      TReturnCode mergeDirectory( ::TDirectory& input, ::TDirectory& output,
                                  EMergeMode mode, bool topLevelDir );
      TReturnCode mergeObject( ::TObject& input, ::TObject& output );
      TReturnCode createMetaDataTools();
      std::vector< ::TBranch* > getMissingBranches( ::TTree* first,
                                                    ::TTree* second ) const;
      std::vector< ::TBranch* > getSkippedBranches( ::TTree* tree ) const;
      TReturnCode addAuxBranch( ::TTree* otree, ::TBranch* ibranch ) const;

      std::vector< ::TFile* > m_input;
      std::vector< bool > m_copiedInput;
      ::TFile* m_output;

      std::set< std::string > m_treesToSkip;

      std::map< std::string, TEvent > m_events;
      std::vector< ::TFile* > m_helperFiles;

      std::set< std::string > m_metaDataToolNames;
      std::map< std::string, THolder > m_metaDataTools;

      TEvent::EAuxMode m_mode;

      ::Long64_t m_entriesToMerge;

      int m_verbosity;

      ClassDef( xAOD::TFileMerger, 0 )

   }; // class TFileMerger

} // namespace xAOD

#endif // XAODROOTACCESS_TOOLS_TFILEMERGER_H

---