TFileMerger.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
 
/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id: TFileMerger.h 784654 2016-11-16 17:17:32Z krasznaa $
#ifndef XAODROOTACCESS_TOOLS_TFILEMERGER_H
#define XAODROOTACCESS_TOOLS_TFILEMERGER_H
 
// System include(s):
#include <string>
#include <vector>
#include <map>
#include <set>
#include <memory>
 
// ROOT include(s):
#include <TObject.h>
 
// Local include(s):
#include "xAODRootAccess/TEvent.h"
#include "AsgMessaging/StatusCode.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();
 
      StatusCode setOutputFileName( const std::string& name,
                                     const std::string& mode = "RECREATE" );
 
      StatusCode addFile( const std::string& name,
                           bool copyLocally = false );
 
      StatusCode addMetaDataTool( const std::string& typeName );
 
      enum EMergeMode {
         kSlowMerge = 0, 
         kFastMerge = 1  
      };
 
      static const ::Long64_t kBigNumber = 1234567890;
 
      StatusCode 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:
      StatusCode closeFiles();
      StatusCode mergeDirectory( ::TDirectory& input, ::TDirectory& output,
                                  EMergeMode mode, bool topLevelDir );
      StatusCode mergeObject( ::TObject& input, ::TObject& output );
      StatusCode createMetaDataTools();
      std::vector< ::TBranch* > getMissingBranches( ::TTree* first,
                                                    ::TTree* second ) const;
      std::vector< ::TBranch* > getSkippedBranches( ::TTree* tree ) const;
      StatusCode 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, std::unique_ptr< 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