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 **************************************************************************/
16 //_________________________________________________________________________
17 // Analysis task that executes the analysis classes
18 // that depend on the CaloTrackCorr frame, frame for Particle identification
19 // with calorimeters and tracks 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
24 // -- Author: Gustavo Conesa (INFN-LNF, LPSC-Grenoble)
30 #include <TInterpreter.h>
31 #include <TClonesArray.h>
32 //#include <Riostream.h>
33 //#include <TObjectTable.h>
36 #include "AliAnalysisTaskCaloTrackCorrelationM.h"
37 #include "AliAnaCaloTrackCorrMaker.h"
38 #include "AliCaloTrackReader.h"
40 #include "AliAnalysisManager.h"
41 #include "AliMultiEventInputHandler.h"
42 #include "AliMixedEvent.h"
43 #include "AliAODEvent.h"
44 #include "AliESDEvent.h"
45 #include "AliAnalysisDataSlot.h"
47 ClassImp(AliAnalysisTaskCaloTrackCorrelationM)
49 //__________________________________________________________________________
50 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM() :
53 fOutputContainer(0x0),
58 // Default constructor
61 //__________________________________________________________________________________________
62 AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM(const char* name) :
63 AliAnalysisTaskME(name),
65 fOutputContainer(0x0),
70 // Default constructor
72 DefineOutput(1, TList::Class());
73 DefineOutput(2, TList::Class()); // will contain cuts or local params
76 //___________________________________________________________________________
77 AliAnalysisTaskCaloTrackCorrelationM::~AliAnalysisTaskCaloTrackCorrelationM()
79 // Remove all pointers
81 // // Do not delete it here, already done somewhere else, need to understand where.
82 // if(fOutputContainer && !AliAnalysisManager::GetAnalysisManager()->IsProofMode()){
83 // fOutputContainer->Clear() ;
84 // delete fOutputContainer ;
92 //__________________________________________________________________
93 void AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects()
95 // Create the output container
96 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - Begin\n");
98 //Get list of aod arrays, add each aod array to analysis frame
99 TClonesArray *array = 0;
100 TList * list = fAna->FillAndGetAODBranchList();
101 if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries());
103 //Put the delta AODs in output file, std or delta
104 if((fAna->GetReader())->WriteDeltaAODToFile()){
105 TString deltaAODName = (fAna->GetReader())->GetDeltaAODFileName();
106 for(Int_t iaod = 0; iaod < list->GetEntries(); iaod++){
107 array = (TClonesArray*) list->At(iaod);
108 if(deltaAODName!="") AddAODBranch("TClonesArray", &array, deltaAODName);//Put it in DeltaAOD file
109 else AddAODBranch("TClonesArray", &array);//Put it in standard AOD file
113 //Histograms container
115 fOutputContainer = fAna->GetOutputContainer();
117 if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries());
119 fOutputContainer->SetOwner(kTRUE);
121 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - End\n");
123 PostData(1,fOutputContainer);
127 //____________________________________________________
128 void AliAnalysisTaskCaloTrackCorrelationM::LocalInit()
130 // Local Initialization
132 //Call the Init to initialize the configuration of the analysis
135 // Create cuts/param objects and publish to slot
136 fCuts = fAna->GetListOfAnalysisCuts();
137 fCuts ->SetOwner(kTRUE);
144 //_______________________________________________
145 void AliAnalysisTaskCaloTrackCorrelationM::Init()
149 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Begin\n");
151 fInputEvent = new AliMixedEvent() ;
153 // Call configuration file if specified
155 if (fConfigName.Length()) {
156 printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data());
157 gROOT->LoadMacro(fConfigName+".C");
158 fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()");
162 printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n");
166 // Add different generator particles to PDG Data Base
167 // to avoid problems when reading MC generator particles
168 AliPDG::AddParticlesToPdgDataBase();
170 //Set in the reader the name of the task in case is needed
171 (fAna->GetReader())->SetTaskName(GetName());
173 // Initialise analysis
177 if((fAna->GetReader())->GetDeltaAODFileName()!="")
178 AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName());
180 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - End\n");
185 //_______________________________________________________________________
186 void AliAnalysisTaskCaloTrackCorrelationM::UserExec(Option_t */*option*/)
188 // Execute analysis for current event
190 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Begin\n");
192 //Get the type of data, check if type is correct
193 Int_t datatype = fAna->GetReader()->GetDataType();
194 if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD &&
195 datatype != AliCaloTrackReader::kMC){
196 printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Wrong type of data\n");
200 Int_t nev = fInputHandler->GetBufferSize();
201 fInputEvent->Reset();
203 for (Int_t iev = 0; iev < nev; iev++) {
204 if (datatype == AliCaloTrackReader::kESD) {
205 AliESDEvent* esd = dynamic_cast<AliESDEvent*>(GetEvent(iev));
206 fInputEvent->AddEvent(esd);
207 } else if (datatype == AliCaloTrackReader::kAOD) {
208 AliAODEvent* aod = dynamic_cast<AliAODEvent*>(GetEvent(iev));
209 fInputEvent->AddEvent(aod);
211 AliFatal("need to implement mixed event for MC") ;
217 fAna->GetReader()->SetInputOutputMCEvent(fInputEvent, AODEvent(), MCEvent());
220 fAna->ProcessEvent((Int_t) Entry(), CurrentFileName());
222 //printf("AliAnalysisTaskCaloTrackCorrelationM::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName());
223 if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - End\n");
225 PostData(1, fOutputContainer);
227 AliAnalysisDataSlot *out0 = GetOutputSlot(0);
228 if (out0 && out0->IsConnected()) PostData(0, fTreeA);
230 //gObjectTable->Print();
234 //________________________________________________________________________
235 void AliAnalysisTaskCaloTrackCorrelationM::Terminate(Option_t */*option*/)
237 // Terminate analysis
240 // Get merged histograms from the output container
241 // Propagate histagrams to maker
242 fAna->Terminate((TList*)GetOutputData(1));