Compilation on Windows/Cygwin
[u/mrichter/AliRoot.git] / PWG4 / AliAnaCaloTrigger.cxx
CommitLineData
6c3efb37 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//_________________________________________________________________________
f38b754b 17// An analysis task to check the trigger data in ESD
18// Creates an ntuple for 2x2 and NxN triggers
19// Each ntuple connects the maximum trigger amplitudes
20// and its positions with reconstructed clusters
6c3efb37 21//
f38b754b 22//*-- Yves Schutz (CERN) & Gustavo Conesa Balbastre (INFN-LNF)
6c3efb37 23//////////////////////////////////////////////////////////////////////////////
24
25#include <TChain.h>
26#include <TFile.h>
27#include <TNtuple.h>
28#include <TVector3.h>
29
f3299f82 30#include "AliAnaCaloTrigger.h"
9ea45617 31#include "AliAnalysisManager.h"
32#include "AliESDEvent.h"
6c3efb37 33#include "AliLog.h"
9ea45617 34#include "AliESDCaloCluster.h"
35
36//______________________________________________________________________________
37AliAnaCaloTrigger::AliAnaCaloTrigger() :
38 fChain(0),
39 fESD(0),
40 fOutputContainer(0),
41 fCalorimeter("PHOS"),
42 fNtTrigger22(0),
43 fNtTriggerNN(0)
44
45{
46 // Default Constructor.
47
48}
6c3efb37 49
50//______________________________________________________________________________
f3299f82 51AliAnaCaloTrigger::AliAnaCaloTrigger(const char *name) :
9ea45617 52 AliAnalysisTask(name,"AnaCaloTrigger"),
6c3efb37 53 fChain(0),
54 fESD(0),
55 fOutputContainer(0),
f3299f82 56 fCalorimeter("PHOS"),
6c3efb37 57 fNtTrigger22(0),
58 fNtTriggerNN(0)
59
60{
61 // Constructor.
62 // Input slot #0 works with an Ntuple
63 DefineInput(0, TChain::Class());
64 // Output slot #0 writes into a TH1 container
65 DefineOutput(0, TObjArray::Class()) ;
66}
463ee300 67//____________________________________________________________________________
68AliAnaCaloTrigger::AliAnaCaloTrigger(const AliAnaCaloTrigger & ct) :
69 AliAnalysisTask(ct), fChain(ct.fChain), fESD(ct.fESD),
70 fOutputContainer(ct.fOutputContainer), fCalorimeter(ct. fCalorimeter),
71 fNtTrigger22(ct.fNtTrigger22), fNtTriggerNN(ct.fNtTriggerNN)
72{
73
74 // cpy ctor
75 SetName (ct.GetName()) ;
76 SetTitle(ct.GetTitle()) ;
77
78}
79
80//_________________________________________________________________________
81AliAnaCaloTrigger & AliAnaCaloTrigger::operator = (const AliAnaCaloTrigger & source)
82{
83 // assignment operator
84
85 if(&source == this) return *this;
86
87 fChain = source.fChain ;
88 fESD = source.fESD ;
89 fOutputContainer = source.fOutputContainer ;
90 fCalorimeter = source. fCalorimeter ;
91 fNtTrigger22 = source.fNtTrigger22 ;
92 fNtTriggerNN = source.fNtTriggerNN ;
93
94 return *this;
95
96}
6c3efb37 97
98//______________________________________________________________________________
f3299f82 99AliAnaCaloTrigger::~AliAnaCaloTrigger()
6c3efb37 100{
101 // dtor
3788def4 102 //fOutputContainer->Clear() ;
103 //delete fOutputContainer ;
104
6c3efb37 105}
106
107
108//______________________________________________________________________________
f3299f82 109void AliAnaCaloTrigger::ConnectInputData(const Option_t*)
6c3efb37 110{
111 // Initialisation of branch container and histograms
112
113 AliInfo(Form("*** Initialization of %s", GetName())) ;
114
115 // Get input data
116 fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
117 if (!fChain) {
118 AliError(Form("Input 0 for %s not found\n", GetName()));
119 return ;
120 }
121
9ea45617 122 fESD = new AliESDEvent();
123 fESD->ReadFromTree(fChain);
124
6c3efb37 125}
126
127//________________________________________________________________________
128
f3299f82 129void AliAnaCaloTrigger::CreateOutputObjects()
6c3efb37 130{
131
9ea45617 132 // Create the outputs containers
133 OpenFile(0) ;
134
6c3efb37 135 // create histograms
f3299f82 136 fNtTrigger22 = new TNtuple(fCalorimeter+"trigger22", "Trigger data 2x2 patch", "a22:a220:enMax:phEnMax:eta22:phi22:etaMax:phiMax:phEtaMax:phPhiMax");
137 fNtTriggerNN = new TNtuple(fCalorimeter+"triggerNN", "Trigger data NxN patch", "aNN:aNN0:enMax:phEnMax:etaNN:phiNN:etaMax:phiMax:phEtaMax:phPhiMax");
9ea45617 138
6c3efb37 139 // create output container
140
141 fOutputContainer = new TObjArray(2) ;
142 fOutputContainer->SetName(GetName()) ;
9ea45617 143
6c3efb37 144 fOutputContainer->AddAt(fNtTrigger22, 0) ;
145 fOutputContainer->AddAt(fNtTriggerNN, 1) ;
146
147}
148
149//______________________________________________________________________________
f3299f82 150void AliAnaCaloTrigger::Exec(Option_t *)
6c3efb37 151{
152 // Processing of one event
153
154 Long64_t entry = fChain->GetReadEntry() ;
155
156 if (!fESD) {
157 AliError("fESD is not connected to the input!") ;
158 return ;
159 }
160
161 if ( !((entry-1)%100) )
162 AliInfo(Form("%s ----> Processing event # %lld", (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
163
f3299f82 164 // Get trigger information of fCalorimeter
165 TArrayF * triggerAmplitudes = 0x0 ;
166 TArrayF * triggerPosition = 0x0 ;
9ea45617 167 Int_t numberOfCaloClusters = fESD->GetNumberOfCaloClusters() ;
f3299f82 168
169 if(fCalorimeter == "PHOS"){
170 triggerAmplitudes = fESD->GetPHOSTriggerAmplitudes();
171 triggerPosition = fESD->GetPHOSTriggerPosition();
f3299f82 172 }
173 else if(fCalorimeter == "EMCAL"){
174 triggerAmplitudes = fESD->GetEMCALTriggerAmplitudes();
175 triggerPosition = fESD->GetEMCALTriggerPosition();
f3299f82 176 }
6797e4bc 177
178 if( triggerAmplitudes && triggerPosition ){
6c3efb37 179 // trigger amplitudes
f38b754b 180 const Float_t ka22 = static_cast<Float_t>(triggerAmplitudes->At(0)) ;
181 const Float_t ka22O = static_cast<Float_t>(triggerAmplitudes->At(1)) ;
182 const Float_t kaNN = static_cast<Float_t>(triggerAmplitudes->At(2)) ;
183 const Float_t kaNNO = static_cast<Float_t>(triggerAmplitudes->At(3)) ;
6c3efb37 184
185 // trigger position
f38b754b 186 const Float_t kx22 = static_cast<Float_t>(triggerPosition->At(0)) ;
187 const Float_t ky22 = static_cast<Float_t>(triggerPosition->At(1)) ;
188 const Float_t kz22 = static_cast<Float_t>(triggerPosition->At(2)) ;
189 const Float_t kxNN = static_cast<Float_t>(triggerPosition->At(3)) ;
190 const Float_t kyNN = static_cast<Float_t>(triggerPosition->At(4)) ;
191 const Float_t kzNN = static_cast<Float_t>(triggerPosition->At(5)) ;
6c3efb37 192
6c3efb37 193 Float_t enMax = 0. ;
194 Float_t phEnMax = 0. ;
4c0b6461 195 Float_t etaMax = 0.5 ;
196 Float_t phiMax = 0. ;
197 Float_t phEtaMax = 0.5 ;
198 Float_t phPhiMax = 0. ;
199
f38b754b 200 TVector3 vpos22(kx22, ky22, kz22) ;
201 TVector3 vposNN(kxNN, kyNN, kzNN) ;
4c0b6461 202 Float_t eta22 = vpos22.Eta() ;
203 Float_t phi22 = vpos22.Phi() * TMath::RadToDeg() + 360. ;
204 Float_t etaNN = vposNN.Eta() ;
205 Float_t phiNN = vposNN.Phi() * TMath::RadToDeg() + 360. ;
6c3efb37 206
f3299f82 207 Int_t icaloCluster ;
6c3efb37 208
f3299f82 209 // loop over the Calorimeters Clusters
6c3efb37 210
9ea45617 211 for(icaloCluster = 0 ; icaloCluster < numberOfCaloClusters ; icaloCluster++) {
212
f3299f82 213 AliESDCaloCluster * cluster = fESD->GetCaloCluster(icaloCluster) ;
9ea45617 214
215 if (cluster && ( (fCalorimeter == "PHOS" && cluster->IsPHOS()) ||
216 (fCalorimeter == "EMCAL" && cluster->IsEMCAL()))) {
217
018c8391 218 Float_t cluEnergy = cluster->E() ;
6c3efb37 219 Float_t pos[3] ;
220 TVector3 vpos ;
221
018c8391 222 cluster->GetPosition( pos ) ;
6c3efb37 223
224 if ( cluEnergy > enMax) {
225 enMax = cluEnergy ;
226 vpos.SetXYZ(pos[0], pos[1], pos[2]) ;
227 etaMax = vpos.Eta() ;
228 phiMax = vpos.Phi() ;
229 }
230
4b707925 231 Double_t * pid = cluster->GetPid() ;
6c3efb37 232
233 if(pid[AliPID::kPhoton] > 0.9) {
234 if ( cluEnergy > phEnMax) {
235 phEnMax = cluEnergy ;
236 vpos.SetXYZ(pos[0], pos[1], pos[2]) ;
237 phEtaMax = vpos.Eta() ;
238 phPhiMax = vpos.Phi() ;
239 }
240 }
6797e4bc 241 }//if cluster
6c3efb37 242
f38b754b 243 fNtTrigger22->Fill(ka22, ka22O, enMax, phEnMax, eta22, phi22, etaMax, phiMax * TMath::RadToDeg() + 360., phEtaMax, phPhiMax * TMath::RadToDeg() + 360.);
244 fNtTriggerNN->Fill(kaNN, kaNNO, enMax, phEnMax, etaNN, phiNN, etaMax, phiMax * TMath::RadToDeg() + 360., phEtaMax, phPhiMax * TMath::RadToDeg() + 360.);
6797e4bc 245 }//CaloCluster loop
6c3efb37 246
6797e4bc 247 }//If trigger arrays filled
6c3efb37 248
249 PostData(0, fOutputContainer);
250
251}
252
253//______________________________________________________________________________
f3299f82 254void AliAnaCaloTrigger::Terminate(Option_t *)
6c3efb37 255{
256 // Processing when the event loop is ended
257
258}