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 **************************************************************************/
17 //_________________________________________________________________________
18 // Analysis task that executes the analysis classes
19 // that depend on the PartCorr frame, frame for Particle identification and correlations.
20 // Specially designed for calorimeters but also can be used for charged tracks
21 // Input of this task is a configuration file that contains all the settings of the analyis
23 // -- Author: Gustavo Conesa (INFN-LNF)
29 #include <TInterpreter.h>
30 #include <TClonesArray.h>
31 //#include <Riostream.h>
32 //#include <TObjectTable.h>
35 #include "AliAnalysisTaskParticleCorrelationM.h"
36 #include "AliAnaPartCorrMaker.h"
37 #include "AliCaloTrackReader.h"
39 #include "AliAnalysisManager.h"
40 #include "AliMultiEventInputHandler.h"
41 #include "AliMixedEvent.h"
42 #include "AliAODEvent.h"
43 #include "AliESDEvent.h"
45 ClassImp(AliAnalysisTaskParticleCorrelationM)
47 ////////////////////////////////////////////////////////////////////////
48 AliAnalysisTaskParticleCorrelationM::AliAnalysisTaskParticleCorrelationM():
51 fOutputContainer(0x0),
52 fConfigName(""), fCuts(0x0), fInputEvent(NULL)
54 // Default constructor
57 //_____________________________________________________
58 AliAnalysisTaskParticleCorrelationM::AliAnalysisTaskParticleCorrelationM(const char* name):
59 AliAnalysisTaskME(name),
61 fOutputContainer(0x0),
62 fConfigName(""), fCuts(0x0), fInputEvent(NULL)
64 // Default constructor
66 DefineOutput(1, TList::Class());
67 DefineOutput(2, TList::Class()); // will contain cuts or local params
70 //_____________________________________________________
71 AliAnalysisTaskParticleCorrelationM::~AliAnalysisTaskParticleCorrelationM()
73 // Remove all pointers
75 // printf("********** Delete Task\n");
76 // // Do not delete it here, already done somewhere else, need to understand where.
77 // if(fOutputContainer){
78 // fOutputContainer->Clear() ;
79 // delete fOutputContainer ;
85 // printf("********** Task deleted \n");
88 //_____________________________________________________
89 void AliAnalysisTaskParticleCorrelationM::UserCreateOutputObjects()
91 // Create the output container
92 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::UserCreateOutputObjects() - Begin\n");
94 //Get list of aod arrays, add each aod array to analysis frame
95 TClonesArray *array = 0;
96 TList * list = fAna->FillAndGetAODBranchList();
97 if (DebugLevel() >= 1) printf("AliAnalysisTaskParticleCorrelationM::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
99 //Put the delta AODs in output file, std or delta
100 if((fAna->GetReader())->WriteDeltaAODToFile()){
101 TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
102 for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++){
103 array = (TClonesArray*) list->At(iaod);
104 if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
105 else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
109 //Histograms container
111 fOutputContainer = fAna->GetOutputContainer();
113 if (DebugLevel() >= 1) printf("AliAnalysisTaskParticleCorrelationM::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
115 fOutputContainer->SetOwner(kTRUE);
117 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::UserCreateOutputObjects() - End\n");
119 PostData(1,fOutputContainer);
122 //_____________________________________________________
123 void AliAnalysisTaskParticleCorrelationM::LocalInit()
125 // Local Initialization
127 //Call the Init to initialize the configuration of the analysis
130 // Create cuts/param objects and publish to slot
131 fCuts = fAna->GetListOfAnalysisCuts();
138 //_____________________________________________________
139 void AliAnalysisTaskParticleCorrelationM::Init()
143 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::Init() - Begin\n");
145 fInputEvent = new AliMixedEvent() ;
147 // Call configuration file if specified
149 if (fConfigName.Length()) {
150 printf("AliAnalysisTaskParticleCorrelationM::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
151 gROOT->LoadMacro(fConfigName+".C");
152 fAna = (AliAnaPartCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
156 printf("AliAnalysisTaskParticleCorrelationM::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
160 // Add different generator particles to PDG Data Base
161 // to avoid problems when reading MC generator particles
162 AliPDG::AddParticlesToPdgDataBase();
164 //Set in the reader the name of the task in case is needed
165 (fAna->GetReader())->SetTaskName(GetName());
167 // Initialise analysis
171 if((fAna->GetReader())->GetDeltaAODFileName()!="")
172 AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
174 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::Init() - End\n");
179 //_____________________________________________________
180 void AliAnalysisTaskParticleCorrelationM::UserExec(Option_t */*option*/)
182 // Execute analysis for current event
184 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::UserExec() - Begin\n");
186 //Get the type of data, check if type is correct
187 Int_t datatype = fAna->GetReader()->GetDataType();
188 if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
189 datatype != AliCaloTrackReader::kMC){
190 printf("AliAnalysisTaskParticleCorrelationM::UserExec() - Wrong type of data\n");
194 Int_t nev = fInputHandler->GetBufferSize();
195 fInputEvent->Reset();
197 for (Int_t iev = 0; iev < nev; iev++) {
198 if (datatype == AliCaloTrackReader::kESD) {
199 AliESDEvent* esd = dynamic_cast<AliESDEvent*>(GetEvent(iev));
200 fInputEvent->AddEvent(esd);
201 } else if (datatype == AliCaloTrackReader::kAOD) {
202 AliAODEvent* aod = dynamic_cast<AliAODEvent*>(GetEvent(iev));
203 fInputEvent->AddEvent(aod);
205 AliFatal("need to implement mixed event for MC") ;
210 fAna->GetReader()->SetInputOutputMCEvent(fInputEvent, AODEvent(), MCEvent());
213 fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
214 //printf("AliAnalysisTaskParticleCorrelationM::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
215 if (DebugLevel() > 1) printf("AliAnalysisTaskParticleCorrelationM::UserExec() - End\n");
217 PostData(1, fOutputContainer);
219 //gObjectTable->Print();
224 //_____________________________________________________
225 void AliAnalysisTaskParticleCorrelationM::Terminate(Option_t */*option*/)
227 // Terminate analysis
230 // Get merged histograms from the output container
231 // Propagate histagrams to maker
232 fAna->Terminate((TList*)GetOutputData(1));