Replace AliTriggerPHOS by AliAnaCaloTrigger
[u/mrichter/AliRoot.git] / PWG4 / AliAnaCaloTrigger.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 //_________________________________________________________________________
17 // An analysis task to check the PHOS/EMCAL simulated trigger
18 //
19 //*-- Yves Schutz & Gustavo Conesa Balbastre
20 //////////////////////////////////////////////////////////////////////////////
21
22 #include <TChain.h>
23 #include <TFile.h> 
24 #include <TNtuple.h>
25 #include <TVector3.h> 
26
27 #include "AliAnaCaloTrigger.h" 
28 #include "AliESD.h" 
29 #include "AliLog.h"
30
31 //______________________________________________________________________________
32 AliAnaCaloTrigger::AliAnaCaloTrigger(const char *name) : 
33   AliAnalysisTask(name,""),  
34   fChain(0),
35   fESD(0), 
36   fOutputContainer(0),
37   fCalorimeter("PHOS"),
38   fNtTrigger22(0), 
39   fNtTriggerNN(0)
40
41 {
42   // Constructor.
43   // Input slot #0 works with an Ntuple
44   DefineInput(0, TChain::Class());
45   // Output slot #0 writes into a TH1 container
46   DefineOutput(0,  TObjArray::Class()) ; 
47 }
48
49 //______________________________________________________________________________
50 AliAnaCaloTrigger::~AliAnaCaloTrigger()
51 {
52   // dtor
53   fOutputContainer->Clear() ; 
54   delete fOutputContainer ;
55   delete fNtTrigger22 ; 
56   delete fNtTriggerNN ; 
57 }
58
59
60 //______________________________________________________________________________
61 void AliAnaCaloTrigger::ConnectInputData(const Option_t*)
62 {
63   // Initialisation of branch container and histograms 
64     
65   AliInfo(Form("*** Initialization of %s", GetName())) ; 
66   
67   // Get input data
68   fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
69   if (!fChain) {
70     AliError(Form("Input 0 for %s not found\n", GetName()));
71     return ;
72   }
73   
74   // One should first check if the branch address was taken by some other task
75   char ** address = (char **)GetBranchAddress(0, "ESD");
76   if (address) {
77     fESD = (AliESD*)(*address);
78   } else {
79     fESD = new AliESD();
80     SetBranchAddress(0, "ESD", &fESD);
81   }
82 }
83
84 //________________________________________________________________________
85
86 void AliAnaCaloTrigger::CreateOutputObjects()
87 {  
88
89   // create histograms 
90   fNtTrigger22 = new TNtuple(fCalorimeter+"trigger22", "Trigger data 2x2 patch", "a22:a220:enMax:phEnMax:eta22:phi22:etaMax:phiMax:phEtaMax:phPhiMax");
91   fNtTriggerNN = new TNtuple(fCalorimeter+"triggerNN", "Trigger data NxN patch", "aNN:aNN0:enMax:phEnMax:etaNN:phiNN:etaMax:phiMax:phEtaMax:phPhiMax");
92
93   // create output container
94   
95   fOutputContainer = new TObjArray(2) ; 
96   fOutputContainer->SetName(GetName()) ; 
97
98   fOutputContainer->AddAt(fNtTrigger22,             0) ; 
99   fOutputContainer->AddAt(fNtTriggerNN,             1) ; 
100
101 }
102
103 //______________________________________________________________________________
104 void AliAnaCaloTrigger::Exec(Option_t *) 
105 {
106   // Processing of one event
107   
108   Long64_t entry = fChain->GetReadEntry() ;
109   
110   if (!fESD) {
111     AliError("fESD is not connected to the input!") ; 
112     return ; 
113   }
114   
115   if ( !((entry-1)%100) ) 
116     AliInfo(Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ; 
117   
118   // Get trigger information of fCalorimeter 
119   TArrayF * triggerAmplitudes = 0x0 ;
120   TArrayF * triggerPosition   = 0x0 ;
121   Int_t firstCaloCluster      = 0 ;
122   Int_t numberOfCaloClusters  = 0 ;
123
124   if(fCalorimeter == "PHOS"){
125     triggerAmplitudes      = fESD->GetPHOSTriggerAmplitudes();
126     triggerPosition        = fESD->GetPHOSTriggerPosition();
127     firstCaloCluster       = fESD->GetFirstPHOSCluster() ;
128     numberOfCaloClusters   = fESD->GetNumberOfPHOSClusters() ;
129   }
130   else if(fCalorimeter == "EMCAL"){
131     triggerAmplitudes    = fESD->GetEMCALTriggerAmplitudes();
132     triggerPosition      = fESD->GetEMCALTriggerPosition();
133     firstCaloCluster     = fESD->GetFirstEMCALCluster() ;
134     numberOfCaloClusters = fESD->GetNumberOfEMCALClusters() ;
135   }
136   
137   // trigger amplitudes
138   const Float_t a22    = static_cast<Float_t>(triggerAmplitudes->At(0)) ; 
139   const Float_t a22O   = static_cast<Float_t>(triggerAmplitudes->At(1)) ; 
140   const Float_t aNN    = static_cast<Float_t>(triggerAmplitudes->At(2)) ; 
141   const Float_t aNNO   = static_cast<Float_t>(triggerAmplitudes->At(3)) ; 
142
143   // trigger position
144   const Float_t x22  =  static_cast<Float_t>(triggerPosition->At(0)) ; 
145   const Float_t y22  =  static_cast<Float_t>(triggerPosition->At(1)) ;
146   const Float_t z22  =  static_cast<Float_t>(triggerPosition->At(2)) ;
147   const Float_t xNN  =  static_cast<Float_t>(triggerPosition->At(3)) ; 
148   const Float_t yNN  =  static_cast<Float_t>(triggerPosition->At(4)) ;
149   const Float_t zNN  =  static_cast<Float_t>(triggerPosition->At(5)) ; 
150   
151  
152    
153   Float_t enMax       = 0. ;
154   Float_t phEnMax     = 0. ;
155   Float_t etaMax      = 0.5 ;
156   Float_t phiMax      = 0. ; 
157   Float_t phEtaMax    = 0.5 ;
158   Float_t phPhiMax    = 0. ; 
159   
160   TVector3 vpos22(x22, y22, z22) ;
161   TVector3 vposNN(xNN, yNN, zNN) ;
162   Float_t eta22 = vpos22.Eta() ; 
163   Float_t phi22 = vpos22.Phi() * TMath::RadToDeg() + 360. ; 
164   Float_t etaNN = vposNN.Eta() ; 
165   Float_t phiNN = vposNN.Phi() * TMath::RadToDeg() + 360. ; 
166
167   Int_t      icaloCluster ; 
168   
169   // loop over the Calorimeters Clusters
170   
171   for(icaloCluster = firstCaloCluster ; icaloCluster < firstCaloCluster + numberOfCaloClusters ; icaloCluster++) {
172     AliESDCaloCluster * cluster = fESD->GetCaloCluster(icaloCluster) ;
173     if (cluster) {
174
175       Float_t cluEnergy = cluster->GetClusterEnergy() ; 
176       Float_t pos[3] ;
177       TVector3 vpos ;
178       
179       cluster->GetGlobalPosition( pos ) ;
180       
181       if ( cluEnergy > enMax) { 
182         enMax = cluEnergy ; 
183         vpos.SetXYZ(pos[0], pos[1], pos[2]) ; 
184         etaMax = vpos.Eta() ; 
185         phiMax = vpos.Phi() ; 
186       }
187
188       Float_t * pid = cluster->GetPid() ;
189       
190       if(pid[AliPID::kPhoton] > 0.9) {
191         if ( cluEnergy > phEnMax) { 
192           phEnMax = cluEnergy ; 
193           vpos.SetXYZ(pos[0], pos[1], pos[2]) ; 
194           phEtaMax = vpos.Eta() ; 
195           phPhiMax = vpos.Phi() ; 
196         }
197       }
198     }
199     
200     fNtTrigger22->Fill(a22, a22O, enMax, phEnMax, eta22, phi22, etaMax, phiMax * TMath::RadToDeg() + 360., phEtaMax, phPhiMax * TMath::RadToDeg() + 360.);
201     fNtTriggerNN->Fill(aNN, aNNO, enMax, phEnMax, etaNN, phiNN, etaMax, phiMax * TMath::RadToDeg() + 360., phEtaMax, phPhiMax * TMath::RadToDeg() + 360.);
202  }
203   
204   
205   PostData(0, fOutputContainer);
206   
207 }
208
209 //______________________________________________________________________________
210 void AliAnaCaloTrigger::Terminate(Option_t *)
211 {
212   // Processing when the event loop is ended
213
214 }