]>
Commit | Line | Data |
---|---|---|
c8fe2783 | 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 | **************************************************************************/ | |
c8fe2783 | 15 | |
16 | //_________________________________________________________________________ | |
17 | // Analysis task that executes the analysis classes | |
f15155ed | 18 | // that depend on the CaloTrackCorr frame, frame for Particle identification |
19 | // with calorimeters and tracks and correlations. | |
c8fe2783 | 20 | // Specially designed for calorimeters but also can be used for charged tracks |
f15155ed | 21 | // Input of this task is a configuration file that contains all the settings |
22 | // of the analysis | |
c8fe2783 | 23 | // |
f15155ed | 24 | // -- Author: Gustavo Conesa (INFN-LNF, LPSC-Grenoble) |
c8fe2783 | 25 | |
26 | #include <cstdlib> | |
27 | ||
28 | // --- Root --- | |
29 | #include <TROOT.h> | |
30 | #include <TInterpreter.h> | |
31 | #include <TClonesArray.h> | |
32 | //#include <Riostream.h> | |
33 | //#include <TObjectTable.h> | |
34 | ||
35 | // --- Analysis --- | |
f15155ed | 36 | #include "AliAnalysisTaskCaloTrackCorrelationM.h" |
37 | #include "AliAnaCaloTrackCorrMaker.h" | |
c8fe2783 | 38 | #include "AliCaloTrackReader.h" |
39 | #include "AliPDG.h" | |
40 | #include "AliAnalysisManager.h" | |
41 | #include "AliMultiEventInputHandler.h" | |
42 | #include "AliMixedEvent.h" | |
43 | #include "AliAODEvent.h" | |
44 | #include "AliESDEvent.h" | |
f2ccb5b8 | 45 | #include "AliAnalysisDataSlot.h" |
c8fe2783 | 46 | |
f15155ed | 47 | ClassImp(AliAnalysisTaskCaloTrackCorrelationM) |
c8fe2783 | 48 | |
f15155ed | 49 | //__________________________________________________________________________ |
50 | AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM() : | |
c8fe2783 | 51 | AliAnalysisTaskME(), |
52 | fAna(0x0), | |
53 | fOutputContainer(0x0), | |
f15155ed | 54 | fConfigName(""), |
55 | fCuts(0x0), | |
56 | fInputEvent(NULL) | |
c8fe2783 | 57 | { |
58 | // Default constructor | |
59 | } | |
60 | ||
f15155ed | 61 | //__________________________________________________________________________________________ |
62 | AliAnalysisTaskCaloTrackCorrelationM::AliAnalysisTaskCaloTrackCorrelationM(const char* name) : | |
c8fe2783 | 63 | AliAnalysisTaskME(name), |
64 | fAna(0x0), | |
65 | fOutputContainer(0x0), | |
f15155ed | 66 | fConfigName(""), |
67 | fCuts(0x0), | |
68 | fInputEvent(NULL) | |
c8fe2783 | 69 | { |
70 | // Default constructor | |
71 | ||
72 | DefineOutput(1, TList::Class()); | |
73 | DefineOutput(2, TList::Class()); // will contain cuts or local params | |
74 | } | |
75 | ||
f15155ed | 76 | //___________________________________________________________________________ |
77 | AliAnalysisTaskCaloTrackCorrelationM::~AliAnalysisTaskCaloTrackCorrelationM() | |
c8fe2783 | 78 | { |
79 | // Remove all pointers | |
80 | ||
f15155ed | 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 ; | |
85 | // } | |
86 | ||
c8fe2783 | 87 | delete fInputEvent ; |
88 | if(fAna) delete fAna; | |
f15155ed | 89 | |
c8fe2783 | 90 | } |
91 | ||
f15155ed | 92 | //__________________________________________________________________ |
93 | void AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() | |
c8fe2783 | 94 | { |
95 | // Create the output container | |
f15155ed | 96 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - Begin\n"); |
c8fe2783 | 97 | |
98 | //Get list of aod arrays, add each aod array to analysis frame | |
99 | TClonesArray *array = 0; | |
f37fa8d2 | 100 | TList * list = fAna->FillAndGetAODBranchList(); |
f15155ed | 101 | if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n AOD branches %d\n",list->GetEntries()); |
102 | ||
f37fa8d2 | 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 | |
110 | } | |
111 | } | |
c8fe2783 | 112 | |
113 | //Histograms container | |
114 | OpenFile(1); | |
115 | fOutputContainer = fAna->GetOutputContainer(); | |
f37fa8d2 | 116 | |
f15155ed | 117 | if (DebugLevel() >= 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - n histograms %d\n",fOutputContainer->GetEntries()); |
f37fa8d2 | 118 | |
c8fe2783 | 119 | fOutputContainer->SetOwner(kTRUE); |
f37fa8d2 | 120 | |
f15155ed | 121 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserCreateOutputObjects() - End\n"); |
122 | ||
c8fe2783 | 123 | PostData(1,fOutputContainer); |
124 | ||
125 | } | |
f15155ed | 126 | |
127 | //____________________________________________________ | |
128 | void AliAnalysisTaskCaloTrackCorrelationM::LocalInit() | |
c8fe2783 | 129 | { |
130 | // Local Initialization | |
131 | ||
132 | //Call the Init to initialize the configuration of the analysis | |
133 | Init(); | |
134 | ||
135 | // Create cuts/param objects and publish to slot | |
136 | fCuts = fAna->GetListOfAnalysisCuts(); | |
138f531a | 137 | fCuts ->SetOwner(kTRUE); |
f15155ed | 138 | |
c8fe2783 | 139 | // Post Data |
140 | PostData(2, fCuts); | |
141 | ||
142 | } | |
143 | ||
f15155ed | 144 | //_______________________________________________ |
145 | void AliAnalysisTaskCaloTrackCorrelationM::Init() | |
c8fe2783 | 146 | { |
147 | // Initialization | |
f15155ed | 148 | |
149 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Begin\n"); | |
c8fe2783 | 150 | |
151 | fInputEvent = new AliMixedEvent() ; | |
152 | ||
153 | // Call configuration file if specified | |
154 | ||
155 | if (fConfigName.Length()) { | |
f15155ed | 156 | printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - ### Configuration file is %s.C ###\n", fConfigName.Data()); |
157 | gROOT->LoadMacro(fConfigName+".C"); | |
158 | fAna = (AliAnaCaloTrackCorrMaker*) gInterpreter->ProcessLine("ConfigAnalysis()"); | |
c8fe2783 | 159 | } |
160 | ||
161 | if(!fAna) { | |
f15155ed | 162 | printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - Analysis maker pointer not initialized, no analysis specified, STOP !\n"); |
163 | abort(); | |
c8fe2783 | 164 | } |
165 | ||
166 | // Add different generator particles to PDG Data Base | |
167 | // to avoid problems when reading MC generator particles | |
168 | AliPDG::AddParticlesToPdgDataBase(); | |
f15155ed | 169 | |
c8fe2783 | 170 | //Set in the reader the name of the task in case is needed |
171 | (fAna->GetReader())->SetTaskName(GetName()); | |
172 | ||
173 | // Initialise analysis | |
174 | fAna->Init(); | |
f15155ed | 175 | |
c8fe2783 | 176 | //Delta AOD |
177 | if((fAna->GetReader())->GetDeltaAODFileName()!="") | |
178 | AliAnalysisManager::GetAnalysisManager()->RegisterExtraFile((fAna->GetReader())->GetDeltaAODFileName()); | |
f15155ed | 179 | |
180 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::Init() - End\n"); | |
c8fe2783 | 181 | |
182 | } | |
183 | ||
184 | ||
f15155ed | 185 | //_______________________________________________________________________ |
186 | void AliAnalysisTaskCaloTrackCorrelationM::UserExec(Option_t */*option*/) | |
c8fe2783 | 187 | { |
188 | // Execute analysis for current event | |
f15155ed | 189 | |
190 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Begin\n"); | |
191 | ||
192 | //Get the type of data, check if type is correct | |
c8fe2783 | 193 | Int_t datatype = fAna->GetReader()->GetDataType(); |
194 | if(datatype != AliCaloTrackReader::kESD && datatype != AliCaloTrackReader::kAOD && | |
195 | datatype != AliCaloTrackReader::kMC){ | |
f15155ed | 196 | printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - Wrong type of data\n"); |
c8fe2783 | 197 | return ; |
198 | } | |
f15155ed | 199 | |
c8fe2783 | 200 | Int_t nev = fInputHandler->GetBufferSize(); |
201 | fInputEvent->Reset(); | |
202 | ||
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); | |
210 | } else { | |
211 | AliFatal("need to implement mixed event for MC") ; | |
212 | } | |
f15155ed | 213 | } |
214 | ||
c8fe2783 | 215 | fInputEvent->Init(); |
f15155ed | 216 | |
c8fe2783 | 217 | fAna->GetReader()->SetInputOutputMCEvent(fInputEvent, AODEvent(), MCEvent()); |
f15155ed | 218 | |
c8fe2783 | 219 | //Process event |
220 | fAna->ProcessEvent((Int_t) Entry(), CurrentFileName()); | |
f15155ed | 221 | |
222 | //printf("AliAnalysisTaskCaloTrackCorrelationM::Current Event %d; Current File Name : %s\n",(Int_t) Entry(), CurrentFileName()); | |
223 | if (DebugLevel() > 1) printf("AliAnalysisTaskCaloTrackCorrelationM::UserExec() - End\n"); | |
224 | ||
c8fe2783 | 225 | PostData(1, fOutputContainer); |
226 | ||
f2ccb5b8 | 227 | AliAnalysisDataSlot *out0 = GetOutputSlot(0); |
228 | if (out0 && out0->IsConnected()) PostData(0, fTreeA); | |
229 | ||
c8fe2783 | 230 | //gObjectTable->Print(); |
f15155ed | 231 | |
c8fe2783 | 232 | } |
233 | ||
f15155ed | 234 | //________________________________________________________________________ |
235 | void AliAnalysisTaskCaloTrackCorrelationM::Terminate(Option_t */*option*/) | |
c8fe2783 | 236 | { |
237 | // Terminate analysis | |
238 | // Do some plots | |
f15155ed | 239 | |
c8fe2783 | 240 | // Get merged histograms from the output container |
241 | // Propagate histagrams to maker | |
242 | fAna->Terminate((TList*)GetOutputData(1)); | |
243 | ||
244 | } | |
245 |