core/include/SCycleBaseExec.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
// $Id: SCycleBaseExec.h 318 2012-06-22 16:32:42Z 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_SCycleBaseExec_H
#define SFRAME_CORE_SCycleBaseExec_H

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

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

// Local include(s):
#include "ISCycleBaseConfig.h"
#include "ISCycleBaseHist.h"
#include "ISCycleBaseNTuple.h"
#include "SCycleBaseBase.h"

// Forward declaration(s):
class TTree;
class SInputData;
class TFile;
class TList;

class SCycleBaseExec : public TSelector,
                       public virtual ISCycleBaseConfig,
                       public virtual ISCycleBaseHist,
                       public virtual ISCycleBaseNTuple,
                       public virtual SCycleBaseBase {

public:
   SCycleBaseExec();

   //                                                                       //
   //   The following are the functions inherited from TSelector.           //
   //                                                                       //

   virtual void   Begin( TTree* );
   virtual void   SlaveBegin( TTree* );
   virtual void   Init( TTree* main_tree );
   virtual Bool_t Notify();
   virtual Bool_t Process( Long64_t entry );
   virtual void   SlaveTerminate();
   virtual void   Terminate();
   virtual Int_t  Version() const { return 2; }

   //                                                                       //
   //   The following are the functions to be implemented in the derived    //
   //   classes.                                                            //
   //                                                                       //


   virtual void BeginCycle() throw( SError ) = 0;

   virtual void EndCycle() throw( SError ) = 0;


   virtual void BeginInputData( const SInputData& ) throw( SError ) = 0;

   virtual void EndInputData  ( const SInputData& ) throw( SError ) = 0;


   virtual void BeginMasterInputData( const SInputData& ) throw( SError ) {}

   virtual void EndMasterInputData( const SInputData& ) throw( SError ) {}


   virtual void BeginInputFile( const SInputData& ) throw( SError ) = 0;


   virtual void ExecuteEvent( const SInputData&, Double_t weight ) throw( SError ) = 0;

private:
   void ReadConfig() throw( SError );
   virtual void ExecuteEvent( Int_t event, Int_t px, Int_t py );

   Long64_t m_nProcessedEvents;
   Long64_t m_nSkippedEvents;

   // variable used for the case of multiple InputData objects with
   // the same type, that are written to the same output file
   Bool_t m_keepOutputFile;
   Bool_t m_firstInputDataOfMany;

   Bool_t m_firstInit;

   TTree*                m_inputTree; 
   SInputData*           m_inputData; 
   std::vector< TTree* > m_outputTrees; 
   TFile*                m_outputFile; 

#ifndef DOXYGEN_IGNORE
   ClassDef( SCycleBaseExec, 0 );
#endif // DOXYGEN_IGNORE

}; // class SCycleBaseExec

#endif // SFRAME_CORE_SCycleBaseExec_H