ANALYSIS/EventMixing/AliMixEventInputHandler.cxx

   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 #include <TFile.h>
  11 #include <TChain.h>
  12 #include <TEntryList.h>
  13 #include "AliLog.h"
  14 #include "AliMixEventPool.h"
  15 #include "AliMixEventInputHandler.h"
  16 #include "AliAnalysisManager.h"
  17
  18 #include "AliMixInputHandlerInfo.h"
  19 #include <TChainElement.h>
  20
  21 ClassImp(AliMixEventInputHandler)
  22
  23 //_____________________________________________________________________________
  24 AliMixEventInputHandler::AliMixEventInputHandler(const Int_t size) :
  25     AliInputEventHandler(),
  26     fBufferSize(size),
  27     fInputHandlers(),
  28     fMixTrees(),
  29     fTreeMap(size),
  30     fMixIntupHandlerInfoTmp(0),
  31     fEntryCounter(0),
  32     fEventPool(0),
  33     fMixEventNumber(0) {
  34 //
  35 // Default constructor.
  36 //
  37   AliDebug(AliLog::kDebug + 10, "<-");
  38   AliDebug(AliLog::kDebug + 10, "->");
  39 }
  40
  41 //_____________________________________________________________________________
  42 AliInputEventHandler* AliMixEventInputHandler::InputEventHandler(const Int_t index) {
  43   //
  44   // Returns input handler
  45   //
  46   AliDebug(AliLog::kDebug, Form("<-"));
  47   if ((index >= 0) && (index < fBufferSize)) {
  48     AliDebug(AliLog::kDebug, Form("->"));
  49     return (AliInputEventHandler*) fInputHandlers.At(index);
  50   }
  51   AliDebug(AliLog::kDebug, Form("->"));
  52   return 0;
  53 }
  54 //_____________________________________________________________________________
  55 void AliMixEventInputHandler::SetInputHandlerForMixing(const AliInputEventHandler*const inHandler) {
  56   //
  57   // Create N (fBufferSize) copies of input handler
  58   //
  59   AliDebug(AliLog::kDebug, Form("<-"));
  60   AliDebug(AliLog::kDebug, Form("Creating %d input event handlers ...", fBufferSize));
  61   for (Int_t i = 0;i < fBufferSize;i++) {
  62     AliDebug(AliLog::kDebug + 5, Form("Adding %d ...", i));
  63     fInputHandlers.Add((AliInputEventHandler*) inHandler->Clone());
  64   }
  65   AliDebug(AliLog::kDebug, Form("->"));
  66 }
  67 //_____________________________________________________________________________
  68 Bool_t AliMixEventInputHandler::Init(Option_t*opt) {
  69   //
  70   // Init() is called for all mix input handlers.
  71   //
  72   AliDebug(AliLog::kDebug, Form("<- \"%s\"", opt));
  73   AliInputEventHandler *eh = 0;
  74   TObjArrayIter next(&fInputHandlers);
  75   while ((eh = (AliInputEventHandler *) next())) {
  76     eh->Init(opt);
  77   }
  78   AliDebug(AliLog::kDebug, Form("->"));
  79   return kTRUE;
  80 }
  81 //_____________________________________________________________________________
  82 Bool_t AliMixEventInputHandler::Init(TTree* tree, Option_t*) {
  83   //
  84   // Init(const char*path) is called for all mix input handlers.
  85   // Create event pool if needed
  86   //
  87   AliDebug(AliLog::kDebug, Form("<- %p", tree));
  88   if (!tree) {
  89     AliDebug(AliLog::kDebug, Form("->"));
  90     return kFALSE;
  91   }
  92   AliDebug(AliLog::kDebug, Form("%s", tree->GetCurrentFile()->GetName()));
  93
  94   // clears array of input handlers
  95   fMixTrees.SetOwner(kTRUE);
  96   fMixTrees.Clear();
  97
  98   // create AliMixInputHandlerInfo
  99   if (!fMixIntupHandlerInfoTmp) fMixIntupHandlerInfoTmp = new AliMixInputHandlerInfo(tree->GetName());
 100
 101   // adds current file
 102   fMixIntupHandlerInfoTmp->AddTreeToChain(tree);
 103   Int_t lastIndex = fMixIntupHandlerInfoTmp->GetChain()->GetListOfFiles()->GetEntries();
 104   TChainElement *che = (TChainElement *)fMixIntupHandlerInfoTmp->GetChain()->GetListOfFiles()->At(lastIndex - 1);
 105   AliMixInputHandlerInfo *mixIHI = 0;
 106   for (Int_t i = 0;i < fInputHandlers.GetEntries();i++) {
 107
 108     AliDebug(AliLog::kDebug, Form("fInputHandlers[%d]", i));
 109     mixIHI = new AliMixInputHandlerInfo(fMixIntupHandlerInfoTmp->GetName(), fMixIntupHandlerInfoTmp->GetTitle());
 110     mixIHI->PrepareEntry(che, -1, InputEventHandler(i));
 111     AliDebug(AliLog::kDebug, Form("chain[%d]->GetEntries() = %lld", i, mixIHI->GetChain()->GetEntries()));
 112     fMixTrees.Add(mixIHI);
 113   }
 114   AliDebug(AliLog::kDebug, Form("fEntryCounter=%lld", fEntryCounter));
 115
 116   if (fEventPool->NeedInit())
 117     fEventPool->Init();
 118
 119   AliDebug(AliLog::kDebug, Form("->"));
 120   return kTRUE;
 121 }
 122 //_____________________________________________________________________________
 123 Bool_t AliMixEventInputHandler::Notify() {
 124   //
 125   // Notify() is called for all mix input handlers
 126   //
 127   AliDebug(AliLog::kDebug, Form("<-"));
 128   AliInputEventHandler *eh = 0;
 129   TObjArrayIter next(&fInputHandlers);
 130   while ((eh = (AliInputEventHandler *) next())) {
 131     eh->Notify();
 132   }
 133   AliDebug(AliLog::kDebug, Form("->"));
 134   return kTRUE;
 135 }
 136
 137 //_____________________________________________________________________________
 138 Bool_t AliMixEventInputHandler::Notify(const char*path) {
 139   //
 140   // Notify(const char*path) is called for all mix input handlers
 141   //
 142   AliDebug(AliLog::kDebug, Form("<- %s", path));
 143   AliInputEventHandler *eh = 0;
 144   TObjArrayIter next(&fInputHandlers);
 145   while ((eh = (AliInputEventHandler *) next())) {
 146     eh->Notify(path);
 147   }
 148   AliDebug(AliLog::kDebug, Form("->"));
 149   return kTRUE;
 150 }
 151 //_____________________________________________________________________________
 152 Bool_t AliMixEventInputHandler::BeginEvent(Long64_t entry) {
 153   //
 154   // BeginEvent(Long64_t entry) is called for all mix input handlers
 155   //
 156
 157   AliDebug(AliLog::kDebug, Form("-> %lld", entry));
 158   AliInputEventHandler *eh = 0;
 159   TObjArrayIter next(&fInputHandlers);
 160   while ((eh = (AliInputEventHandler *) next())) {
 161     eh->BeginEvent(entry);
 162   }
 163   AliDebug(AliLog::kDebug, Form("->"));
 164   return kTRUE;
 165 }
 166
 167
 168 //_____________________________________________________________________________
 169 Bool_t AliMixEventInputHandler::GetEntry() {
 170   //
 171   // Sets correct events to every mix events
 172   //
 173   AliDebug(AliLog::kDebug, Form("<-"));
 174
 175   AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
 176   AliInputEventHandler *inEvHMain = dynamic_cast<AliInputEventHandler *>(mgr->GetInputEventHandler());
 177
 178   Long64_t zeroChainEntries = fMixIntupHandlerInfoTmp->GetChain()->GetEntries() - inEvHMain->GetTree()->GetEntries();
 179
 180   // if fEntryCounter is 0 just add entry
 181   if (!fEntryCounter) {
 182     if (inEvHMain) {
 183
 184       fEventPool->AddEntry(inEvHMain->GetTree()->GetReadEntry() + zeroChainEntries, inEvHMain->GetEvent());
 185     }
 186     return kTRUE;
 187   }
 188
 189   AliDebug(AliLog::kDebug, Form("++++++++++++++ BEGIN SETUP EVENT %lld +++++++++++++++++++", fEntryCounter));
 190
 191   fMixEventNumber = 0;
 192   TEntryList *el = fEventPool->FindEntryList(inEvHMain->GetEvent());
 193   Long64_t elNum = 0;
 194   if (el)
 195     elNum = el->GetN();
 196   else
 197     AliDebug(AliLog::kDebug, "el is null");
 198
 199   AliInputEventHandler *eh = 0;
 200   AliMixInputHandlerInfo *mihi = 0;
 201   TObjArrayIter next(&fInputHandlers);
 202   Int_t counter = 0;
 203   Long64_t entryMix = 0;
 204   while ((eh = (AliInputEventHandler *) next())) {
 205     if (fEventPool->GetListOfEventCuts()->GetEntries() > 0) {
 206       entryMix = -1;
 207       if (el && el->GetN() >= fBufferSize) {
 208         Long64_t entryInEntryList =  elNum - 1 - counter;
 209         if (entryInEntryList < 0) break;
 210         entryMix = el->GetEntry(entryInEntryList);
 211       }
 212     } else {
 213       entryMix = fEntryCounter - 1 - counter ;
 214     }
 215
 216     AliDebug(AliLog::kDebug, Form("Handler[%d] entryMix %lld ", counter, entryMix));
 217     if (entryMix < 0) break;
 218
 219     mihi = (AliMixInputHandlerInfo*) fMixTrees.At(counter);
 220     TChainElement *te = fMixIntupHandlerInfoTmp->GetEntryInTree(entryMix);
 221     mihi->PrepareEntry(te, entryMix, InputEventHandler(counter));
 222     fMixEventNumber++;
 223     counter++;
 224   }
 225
 226   if (inEvHMain) {
 227     fEventPool->AddEntry(inEvHMain->GetTree()->GetReadEntry() + zeroChainEntries, inEvHMain->GetEvent());
 228   }
 229
 230   AliDebug(AliLog::kDebug, Form("fEntryCounter=%lld fMixEventNumber=%d", fEntryCounter, fMixEventNumber));
 231   AliDebug(AliLog::kDebug, Form("++++++++++++++ END SETUP EVENT %lld +++++++++++++++++++", fEntryCounter));
 232   AliDebug(AliLog::kDebug, Form("->"));
 233   return kTRUE;
 234 }
 235 //_____________________________________________________________________________
 236 Bool_t AliMixEventInputHandler::FinishEvent() {
 237   //
 238   // FinishEvent() is called for all mix input handlers
 239   //
 240   AliDebug(AliLog::kDebug, Form("<-"));
 241   AliInputEventHandler *eh = 0;
 242   TObjArrayIter next(&fInputHandlers);
 243   while ((eh = (AliInputEventHandler *) next())) {
 244     eh->FinishEvent();
 245   }
 246   fEntryCounter++;
 247   AliDebug(AliLog::kDebug, Form("->"));
 248   return kTRUE;
 249 }