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