Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / PHOS / AliPHOSESDQA.cxx
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 //______________________________________________________________________________
41 AliPHOSESDQA::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 //______________________________________________________________________________
61 AliPHOSESDQA::~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 //______________________________________________________________________________
79 void 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 //________________________________________________________________________
103 void 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 //______________________________________________________________________________
134 void 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) ) 
146     AliDebug(AliQAv1::GetQADebugLevel(), Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ; 
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 //______________________________________________________________________________
204 void 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 ; 
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   
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 }