AliPhysics  ff1d528 (ff1d528)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AliAnalysisTaskChargedParticlesRef.cxx
Go to the documentation of this file.
1 /**************************************************************************
2  * Copyright(c) 1998-2015, 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 #include <algorithm>
16 #include <array>
17 #include <functional>
18 #include <iostream>
19 #include <map>
20 
21 #include <TMath.h>
22 #include <THistManager.h>
23 #include <TLinearBinning.h>
24 
25 #include "AliAnalysisDataContainer.h"
26 #include "AliAnalysisManager.h"
27 #include "AliAnalysisUtils.h"
28 #include "AliAODInputHandler.h"
29 #include "AliAODTrack.h"
31 #include "AliEMCALGeometry.h"
32 #include "AliEMCALRecoUtils.h"
33 #include "AliEMCALTriggerPatchInfo.h"
34 #include "AliEmcalTrackSelection.h"
35 #include "AliESDtrackCuts.h"
36 #include "AliESDEvent.h"
37 #include "AliInputEventHandler.h"
38 #include "AliPIDResponse.h"
39 #include "AliTOFPIDResponse.h"
40 #include "AliVEvent.h"
41 #include "AliVEventHandler.h"
42 #include "AliVVertex.h"
43 
46 
50 
51 namespace EMCalTriggerPtAnalysis {
52 
53 AliAnalysisTaskChargedParticlesRef::AliAnalysisTaskChargedParticlesRef() :
55  fTrackCuts(nullptr),
56  fYshift(0.465),
57  fEtaSign(1),
58  fMinPt(0.1),
59  fEtaLabCut(-0.5, 0.5),
60  fEtaCmsCut(-2., 2.),
61  fPhiCut(0., TMath::TwoPi()),
62  fStudyPID(false),
63  fStudyEMCALgeo(false),
64  fEnableSumw2(false)
65 {
66 }
67 
70  fTrackCuts(nullptr),
71  fYshift(0.465),
72  fEtaSign(1),
73  fMinPt(0.1),
74  fEtaLabCut(-0.5, 0.5),
75  fEtaCmsCut(-2., 2.),
76  fPhiCut(0., TMath::TwoPi()),
77  fStudyPID(false),
78  fStudyEMCALgeo(false),
79  fEnableSumw2(false)
80 {
81  SetNeedEmcalGeom(true);
82  SetCaloTriggerPatchInfoName("EmcalTriggers");
83 }
84 
86  //if(fTrackCuts) delete fTrackCuts;
87 }
88 
90  if(!fTrackCuts) InitializeTrackCuts("standard", fInputHandler->IsA() == AliAODInputHandler::Class());
92 }
93 
95 
96  PtBinning newbinning;
97  TString optionstring = fEnableSumw2 ? "s" : "";
98 
99  TLinearBinning etabinning(64, -0.8, 0.8), phibinning(100, 0., 2*TMath::Pi()), chargebinning(2, -1.5, 1.5);
100  const TBinning *binning4D[4] = {&newbinning, &etabinning, &phibinning, &chargebinning};
101 
102  // Binning for the PID histogram
103  const int kdimPID = 3;
104  const int knbinsPID[kdimPID] = {1000, 200, 300};
105  const double kminPID[kdimPID] = {-100., 0., 0.}, kmaxPID[kdimPID] = {100., 200., 1.5};
106  for(auto trg : GetSupportedTriggers()){
107  fHistos->CreateTH1("hEventCount" + trg, "Event Counter for trigger class " + trg, 1, 0.5, 1.5, optionstring);
108  fHistos->CreateTH1("hVertexBefore" + trg, "Vertex distribution before z-cut for trigger class " + trg, 500, -50, 50, optionstring);
109  fHistos->CreateTH1("hVertexAfter" + trg, "Vertex distribution after z-cut for trigger class " + trg, 100, -10, 10, optionstring);
110 
111 
112  fHistos->CreateTHnSparse("hPtEtaPhiAll" + trg, "p_{t}-#eta-#phi distribution of all accepted tracks for trigger " + trg + " ; p_{t} (GeV/c); #eta; #phi; charge", 4, binning4D, optionstring);
113  fHistos->CreateTHnSparse("hPtEtaPhiCent" + trg, "p_{t}-#eta-#phi distribution of all accepted tracks for trigger " + trg + "; p_{t} (GeV/c); #eta; #phi; charge", 4, binning4D, optionstring);
114  if(fStudyEMCALgeo) {
115  fHistos->CreateTHnSparse("hPtEtaPhiEMCALAll" + trg, "p_{t}-#eta-#phi distribution of all accepted tracks pointing to the EMCAL for trigger " + trg + "; p_{t} (GeV/c); #eta; #phi; charge", 4, binning4D, optionstring);
116  fHistos->CreateTHnSparse("hPtEtaPhiEMCALCent" + trg, "p_{t}-#eta-#phi distribution of all accepted tracks pointing to the EMCAL for trigger " + trg + "; p_{t} (GeV/c); #eta; #phi; charge", 4, binning4D, optionstring);
117  }
118 
119  if(fStudyPID){
120  fHistos->CreateTH2(Form("hTPCdEdxEMCAL%s", trg.Data()), Form("TPC dE/dx of charged particles in the EMCAL region for trigger %s", trg.Data()), 400, -20., 20., 200, 0., 200., optionstring);
121  fHistos->CreateTH2(Form("hTOFBetaEMCAL%s", trg.Data()), Form("TOF beta of charged particles in the EMCAL region for trigger %s", trg.Data()), 400, -20., 20., 150, 0., 1.5, optionstring);
122  fHistos->CreateTHnSparse(Form("hPIDcorrEMCAL%s", trg.Data()), Form("Correlation of PID observables for Trigger %s", trg.Data()), kdimPID, knbinsPID, kminPID, kmaxPID, optionstring);
123  }
124  }
125 }
126 
128  Bool_t hasPIDresponse = fInputHandler->GetPIDResponse() != nullptr;
129  if(fStudyPID && !hasPIDresponse) AliErrorStream() << "PID requested but PID response not available" << std::endl;
130 
131  // Loop over tracks, fill select particles
132  // Histograms
133  // - Full eta_{lab} (-0.8, 0.8), new binning
134  // - Full eta_{lab} (-0.8, 0.8), old binning
135  // - Eta distribution for tracks above 1, 2, 5, 10 GeV/c without eta cut
136  // - Central eta_{cms} (-0.3, -0.2), new binning,
137  // - Central eta_{cms} (-0.8, -0.2), old binning,
138  // - Eta distribution for tracks above 1, 2, 5, 10 GeV/c
139  // - Eta distribution for tracks above 1, 2, 5, 10 GeV/c with eta cut
140  AliVTrack *checktrack(nullptr);
141  int ptmin[5] = {1,2,5,10,20}; // for eta distributions
142  Bool_t isEMCAL(kFALSE);
143  Double_t etaEMCAL(0.), phiEMCAL(0.);
144  for(int itrk = 0; itrk < fInputEvent->GetNumberOfTracks(); ++itrk){
145  checktrack = dynamic_cast<AliVTrack *>(fInputEvent->GetTrack(itrk));
146  if(!checktrack) continue;
147  if(!fEtaLabCut.IsInRange(checktrack->Eta())) continue;
148  if(!fPhiCut.IsInRange(checktrack->Phi())) continue;
149  if(TMath::Abs(checktrack->Pt()) < fMinPt) continue;
150 
151  if(fStudyEMCALgeo){
152  if(checktrack->IsA() == AliESDtrack::Class()){
153  AliESDtrack copytrack(*(static_cast<AliESDtrack *>(checktrack)));
154  AliEMCALRecoUtils::ExtrapolateTrackToEMCalSurface(&copytrack);
155  etaEMCAL = copytrack.GetTrackEtaOnEMCal();
156  phiEMCAL = copytrack.GetTrackPhiOnEMCal();
157  } else {
158  AliAODTrack copytrack(*(static_cast<AliAODTrack *>(checktrack)));
159  AliEMCALRecoUtils::ExtrapolateTrackToEMCalSurface(&copytrack);
160  etaEMCAL = copytrack.GetTrackEtaOnEMCal();
161  phiEMCAL = copytrack.GetTrackPhiOnEMCal();
162  }
163  Int_t supermoduleID = -1;
164  isEMCAL = fGeom->SuperModuleNumberFromEtaPhi(etaEMCAL, phiEMCAL, supermoduleID);
165  // Exclude supermodules 10 and 11 as they did not participate in the trigger
166  isEMCAL = isEMCAL && supermoduleID < 10;
167  }
168 
169  // Calculate eta in cms frame according
170  // EPJC74 (2014) 3054:
171  // eta_cms = - eta_lab - |yshift|
172  Double_t etacent = -1. * checktrack->Eta() - TMath::Abs(fYshift);
173  etacent *= fEtaSign;
174 
175  if(!fEtaCmsCut.IsInRange(etacent)) continue; // Apply eta-cent cut
176  if(!fTrackCuts->IsTrackAccepted(checktrack)) continue;
177 
178  // Charge separation
179  bool posCharge = checktrack->Charge() > 0;
180 
181  for(const auto &t : fSelectedTriggers){
182  FillTrackHistos(t, posCharge, checktrack->Pt(), checktrack->Eta() * fEtaSign, etacent, checktrack->Phi(), isEMCAL);
183  if(fStudyPID && hasPIDresponse)
184  if(isEMCAL) FillPIDHistos(t, *checktrack);
185  }
186  }
187  return true;
188 }
189 
191  // Apply vertex z cut
192  for(const auto &t : fSelectedTriggers){
193  Double_t weight = GetTriggerWeight(t);
194  fHistos->FillTH1(Form("hVertexBefore%s", t.Data()), fVertex[2], weight);
195  }
196 }
197 
199  for(const auto &t : fSelectedTriggers) {
200  Double_t weight = GetTriggerWeight(t);
201  // Fill Event counter and reference vertex distributions after event selection
202  fHistos->FillTH1(Form("hEventCount%s", t.Data()), 1, weight);
203  fHistos->FillTH1(Form("hVertexAfter%s", t.Data()), fVertex[2], weight);
204  }
205 }
206 
208  const TString &eventclass,
209  Bool_t posCharge,
210  Double_t pt,
211  Double_t etalab,
212  Double_t etacent,
213  Double_t phi,
214  Bool_t inEmcal
215  )
216 {
217  Double_t weight = GetTriggerWeight(eventclass);
218  AliDebugStream(1) << GetName() << ": Using weight " << weight << " for trigger " << eventclass << " in particle histograms." << std::endl;
219  double kinepointall[4] = {TMath::Abs(pt), etalab, phi, posCharge ? 1. : -1.}, kinepointcent[4] = {TMath::Abs(pt), etacent, phi, posCharge ? 1. : -1.};
220  fHistos->FillTHnSparse("hPtEtaPhiAll" + eventclass, kinepointall, weight);
221  fHistos->FillTHnSparse("hPtEtaPhiCent" + eventclass, kinepointcent, weight);
222  if(fStudyEMCALgeo && inEmcal){
223  fHistos->FillTHnSparse("hPtEtaPhiEMCALAll" + eventclass, kinepointall, weight);
224  fHistos->FillTHnSparse("hPtEtaPhiEMCALCent" + eventclass, kinepointall, weight);
225  }
226 }
227 
229  const TString &eventclass,
230  const AliVTrack &trk
231 ) {
232  Double_t weight = GetTriggerWeight(eventclass);
233  AliDebugStream(1) << GetName() << ": Using weight " << weight << " for trigger " << eventclass << " in PID histograms." << std::endl;
234  AliPIDResponse *pid = fInputHandler->GetPIDResponse();
235  if(TMath::Abs(trk.Eta()) > 0.5) return;
236  if(!((trk.GetStatus() & AliVTrack::kTOFout) && (trk.GetStatus() & AliVTrack::kTIME))) return;
237 
238  double poverz = TMath::Abs(trk.P())/static_cast<double>(trk.Charge());
239  fHistos->FillTH2("hTPCdEdxEMCAL" + eventclass, poverz, trk.GetTPCsignal(), weight);
240  // correct for units - TOF in ps, track length in cm
241  Double_t trtime = (trk.GetTOFsignal() - pid->GetTOFResponse().GetTimeZero()) * 1e-12;
242  Double_t v = trk.GetIntegratedLength()/(100. * trtime);
243  Double_t beta = v / TMath::C();
244  fHistos->FillTH2("hTOFBetaEMCAL" + eventclass, poverz, beta, weight);
245  double datapoint[3] = {poverz, trk.GetTPCsignal(), beta};
246  fHistos->FillTHnSparse("hPIDcorrEMCAL" + eventclass, datapoint, weight);
247 }
248 
251 }
252 
254  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
255 
256  TString taskname = "chargedParticleQA_" + suffix;
257 
259  mgr->AddTask(task);
260 
261  TString outfile(mgr->GetCommonFileName());
262  outfile += ":ChargedParticleQA_" + suffix;
263  TString containername = "TrackResults_" + suffix;
264 
265  task->ConnectInput(0, mgr->GetCommonInputContainer());
266  mgr->ConnectOutput(task, 1, mgr->CreateContainer(containername.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, outfile.Data()));
267 
268  return task;
269 }
270 
272  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
273 
275  mgr->AddTask(task);
276 
277  // Set Energy thresholds for additional patch selection:
278  // These are events with offline patches of a given type where the trigger reached already the plateau
279  // These numers are determined as:
280  // EMC7: 5 GeV
281  // EG1: 14 GeV
282  // EG2: 8 GeV
283  // EJ1: 22 GeV
284  // EJ2: 12 GeV
285  mgr->AddTask(task);
288  );
291  cutname,
292  mgr->GetInputEventHandler()->IsA() == AliAODInputHandler::Class()
293  )
294  );
295 
296  TString outfile(mgr->GetCommonFileName());
297  outfile += ":ChargedParticleQA_%s" + cutname;
298 
299  task->ConnectInput(0, mgr->GetCommonInputContainer());
300  mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("TrackResults_%s", cutname.Data()), TList::Class(), AliAnalysisManager::kOutputContainer, outfile.Data()));
301 
302  return task;
303 }
304 
307 {
308  this->SetMinimum(0.);
309  this->AddStep(1, 0.05);
310  this->AddStep(2, 0.1);
311  this->AddStep(4, 0.2);
312  this->AddStep(7, 0.5);
313  this->AddStep(16, 1);
314  this->AddStep(36, 2);
315  this->AddStep(40, 4);
316  this->AddStep(50, 5);
317  this->AddStep(100, 10);
318  this->AddStep(200, 20);
319 }
320 
321 } /* namespace EMCalTriggerPtAnalysis */
std::vector< TString > fSelectedTriggers
! Triggers selected for given event
virtual bool Run()
Simple task testing particle spectra in triggered events.
double Double_t
Definition: External.C:58
Double_t fEtaSign
Sign of the eta distribution (swaps when beam directions swap): p-Pb: +1, Pb-p: -1.
Class creating a linear binning, used in the histogram manager.
static AliAnalysisTaskChargedParticlesRef * AddTaskChargedParticlesRef(const TString &suffix)
Pre-configure task so that it can be used in subwagons.
void FillTH2(const char *hname, double x, double y, double weight=1., Option_t *opt="")
Fill a 2D histogram within the container.
virtual void SaveQAObjects(TList *outputList)
void AddStep(Double_t max, Double_t binwidth)
virtual void UserFillHistosBeforeEventSelection()
Implementation of framework function UserFillHistosBeforeEventSelection.
Interface for binnings used by the histogram handler.
Definition: TBinning.h:21
void SetOfflineTriggerSelection(AliEmcalTriggerOfflineSelection *sel)
Set an offline trigger selection.
void SetCaloTriggerPatchInfoName(const char *n)
static AliEmcalTrackSelection * TrackCutsFactory(TString name, Bool_t isAOD)
Fully-configure EMCAL track selection independent of the data type.
Bool_t fStudyEMCALgeo
Add histograms for tracks pointing to the EMCAL acceptance.
void FillTHnSparse(const char *name, const double *x, double weight=1., Option_t *opt="")
static AliAnalysisTaskChargedParticlesRef * AddTaskChargedParticlesRefDefault(const TString &cutname="standard")
Fully configure task, not intended for subwagons. Using a default cut configuration.
TH2 * CreateTH2(const char *name, const char *title, int nbinsx, double xmin, double xmax, int nbinsy, double ymin, double ymax, Option_t *opt="")
Create a new TH2 within the container.
int Int_t
Definition: External.C:63
AliEMCALGeometry * fGeom
!emcal geometry
void FillTrackHistos(const TString &eventclass, Bool_t posCharge, Double_t pt, Double_t eta, Double_t etacent, Double_t phi, Bool_t inEmcal)
Fill track (kinematic) histograms.
void SetEMCALTrackSelection(AliEmcalTrackSelection *sel)
Set the virtual track selection.
TH1 * CreateTH1(const char *name, const char *title, int nbins, double xmin, double xmax, Option_t *opt="")
Create a new TH1 within the container.
Helper class creating user defined custom binning.
const Double_t ptmin
void FillTH1(const char *hname, double x, double weight=1., Option_t *opt="")
Fill a 1D histogram within the container.
Declaration of class AliEMCalTriggerExtraCuts.
AliEmcalList * fOutput
!output list
ClassImp(AliAnalysisTaskCRC) AliAnalysisTaskCRC
Double_t fVertex[3]
!event vertex
void FillPIDHistos(const TString &eventclass, const AliVTrack &track)
Fill PID-related histograms.
void SetNeedEmcalGeom(Bool_t n)
void InitializeTrackCuts(TString cutname, bool isAOD)
Set the track selection.
bool Bool_t
Definition: External.C:53
virtual void UserFillHistosAfterEventSelection()
Implementation of framework function UserFillHistosAfterEventSelection.
THnSparse * CreateTHnSparse(const char *name, const char *title, int ndim, const int *nbins, const double *min, const double *max, Option_t *opt="")
Create a new THnSparse within the container.
static AliEmcalTriggerOfflineSelection * TriggerSelectionFactory(Double_t el0, Double_t eg1, Double_t eg2, Double_t ej1, Double_t ej2, AliEmcalTriggerOfflineSelection::EmcalEnergyDefinition_t endef=AliEmcalTriggerOfflineSelection::kFEEEnergy)
Configures EMCAL trigger offline selection used to restrict EMCAL triggered sample.
virtual bool IsTrackAccepted(AliVTrack *const trk)=0
Interface for track selection code.
void SetMinimum(Double_t min)