1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
20 #include <TInterpreter.h>
25 #include "AliAnalysisTaskSE.h"
26 #include "AliAnalysisManager.h"
27 #include "AliAnalysisCuts.h"
28 #include "AliAnalysisDataSlot.h"
30 #include "AliESDEvent.h"
31 #include "AliESDfriend.h"
33 #include "AliAODEvent.h"
34 #include "AliAODHeader.h"
35 #include "AliAODTracklets.h"
36 #include "AliAODCaloCells.h"
37 #include "AliAODMCParticle.h"
38 #include "AliVEvent.h"
39 #include "AliAODHandler.h"
40 #include "AliAODInputHandler.h"
41 #include "AliMCEventHandler.h"
42 #include "AliInputEventHandler.h"
43 #include "AliESDInputHandler.h"
44 #include "AliMCEvent.h"
49 ClassImp(AliAnalysisTaskSE)
51 ////////////////////////////////////////////////////////////////////////
52 AliAODHeader* AliAnalysisTaskSE::fgAODHeader = NULL;
53 TClonesArray* AliAnalysisTaskSE::fgAODTracks = NULL;
54 TClonesArray* AliAnalysisTaskSE::fgAODVertices = NULL;
55 TClonesArray* AliAnalysisTaskSE::fgAODV0s = NULL;
56 TClonesArray* AliAnalysisTaskSE::fgAODPMDClusters = NULL;
57 TClonesArray* AliAnalysisTaskSE::fgAODJets = NULL;
58 TClonesArray* AliAnalysisTaskSE::fgAODFMDClusters = NULL;
59 TClonesArray* AliAnalysisTaskSE::fgAODCaloClusters = NULL;
60 TClonesArray* AliAnalysisTaskSE::fgAODMCParticles = NULL;
61 AliAODTracklets* AliAnalysisTaskSE::fgAODTracklets = NULL;
62 AliAODCaloCells* AliAnalysisTaskSE::fgAODEmcalCells = NULL;
63 AliAODCaloCells* AliAnalysisTaskSE::fgAODPhosCells = NULL;
65 AliAnalysisTaskSE::AliAnalysisTaskSE():
75 fCurrentRunNumber(-1),
79 // Default constructor
82 AliAnalysisTaskSE::AliAnalysisTaskSE(const char* name):
83 AliAnalysisTask(name, "AnalysisTaskSE"),
92 fCurrentRunNumber(-1),
96 // Default constructor
97 DefineInput (0, TChain::Class());
98 DefineOutput(0, TTree::Class());
101 AliAnalysisTaskSE::AliAnalysisTaskSE(const AliAnalysisTaskSE& obj):
102 AliAnalysisTask(obj),
111 fCurrentRunNumber(-1),
118 fInputEvent = obj.fInputEvent;
119 fESDfriend = obj.fESDfriend;
120 fInputHandler = obj.fInputHandler;
121 fOutputAOD = obj.fOutputAOD;
122 fMCEvent = obj.fMCEvent;
124 fCurrentRunNumber = obj.fCurrentRunNumber;
125 fHistosQA = obj.fHistosQA;
130 AliAnalysisTaskSE& AliAnalysisTaskSE::operator=(const AliAnalysisTaskSE& other)
133 AliAnalysisTask::operator=(other);
134 fDebug = other.fDebug;
135 fEntry = other.fEntry;
136 fInputEvent = other.fInputEvent;
137 fESDfriend = other.fESDfriend;
138 fInputHandler = other.fInputHandler;
139 fOutputAOD = other.fOutputAOD;
140 fMCEvent = other.fMCEvent;
141 fTreeA = other.fTreeA;
142 fCurrentRunNumber = other.fCurrentRunNumber;
143 fHistosQA = other.fHistosQA;
144 fSelectCollisions = other.fSelectCollisions;
149 void AliAnalysisTaskSE::ConnectInputData(Option_t* /*option*/)
151 // Connect the input data
152 if (fDebug > 1) printf("AnalysisTaskSE::ConnectInputData() \n");
156 fInputHandler = (AliInputEventHandler*)
157 ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
161 AliMCEventHandler* mcH = 0;
162 mcH = (AliMCEventHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler());
164 fMCEvent = mcH->MCEvent();
168 if ((fInputHandler->GetTree())->GetBranch("ESDfriend."))
169 fESDfriend = ((AliESDInputHandler*)fInputHandler)->GetESDfriend();
171 fInputEvent = fInputHandler->GetEvent();
172 } else if( fMCEvent ) {
173 AliWarning("No Input Event Handler connected, only MC Truth Event Handler") ;
175 AliError("No Input Event Handler connected") ;
180 void AliAnalysisTaskSE::CreateOutputObjects()
182 // Create the output container
185 if (fDebug > 1) printf("AnalysisTaskSE::CreateOutPutData() \n");
187 AliAODHandler* handler = (AliAODHandler*)
188 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
190 Bool_t merging = kFALSE;
191 AliAODInputHandler* aodIH = static_cast<AliAODInputHandler*>((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
193 if (aodIH->GetMergeEvents()) merging = kTRUE;
197 // Check if AOD replication has been required
200 fOutputAOD = handler->GetAOD();
201 fTreeA = handler->GetTree();
202 if (fOutputAOD && !(handler->IsStandard())) {
203 if ((handler->NeedsHeaderReplication()) && !(fgAODHeader))
205 if (fDebug > 1) AliInfo("Replicating header");
206 fgAODHeader = new AliAODHeader;
207 handler->AddBranch("AliAODHeader", &fgAODHeader);
209 if ((handler->NeedsTracksBranchReplication() || merging) && !(fgAODTracks))
211 if (fDebug > 1) AliInfo("Replicating track branch\n");
212 fgAODTracks = new TClonesArray("AliAODTrack",500);
213 fgAODTracks->SetName("tracks");
214 handler->AddBranch("TClonesArray", &fgAODTracks);
216 if ((handler->NeedsVerticesBranchReplication() || merging) && !(fgAODVertices))
218 if (fDebug > 1) AliInfo("Replicating vertices branch\n");
219 fgAODVertices = new TClonesArray("AliAODVertex",500);
220 fgAODVertices->SetName("vertices");
221 handler->AddBranch("TClonesArray", &fgAODVertices);
223 if ((handler->NeedsV0sBranchReplication()) && !(fgAODV0s))
225 if (fDebug > 1) AliInfo("Replicating V0s branch\n");
226 fgAODV0s = new TClonesArray("AliAODv0",500);
227 fgAODV0s->SetName("v0s");
228 handler->AddBranch("TClonesArray", &fgAODV0s);
230 if ((handler->NeedsTrackletsBranchReplication()) && !(fgAODTracklets))
232 if (fDebug > 1) AliInfo("Replicating Tracklets branch\n");
233 fgAODTracklets = new AliAODTracklets("tracklets","tracklets");
234 handler->AddBranch("AliAODTracklets", &fgAODTracklets);
236 if ((handler->NeedsPMDClustersBranchReplication()) && !(fgAODPMDClusters))
238 if (fDebug > 1) AliInfo("Replicating PMDClusters branch\n");
239 fgAODPMDClusters = new TClonesArray("AliAODPmdCluster",500);
240 fgAODPMDClusters->SetName("pmdClusters");
241 handler->AddBranch("TClonesArray", &fgAODPMDClusters);
243 if ((handler->NeedsJetsBranchReplication() || merging) && !(fgAODJets))
245 if (fDebug > 1) AliInfo("Replicating Jets branch\n");
246 fgAODJets = new TClonesArray("AliAODJet",500);
247 fgAODJets->SetName("jets");
248 handler->AddBranch("TClonesArray", &fgAODJets);
250 if ((handler->NeedsFMDClustersBranchReplication()) && !(fgAODFMDClusters))
252 AliInfo("Replicating FMDClusters branch\n");
253 fgAODFMDClusters = new TClonesArray("AliAODFmdCluster",500);
254 fgAODFMDClusters->SetName("fmdClusters");
255 handler->AddBranch("TClonesArray", &fgAODFMDClusters);
257 if ((handler->NeedsCaloClustersBranchReplication() || merging) && !(fgAODCaloClusters))
259 if (fDebug > 1) AliInfo("Replicating CaloClusters branch\n");
260 fgAODCaloClusters = new TClonesArray("AliAODCaloCluster",500);
261 fgAODCaloClusters->SetName("caloClusters");
262 handler->AddBranch("TClonesArray", &fgAODCaloClusters);
264 fgAODEmcalCells = new AliAODCaloCells();
265 fgAODEmcalCells->SetName("emcalCells");
266 handler->AddBranch("AliAODCaloCells", &fgAODEmcalCells);
268 fgAODPhosCells = new AliAODCaloCells();
269 fgAODPhosCells->SetName("phosCells");
270 handler->AddBranch("AliAODCaloCells", &fgAODPhosCells);
272 if ((handler->NeedsMCParticlesBranchReplication() || merging) && !(fgAODMCParticles))
274 if (fDebug > 1) AliInfo("Replicating MCParticles branch\n");
275 fgAODMCParticles = new TClonesArray("AliAODMCParticle",500);
276 fgAODMCParticles->SetName("mcparticles");
277 handler->AddBranch("TClonesArray", &fgAODMCParticles);
279 // cache the pointerd in the AODEvent
280 fOutputAOD->GetStdContent();
283 AliWarning("No AOD Event Handler connected.") ;
285 UserCreateOutputObjects();
288 void AliAnalysisTaskSE::Exec(Option_t* option)
291 // Exec analysis of one event
292 if (fDebug > 1) AliInfo("AliAnalysisTaskSE::Exec() \n");
294 AliAODHandler* handler = (AliAODHandler*)
295 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
296 AliAODInputHandler* aodH = dynamic_cast<AliAODInputHandler*>(fInputHandler);
298 // Was event selected ?
299 Bool_t isSelected = kTRUE;
300 if( fInputHandler && fInputHandler->GetEventSelection() && fSelectCollisions) {
301 isSelected = fInputHandler->IsEventSelected();
304 if (handler) handler->SetFillAOD(isSelected);
306 if( fInputHandler ) fEntry = fInputHandler->GetReadEntry();
309 // Notify the change of run number
310 if (InputEvent()->GetRunNumber() != fCurrentRunNumber) {
311 fCurrentRunNumber = InputEvent()->GetRunNumber();
316 fEntry = fMCEvent->Header()->GetEvent();
317 if ( !((Entry()-1)%100) && fDebug > 0)
318 AliInfo(Form("%s ----> Processing event # %lld", CurrentFileName(), Entry()));
323 if (handler && aodH) {
324 fMCEvent = aodH->MCEvent();
325 Bool_t merging = aodH->GetMergeEvents();
327 if (!(handler->IsStandard()) && !(handler->AODIsReplicated())) {
328 if ((handler->NeedsHeaderReplication()) && (fgAODHeader))
330 // copy the contents by assigment
331 *fgAODHeader = *(dynamic_cast<AliAODHeader*>(InputEvent()->GetHeader()));
333 if ((handler->NeedsTracksBranchReplication() || merging) && (fgAODTracks))
335 TClonesArray* tracks = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetTracks();
336 new (fgAODTracks) TClonesArray(*tracks);
338 if ((handler->NeedsVerticesBranchReplication() || merging) && (fgAODVertices))
340 TClonesArray* vertices = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetVertices();
341 new (fgAODVertices) TClonesArray(*vertices);
343 if ((handler->NeedsV0sBranchReplication()) && (fgAODV0s))
345 TClonesArray* v0s = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetV0s();
346 new (fgAODV0s) TClonesArray(*v0s);
348 if ((handler->NeedsTrackletsBranchReplication()) && (fgAODTracklets))
350 *fgAODTracklets = *(dynamic_cast<AliAODEvent*>(InputEvent()))->GetTracklets();
352 if ((handler->NeedsPMDClustersBranchReplication()) && (fgAODPMDClusters))
354 TClonesArray* pmdClusters = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetPmdClusters();
355 new (fgAODPMDClusters) TClonesArray(*pmdClusters);
357 if ((handler->NeedsJetsBranchReplication() || merging) && (fgAODJets))
359 TClonesArray* jets = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetJets();
360 new (fgAODJets) TClonesArray(*jets);
362 if ((handler->NeedsFMDClustersBranchReplication()) && (fgAODFMDClusters))
364 TClonesArray* fmdClusters = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetFmdClusters();
365 new (fgAODFMDClusters) TClonesArray(*fmdClusters);
367 if ((handler->NeedsCaloClustersBranchReplication() || merging) && (fgAODCaloClusters))
369 TClonesArray* caloClusters = (dynamic_cast<AliAODEvent*>(InputEvent()))->GetCaloClusters();
370 new (fgAODCaloClusters) TClonesArray(*caloClusters);
373 if ((handler->NeedsMCParticlesBranchReplication() || merging) && (fgAODMCParticles))
375 TClonesArray* mcParticles = (TClonesArray*) ((dynamic_cast<AliAODEvent*>(InputEvent()))->FindListObject("mcparticles"));
376 new (fgAODMCParticles) TClonesArray(*mcParticles);
379 // Additional merging if needed
382 TClonesArray* mcparticles = (TClonesArray*) ((aodH->GetEventToMerge())->FindListObject("mcparticles"));
383 Int_t npart = mcparticles->GetEntries();
384 Int_t nc = fgAODMCParticles->GetEntries();
387 for (Int_t i = 0; i < npart; i++) {
388 AliAODMCParticle* particle = (AliAODMCParticle*) mcparticles->At(i);
389 new((*fgAODMCParticles)[nc++]) AliAODMCParticle(*particle);
393 TClonesArray* tracks = aodH->GetEventToMerge()->GetTracks();
394 Int_t ntr = tracks->GetEntries();
395 nc = fgAODTracks->GetEntries();
396 for (Int_t i = 0; i < ntr; i++) {
397 AliAODTrack* track = (AliAODTrack*) tracks->At(i);
398 AliAODTrack* newtrack = new((*fgAODTracks)[nc++]) AliAODTrack(*track);
400 newtrack->SetLabel(newtrack->GetLabel() + nc0);
403 for (Int_t i = 0; i < nc; i++)
405 AliAODTrack* track = (AliAODTrack*) fgAODTracks->At(i);
406 track->ResetBit(kIsReferenced);
407 track->SetUniqueID(0);
412 TClonesArray* clusters = aodH->GetEventToMerge()->GetCaloClusters();
413 Int_t ncl = clusters->GetEntries();
414 nc = fgAODCaloClusters->GetEntries();
415 for (Int_t i = 0; i < ncl; i++) {
416 AliAODCaloCluster* cluster = (AliAODCaloCluster*) clusters->At(i);
417 new((*fgAODCaloClusters)[nc++]) AliAODCaloCluster(*cluster);
420 AliAODCaloCells* cellsA = aodH->GetEventToMerge()->GetEMCALCells();
421 Int_t ncells = cellsA->GetNumberOfCells();
422 nc = fgAODEmcalCells->GetNumberOfCells();
424 for (Int_t i = 0; i < ncells; i++) {
425 Int_t cn = cellsA->GetCellNumber(i);
426 Int_t pos = fgAODEmcalCells->GetCellPosition(cn);
428 Double_t amp = cellsA->GetAmplitude(i) + fgAODEmcalCells->GetAmplitude(pos);
429 fgAODEmcalCells->SetCell(pos, cn, amp);
431 Double_t amp = cellsA->GetAmplitude(i);
432 fgAODEmcalCells->SetCell(nc++, cn, amp);
433 fgAODEmcalCells->Sort();
440 handler->SetAODIsReplicated();
444 // Call the user analysis
445 if (!fSelectCollisions || isSelected)
448 // Added protection in case the derived task is not an AOD producer.
449 AliAnalysisDataSlot *out0 = GetOutputSlot(0);
450 if (out0 && out0->IsConnected()) PostData(0, fTreeA);
453 const char* AliAnalysisTaskSE::CurrentFileName()
455 // Returns the current file name
457 return fInputHandler->GetTree()->GetCurrentFile()->GetName();
459 return ((AliMCEventHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler()))->TreeK()->GetCurrentFile()->GetName();
463 void AliAnalysisTaskSE::AddAODBranch(const char* cname, void* addobj, const char *fname)
465 // Add a new branch to the aod tree
466 AliAODHandler* handler = (AliAODHandler*)
467 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
469 handler->AddBranch(cname, addobj, fname);
473 Bool_t AliAnalysisTaskSE::IsStandardAOD() const
475 // Check if the output AOD handler is configured for standard or delta AOD.
476 // Users should first check that AODEvent() returns non-null.
477 AliAODHandler* handler = (AliAODHandler*)
478 ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
480 Error("IsStandardAOD", "No AOD handler. Please use AODEvent() to check this first");
483 return handler->IsStandard();
486 Bool_t AliAnalysisTaskSE::Notify()
488 return (UserNotify());