Connection to tree in Notify() seems to be safer.
[u/mrichter/AliRoot.git] / 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