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