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