char * to TString
[u/mrichter/AliRoot.git] / ESDCheck / AliEMCALQATask.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 EMCAL 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>
1f588058 34#include <TString.h>
1dfe075f 35
36#include "AliEMCALQATask.h"
37#include "AliESD.h"
38#include "AliLog.h"
39
40//______________________________________________________________________________
41AliEMCALQATask::AliEMCALQATask(const char *name) :
42 AliAnalysisTask(name,""),
43 fChain(0),
44 fESD(0),
45 fhEMCALPos(0),
46 fhEMCAL(0),
47 fhEMCALEnergy(0),
48 fhEMCALDigits(0),
49 fhEMCALRecParticles(0),
50 fhEMCALPhotons(0),
51 fhEMCALInvariantMass(0),
52 fhEMCALDigitsEvent(0)
53{
54 // Constructor.
55 // Input slot #0 works with an Ntuple
56 DefineInput(0, TChain::Class());
57 // Output slot #0 writes into a TH1 container
58 DefineOutput(0, TObjArray::Class()) ;
59}
60
61//______________________________________________________________________________
62AliEMCALQATask::~AliEMCALQATask()
63{
64 // dtor
65 fOutputContainer->Clear() ;
66 delete fOutputContainer ;
67
68 delete fhEMCALPos ;
69 delete fhEMCAL ;
70 delete fhEMCALEnergy ;
71 delete fhEMCALDigits ;
72 delete fhEMCALRecParticles ;
73 delete fhEMCALPhotons ;
74 delete fhEMCALInvariantMass ;
75 delete fhEMCALDigitsEvent ;
76}
77
78//______________________________________________________________________________
c52c2132 79void AliEMCALQATask::ConnectInputData(const Option_t*)
1dfe075f 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
c52c2132 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);
1dfe075f 99 }
c52c2132 100}
1dfe075f 101
c52c2132 102//______________________________________________________________________________
103void AliEMCALQATask::CreateOutputObjects()
104{
105// create histograms
1e20f195 106
107 OpenFile(0) ;
108
1dfe075f 109 fhEMCALPos = new TNtuple("EMCALPos" , "Position in EMCAL" , "x:y:z");
110 fhEMCAL = new TNtuple("EMCAL" , "EMCAL" , "event:digits:clusters:photons");
111 fhEMCALEnergy = new TH1D("EMCALEnergy" , "EMCALEnergy" , 1000, 0., 10. ) ;
112 fhEMCALDigits = new TH1I("EMCALDigitsCluster" , "EMCALDigits" , 20 , 0 , 20 ) ;
113 fhEMCALRecParticles = new TH1D("EMCALRecParticles" , "EMCALRecParticles", 20 , 0., 20. ) ;
114 fhEMCALPhotons = new TH1I("EMCALPhotons" , "EMCALPhotons" , 20 , 0 , 20 ) ;
115 fhEMCALInvariantMass = new TH1D("EMCALInvariantMass" , "EMCALInvariantMass", 400, 0., 400.) ;
116 fhEMCALDigitsEvent = new TH1I("EMCALDigitsEvent" , "EMCALDigitsEvent" , 30 , 0 , 30 ) ;
117
118 // create output container
119
120 fOutputContainer = new TObjArray(8) ;
121 fOutputContainer->SetName(GetName()) ;
122
123 fOutputContainer->AddAt(fhEMCALPos, 0) ;
124 fOutputContainer->AddAt(fhEMCAL, 1) ;
125 fOutputContainer->AddAt(fhEMCALEnergy, 2) ;
126 fOutputContainer->AddAt(fhEMCALDigits, 3) ;
127 fOutputContainer->AddAt(fhEMCALRecParticles, 4) ;
128 fOutputContainer->AddAt(fhEMCALPhotons, 5) ;
129 fOutputContainer->AddAt(fhEMCALInvariantMass, 6) ;
130 fOutputContainer->AddAt(fhEMCALDigitsEvent, 7) ;
131}
132
133//______________________________________________________________________________
134void AliEMCALQATask::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) )
146 AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
147
148 //************************ EMCAL *************************************
149 Int_t firstEmcalCluster = fESD->GetFirstEMCALCluster() ;
150 const Int_t mumberOfEmcalClusters = fESD->GetNumberOfEMCALClusters() ;
151
152 TVector3 ** emcalVector = new TVector3*[mumberOfEmcalClusters] ;
153 Float_t * emcalPhotonsEnergy = new Float_t[mumberOfEmcalClusters] ;
154 Int_t emcalCluster ;
155 Int_t numberOfEmcalClustersv1 = 0 ;
156 Int_t numberOfPhotonsInEmcal = 0 ;
157 Int_t numberOfDigitsInEmcal = 0 ;
158
159
160 // loop over all the EMCAL Cluster
161 for(emcalCluster = firstEmcalCluster ; emcalCluster < firstEmcalCluster + mumberOfEmcalClusters ; emcalCluster++) {
162 AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(emcalCluster) ;
163 if (caloCluster) {
164 Float_t pos[3] ;
1dfe075f 165 if(caloCluster->GetClusterType() == AliESDCaloCluster::kClusterv1) {
1f588058 166 caloCluster->GetPosition(pos) ;
11c5d692 167 fhEMCALPos->Fill(pos[0],pos[1],pos[2]) ;
1f588058 168 fhEMCALEnergy->Fill(caloCluster->E()) ;
1dfe075f 169 fhEMCALDigits->Fill(entry, caloCluster->GetNumberOfDigits()) ;
170 numberOfEmcalClustersv1++ ;
171 numberOfDigitsInEmcal += caloCluster->GetNumberOfDigits() ;
172 // Float_t * pid = clus->GetPid() ;
173 // if(pid[AliPID::kPhoton]>0.9){
174 emcalVector[numberOfPhotonsInEmcal] = new TVector3(pos[0],pos[1],pos[2]) ;
1f588058 175 emcalPhotonsEnergy[numberOfPhotonsInEmcal] = caloCluster->E() ;
1dfe075f 176 numberOfPhotonsInEmcal++ ;
177 }
178 }
179 } // EMCAL clusters loop
180
181 fhEMCALRecParticles->Fill(numberOfEmcalClustersv1);
182 fhEMCALPhotons->Fill(numberOfPhotonsInEmcal);
183 fhEMCALDigitsEvent->Fill(numberOfDigitsInEmcal);
184 fhEMCAL->Fill(entry, numberOfDigitsInEmcal, numberOfEmcalClustersv1, numberOfPhotonsInEmcal) ;
185
186 // invariant Mass
187 if (numberOfPhotonsInEmcal > 1 ) {
188 Int_t emcalPhoton1, emcalPhoton2 ;
189 for(emcalPhoton1 = 0 ; emcalPhoton1 < numberOfPhotonsInEmcal ; emcalPhoton1++) {
190 for(emcalPhoton2 = emcalPhoton1 + 1 ; emcalPhoton2 < numberOfPhotonsInEmcal ; emcalPhoton2++) {
191 Float_t tempMass = TMath::Sqrt( 2 * emcalPhotonsEnergy[emcalPhoton1] * emcalPhotonsEnergy[emcalPhoton2] *
192 ( 1 - TMath::Cos( emcalVector[emcalPhoton1]->Angle(*emcalVector[emcalPhoton2]))
193 )
194 );
195 fhEMCALInvariantMass->Fill(tempMass*1000.);
196 }
197 }
198 }
199
200 PostData(0, fOutputContainer);
201
202 delete [] emcalVector ;
203 delete [] emcalPhotonsEnergy ;
204}
205
206//______________________________________________________________________________
207void AliEMCALQATask::Terminate(Option_t *)
208{
209 // Processing when the event loop is ended
c52c2132 210 fOutputContainer = (TObjArray*)GetOutputData(0);
211 fhEMCALEnergy = (TH1D*)fOutputContainer->At(2);
212 fhEMCALDigits = (TH1I*)fOutputContainer->At(3);
213 fhEMCALRecParticles = (TH1D*)fOutputContainer->At(4);
214 fhEMCALPhotons = (TH1I*)fOutputContainer->At(5);
215 fhEMCALInvariantMass = (TH1D*)fOutputContainer->At(6);
216 fhEMCALDigitsEvent = (TH1I*)fOutputContainer->At(7);
84eb42a1 217
2704006a 218 Bool_t problem = kFALSE ;
84eb42a1 219 AliInfo(Form(" *** %s Report:", GetName())) ;
220 printf(" EMCALEnergy Mean : %5.3f , RMS : %5.3f \n", fhEMCALEnergy->GetMean(), fhEMCALEnergy->GetRMS() ) ;
221 printf(" EMCALDigits Mean : %5.3f , RMS : %5.3f \n", fhEMCALDigits->GetMean(), fhEMCALDigits->GetRMS() ) ;
222 printf(" EMCALRecParticles Mean : %5.3f , RMS : %5.3f \n", fhEMCALRecParticles->GetMean(), fhEMCALRecParticles->GetRMS() ) ;
223 printf(" EMCALPhotons Mean : %5.3f , RMS : %5.3f \n", fhEMCALPhotons->GetMean(), fhEMCALPhotons->GetRMS() ) ;
224 printf(" EMCALInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhEMCALInvariantMass->GetMean(), fhEMCALInvariantMass->GetRMS() ) ;
225 printf(" EMCALDigitsEvent Mean : %5.3f , RMS : %5.3f \n", fhEMCALDigitsEvent->GetMean(), fhEMCALDigitsEvent->GetRMS() ) ;
1dfe075f 226
227 TCanvas * cEMCAL = new TCanvas("EMCAL", "EMCAL ESD Test", 400, 10, 600, 700);
228 cEMCAL->Divide(3, 2) ;
229
230 cEMCAL->cd(1) ;
84eb42a1 231 if ( fhEMCALEnergy->GetMaximum() > 0. )
232 gPad->SetLogy();
1dfe075f 233 fhEMCALEnergy->SetAxisRange(0, 25.);
234 fhEMCALEnergy->SetXTitle("Energy (GeV)");
235 fhEMCALEnergy->Draw();
236
237 cEMCAL->cd(2) ;
84eb42a1 238 if ( fhEMCALDigits->GetMaximum() > 0. )
239 gPad->SetLogy();
1dfe075f 240 fhEMCALDigits->SetAxisRange(0, 25.);
241 fhEMCALDigits->SetXTitle("DigitsPerCluster");
242 fhEMCALDigits->Draw();
243
244 cEMCAL->cd(3) ;
84eb42a1 245 if ( fhEMCALRecParticles->GetMaximum() > 0. )
246 gPad->SetLogy();
1dfe075f 247 fhEMCALRecParticles->SetAxisRange(0, 25.);
248 fhEMCALRecParticles->SetXTitle("RecParticles");
249 fhEMCALRecParticles->Draw();
250
251 cEMCAL->cd(4) ;
84eb42a1 252 if ( fhEMCALPhotons->GetMaximum() > 0. )
253 gPad->SetLogy();
1dfe075f 254 fhEMCALPhotons->SetAxisRange(0, 25.);
255 fhEMCALPhotons->SetXTitle("Photons");
256 fhEMCALPhotons->Draw();
257
258 cEMCAL->cd(5) ;
259 fhEMCALInvariantMass->SetXTitle("InvariantMass (MeV/c²)");
260 fhEMCALInvariantMass->Draw();
261
262 cEMCAL->cd(6) ;
84eb42a1 263 if ( fhEMCALDigitsEvent->GetMaximum() > 0. )
264 gPad->SetLogy();
1dfe075f 265 fhEMCALDigitsEvent->SetAxisRange(0, 40.);
266 fhEMCALDigitsEvent->SetXTitle("DigitsPerEvent");
267 fhEMCALDigitsEvent->Draw();
268
269 cEMCAL->Print("EMCAL.eps");
270
271 char line[1024] ;
84eb42a1 272 sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
1dfe075f 273 gROOT->ProcessLine(line);
274 sprintf(line, ".!rm -fR *.eps");
275 gROOT->ProcessLine(line);
276
2704006a 277 AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
278
1f588058 279 TString report ;
2704006a 280 if(problem)
281 report="Problems found, please check!!!";
282 else
283 report="OK";
284
1f588058 285 AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report.Data())) ;
1dfe075f 286}