Example train to produce HF candidates
[u/mrichter/AliRoot.git] / ESDCheck / AliPHOSQATask.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 // 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
25 //
26 //*-- Yves Schutz 
27 //////////////////////////////////////////////////////////////////////////////
28
29 #include <TCanvas.h>
30 #include <TChain.h>
31 #include <TFile.h> 
32 #include <TH1.h>
33 #include <TNtuple.h>
34 #include <TROOT.h> 
35 #include <TVector3.h> 
36 #include <TString.h> 
37
38 #include "AliPHOSQATask.h" 
39 #include "AliESD.h" 
40 #include "AliLog.h"
41
42 //______________________________________________________________________________
43 AliPHOSQATask::AliPHOSQATask(const char *name) : 
44   AliAnalysisTask(name,""),  
45   fChain(0),
46   fESD(0), 
47   fOutputContainer(0),
48   fhPHOSPos(0),
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
64 //____________________________________________________________________________
65 AliPHOSQATask::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 //_____________________________________________________________________________
83 AliPHOSQATask& AliPHOSQATask::operator = (const AliPHOSQATask& ap)
84 {
85 // assignment operator
86
87   this->~AliPHOSQATask();
88   new(this) AliPHOSQATask(ap);
89   return *this;
90 }
91
92 //______________________________________________________________________________
93 AliPHOSQATask::~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 //______________________________________________________________________________
111 void AliPHOSQATask::ConnectInputData(const Option_t*)
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   
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);
131   }
132 }
133
134 //________________________________________________________________________
135 void AliPHOSQATask::CreateOutputObjects()
136 {  
137   // create histograms 
138   
139   OpenFile(0) ; 
140
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 //______________________________________________________________________________
166 void 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() ;
183   const Int_t kNumberOfPhosClusters   = fESD->GetNumberOfPHOSClusters() ;
184   
185   TVector3 ** phosVector       = new TVector3*[kNumberOfPhosClusters] ;
186   Float_t  * phosPhotonsEnergy = new Float_t[kNumberOfPhosClusters] ;
187   Int_t      phosCluster ; 
188   Int_t      numberOfPhotonsInPhos  = 0 ;
189   Int_t      numberOfDigitsInPhos   = 0 ;
190   
191   // loop over the PHOS Cluster
192   for(phosCluster = firstPhosCluster ; phosCluster < firstPhosCluster + kNumberOfPhosClusters ; phosCluster++) {
193     AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(phosCluster) ;
194     if (caloCluster) {
195       Float_t pos[3] ;
196       caloCluster->GetPosition( pos ) ;
197       fhPHOSEnergy->Fill( caloCluster->E() ) ;
198       fhPHOSPos->Fill( pos[0], pos[1], pos[2] ) ;
199       fhPHOSDigits->Fill(entry, caloCluster->GetNumberOfDigits() ) ;
200       numberOfDigitsInPhos += caloCluster->GetNumberOfDigits() ;
201       Double_t * pid = caloCluster->GetPid() ;
202       if(pid[AliPID::kPhoton] > 0.9) {
203         phosVector[numberOfPhotonsInPhos] = new TVector3(pos[0],pos[1],pos[2]) ;
204         phosPhotonsEnergy[numberOfPhotonsInPhos]=caloCluster->E() ;
205         numberOfPhotonsInPhos++;
206       }
207     }
208   } //PHOS clusters
209     
210   fhPHOSRecParticles->Fill(kNumberOfPhosClusters);
211   fhPHOSPhotons->Fill(numberOfPhotonsInPhos);
212   fhPHOSDigitsEvent->Fill(numberOfDigitsInPhos);
213   fhPHOS->Fill(entry, numberOfDigitsInPhos, kNumberOfPhosClusters, numberOfPhotonsInPhos) ; 
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 //______________________________________________________________________________
236 void AliPHOSQATask::Terminate(Option_t *)
237 {
238   // Processing when the event loop is ended
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
250   Bool_t problem = kFALSE ; 
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()    ) ;
258
259   TCanvas  * cPHOS = new TCanvas("cPHOS", "PHOS ESD Test", 400, 10, 600, 700) ;
260   cPHOS->Divide(3, 2);
261
262   cPHOS->cd(1) ; 
263   if ( fhPHOSEnergy->GetMaximum() > 0. ) 
264     gPad->SetLogy();
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) ; 
275   if ( fhPHOSRecParticles->GetMaximum() > 0. ) 
276     gPad->SetLogy();
277   fhPHOSRecParticles->SetAxisRange(0, 25.);
278   fhPHOSRecParticles->SetLineColor(2);
279   fhPHOSRecParticles->Draw();
280
281   cPHOS->cd(4) ; 
282   if ( fhPHOSPhotons->GetMaximum() > 0. ) 
283     gPad->SetLogy();
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) ; 
293   if ( fhPHOSDigitsEvent->GetMaximum() > 0. ) 
294     gPad->SetLogy();
295   fhPHOSDigitsEvent->SetAxisRange(0,40.);
296   fhPHOSDigitsEvent->SetLineColor(2);
297   fhPHOSDigitsEvent->Draw();
298  
299   cPHOS->Print("PHOS.eps");
300  
301   char line[1024] ; 
302   sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ; 
303   gROOT->ProcessLine(line);
304   sprintf(line, ".!rm -fR *.eps"); 
305   gROOT->ProcessLine(line);
306  
307   AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
308
309   TString report ; 
310   if(problem)
311     report="Problems found, please check!!!";  
312   else 
313     report="OK";
314
315   AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report.Data())) ; 
316 }