ANALYSIS/EventMixing/AliMixEventInputHandler.h

   1 //
   2 // Class AliMixEventInputHandler
   3 //
   4 // Mixing input handler prepare N events before UserExec
   5 // TODO example
   6 // author:
   7 //        Martin Vala (martin.vala@cern.ch)
   8 //
   9
  10 #ifndef ALIMIXEVENTINPUTHANDLER_H
  11 #define ALIMIXEVENTINPUTHANDLER_H
  12
  13 #include <TObjArray.h>
  14
  15 #include "AliInputEventHandler.h"
  16 #include <TArrayI.h>
  17
  18 class TChain;
  19 class AliMixEventPool;
  20 class AliMixInputHandlerInfo;
  21 class AliMixEventInputHandler : public AliInputEventHandler {
  22
  23 public:
  24   AliMixEventInputHandler(const Int_t size=1);
  25   virtual ~AliMixEventInputHandler() {;};
  26
  27   // From the interface
  28   virtual Bool_t  Init(Option_t* opt /*opt*/);
  29   virtual Bool_t  Init(TTree* tree, Option_t* /*opt*/);
  30   virtual Bool_t  BeginEvent(Long64_t entry /*entry*/);
  31   virtual Bool_t  GetEntry();
  32   virtual Bool_t  FinishEvent();
  33   virtual Bool_t  Notify();
  34   virtual Bool_t  Notify(const char *path);
  35
  36
  37   void SetInputHandlerForMixing(const AliInputEventHandler * const inHandler);
  38   void SetEventPool(AliMixEventPool * const evPool) {fEventPool = evPool;}
  39
  40   AliInputEventHandler * InputEventHandler(const Int_t index);
  41   AliMixEventPool *GetEventPool() const { return fEventPool;}
  42   Int_t           BufferSize() const {return fBufferSize;}
  43   Int_t           MixedEventNumber() const {return fMixEventNumber;}
  44   Long64_t        EntryAll() const { return fEntryCounter;}
  45 protected:
  46
  47   Int_t             fBufferSize;          // Size of the buffer
  48   TObjArray         fInputHandlers;       // buffer of input handlers
  49   TObjArray         fMixTrees;            // buffer of input handlers
  50   TArrayI           fTreeMap;             // tree map
  51   AliMixInputHandlerInfo *fMixIntupHandlerInfoTmp;    //! mix input handler info full chain
  52   Long64_t          fEntryCounter;        // entry counter
  53   AliMixEventPool  *fEventPool;           // event pool
  54
  55   Int_t             fMixEventNumber;      // number mix
  56
  57 private:
  58
  59   AliMixEventInputHandler(const AliMixEventInputHandler& handler);
  60   AliMixEventInputHandler& operator=(const AliMixEventInputHandler& handler);
  61
  62   ClassDef(AliMixEventInputHandler, 1)
  63 };
  64
  65 #endif