]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG4/AliAnalysisTaskPHOSExample.cxx
appropriately modified builders for package and PAR library
[u/mrichter/AliRoot.git] / PWG4 / AliAnalysisTaskPHOSExample.cxx
CommitLineData
d92b41ad 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 **************************************************************************/
15
16/* $Id: */
17
18//_________________________________________________________________________
19// A basic analysis task to analyse photon detected by PHOS
20// A basic analysis task to analyse photon detected by PHOS
21// A basic analysis task to analyse photon detected by PHOS
22// Adapted for AliAnalysisTaskSE and AOD production
23// by Gustavo Conesa
24//
25//*-- Yves Schutz
26//////////////////////////////////////////////////////////////////////////////
27
28#include <TCanvas.h>
29#include <TH1.h>
30#include <TROOT.h>
31#include <TLegend.h>
32#include <TNtuple.h>
33#include <TVector3.h>
34
35#include "AliAnalysisTaskPHOSExample.h"
36#include "AliAnalysisManager.h"
37#include "AliESDEvent.h"
38#include "AliESDCaloCluster.h"
39#include "AliAODEvent.h"
40#include "AliAODPhoton.h"
41#include "AliLog.h"
42#include "AliESDVertex.h"
43
44//______________________________________________________________________________
45AliAnalysisTaskPHOSExample::AliAnalysisTaskPHOSExample() :
46 fDebug(0),
47 fAODPhotons(0x0),
48 fPhotonsInPhos(0),
49 fPhotonId(1.0),
50 fOutputList(0x0),
51 fhPHOSPos(0),
52 fhPHOS(0),
53 fhPHOSEnergy(0),
54 fhPHOSDigits(0),
55 fhPHOSRecParticles(0),
56 fhPHOSPhotons(0),
57 fhPHOSInvariantMass(0),
58 fhPHOSDigitsEvent(0)
59{
60 //Default constructor
61}
62//______________________________________________________________________________
63AliAnalysisTaskPHOSExample::AliAnalysisTaskPHOSExample(const char *name) :
64 AliAnalysisTaskSE(name),
65 fDebug(0),
66 fAODPhotons(0x0),
67 fPhotonsInPhos(0),
68 fPhotonId(1.0),
69 fOutputList(0x0),
70 fhPHOSPos(0),
71 fhPHOS(0),
72 fhPHOSEnergy(0),
73 fhPHOSDigits(0),
74 fhPHOSRecParticles(0),
75 fhPHOSPhotons(0),
76 fhPHOSInvariantMass(0),
77 fhPHOSDigitsEvent(0)
78{
79 // Constructor.
80
81 // Output slots
82 DefineOutput(1, TList::Class()) ;
83}
84
85//____________________________________________________________________________
86AliAnalysisTaskPHOSExample::AliAnalysisTaskPHOSExample(const AliAnalysisTaskPHOSExample& ap) :
87 AliAnalysisTaskSE(ap.GetName()),
88 fDebug(ap.fDebug),
89 fAODPhotons(ap.fAODPhotons),
90 fPhotonsInPhos(ap.fPhotonsInPhos),
91 fPhotonId(ap.fPhotonId),
92 fOutputList(ap.fOutputList),
93 fhPHOSPos(ap.fhPHOSPos),
94 fhPHOS(ap.fhPHOS),
95 fhPHOSEnergy(ap.fhPHOSEnergy),
96 fhPHOSDigits(ap.fhPHOSDigits),
97 fhPHOSRecParticles(ap.fhPHOSRecParticles),
98 fhPHOSPhotons(ap.fhPHOSPhotons),
99 fhPHOSInvariantMass(ap.fhPHOSInvariantMass),
100 fhPHOSDigitsEvent(ap.fhPHOSDigitsEvent)
101{
102 // cpy ctor
103}
104
105//_____________________________________________________________________________
106AliAnalysisTaskPHOSExample& AliAnalysisTaskPHOSExample::operator = (const AliAnalysisTaskPHOSExample& ap)
107{
108// assignment operator
109
110 this->~AliAnalysisTaskPHOSExample();
111 new(this) AliAnalysisTaskPHOSExample(ap);
112 return *this;
113}
114
115//______________________________________________________________________________
116AliAnalysisTaskPHOSExample::~AliAnalysisTaskPHOSExample()
117{
118 // dtor
119 if(fOutputList) {
120 fOutputList->Clear() ;
121 delete fOutputList ;
122 }
123}
124
125
126//________________________________________________________________________
127void AliAnalysisTaskPHOSExample::UserCreateOutputObjects()
128{
129 // Create the outputs containers
130 //AODs
131 fAODPhotons = new TClonesArray("AliAODPhoton", 0);
132 fAODPhotons->SetName("Photons");
133 AddAODBranch("TClonesArray", fAODPhotons);
134
135 OpenFile(1) ;
136
137 fhPHOSPos = new TNtuple("PHOSPos" , "Position in PHOS" , "x:y:z");
138 fhPHOS = new TNtuple("PHOS" , "PHOS" , "event:digits:clusters:photons");
139 fhPHOSEnergy = new TH1D("PHOSEnergy" , "PHOSEnergy" , 100, 0., 100. ) ;
140 fhPHOSDigits = new TH1I("PHOSDigitsCluster" , "PHOSDigits" , 20 , 0 , 20 ) ;
141 fhPHOSRecParticles = new TH1D("PHOSRecParticles" , "PHOSRecParticles" , 20 , 0., 20. ) ;
142 fhPHOSPhotons = new TH1I("PHOSPhotons" , "PHOSPhotons" , 20 , 0 , 20 ) ;
143 fhPHOSInvariantMass = new TH1D("PHOSInvariantMass" , "PHOSInvariantMass" , 400, 0., 400.) ;
144 fhPHOSDigitsEvent = new TH1I("PHOSDigitsEvent" , "PHOSDigitsEvent" , 30 , 0 , 30 ) ;
145
146 // create output container
147
148 fOutputList = new TList() ;
149 fOutputList->SetName(GetName()) ;
150
151 fOutputList->AddAt(fhPHOSPos, 0) ;
152 fOutputList->AddAt(fhPHOS, 1) ;
153 fOutputList->AddAt(fhPHOSEnergy, 2) ;
154 fOutputList->AddAt(fhPHOSDigits, 3) ;
155 fOutputList->AddAt(fhPHOSRecParticles, 4) ;
156 fOutputList->AddAt(fhPHOSPhotons, 5) ;
157 fOutputList->AddAt(fhPHOSInvariantMass, 6) ;
158 fOutputList->AddAt(fhPHOSDigitsEvent, 7) ;
159}
160
161//______________________________________________________________________________
162void AliAnalysisTaskPHOSExample::UserExec(Option_t *)
163{
164 // Processing of one event
165
166 // if ( !((Entry()-1)%100) )
167 AliInfo(Form(" Processing event # %lld", Entry())) ;
168 AliESDEvent* esd = (AliESDEvent*)InputEvent();
169
170 //************************ PHOS *************************************
171 TRefArray * caloClustersArr = new TRefArray();
172 esd->GetPHOSClusters(caloClustersArr);
173
174 const Int_t kNumberOfPhosClusters = caloClustersArr->GetEntries() ;
175
176 TVector3 ** phosVector = new TVector3*[kNumberOfPhosClusters] ;
177 Float_t * phosPhotonsEnergy = new Float_t[kNumberOfPhosClusters] ;
178 Int_t phosCluster ;
179 Int_t numberOfDigitsInPhos = 0 ;
180 Double_t v[3] ; //vertex ;
181 esd->GetVertex()->GetXYZ(v) ;
182
183 fPhotonsInPhos = 0 ;
184 // loop over the PHOS Cluster
185 for(phosCluster = 0 ; phosCluster < kNumberOfPhosClusters ; phosCluster++) {
186 AliESDCaloCluster * caloCluster = (AliESDCaloCluster *) caloClustersArr->At(phosCluster) ;
187
188 //AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(phosCluster) ;
189 if (caloCluster) {
190 Float_t pos[3] ;
191 caloCluster->GetPosition( pos ) ;
192 fhPHOSEnergy->Fill( caloCluster->E() ) ;
193 fhPHOSPos->Fill( pos[0], pos[1], pos[2] ) ;
194 fhPHOSDigits->Fill(Entry(), caloCluster->GetNCells() ) ;
195 numberOfDigitsInPhos += caloCluster->GetNCells() ;
196 Double_t * pid = caloCluster->GetPid() ;
197 if(pid[AliPID::kPhoton] > GetPhotonId() ) {
198 phosVector[fPhotonsInPhos] = new TVector3(pos[0],pos[1],pos[2]) ;
199 phosPhotonsEnergy[fPhotonsInPhos]=caloCluster->E() ;
200 TLorentzVector momentum ;
201 caloCluster->GetMomentum(momentum, v);
202 new ((*fAODPhotons)[fPhotonsInPhos++]) AliAODPhoton (momentum);
203 }
204 }
205 } //PHOS clusters
206
207 fhPHOSRecParticles->Fill(kNumberOfPhosClusters);
208 fhPHOSPhotons->Fill(fPhotonsInPhos);
209 fhPHOSDigitsEvent->Fill(numberOfDigitsInPhos);
210 fhPHOS->Fill(Entry(), numberOfDigitsInPhos, kNumberOfPhosClusters, fPhotonsInPhos) ;
211
212 // invariant Mass
213 if (fPhotonsInPhos > 1 ) {
214 Int_t phosPhoton1, phosPhoton2 ;
215 for(phosPhoton1 = 0 ; phosPhoton1 < fPhotonsInPhos ; phosPhoton1++) {
216 for(phosPhoton2 = phosPhoton1 + 1 ; phosPhoton2 < fPhotonsInPhos ; phosPhoton2++) {
217 Float_t tempMass = TMath::Sqrt( 2 * phosPhotonsEnergy[phosPhoton1] * phosPhotonsEnergy[phosPhoton2] *
218 ( 1 - TMath::Cos(phosVector[phosPhoton1]->Angle(*phosVector[phosPhoton2])) )
219 );
220 fhPHOSInvariantMass->Fill(tempMass*1000.);
221 }
222 }
223 }
224
225 PostData(1, fOutputList);
226
227 delete [] phosVector ;
228 delete [] phosPhotonsEnergy ;
229
230}
231
232
233//______________________________________________________________________________
234void AliAnalysisTaskPHOSExample::Init()
235{
236 // Intialisation of parameters
237 AliInfo("Doing initialisation") ;
238 SetPhotonId(0.9) ;
239}
240
241//______________________________________________________________________________
242void AliAnalysisTaskPHOSExample::Terminate(Option_t *)
243{
244 // Processing when the event loop is ended
245
246 Bool_t problem = kFALSE ;
247 AliInfo(Form(" *** %s Report:", GetName())) ;
248 printf(" PHOSEnergy Mean : %5.3f , RMS : %5.3f \n", fhPHOSEnergy->GetMean(), fhPHOSEnergy->GetRMS() ) ;
249 printf(" PHOSDigits Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigits->GetMean(), fhPHOSDigits->GetRMS() ) ;
250 printf(" PHOSRecParticles Mean : %5.3f , RMS : %5.3f \n", fhPHOSRecParticles->GetMean(), fhPHOSRecParticles->GetRMS() ) ;
251 printf(" PHOSPhotons Mean : %5.3f , RMS : %5.3f \n", fhPHOSPhotons->GetMean(), fhPHOSPhotons->GetRMS() ) ;
252 printf(" PHOSInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhPHOSInvariantMass->GetMean(), fhPHOSInvariantMass->GetRMS() ) ;
253 printf(" PHOSDigitsEvent Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigitsEvent->GetMean(), fhPHOSDigitsEvent->GetRMS() ) ;
254
255 TCanvas * cPHOS = new TCanvas("cPHOS", "PHOS ESD Test", 400, 10, 600, 700) ;
256 cPHOS->Divide(3, 2);
257
258 cPHOS->cd(1) ;
259 if ( fhPHOSEnergy->GetMaximum() > 0. )
260 gPad->SetLogy();
261 fhPHOSEnergy->SetAxisRange(0, 25.);
262 fhPHOSEnergy->SetLineColor(2);
263 fhPHOSEnergy->Draw();
264
265 cPHOS->cd(2) ;
266 fhPHOSDigits->SetAxisRange(0,25.);
267 fhPHOSDigits->SetLineColor(2);
268 fhPHOSDigits->Draw();
269
270 cPHOS->cd(3) ;
271 if ( fhPHOSRecParticles->GetMaximum() > 0. )
272 gPad->SetLogy();
273 fhPHOSRecParticles->SetAxisRange(0, 25.);
274 fhPHOSRecParticles->SetLineColor(2);
275 fhPHOSRecParticles->Draw();
276
277 cPHOS->cd(4) ;
278 if ( fhPHOSPhotons->GetMaximum() > 0. )
279 gPad->SetLogy();
280 fhPHOSPhotons->SetAxisRange(0,25.);
281 fhPHOSPhotons->SetLineColor(2);
282 fhPHOSPhotons->Draw();
283
284 cPHOS->cd(5) ;
285 fhPHOSInvariantMass->SetLineColor(2);
286 fhPHOSInvariantMass->Draw();
287
288 cPHOS->cd(6) ;
289 if ( fhPHOSDigitsEvent->GetMaximum() > 0. )
290 gPad->SetLogy();
291 fhPHOSDigitsEvent->SetAxisRange(0,40.);
292 fhPHOSDigitsEvent->SetLineColor(2);
293 fhPHOSDigitsEvent->Draw();
294
295 cPHOS->Print("PHOS.eps");
296
297 char line[1024] ;
298 sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
299 gROOT->ProcessLine(line);
300 sprintf(line, ".!rm -fR *.eps");
301 gROOT->ProcessLine(line);
302
303 AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
304
305 char * report ;
306 if(problem)
307 report="Problems found, please check!!!";
308 else
309 report="OK";
310
311 AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report)) ;
312}