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