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 /* History of cvs commits:
20 * Revision 1.4 2007/10/29 13:48:42 gustavo
21 * Corrected coding violations
23 * Revision 1.2 2007/08/17 12:40:04 schutz
24 * New analysis classes by Gustavo Conesa
26 * Revision 1.1.2.1 2007/07/26 10:32:09 schutz
27 * new analysis classes in the the new analysis framework
32 //_________________________________________________________________________
33 // Base class for gamma and correlation analysis
34 // It is called by the task class AliAnalysisGammaTask and it connects the input (ESD/AOD/MonteCarlo)
35 // got with AliGammaReader (produces TClonesArrays of TParticles), with the analysis classes
36 // AliAnaGammaDirect, AliAnaGammaCorrelation ....
38 //*-- Author: Gustavo Conesa (INFN-LNF)
40 // --- ROOT system ---
42 #include <TParticle.h>
45 //---- AliRoot system ----
46 #include "AliAnaGamma.h"
47 #include "AliGammaReader.h"
48 #include "AliAnaGammaDirect.h"
49 #include "AliAnaGammaCorrelation.h"
50 #include "AliAnaGammaSelection.h"
51 #include "AliNeutralMesonSelection.h"
52 #include "AliAODCaloCluster.h"
53 #include "AliAODEvent.h"
54 #include "Riostream.h"
60 //____________________________________________________________________________
61 AliAnaGamma::AliAnaGamma() :
63 fOutputContainer(0x0),
64 fAnaType(0), fCalorimeter(0), fData(0x0), fKine(0x0),
65 fReader(0x0), fGammaDirect(0x0), fGammaCorrelation(0x0), fGammaSelection(0x0),
66 fNeutralMesonSelection(0x0), fAODclusters(0x0), fNAODclusters(0)
70 //Initialize parameters, pointers and histograms
72 fReader = new AliGammaReader();
74 fGammaDirect = new AliAnaGammaDirect();
75 if(!fGammaCorrelation)
76 fGammaCorrelation = new AliAnaGammaCorrelation();
78 fGammaSelection = new AliAnaGammaSelection();
79 if(!fNeutralMesonSelection)
80 fNeutralMesonSelection = new AliNeutralMesonSelection();
88 //____________________________________________________________________________
89 AliAnaGamma::AliAnaGamma(const AliAnaGamma & g) :
91 fOutputContainer(g. fOutputContainer),
92 fAnaType(g.fAnaType), fCalorimeter(g.fCalorimeter),
93 fData(g.fData), fKine(g.fKine),fReader(g.fReader),
94 fGammaDirect(g.fGammaDirect), fGammaCorrelation(g.fGammaCorrelation),
95 fGammaSelection(g.fGammaSelection),
96 fNeutralMesonSelection(g.fNeutralMesonSelection),
97 fAODclusters(g. fAODclusters), fNAODclusters(g.fNAODclusters)
103 //_________________________________________________________________________
104 AliAnaGamma & AliAnaGamma::operator = (const AliAnaGamma & source)
106 // assignment operator
108 if(this == &source)return *this;
109 ((TObject *)this)->operator=(source);
111 fOutputContainer = source.fOutputContainer ;
112 fAnaType = source.fAnaType;
113 fCalorimeter = source.fCalorimeter ;
114 fData = source.fData ;
115 fKine = source.fKine ;
116 fReader = source.fReader ;
117 fGammaDirect = source.fGammaDirect ;
118 fGammaCorrelation = source.fGammaCorrelation ;
119 fGammaSelection = source.fGammaSelection ;
120 fNeutralMesonSelection = source.fNeutralMesonSelection ;
126 //____________________________________________________________________________
127 AliAnaGamma::~AliAnaGamma()
129 // Remove all pointers.
131 // Protection added in case of NULL pointers (MG)
132 if (fOutputContainer) {
133 fOutputContainer->Clear();
134 delete fOutputContainer ;
137 if (fData) delete fData ;
138 if (fKine) delete fKine ;
139 if (fReader) delete fReader ;
140 if (fGammaDirect) delete fGammaDirect ;
141 if (fGammaCorrelation) delete fGammaCorrelation ;
142 if (fGammaSelection) delete fGammaSelection ;
143 if (fNeutralMesonSelection) delete fNeutralMesonSelection ;
147 //________________________________________________________________________
148 void AliAnaGamma::Init()
151 //Init container histograms and other common variables
153 //Histograms container
154 fOutputContainer = new TList ;
156 //Fill container with appropriate histograms
158 //Set selection analysis histograms
159 TList * selectcontainer = fGammaSelection->GetCreateOutputObjects();
160 for(Int_t i = 0; i < selectcontainer->GetEntries(); i++){
162 TString name = (selectcontainer->At(i))->GetName();
163 if(!fReader->IsEMCALOn() && name.Contains("EMCAL")) add = kFALSE;
164 if(!fReader->IsPHOSOn() && name.Contains("PHOS")) add = kFALSE;
165 if(!fReader->IsCTSOn() && !fGammaSelection->FillCTS() && name.Contains("CTS")) add = kFALSE;
166 if(add) fOutputContainer->Add(selectcontainer->At(i)) ;
167 } //Set selection analysis histograms
170 //Set prompt photon analysis histograms
171 TList * promptcontainer = fGammaDirect->GetCreateOutputObjects();
172 for(Int_t i = 0; i < promptcontainer->GetEntries(); i++)
173 fOutputContainer->Add(promptcontainer->At(i)) ;
175 //Check if selected options are correct or set them when necessary
176 if(fReader->GetDataType() == AliGammaReader::kMCData){
177 fGammaDirect->SetMC();//Only useful with AliGammaMCDataReader, by default kFALSE
178 fGammaSelection->SetMC();//Only useful with AliGammaMCDataReader, by default kFALSE
181 if(fAnaType == kCorrelation){
183 //Check if selected options are correct
184 if (fGammaDirect->GetICMethod()==AliAnaGammaDirect::kSeveralIC)
185 AliFatal("Correlation not allowed with multiple isolation cuts, kCorrelation and kSeveralIC do not go together");
187 if(fGammaCorrelation->GetCorrelationType() == AliAnaGammaCorrelation::kParton &&
188 fReader->GetDataType() != AliGammaReader::kMC)
189 AliFatal("kParton must be analyzed with data kMC");
191 //Set the parameters for the neutral pair selection depending on the analysis,
192 fNeutralMesonSelection->SetDeltaPhiCutRange(fGammaCorrelation->GetDeltaPhiMinCut(),
193 fGammaCorrelation->GetDeltaPhiMaxCut());
196 if(fGammaCorrelation->GetCorrelationType() == AliAnaGammaCorrelation::kHadron){
197 fNeutralMesonSelection->SetPhiPtSelection(AliNeutralMesonSelection::kSelectPhiMinPt);
198 fNeutralMesonSelection->SetMinPt(fGammaCorrelation->GetMinPtHadron());
202 if(fGammaCorrelation->GetCorrelationType() == AliAnaGammaCorrelation::kJetLeadCone){
203 fNeutralMesonSelection->SetPhiPtSelection(AliNeutralMesonSelection::kSelectPhiPtRatio);
204 fNeutralMesonSelection->SetRatioCutRange(fGammaCorrelation->GetRatioMinCut(),
205 fGammaCorrelation->GetRatioMaxCut());
208 //Set the neutral mesosn selection histograms
209 TList * neutralmesoncontainer = fNeutralMesonSelection->GetCreateOutputObjects();
210 if(fNeutralMesonSelection->AreNeutralMesonSelectionHistosKept()){
211 for(Int_t i = 0; i < neutralmesoncontainer->GetEntries(); i++)
212 fOutputContainer->Add(neutralmesoncontainer->At(i)) ;
215 //Set correlation histograms
216 TList * correlationcontainer = fGammaCorrelation->GetCreateOutputObjects();
217 for(Int_t i = 0; i < correlationcontainer->GetEntries(); i++)
218 fOutputContainer->Add(correlationcontainer->At(i)) ;
219 fGammaCorrelation->SetOutputContainer(fOutputContainer);
220 fGammaCorrelation->SetNeutralMesonSelection(fNeutralMesonSelection);
226 //____________________________________________________________________________
227 void AliAnaGamma::InitParameters()
233 fCalorimeter = "EMCAL";
237 //__________________________________________________________________
238 void AliAnaGamma::Print(const Option_t * opt) const
241 //Print some relevant parameters set for the analysis
245 Info("Print", "%s %s", GetName(), GetTitle() ) ;
246 printf("Analysis type = %d\n", fAnaType) ;
247 printf("Calorimeter = %s\n", fCalorimeter.Data()) ;
253 fGammaDirect->Print("");
254 }// case kIsolationCut
259 fGammaCorrelation->Print("");
260 }// case kCorrelation
267 //____________________________________________________________________________
268 Bool_t AliAnaGamma::ProcessEvent(Long64_t entry){
270 AliDebug(1,Form("Entry %d",entry));
271 //cout<<"Event >>>>>>>>>>>>> "<<entry<<endl;
272 if(!fOutputContainer)
273 AliFatal("Histograms not initialized");
275 //CreateTLists with arrays of TParticles. Filled with particles only relevant for the analysis.
277 TClonesArray * plCTS = new TClonesArray("TParticle",1000); // All particles refitted in Central Tracking System (ITS+TPC)
278 TClonesArray * plEMCAL = new TClonesArray("TParticle",1000); // All particles measured in Jet Calorimeter (EMCAL)
279 TClonesArray * plPHOS = new TClonesArray("TParticle",1000); // All particles measured Gamma calorimeter
280 TClonesArray * plPrimCTS = new TClonesArray("TParticle",1000); // primary tracks
281 TClonesArray * plPrimEMCAL = new TClonesArray("TParticle",1000); // primary emcal clusters
282 TClonesArray * plPrimPHOS = new TClonesArray("TParticle",1000); // primary phos clusters
283 TClonesArray * plParton = new TClonesArray("TParticle",1000); // All partons
284 //Fill lists with photons, neutral particles and charged particles
285 //look for the highest energy photon in the event inside fCalorimeter
287 //Fill particle lists
288 if(fReader->GetDataType() == AliGammaReader::kData){
289 AliDebug(1,"Data analysis");
290 fReader->CreateParticleList(fData, NULL,plCTS,plEMCAL,plPHOS,NULL,NULL,NULL);
292 else if( fReader->GetDataType()== AliGammaReader::kMC){
293 AliDebug(1,"Kinematics analysis");
294 fReader->CreateParticleList(fKine, NULL,plCTS,plEMCAL,plPHOS,plParton,NULL,NULL);
296 else if(fReader->GetDataType() == AliGammaReader::kMCData) {
297 AliDebug(1,"Data + Kinematics analysis");
298 fReader->CreateParticleList(fData, fKine,plCTS,plEMCAL,plPHOS,plPrimCTS,plPrimEMCAL,plPrimPHOS);
301 AliError("Option not implemented");
303 //Fill AOD with calorimeter
304 //Temporal solution, just for testing
305 //FillAODs(plPHOS,plEMCAL);
307 //Select particles to do the final analysis.
308 if(fReader->IsEMCALOn()) fGammaSelection->Selection("EMCAL",plEMCAL,plPrimEMCAL);
309 if(fReader->IsPHOSOn()) fGammaSelection->Selection("PHOS",plPHOS,plPrimPHOS);
310 if(fReader->IsCTSOn() && fGammaSelection->FillCTS()) fGammaSelection->Selection("CTS",plCTS,plPrimCTS);
313 //Search highest energy prompt gamma in calorimeter
314 if(fCalorimeter == "PHOS")
315 MakeAnalysis(plPHOS, plEMCAL, plCTS, plParton, plPrimPHOS) ;
316 else if (fCalorimeter == "EMCAL")
317 MakeAnalysis(plEMCAL, plPHOS, plCTS,plParton, plPrimEMCAL) ;
319 AliFatal("Wrong calorimeter name");
326 plPrimEMCAL->Clear() ;
327 plPrimPHOS->Clear() ;
341 //____________________________________________________________________________
342 void AliAnaGamma::MakeAnalysis(TClonesArray * plCalo, TClonesArray * plNe, TClonesArray * plCTS, TClonesArray * plParton, TClonesArray * plPrimCalo) {
344 TParticle * pGamma = new TParticle ;
345 Bool_t isInCalo = kFALSE ;
350 //Only Prompt photon analysis
353 AliDebug(1,"kPrompt analysis");
354 switch(fGammaDirect->GetICMethod())
357 case AliAnaGammaDirect::kSeveralIC:
359 fGammaDirect->MakeSeveralICAnalysis(plCalo, plCTS);
360 AliDebug(1,"kSeveralIC analysis");
366 fGammaDirect->GetPromptGamma(plCalo, plCTS,plPrimCalo, pGamma,isInCalo);
368 AliDebug(1,"Prompt gamma not found");
375 //Correlate prompt photon with something: parton, hadron, jet.
378 AliDebug(1,"kCorrelation analysis");
380 fGammaDirect->GetPromptGamma(plCalo, plCTS,plPrimCalo, pGamma,isInCalo);
382 if(isInCalo){//If prompt photon found, do correlation
384 switch(fGammaCorrelation->GetCorrelationType())
386 case AliAnaGammaCorrelation::kParton:
388 AliDebug(1,"kParton correlation");
389 fGammaCorrelation->MakeGammaCorrelation(pGamma, plParton, NULL);
393 case AliAnaGammaCorrelation::kHadron:
395 AliDebug(1,"kHadron correlation");
396 fGammaCorrelation->MakeGammaCorrelation(pGamma, plCTS, plNe);
400 case AliAnaGammaCorrelation::kJetLeadCone:
402 AliDebug(1,"kJetLeadCone correlation");
403 fGammaCorrelation->MakeGammaCorrelation(pGamma, plCTS, plNe);
404 }// case kJetLeadCone
407 case AliAnaGammaCorrelation::kJetFinder:
409 AliDebug(1,"kJetFinder correlation");
410 printf("Analysis not implemented \n");
413 }// switch correlation
415 else AliDebug(2,"Prompt gamma not found");
416 }// case kCorrelation
425 //____________________________________________________
426 void AliAnaGamma::AddCluster(AliAODCaloCluster p)
428 // Add new jet to the list
429 cout<<"AOD list pointer "<<fAODclusters<<" nAODclusters "<<fNAODclusters<<endl;
430 cout<<"list entries "<<fAODclusters->GetEntries()<<endl;
431 new ((*fAODclusters)[fNAODclusters++]) AliAODCaloCluster(p);
434 //___________________________________________________
435 void AliAnaGamma::ConnectAOD(AliAODEvent* aod)
437 // Connect to the AOD
438 fAODclusters = aod->GetCaloClusters();
441 //____________________________________________________________________________
442 void AliAnaGamma::FillAODs(TClonesArray * plPHOS, TClonesArray * plEMCAL){
443 //Fill AOD caloClusters
444 //Temporal method, just for testing AOD creation
446 //Fill AOD with PHOS clusters
447 Int_t nphos = plPHOS->GetEntries() ;
448 cout<<"PHOS entries "<<nphos<<endl;
449 for(Int_t ipr = 0;ipr < nphos ; ipr ++ ){
450 TParticle * particle = dynamic_cast<TParticle *>(plPHOS->At(ipr)) ;
451 Float_t pos []= {0,0,0};
452 AliAODCaloCluster phos(ipr,0,0x0,particle->Energy(),pos,0x0,AliAODCluster::kPHOSNeutral,0);
456 //Fill AOD with EMCAL clusters
457 cout<<"EMCAL entries "<<plEMCAL->GetEntries()<<endl;
458 for(Int_t ipr = 0;ipr < plEMCAL->GetEntries() ; ipr ++ ){
459 TParticle * particle = dynamic_cast<TParticle *>(plEMCAL->At(ipr)) ;
460 Float_t pos []= {0,0,0};
461 AliAODCaloCluster emcal(ipr+nphos,0,0x0,particle->Energy(),pos,0x0,AliAODCluster::kEMCALClusterv1,0);