core/include/SFileMerger.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
// $Id: SFileMerger.h 335 2012-11-21 14:11:47Z krasznaa $
/***************************************************************************
 * @Project: SFrame - ROOT-based analysis framework for ATLAS
 * @Package: Core
 *
 * @author Stefan Ask       <Stefan.Ask@cern.ch>           - Manchester
 * @author David Berge      <David.Berge@cern.ch>          - CERN
 * @author Johannes Haller  <Johannes.Haller@cern.ch>      - Hamburg
 * @author A. Krasznahorkay <Attila.Krasznahorkay@cern.ch> - CERN/Debrecen
 *
 ***************************************************************************/

#ifndef SFRAME_CORE_SFileMerger_H
#define SFRAME_CORE_SFileMerger_H

// STL include(s):
#include <vector>

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

// Local include(s):
#include "SError.h"
#include "SLogger.h"

// Forward declaration(s):
class TFile;

class SFileMerger {

public:
   SFileMerger();
   ~SFileMerger();

   Bool_t AddFile( const TString& fileName ) throw( SError );
   Bool_t OutputFile( const TString& fileName,
                      const TString& mode = "UPDATE" ) throw( SError );

   Bool_t Merge() throw( SError );

private:
   void CloseFiles();
   void MergeDirectory( TDirectory* input, TDirectory* output ) throw( SError );
   void MergeObjects( TObject* in, TObject* out ) throw( SError );

   std::vector< TFile* > m_inputFiles; 
   TFile*                m_outputFile; 

   mutable SLogger m_logger; 

}; // class SFileMerger

#endif // SFRAME_CORE_SFileMerger_H