ANALYSIS/AliMultiEventInputHandler.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 //-------------------------------------------------------------------------
  19 //     Event handler for multiple VEvent input.
  20 //     This class handles multiple inputs for event mixing.
  21 //     Author: Andreas Morsch, CERN
  22 //-------------------------------------------------------------------------
  23
  24 #include "AliMultiEventInputHandler.h"
  25 #include "AliVEvent.h"
  26 #include "AliAODEvent.h"
  27 #include "AliESDEvent.h"
  28 #include "AliVEventPool.h"
  29 #include "AliLog.h"
  30 #include <TObjArray.h>
  31 #include <TTree.h>
  32
  33
  34 ClassImp(AliMultiEventInputHandler)
  35
  36 AliMultiEventInputHandler::AliMultiEventInputHandler() :
  37     AliInputEventHandler(),
  38     fBufferSize(0),
  39     fFormat(1),
  40     fNBuffered(0),
  41     fIndex(0),
  42     fCurrentBin(0),
  43     fEventPool(0),
  44     fEventBuffer(0)
  45 {
  46   // Default constructor
  47 }
  48
  49 //______________________________________________________________________________
  50 AliMultiEventInputHandler::AliMultiEventInputHandler(Int_t size, Int_t format) :
  51     AliInputEventHandler(),
  52     fBufferSize(size),
  53     fFormat(format),
  54     fNBuffered(0),
  55     fIndex(0),
  56     fCurrentBin(0),
  57     fEventPool(0),
  58     fEventBuffer(new AliVEvent*[size])
  59 {
  60   // Default constructor
  61     for (Int_t i = 0; i < size; i++)
  62         if (fFormat == 1) {
  63             fEventBuffer[i] = new AliAODEvent();
  64         } else if (fFormat == 0) {
  65             fEventBuffer[i] = new AliESDEvent();
  66         } else{
  67             AliWarning(Form("Unknown Format %5d", fFormat));
  68         }
  69 }
  70
  71 //______________________________________________________________________________
  72 AliMultiEventInputHandler::AliMultiEventInputHandler(const char* name, const char* title, Int_t size, Int_t format):
  73     AliInputEventHandler(name, title),
  74     fBufferSize(size),
  75     fFormat(format),
  76     fNBuffered(0),
  77     fIndex(0),
  78     fCurrentBin(0),
  79     fEventPool(0),
  80     fEventBuffer(new AliVEvent*[size])
  81 {
  82     // Constructor
  83     for (Int_t i = 0; i < size; i++)
  84         if (fFormat == 1) {
  85             fEventBuffer[i] = new AliAODEvent();
  86         } else if (fFormat == 0) {
  87             fEventBuffer[i] = new AliESDEvent();
  88         } else{
  89             AliWarning(Form("Unknown Format %5d", fFormat));
  90         }
  91 }
  92
  93 //______________________________________________________________________________
  94 AliMultiEventInputHandler::~AliMultiEventInputHandler()
  95 {
  96 // Destructor
  97 }
  98
  99 Bool_t AliMultiEventInputHandler::Init(TTree* tree, Option_t* /*opt*/)
 100 {
 101     // Initialisation necessary for each new tree
 102     fTree = tree;
 103     if (!fTree) return kFALSE;
 104     for (Int_t i = 0; i < fBufferSize; i++)
 105         fEventBuffer[i]->Clear();
 106     fIndex     = 0;
 107     fNBuffered = 1;
 108     return kTRUE;
 109 }
 110
 111
 112 Bool_t AliMultiEventInputHandler::Notify(const char */*path*/)
 113 {
 114     // Connect to new tree
 115     fEventBuffer[0]->ReadFromTree(fTree, "");
 116     return (kTRUE);
 117 }
 118
 119 Bool_t AliMultiEventInputHandler::BeginEvent(Long64_t /*entry*/)
 120 {
 121     // Actions before analysis of each event
 122     //
 123     // Reset the number of events buffered for this bin to 0
 124     if (fCurrentBin != (fEventPool->BinNumber())) {
 125         fCurrentBin = fEventPool->BinNumber();
 126         fNBuffered = 0;
 127     }
 128     return kTRUE;
 129 }
 130
 131 Bool_t AliMultiEventInputHandler::FinishEvent()
 132 {
 133     //
 134     // Connect the next event in the buffer to the tree
 135     fIndex++;
 136
 137     fIndex %= fBufferSize;
 138     AliInfo(Form("Connecting buffer entry %5d", fIndex));
 139     fEventBuffer[fIndex]->Clear();
 140     fEventBuffer[fIndex]->ReadFromTree(fTree, "reconnect");
 141
 142     fNBuffered++;
 143     if (fNBuffered > fBufferSize) fNBuffered = fBufferSize;
 144
 145     return (kTRUE);
 146 }
 147
 148 AliVEvent* AliMultiEventInputHandler::GetEvent(Int_t iev) const
 149 {
 150     // Get event number iev from buffer
 151     if ((iev < 0) || (iev >= fBufferSize))
 152     {
 153         AliWarning(Form("Event number out of range: %10d", iev));
 154         return 0;
 155     }
 156
 157     iev = fIndex - (fBufferSize - 1 - iev);
 158     if (iev < 0) iev += fBufferSize;
 159     AliInfo(Form("Event index in buffer is %5d", iev));
 160     return (fEventBuffer[iev]);
 161 }
 162