2 // Class AliMixEventInputHandler
4 // Mixing input handler prepare N events before UserExec
7 // Martin Vala (martin.vala@cern.ch)
10 #ifndef ALIMIXINPUTEVENTHANDLER_H
11 #define ALIMIXINPUTEVENTHANDLER_H
13 #include <TObjArray.h>
16 #include <AliVEvent.h>
18 #include "AliMultiInputEventHandler.h"
21 class AliMixEventPool;
22 class AliMixInputHandlerInfo;
23 class AliInputEventHandler;
24 class AliMixInputEventHandler : public AliMultiInputEventHandler {
27 AliMixInputEventHandler(const Int_t size = 1, const Int_t mixNum = 1);
30 virtual Bool_t Init(Option_t *opt) { return AliMultiInputEventHandler::Init(opt); }
31 virtual Bool_t Init(TTree *tree, Option_t *opt);
32 virtual Bool_t Notify();
33 virtual Bool_t Notify(const char *path);
34 virtual Bool_t BeginEvent(Long64_t entry);
35 virtual Bool_t GetEntry();
36 virtual Bool_t FinishEvent();
38 // removing default impementation
39 virtual void AddInputEventHandler(AliVEventHandler */*inHandler*/);
41 void SetInputHandlerForMixing(const AliInputEventHandler *const inHandler);
42 void SetEventPool(AliMixEventPool *const evPool) { fEventPool = evPool; }
44 AliInputEventHandler *InputEventHandler(const Int_t index);
45 AliMixEventPool *GetEventPool() const { return fEventPool; }
46 Int_t BufferSize() const { return fBufferSize; }
47 Int_t NumberMixedTimes() const { return fNumberMixed; }
48 Int_t MixNumber() const { return fMixNumber; }
49 Long64_t EntryAll() const { return fEntryCounter; }
50 void UseDefaultProcess(Bool_t b = kTRUE) { fUseDefautProcess = b; }
51 void UsePreMixEvents(Bool_t b = kTRUE) { fUsePreMixEvents = b; }
52 void SetMixNumber(const Int_t mixNum);
54 void SetCurrentBinIndex(Int_t const index) { fCurrentBinIndex = index; }
55 void SetCurrentEntry(Long64_t const entry) { fCurrentEntry = entry ; }
56 void SetCurrentEntryMain(Long64_t const entry) { fCurrentEntryMain = entry ; }
57 void SetCurrentEntryMix(Long64_t const entry) { fCurrentEntryMix = entry ; }
58 void SetNumberMixed(Int_t const index) { fNumberMixed = index; }
60 Int_t CurrentBinIndex() const { return fCurrentBinIndex; }
61 Long64_t CurrentEntry() const { return fCurrentEntry; }
62 Long64_t CurrentEntryMain() const { return fCurrentEntryMain; }
63 Long64_t CurrentEntryMix() const { return fCurrentEntryMix; }
64 Int_t NumberMixed() const { return fNumberMixed; }
66 void SelectCollisionCandidates(UInt_t offlineTriggerMask = AliVEvent::kMB) {fOfflineTriggerMask = offlineTriggerMask;}
67 Bool_t IsEventCurrentSelected();
70 TObjArray fMixTrees; // buffer of input handlers
71 TArrayI fTreeMap; // tree map
72 AliMixInputHandlerInfo *fMixIntupHandlerInfoTmp;//! mix input handler info full chain
73 Long64_t fEntryCounter; // entry counter
74 AliMixEventPool *fEventPool; // event pool
75 Int_t fNumberMixed; // number of mixed events with current event
76 Int_t fMixNumber; // user's mix number request
80 Bool_t fUseDefautProcess; // use default process
81 Bool_t fUsePreMixEvents; // use pre mixing events
84 Long64_t fCurrentEntry; //! current entry number (adds 1 for every event processed on each worker)
85 Long64_t fCurrentEntryMain; //! current entry in chain of processed files
86 Long64_t fCurrentEntryMix; //! current mixed entry in chain of processed files
87 Int_t fCurrentBinIndex; //! current bin index
89 UInt_t fOfflineTriggerMask; // Task processes collision candidates only
91 virtual Bool_t MixStd();
92 virtual Bool_t MixBuffer();
93 virtual Bool_t MixEventsMoreTimesWithOneEvent();
94 virtual Bool_t MixEventsMoreTimesWithBuffer();
96 void UserExecMixAllTasks(Long64_t entryCounter, Int_t idEntryList, Long64_t entryMainReal, Long64_t entryMixReal, Int_t numMixed);
98 AliMixInputEventHandler(const AliMixInputEventHandler& handler);
99 AliMixInputEventHandler &operator=(const AliMixInputEventHandler &handler);
101 ClassDef(AliMixInputEventHandler, 2)