AliPhysics  ec7afe5 (ec7afe5)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AliAnalysisTaskEmcalLight.h
Go to the documentation of this file.
1 #ifndef ALIANALYSISTASKEMCALLIGHT_H
2 #define ALIANALYSISTASKEMCALLIGHT_H
3 /* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice */
5 
6 class TClonesArray;
7 class TString;
8 class TList;
9 class AliEmcalParticle;
10 class AliMCParticle;
11 class AliVCluster;
12 class AliVTrack;
13 class AliVParticle;
14 class AliVCaloCells;
15 class TH1;
16 class TProfile;
17 class AliEMCALGeometry;
18 class AliGenPythiaEventHeader;
19 class AliVCaloTrigger;
20 class AliAnalysisUtils;
21 class AliEMCALTriggerPatchInfo;
22 class AliAODTrack;
23 
24 #include <map>
25 #include <set>
26 #include <string>
27 
28 #include "Rtypes.h"
29 
30 #include "AliParticleContainer.h"
31 #include "AliMCParticleContainer.h"
32 #include "AliTrackContainer.h"
33 #include "AliClusterContainer.h"
34 
35 #include "AliAnalysisTaskSE.h"
70  public:
71 
76  enum EDataType_t {
80  };
81 
86  enum EBeamType_t {
87  kNA = -1,
88  kpp = 0,
89  kAA = 1,
90  kpA = 2
91  };
92 
94  AliAnalysisTaskEmcalLight(const char *name, Bool_t histo=kFALSE);
96 
97  // Containers
98  AliParticleContainer *AddParticleContainer(std::string branchName, std::string contName="");
99  AliClusterContainer *AddClusterContainer(std::string branchName, std::string contName="");
100  void AdoptParticleContainer(AliParticleContainer* cont) { fParticleCollArray[cont->GetName()] = cont; }
101  void AdoptClusterContainer(AliClusterContainer* cont) { fClusterCollArray[cont->GetName()] = cont; }
102  AliParticleContainer *GetParticleContainer(std::string name) const;
103  AliClusterContainer *GetClusterContainer(std::string name) const;
104  AliMCParticleContainer *GetMCParticleContainer(std::string name) const { return dynamic_cast<AliMCParticleContainer*>(GetParticleContainer(name)); }
105  AliTrackContainer *GetTrackContainer(std::string name) const { return dynamic_cast<AliTrackContainer*>(GetParticleContainer(name)) ; }
106  void RemoveParticleContainer(std::string name) { fParticleCollArray.erase(name) ; }
107  void RemoveClusterContainer(std::string name) { fClusterCollArray.erase(name) ; }
108 
109  // Other input data
110  void SetCaloCellsName(const char *n) { fCaloCellsName = n ; }
112  void SetCaloTriggersName(const char *n) { fCaloTriggersName = n ; }
113  void SetCentralityEstimator(const char *c) { fCentEst = c ; }
114  void SetIsPythia(Bool_t i) { fIsPythia = i ; }
116 
117  // Task configuration
121  void SetCentBins(const std::vector<double>& bins) { fCentBins = std::vector<double>(bins) ; }
122  Int_t GetNCentBins() const { return fCentBins.size() > 1 ? fCentBins.size() - 1 : 1; }
124 
125  // Event selection
127  void SetCentRange(Double_t min, Double_t max) { fMinCent = min ; fMaxCent = max ; }
128  void SetVzRange(Double_t min, Double_t max) { fMinVz = min ; fMaxVz = max ; }
130  void SetMinNVertCont(Int_t cut) { fMinNVertCont = cut ; }
131  void SetPtHardRange(Double_t min, Double_t max) { fMinPtHard = min ; fMaxPtHard = max ; }
133  void AddAcceptedTriggerClass(const char* trigClass) { fAcceptedTriggerClasses.insert(trigClass) ; }
134  void AddRejectedTriggerClass(const char* trigClass) { fRejectedTriggerClasses.insert(trigClass) ; }
137  void SetMCFilter() { fMCRejectFilter = kTRUE ; }
138  void ResetMCFilter() { fMCRejectFilter = kFALSE ; }
145 
146  protected:
147  void SetRejectionReasonLabels(TAxis* axis);
148  void AddObjectToEvent(TObject *obj, Bool_t attempt = kFALSE);
149  TClonesArray *GetArrayFromEvent(const char *name, const char *clname=0);
151  Bool_t PythiaInfoFromFile(const char* currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard);
152  Bool_t IsTrackInEmcalAcceptance(AliVParticle* part, Double_t edges=0.9) const;
154 
155  // Overloaded AliAnalysisTaskSE methods
157  void UserExec(Option_t *option);
158  Bool_t UserNotify();
159 
160  // Virtual functions, to be overloaded in derived classes
161  virtual void ExecOnce();
162  virtual Bool_t FillGeneralHistograms(Bool_t eventSelected);
163  virtual Bool_t IsEventSelected();
164  virtual Bool_t RetrieveEventObjects();
170  virtual Bool_t FillHistograms() { return kTRUE ; }
178  virtual Bool_t Run() { return kTRUE ; }
179 
180  // Static utilities
181  static void GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff);
182  static Byte_t GetTrackType(const AliVTrack *t);
183  static Byte_t GetTrackType(const AliAODTrack *aodTrack, UInt_t filterBit1, UInt_t filterBit2);
184  static Double_t DeltaPhi(Double_t phia, Double_t phib, Double_t rMin = -TMath::Pi()/2, Double_t rMax = 3*TMath::Pi()/2);
186  static void GenerateFixedBinArray(Int_t n, Double_t min, Double_t max, Double_t* array);
187  static Double_t GetParallelFraction(AliVParticle* part1, AliVParticle* part2);
188  static Double_t GetParallelFraction(const TVector3& vect1, AliVParticle* part2);
189  static EBeamType_t BeamTypeFromRunNumber(Int_t runnumber);
190 
192 
193  // Task configuration
198  std::vector<double> fCentBins;
200 
201  // Input data
207 
208  std::map<std::string,
210  std::map<std::string,
212 
213  // Event selection
224  std::set<std::string> fAcceptedTriggerClasses;
225  std::set<std::string> fRejectedTriggerClasses;
231 
232  // Service fields
235  AliEMCALGeometry *fGeom;
236  AliVCaloCells *fCaloCells;
237  AliVCaloTrigger *fCaloTriggers;
238  TClonesArray *fTriggerPatchInfo;
249  std::vector<std::string> fFiredTriggerClasses;
251  AliGenPythiaEventHeader *fPythiaHeader;
256 
257  // Output
277  TProfile *fHistXsection;
278 
279  private:
280  AliAnalysisTaskEmcalLight(const AliAnalysisTaskEmcalLight&); // not implemented
282 
284  ClassDef(AliAnalysisTaskEmcalLight, 3);
286 };
287 
297 {
298  Double_t dphi = -999;
299  const Double_t tpi = TMath::TwoPi();
300 
301  if (phia < 0) phia += tpi;
302  else if (phia > tpi) phia -= tpi;
303  if (phib < 0) phib += tpi;
304  else if (phib > tpi) phib -= tpi;
305  dphi = phib - phia;
306  if (dphi < rangeMin) dphi += tpi;
307  else if (dphi > rangeMax) dphi -= tpi;
308 
309  return dphi;
310 }
311 
322 {
323  Double_t binWidth = (max-min)/n;
324  array[0] = min;
325  for (Int_t i = 1; i <= n; i++) {
326  array[i] = array[i-1]+binWidth;
327  }
328 }
329 
340 {
341  Double_t *array = new Double_t[n+1];
342  GenerateFixedBinArray(n, min, max, array);
343  return array;
344 }
345 
346 #endif
Bool_t fSwitchOffLHC15oFaultyBranches
Switch off faulty tree branches in LHC15o AOD trees.
AliClusterContainer * AddClusterContainer(std::string branchName, std::string contName="")
Float_t fPtHardAndJetPtFactor
Factor between ptHard and jet pT to reject/accept event.
Double_t fVertexSPD[3]
!event Svertex
TString fCaloTriggersName
name of calo triggers collection
AliEMCALGeometry * fGeom
!emcal geometry
EBeamType_t fBeamType
!event beam type
double Double_t
Definition: External.C:58
TClonesArray * GetArrayFromEvent(const char *name, const char *clname=0)
EDataType_t
Switch for the data type.
Float_t fXsection
!x-section from pythia header
Double_t fEPV0A
!event plane V0A
TH1 * fHistZVertexNoSel
!z vertex position (no event selection)
Double_t fEPV0
!event plane V0
TH1 * fHistEventCount
!incoming and selected events
TProfile * fHistXsectionVsPtHardNoSel
!x section from pythia header (no event selection)
Container with name, TClonesArray and cuts for particles.
Bool_t fMCRejectFilter
enable the filtering of events by tail rejection
TString fCaloTriggerPatchInfoName
trigger patch info array name
Double_t fMinVz
min vertex for event selection
void AddRejectedTriggerClass(const char *trigClass)
TCanvas * c
Definition: TestFitELoss.C:172
TString fCentEst
name of the centrality estimator
TH1 * fHistTrialsVsPtHardNoSel
!total number of trials per pt hard bin after selection (no event selection)
UInt_t fTriggerSelectionBitMap
trigger selection bit map
TH1 * fHistCentrality
!event centrality distribution
std::vector< double > fCentBins
how many centrality bins
static Byte_t GetTrackType(const AliVTrack *t)
TH1 * fHistTriggerClasses
!number of events in each trigger class
TString fCaloCellsName
name of calo cell collection
TH1 * fHistEvents
!total number of events per pt hard bin
Bool_t IsTrackInEmcalAcceptance(AliVParticle *part, Double_t edges=0.9) const
Double_t fMinCent
min centrality for event selection
std::set< std::string > fRejectedTriggerClasses
list of accepted trigger classes
void SetCentRange(Double_t min, Double_t max)
Double_t fMaxMinimumBiasPtHard
maximum pt hard for the minimum bias pt hard bin (MC)
Container for particles within the EMCAL framework.
void SetPtHardRange(Double_t min, Double_t max)
static void GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff)
void AddObjectToEvent(TObject *obj, Bool_t attempt=kFALSE)
static Double_t fgkEMCalDCalPhiDivide
phi value used to distinguish between DCal and EMCal
EBeamType_t fForceBeamType
forced beam type
std::vector< std::string > fFiredTriggerClasses
!trigger classes fired by the current event
Bool_t fCreateHisto
whether or not create histograms
TH1 * fHistTrialsVsPtHard
!total number of trials per pt hard bin after selection
TH1 * fHistEventRejection
!book keep reasons for rejecting event
int Int_t
Definition: External.C:63
TH1 * fHistTrials
!trials from pyxsec.root
void AddAcceptedTriggerClass(const char *trigClass)
Double_t fMinPtHard
select minimum pt hard (MC)
unsigned int UInt_t
Definition: External.C:33
float Float_t
Definition: External.C:68
void RemoveParticleContainer(std::string name)
Base task in the EMCAL framework (lighter version of AliAnalysisTaskEmcal)
TH1 * fHistEventsVsPtHard
!total number of events per pt hard bin after selection
TH1 * fHistTriggerClassesNoSel
!number of events in each trigger class (no event selection)
Float_t fPtHardAndClusterPtFactor
Factor between ptHard and cluster pT to reject/accept event.
Double_t fMaxVzDiff
upper limit for distance between primary and SPD vertex
Double_t fVertex[3]
!event vertex
static Double_t * GenerateFixedBinArray(Int_t n, Double_t min, Double_t max)
TH1 * fHistCentralityNoSel
!event centrality distribution (no event selection)
AliClusterContainer * GetClusterContainer(std::string name) const
void AdoptClusterContainer(AliClusterContainer *cont)
Bool_t fIsPythia
if it is a PYTHIA production
TProfile * fHistXsection
!x section from pyxsec.root
Int_t fCentBin
!event centrality bin
unsigned long ULong_t
Definition: External.C:38
Double_t fMaxCent
max centrality for event selection
ULong_t fFiredTriggerBitMap
!bit map of fired triggers
Double_t fMinNVertCont
minumum number of vertex contributors
Int_t fNVertSPDCont
!event SPD vertex number of contributors
std::set< std::string > fAcceptedTriggerClasses
list of accepted trigger classes
static Double_t DeltaPhi(Double_t phia, Double_t phib, Double_t rMin=-TMath::Pi()/2, Double_t rMax=3 *TMath::Pi()/2)
TH1 * fHistEventPlaneNoSel
!event plane distribution (no event selection)
void SetCentralityEstimator(const char *c)
Bool_t PythiaInfoFromFile(const char *currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard)
Bool_t fLocalInitialized
!whether or not the task has been already initialized
void SetCentBins(const std::vector< double > &bins)
static EBeamType_t BeamTypeFromRunNumber(Int_t runnumber)
void SetCaloTriggerPatchInfoName(const char *n)
AliParticleContainer * AddParticleContainer(std::string branchName, std::string contName="")
void RemoveClusterContainer(std::string name)
std::map< std::string, AliParticleContainer * > fParticleCollArray
particle/track collection array
void SetVzRange(Double_t min, Double_t max)
TH1 * fHistEventsVsPtHardNoSel
!total number of events per pt hard bin after selection (no event selection)
Int_t fNVertCont
!event vertex number of contributors
Bool_t fGeneralHistograms
whether or not it should fill some general histograms
AliGenPythiaEventHeader * fPythiaHeader
!event Pythia header
const char Option_t
Definition: External.C:48
Float_t fPtHardAndTrackPtFactor
Factor between ptHard and track pT to reject/accept event.
AliParticleContainer * GetParticleContainer(std::string name) const
bool Bool_t
Definition: External.C:53
AliMCParticleContainer * GetMCParticleContainer(std::string name) const
TH1 * fHistEventPlane
!event plane distribution
std::map< std::string, AliClusterContainer * > fClusterCollArray
cluster collection array
EBeamType_t
Switch for the beam type.
AliVCaloTrigger * fCaloTriggers
!calo triggers
TProfile * fHistXsectionVsPtHard
!x section from pythia header
AliTrackContainer * GetTrackContainer(std::string name) const
Container structure for EMCAL clusters.
Container for MC-true particles within the EMCAL framework.
TH1 * fHistZVertex
!z vertex position
AliAnalysisTaskEmcalLight & operator=(const AliAnalysisTaskEmcalLight &)
void AdoptParticleContainer(AliParticleContainer *cont)
virtual Bool_t FillGeneralHistograms(Bool_t eventSelected)
Double_t fMaxVz
max vertex for event selection
Double_t fCent
!event centrality
Double_t fEPV0C
!event plane V0C
Definition: External.C:196
Bool_t fUseNewCentralityEstimation
Use new centrality estimation (for 2015 data)
TClonesArray * fTriggerPatchInfo
!trigger patch info array
Double_t fMaxPtHard
select maximum pt hard (MC)
Bool_t fNeedEmcalGeom
whether or not the task needs the emcal geometry
static Double_t GetParallelFraction(AliVParticle *part1, AliVParticle *part2)
EDataType_t fDataType
!data type (ESD or AOD)