Replace AliTriggerPHOS by AliAnaCaloTrigger
[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//_________________________________________________________________________
f3299f82 17// An analysis task to check the PHOS/EMCAL simulated trigger
6c3efb37 18//
f3299f82 19//*-- Yves Schutz & Gustavo Conesa Balbastre
6c3efb37 20//////////////////////////////////////////////////////////////////////////////
21
22#include <TChain.h>
23#include <TFile.h>
24#include <TNtuple.h>
25#include <TVector3.h>
26
f3299f82 27#include "AliAnaCaloTrigger.h"
6c3efb37 28#include "AliESD.h"
29#include "AliLog.h"
30
31//______________________________________________________________________________
f3299f82 32AliAnaCaloTrigger::AliAnaCaloTrigger(const char *name) :
6c3efb37 33 AliAnalysisTask(name,""),
34 fChain(0),
35 fESD(0),
36 fOutputContainer(0),
f3299f82 37 fCalorimeter("PHOS"),
6c3efb37 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//______________________________________________________________________________
f3299f82 50AliAnaCaloTrigger::~AliAnaCaloTrigger()
6c3efb37 51{
52 // dtor
53 fOutputContainer->Clear() ;
54 delete fOutputContainer ;
55 delete fNtTrigger22 ;
56 delete fNtTriggerNN ;
57}
58
59
60//______________________________________________________________________________
f3299f82 61void AliAnaCaloTrigger::ConnectInputData(const Option_t*)
6c3efb37 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
f3299f82 86void AliAnaCaloTrigger::CreateOutputObjects()
6c3efb37 87{
88
89 // create histograms
f3299f82 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");
6c3efb37 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//______________________________________________________________________________
f3299f82 104void AliAnaCaloTrigger::Exec(Option_t *)
6c3efb37 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
f3299f82 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 }
6c3efb37 136
137 // trigger amplitudes
6c3efb37 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
4c0b6461 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)) ;
6c3efb37 150
f3299f82 151
6c3efb37 152
153 Float_t enMax = 0. ;
154 Float_t phEnMax = 0. ;
4c0b6461 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. ;
6c3efb37 166
f3299f82 167 Int_t icaloCluster ;
6c3efb37 168
f3299f82 169 // loop over the Calorimeters Clusters
6c3efb37 170
f3299f82 171 for(icaloCluster = firstCaloCluster ; icaloCluster < firstCaloCluster + numberOfCaloClusters ; icaloCluster++) {
172 AliESDCaloCluster * cluster = fESD->GetCaloCluster(icaloCluster) ;
173 if (cluster) {
6c3efb37 174
f3299f82 175 Float_t cluEnergy = cluster->GetClusterEnergy() ;
6c3efb37 176 Float_t pos[3] ;
177 TVector3 vpos ;
178
f3299f82 179 cluster->GetGlobalPosition( pos ) ;
6c3efb37 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
f3299f82 188 Float_t * pid = cluster->GetPid() ;
6c3efb37 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
4c0b6461 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.);
6c3efb37 202 }
203
204
205 PostData(0, fOutputContainer);
206
207}
208
209//______________________________________________________________________________
f3299f82 210void AliAnaCaloTrigger::Terminate(Option_t *)
6c3efb37 211{
212 // Processing when the event loop is ended
213
214}