]>
Commit | Line | Data |
---|---|---|
6d3a7bbf | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, 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 | // | |
20 | // Base class for filtering friends | |
21 | // | |
22 | ////////////////////////////////////////////////////////////////////////// | |
23 | ||
24 | #include <TChain.h> | |
25 | #include <TFile.h> | |
26 | #include <TList.h> | |
27 | ||
28 | #include "AliAnalysisTaskFilter.h" | |
29 | #include "AliAnalysisManager.h" | |
30 | #include "AliAnalysisDataSlot.h" | |
31 | #include "AliESDEvent.h" | |
32 | #include "AliESD.h" | |
33 | #include "AliVEvent.h" | |
34 | #include "AliESDHandler.h" | |
35 | #include "AliInputEventHandler.h" | |
36 | #include "AliLog.h" | |
37 | #include "AliESDfriend.h" | |
38 | #include "AliESDfriendTrack.h" | |
39 | ||
40 | ||
41 | ClassImp(AliAnalysisTaskFilter) | |
42 | ||
43 | //////////////////////////////////////////////////////////////////////// | |
44 | ||
45 | AliAnalysisTaskFilter::AliAnalysisTaskFilter(): | |
46 | AliAnalysisTask(), | |
47 | fDebug(0), | |
48 | fEntry(0), | |
49 | fInputEvent(0x0), | |
50 | fInputHandler(0x0), | |
6d3a7bbf | 51 | fOutputESDfriend(0x0), |
03a8000a | 52 | fTreeEF(0x0) |
6d3a7bbf | 53 | { |
54 | // | |
55 | // Default constructor | |
56 | // | |
57 | } | |
58 | ||
59 | //______________________________________________________________________ | |
60 | ||
61 | AliAnalysisTaskFilter::AliAnalysisTaskFilter(const char* name): | |
62 | AliAnalysisTask(name, "AnalysisTaskFilter"), | |
63 | fDebug(0), | |
64 | fEntry(0), | |
65 | fInputEvent(0x0), | |
66 | fInputHandler(0x0), | |
6d3a7bbf | 67 | fOutputESDfriend(0x0), |
03a8000a | 68 | fTreeEF(0x0) |
6d3a7bbf | 69 | { |
70 | // | |
71 | // Default constructor | |
72 | // | |
73 | ||
74 | DefineInput (0, TChain::Class()); | |
75 | DefineOutput(0, TTree::Class()); | |
76 | } | |
77 | ||
78 | //______________________________________________________________________ | |
79 | ||
80 | AliAnalysisTaskFilter::AliAnalysisTaskFilter(const AliAnalysisTaskFilter& obj): | |
81 | AliAnalysisTask(obj), | |
82 | fDebug(0), | |
83 | fEntry(0), | |
84 | fInputEvent(0x0), | |
85 | fInputHandler(0x0), | |
6d3a7bbf | 86 | fOutputESDfriend(0x0), |
03a8000a | 87 | fTreeEF(0x0) |
6d3a7bbf | 88 | { |
89 | // | |
90 | // Copy constructor | |
91 | // | |
92 | ||
93 | fDebug = obj.fDebug; | |
94 | fEntry = obj.fEntry; | |
95 | fInputEvent = obj.fInputEvent; | |
96 | fInputHandler = obj.fInputHandler; | |
6d3a7bbf | 97 | fOutputESDfriend = obj.fOutputESDfriend; |
03a8000a | 98 | fTreeEF = obj.fTreeEF; |
6d3a7bbf | 99 | } |
100 | ||
101 | ||
102 | //______________________________________________________________________ | |
103 | ||
104 | AliAnalysisTaskFilter& AliAnalysisTaskFilter::operator=(const AliAnalysisTaskFilter& other) | |
105 | { | |
106 | // | |
107 | // Assignment | |
108 | // | |
109 | ||
110 | AliAnalysisTask::operator=(other); | |
111 | fDebug = other.fDebug; | |
112 | fEntry = other.fEntry; | |
113 | fInputEvent = other.fInputEvent; | |
114 | fInputHandler = other.fInputHandler; | |
6d3a7bbf | 115 | fOutputESDfriend = other.fOutputESDfriend; |
03a8000a | 116 | fTreeEF = other.fTreeEF; |
6d3a7bbf | 117 | return *this; |
118 | } | |
119 | ||
120 | ||
121 | //______________________________________________________________________ | |
122 | ||
123 | void AliAnalysisTaskFilter::ConnectInputData(Option_t* /*option*/) | |
124 | { | |
125 | // | |
126 | // Connect the input data | |
127 | // | |
128 | ||
129 | if (fDebug > 1) printf("AnalysisTaskFilter::ConnectInputData() \n"); | |
130 | fInputHandler = (AliInputEventHandler*) | |
131 | ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); | |
132 | if (fInputHandler) { | |
133 | fInputEvent = fInputHandler->GetEvent(); | |
134 | } else { | |
135 | AliError("No Input Event Handler connected") ; | |
136 | return ; | |
137 | } | |
138 | } | |
139 | ||
140 | //______________________________________________________________________ | |
141 | ||
142 | void AliAnalysisTaskFilter::CreateOutputObjects() | |
143 | { | |
144 | // | |
145 | // Create the output container | |
146 | // | |
147 | ||
148 | if (fDebug > 1) printf("AnalysisTaskFilter::CreateOutPutData() \n"); | |
149 | ||
150 | AliESDHandler* handler = (AliESDHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); | |
151 | ||
152 | if (handler) { | |
03a8000a | 153 | fTreeEF = handler->GetTree(); |
6d3a7bbf | 154 | } |
155 | else { | |
156 | AliWarning("No AOD Event Handler connected.") ; | |
157 | } | |
158 | ||
159 | UserCreateOutputObjects(); | |
160 | } | |
161 | ||
162 | //______________________________________________________________________ | |
163 | ||
164 | void AliAnalysisTaskFilter::Exec(Option_t* option) | |
165 | { | |
166 | // | |
167 | // Exec analysis of one event | |
168 | // | |
169 | ||
170 | if (fDebug > 1) AliInfo("AliAnalysisTaskFilter::Exec() \n"); | |
171 | ||
172 | if( fInputHandler ) { | |
173 | fEntry = fInputHandler->GetReadEntry(); | |
174 | } | |
175 | ||
176 | ||
177 | if ( !((Entry()-1)%100) && fDebug > 0) { | |
178 | AliInfo(Form("%s ----> Processing event # %lld", CurrentFileName(), Entry())); | |
179 | } | |
180 | AliESDHandler* handler = (AliESDHandler*)((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); | |
181 | ||
182 | if (UserSelectESDfriendForCurrentEvent()){ | |
183 | // Call the user analysis only if the event was selected | |
184 | fOutputESDfriend = handler->GetESDfriend(); | |
185 | UserExec(option); | |
186 | } | |
03a8000a | 187 | else { |
188 | // Set null pointer | |
189 | fOutputESDfriend = 0x0; | |
190 | ||
191 | } | |
6d3a7bbf | 192 | |
193 | // Added protection in case the derived task is not an AOD producer. | |
194 | AliAnalysisDataSlot *out0 = GetOutputSlot(0); | |
03a8000a | 195 | if (out0 && out0->IsConnected()) PostData(0, fTreeEF); |
6d3a7bbf | 196 | } |
197 | ||
198 | //______________________________________________________________________ | |
199 | ||
200 | const char* AliAnalysisTaskFilter::CurrentFileName() | |
201 | { | |
202 | // Returns the current file name | |
203 | if( fInputHandler ){ | |
204 | return fInputHandler->GetTree()->GetCurrentFile()->GetName(); | |
205 | } | |
206 | else return ""; | |
207 | } | |
208 | ||
209 | //______________________________________________________________________ | |
210 | ||
211 | void AliAnalysisTaskFilter::AddFriendTrackAt(AliESDfriendTrack* t, Int_t index) | |
212 | { | |
213 | // | |
214 | // Adds the friend track at the i-th position in the TClonesArray | |
215 | // of the ESD friend tracks | |
216 | // | |
217 | ||
218 | AliESDfriendTrack* currentTrack = (AliESDfriendTrack*)fOutputESDfriend->GetTrack(index); | |
219 | if(currentTrack && currentTrack->GetCalibObject(0)){ | |
220 | AliWarning("Friend already there"); | |
221 | } | |
222 | else { | |
223 | // AliInfo("Adding track"); | |
224 | fOutputESDfriend->AddTrackAt(t,index); | |
225 | } | |
226 | } |