PROOF-aware version of the analysis framework (Andrei)
[u/mrichter/AliRoot.git] / ESDCheck / AliPHOSQATask.cxx
CommitLineData
1dfe075f 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 **************************************************************************/
0b28fd57 15
16/* $Id$ */
17
1dfe075f 18//_________________________________________________________________________
19// An analysis task to check the PHOS photon data in simulated data
20//
21//*-- Yves Schutz
22//////////////////////////////////////////////////////////////////////////////
23
0b28fd57 24#include <TCanvas.h>
1dfe075f 25#include <TChain.h>
0b28fd57 26#include <TFile.h>
1dfe075f 27#include <TH1.h>
28#include <TH1F.h>
29#include <TH1I.h>
1dfe075f 30#include <TLegend.h>
0b28fd57 31#include <TNtuple.h>
32#include <TROOT.h>
1dfe075f 33#include <TVector3.h>
1dfe075f 34
35#include "AliPHOSQATask.h"
36#include "AliESD.h"
37#include "AliLog.h"
38
39//______________________________________________________________________________
40AliPHOSQATask::AliPHOSQATask(const char *name) :
41 AliAnalysisTask(name,""),
42 fChain(0),
43 fESD(0),
44 fhPHOS(0),
45 fhPHOSEnergy(0),
46 fhPHOSDigits(0),
47 fhPHOSRecParticles(0),
48 fhPHOSPhotons(0),
49 fhPHOSInvariantMass(0),
50 fhPHOSDigitsEvent(0)
51{
52 // Constructor.
53 // Input slot #0 works with an Ntuple
54 DefineInput(0, TChain::Class());
55 // Output slot #0 writes into a TH1 container
56 DefineOutput(0, TObjArray::Class()) ;
57}
58
59//______________________________________________________________________________
60AliPHOSQATask::~AliPHOSQATask()
61{
62 // dtor
63 fOutputContainer->Clear() ;
64 delete fOutputContainer ;
65
66 delete fhPHOSPos ;
67 delete fhPHOS ;
68 delete fhPHOSEnergy ;
69 delete fhPHOSDigits ;
70 delete fhPHOSRecParticles ;
71 delete fhPHOSPhotons ;
72 delete fhPHOSInvariantMass ;
73 delete fhPHOSDigitsEvent ;
74}
75
76
77//______________________________________________________________________________
c52c2132 78void AliPHOSQATask::ConnectInputData(const Option_t*)
1dfe075f 79{
80 // Initialisation of branch container and histograms
81
82 AliInfo(Form("*** Initialization of %s", GetName())) ;
83
84 // Get input data
85 fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
86 if (!fChain) {
87 AliError(Form("Input 0 for %s not found\n", GetName()));
88 return ;
89 }
90
c52c2132 91 // One should first check if the branch address was taken by some other task
92 char ** address = (char **)GetBranchAddress(0, "ESD");
93 if (address) {
94 fESD = (AliESD*)(*address);
95 } else {
96 fESD = new AliESD();
97 SetBranchAddress(0, "ESD", &fESD);
1dfe075f 98 }
c52c2132 99}
100
101//________________________________________________________________________
102void AliPHOSQATask::CreateOutputObjects()
103{
1dfe075f 104 // create histograms
105
106 fhPHOSPos = new TNtuple("PHOSPos" , "Position in PHOS" , "x:y:z");
107 fhPHOS = new TNtuple("PHOS" , "PHOS" , "event:digits:clusters:photons");
108 fhPHOSEnergy = new TH1D("PHOSEnergy" , "PHOSEnergy" , 1000, 0., 10. ) ;
109 fhPHOSDigits = new TH1I("PHOSDigitsCluster" , "PHOSDigits" , 20 , 0 , 20 ) ;
110 fhPHOSRecParticles = new TH1D("PHOSRecParticles" , "PHOSRecParticles" , 20 , 0., 20. ) ;
111 fhPHOSPhotons = new TH1I("PHOSPhotons" , "PHOSPhotons" , 20 , 0 , 20 ) ;
112 fhPHOSInvariantMass = new TH1D("PHOSInvariantMass" , "PHOSInvariantMass" , 400, 0., 400.) ;
113 fhPHOSDigitsEvent = new TH1I("PHOSDigitsEvent" , "PHOSDigitsEvent" , 30 , 0 , 30 ) ;
114
115 // create output container
116
117 fOutputContainer = new TObjArray(8) ;
118 fOutputContainer->SetName(GetName()) ;
119
120 fOutputContainer->AddAt(fhPHOSPos, 0) ;
121 fOutputContainer->AddAt(fhPHOS, 1) ;
122 fOutputContainer->AddAt(fhPHOSEnergy, 2) ;
123 fOutputContainer->AddAt(fhPHOSDigits, 3) ;
124 fOutputContainer->AddAt(fhPHOSRecParticles, 4) ;
125 fOutputContainer->AddAt(fhPHOSPhotons, 5) ;
126 fOutputContainer->AddAt(fhPHOSInvariantMass, 6) ;
127 fOutputContainer->AddAt(fhPHOSDigitsEvent, 7) ;
128}
129
130//______________________________________________________________________________
131void AliPHOSQATask::Exec(Option_t *)
132{
133 // Processing of one event
134
135 Long64_t entry = fChain->GetReadEntry() ;
136
137 if (!fESD) {
138 AliError("fESD is not connected to the input!") ;
139 return ;
140 }
141
142 if ( !((entry-1)%100) )
143 AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
144
145 //************************ PHOS *************************************
146
147 Int_t firstPhosCluster = fESD->GetFirstPHOSCluster() ;
148 const Int_t numberOfPhosClusters = fESD->GetNumberOfPHOSClusters() ;
149
150 TVector3 ** phosVector = new TVector3*[numberOfPhosClusters] ;
151 Float_t * phosPhotonsEnergy = new Float_t[numberOfPhosClusters] ;
152 Int_t phosCluster ;
153 Int_t numberOfPhotonsInPhos = 0 ;
154 Int_t numberOfDigitsInPhos = 0 ;
155
156 // loop over the PHOS Cluster
157 for(phosCluster = firstPhosCluster ; phosCluster < firstPhosCluster + numberOfPhosClusters ; phosCluster++) {
158 AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(phosCluster) ;
159 if (caloCluster) {
160 Float_t pos[3] ;
161 caloCluster->GetGlobalPosition( pos ) ;
162 fhPHOSEnergy->Fill( caloCluster->GetClusterEnergy() ) ;
163 fhPHOSPos->Fill( pos[0], pos[1], pos[2] ) ;
164 fhPHOSDigits->Fill(entry, caloCluster->GetNumberOfDigits() ) ;
165 numberOfDigitsInPhos += caloCluster->GetNumberOfDigits() ;
166 Float_t * pid = caloCluster->GetPid() ;
167 if(pid[AliPID::kPhoton] > 0.9) {
168 phosVector[numberOfPhotonsInPhos] = new TVector3(pos[0],pos[1],pos[2]) ;
169 phosPhotonsEnergy[numberOfPhotonsInPhos]=caloCluster->GetClusterEnergy() ;
170 numberOfPhotonsInPhos++;
171 }
172 }
173 } //PHOS clusters
174
175 fhPHOSRecParticles->Fill(numberOfPhosClusters);
176 fhPHOSPhotons->Fill(numberOfPhotonsInPhos);
177 fhPHOSDigitsEvent->Fill(numberOfDigitsInPhos);
178 fhPHOS->Fill(entry, numberOfDigitsInPhos, numberOfPhosClusters, numberOfPhotonsInPhos) ;
179
180 // invariant Mass
181 if (numberOfPhotonsInPhos > 1 ) {
182 Int_t phosPhoton1, phosPhoton2 ;
183 for(phosPhoton1 = 0 ; phosPhoton1 < numberOfPhotonsInPhos ; phosPhoton1++) {
184 for(phosPhoton2 = phosPhoton1 + 1 ; phosPhoton2 < numberOfPhotonsInPhos ; phosPhoton2++) {
185 Float_t tempMass = TMath::Sqrt( 2 * phosPhotonsEnergy[phosPhoton1] * phosPhotonsEnergy[phosPhoton2] *
186 ( 1 - TMath::Cos(phosVector[phosPhoton1]->Angle(*phosVector[phosPhoton2])) )
187 );
188 fhPHOSInvariantMass->Fill(tempMass*1000.);
189 }
190 }
191 }
192
193 PostData(0, fOutputContainer);
194
195 delete [] phosVector ;
196 delete [] phosPhotonsEnergy ;
197
198}
199
200//______________________________________________________________________________
201void AliPHOSQATask::Terminate(Option_t *)
202{
203 // Processing when the event loop is ended
c52c2132 204
205 fOutputContainer = (TObjArray*)GetOutputData(0);
206 fhPHOSPos = (TNtuple*)fOutputContainer->At(0);
207 fhPHOS = (TNtuple*)fOutputContainer->At(1);
208 fhPHOSEnergy = (TH1D*)fOutputContainer->At(2);
209 fhPHOSDigits = (TH1I*)fOutputContainer->At(3);
210 fhPHOSRecParticles = (TH1D*)fOutputContainer->At(4);
211 fhPHOSPhotons = (TH1I*)fOutputContainer->At(5);
212 fhPHOSInvariantMass = (TH1D*)fOutputContainer->At(6);
213 fhPHOSDigitsEvent = (TH1I*)fOutputContainer->At(7);
214
1dfe075f 215 printf("PHOSEnergy Mean : %5.3f , RMS : %5.3f \n", fhPHOSEnergy->GetMean(), fhPHOSEnergy->GetRMS() ) ;
216 printf("PHOSDigits Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigits->GetMean(), fhPHOSDigits->GetRMS() ) ;
217 printf("PHOSRecParticles Mean : %5.3f , RMS : %5.3f \n", fhPHOSRecParticles->GetMean(), fhPHOSRecParticles->GetRMS() ) ;
218 printf("PHOSPhotons Mean : %5.3f , RMS : %5.3f \n", fhPHOSPhotons->GetMean(), fhPHOSPhotons->GetRMS() ) ;
219 printf("PHOSInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhPHOSInvariantMass->GetMean(), fhPHOSInvariantMass->GetRMS() ) ;
220 printf("PHOSDigitsEvent Mean : %5.3f , RMS : %5.3f \n", fhPHOSDigitsEvent->GetMean(), fhPHOSDigitsEvent->GetRMS() ) ;
221
222 TCanvas * cPHOS = new TCanvas("cPHOS", "PHOS ESD Test", 400, 10, 600, 700) ;
223 cPHOS->Divide(3, 2);
224
225 cPHOS->cd(1) ;
226 gPad->SetLogy();
227 fhPHOSEnergy->SetAxisRange(0, 25.);
228 fhPHOSEnergy->SetLineColor(2);
229 fhPHOSEnergy->Draw();
230
231 cPHOS->cd(2) ;
232 fhPHOSDigits->SetAxisRange(0,25.);
233 fhPHOSDigits->SetLineColor(2);
234 fhPHOSDigits->Draw();
235
236 cPHOS->cd(3) ;
237 gPad->SetLogy();
238 fhPHOSRecParticles->SetAxisRange(0, 25.);
239 fhPHOSRecParticles->SetLineColor(2);
240 fhPHOSRecParticles->Draw();
241
242 cPHOS->cd(4) ;
243 gPad->SetLogy();
244 fhPHOSPhotons->SetAxisRange(0,25.);
245 fhPHOSPhotons->SetLineColor(2);
246 fhPHOSPhotons->Draw();
247
248 cPHOS->cd(5) ;
249 fhPHOSInvariantMass->SetLineColor(2);
250 fhPHOSInvariantMass->Draw();
251
252 cPHOS->cd(6) ;
253 gPad->SetLogy();
254 fhPHOSDigitsEvent->SetAxisRange(0,40.);
255 fhPHOSDigitsEvent->SetLineColor(2);
256 fhPHOSDigitsEvent->Draw();
257
258 cPHOS->Print("PHOS.eps");
259
260 char line[1024] ;
261 sprintf(line, ".!tar -zcvf %s.tar.gz *.eps", GetName()) ;
262 gROOT->ProcessLine(line);
263 sprintf(line, ".!rm -fR *.eps");
264 gROOT->ProcessLine(line);
265
266 AliInfo(Form("!!! All the eps files are in %s.tar.gz !!! \n", GetName())) ;
267}