AliPhysics  8bb951a (8bb951a)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AliAnalysisTaskEmcal.h
Go to the documentation of this file.
1 #ifndef ALIANALYSISTASKEMCAL_H
2 #define ALIANALYSISTASKEMCAL_H
3 /* Copyright(c) 1998-1999, 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 class AliEmcalPythiaInfo;
24 
25 #include "Rtypes.h"
26 
27 #include "AliParticleContainer.h"
28 #include "AliMCParticleContainer.h"
29 #include "AliTrackContainer.h"
30 #include "AliClusterContainer.h"
31 
32 #include "AliAnalysisTaskSE.h"
66 class AliAnalysisTaskEmcal : public AliAnalysisTaskSE {
67  public:
68 
73  enum BeamType {
74  kNA = -1,
75  kpp = 0,
76  kAA = 1,
77  kpA = 2
78  };
79 
84  enum TriggerType {
85  kND = -1,
86  kJ1 = 0,
87  kJ2 = 1,
88  kG1 = 2,
89  kG2 = 3,
90  kL0 = 4
91  };
92 
103  };
104 
112  };
113 
115  AliAnalysisTaskEmcal(const char *name, Bool_t histo=kFALSE);
116  virtual ~AliAnalysisTaskEmcal();
117 
119  AliTrackContainer *AddTrackContainer(const char *n);
126  AliParticleContainer *GetParticleContainer(Int_t i=0) const;
127  AliParticleContainer *GetParticleContainer(const char* name) const;
128  AliClusterContainer *GetClusterContainer(Int_t i=0) const;
129  AliClusterContainer *GetClusterContainer(const char* name) const;
131  AliMCParticleContainer *GetMCParticleContainer(const char* name) const { return dynamic_cast<AliMCParticleContainer*>(GetParticleContainer(name)); }
132  AliTrackContainer *GetTrackContainer(Int_t i=0) const { return dynamic_cast<AliTrackContainer*>(GetParticleContainer(i)) ; }
133  AliTrackContainer *GetTrackContainer(const char* name) const { return dynamic_cast<AliTrackContainer*>(GetParticleContainer(name)) ; }
134  void RemoveParticleContainer(Int_t i=0) { fParticleCollArray.RemoveAt(i) ; }
135  void RemoveClusterContainer(Int_t i=0) { fClusterCollArray.RemoveAt(i) ; }
136  void SetCaloCellsName(const char *n) { fCaloCellsName = n ; }
138  void SetCaloTriggersName(const char *n) { fCaloTriggersName = n ; }
139  void SetCentRange(Double_t min, Double_t max) { fMinCent = min ; fMaxCent = max ; }
140  void SetCentralityEstimator(const char *c) { fCentEst = c ; }
141  void SetClusName(const char *n) { AddClusterContainer(n) ; }
142  void SetClusPtCut(Double_t cut, Int_t c=0);
143  void SetClusTimeCut(Double_t min, Double_t max, Int_t c=0);
144  void SetEventPlaneVsEmcal(Double_t ep) { fEventPlaneVsEmcal = ep ; }
146  void SetHistoBins(Int_t nbins, Double_t min, Double_t max) { fNbins = nbins; fMinBinPt = min; fMaxBinPt = max ; }
147  void SetIsEmbedded(Bool_t i) { fIsEmbedded = i ; }
148  void SetIsPythia(Bool_t i) { fIsPythia = i ; }
150  void SetMCLabelShift(Int_t s) { fMCLabelShift = s ; }
151  void SetMinMCLabel(Int_t s) { fMinMCLabel = s ; }
152  void SetMinNTrack(Int_t min) { fMinNTrack = min ; }
153  void SetZvertexDiffValue(Double_t cut) { fZvertexDiff = cut ; }
154  void SetMinPtTrackInEmcal(Double_t min) { fMinPtTrackInEmcal = min ; }
155  virtual void SetNCentBins(Int_t n) { fNcentBins = n ; }
156  void SetNeedEmcalGeom(Bool_t n) { fNeedEmcalGeom = n ; }
157  void SetOffTrigger(UInt_t t) { fOffTrigger = t ; }
158  void SetTrackEtaLimits(Double_t min, Double_t max, Int_t c=0);
159  void SetTrackPhiLimits(Double_t min, Double_t max, Int_t c=0);
160  void SetTrackPtCut(Double_t cut, Int_t c=0);
161  void SetTracksName(const char *n) { AddParticleContainer(n) ; }
162  void SetTrigClass(const char *n) { fTrigClass = n ; }
164  void SetUseAliAnaUtils(Bool_t b, Bool_t bRejPilup = kTRUE) { fUseAliAnaUtils = b ; fRejectPileup = bRejPilup ; }
165  void SetVzRange(Double_t min, Double_t max) { fMinVz = min ; fMaxVz = max ; }
170  void SetPythiaInfoName(const char *n) { fPythiaInfoName = n ; }
171  const TString& GetPythiaInfoName() const { return fPythiaInfoName ; }
172  const AliEmcalPythiaInfo *GetPythiaInfo() const { return fPythiaInfo ; }
173 
174  protected:
175  void LoadPythiaInfo(AliVEvent *event);
176  void SetRejectionReasonLabels(TAxis* axis);
177  Bool_t AcceptCluster(AliVCluster *clus, Int_t c = 0) const;
178  Bool_t AcceptTrack(AliVParticle *track, Int_t c = 0) const;
179  void AddObjectToEvent(TObject *obj, Bool_t attempt = kFALSE);
180  AliVParticle *GetAcceptParticleFromArray(Int_t p, Int_t c=0) const;
181  AliVCluster *GetAcceptClusterFromArray(Int_t cl, Int_t c=0) const;
182  TClonesArray *GetArrayFromEvent(const char *name, const char *clname=0);
184  TClonesArray *GetParticleArray(Int_t i=0) const;
185  TClonesArray *GetClusterArray(Int_t i=0) const;
186  Int_t GetNParticles(Int_t i=0) const;
187  Int_t GetNClusters(Int_t i=0) const;
188  AliEMCALTriggerPatchInfo *GetMainTriggerPatch(TriggerCategory triggersel = kTriggerLevel1Jet, Bool_t doOfflinSimple = kFALSE);
189  Bool_t HasTriggerType(TriggerType triggersel);
190  ULong_t GetTriggerList();
191  Bool_t PythiaInfoFromFile(const char* currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard);
192  Bool_t IsTrackInEmcalAcceptance(AliVParticle* part, Double_t edges=0.9) const;
193 
194  void GeneratePythiaInfoObject(AliMCEvent* mcEvent);
195 
196  // Overloaded AliAnalysisTaskSE methods
198  void UserExec(Option_t *option);
199  Bool_t UserNotify();
200 
201  // Virtual functions, to be overloaded in derived classes
202  virtual void ExecOnce();
203  virtual Bool_t FillGeneralHistograms();
204  virtual Bool_t IsEventSelected();
205  virtual Bool_t RetrieveEventObjects();
211  virtual Bool_t FillHistograms() { return kTRUE ; }
219  virtual Bool_t Run() { return kTRUE ; }
220 
221  // Static utilities
222  static void GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff);
223  static Byte_t GetTrackType(const AliVTrack *t);
224  static Byte_t GetTrackType(const AliAODTrack *aodTrack, UInt_t filterBit1, UInt_t filterBit2);
225  static Double_t DeltaPhi(Double_t phia, Double_t phib, Double_t rMin = -TMath::Pi()/2, Double_t rMax = 3*TMath::Pi()/2);
226  static Double_t* GenerateFixedBinArray(Int_t n, Double_t min, Double_t max);
227  static void GenerateFixedBinArray(Int_t n, Double_t min, Double_t max, Double_t* array);
228  static Double_t GetParallelFraction(AliVParticle* part1, AliVParticle* part2);
229  static Double_t GetParallelFraction(const TVector3& vect1, AliVParticle* part2);
230 
231  static Double_t fgkEMCalDCalPhiDivide;
232 
233  // Task configuration
234  TString fPythiaInfoName;
237  Bool_t fInitialized;
238  Bool_t fCreateHisto;
239  TString fCaloCellsName;
242  Double_t fMinCent;
243  Double_t fMaxCent;
244  Double_t fMinVz;
245  Double_t fMaxVz;
246  Double_t fTrackPtCut;
247  Int_t fMinNTrack;
248  Double_t fZvertexDiff;
250  Bool_t fRejectPileup;
252  UInt_t fOffTrigger;
253  TString fTrigClass;
255  Int_t fNbins;
256  Double_t fMinBinPt;
257  Double_t fMaxBinPt;
260  Double_t fMinEventPlane;
261  Double_t fMaxEventPlane;
262  TString fCentEst;
263  Bool_t fIsEmbedded;
264  Bool_t fIsPythia;
266  Int_t fMinMCLabel;
268  Int_t fNcentBins;
269  Bool_t fNeedEmcalGeom;
270  TObjArray fParticleCollArray;
271  TObjArray fClusterCollArray;
272  ULong_t fTriggers;
276 
277  // Service fields
278  AliAnalysisUtils *fAliAnalysisUtils;
279  Bool_t fIsEsd;
280  AliEMCALGeometry *fGeom;
281  TClonesArray *fTracks;
282  TClonesArray *fCaloClusters;
283  AliVCaloCells *fCaloCells;
284  AliVCaloTrigger *fCaloTriggers;
285  TClonesArray *fTriggerPatchInfo;
286  Double_t fCent;
287  Int_t fCentBin;
288  Double_t fEPV0;
289  Double_t fEPV0A;
290  Double_t fEPV0C;
291  Double_t fVertex[3];
292  Double_t fVertexSPD[3];
293  Int_t fNVertCont;
296  AliGenPythiaEventHeader *fPythiaHeader;
297  Double_t fPtHard;
298  Int_t fPtHardBin;
299  Int_t fNTrials;
300  Float_t fXsection;
302 
303  // Output
304  TList *fOutput;
309  TH1 *fHistTrials;
310  TH1 *fHistEvents;
311  TProfile *fHistXsection;
312  TH1 *fHistPtHard;
318 
319  private:
320  AliAnalysisTaskEmcal(const AliAnalysisTaskEmcal&); // not implemented
321  AliAnalysisTaskEmcal &operator=(const AliAnalysisTaskEmcal&); // not implemented
322 
324  ClassDef(AliAnalysisTaskEmcal, 14) // EMCAL base analysis task
326 };
327 
336 inline Double_t AliAnalysisTaskEmcal::DeltaPhi(Double_t phia, Double_t phib, Double_t rangeMin, Double_t rangeMax)
337 {
338  Double_t dphi = -999;
339  const Double_t tpi = TMath::TwoPi();
340 
341  if (phia < 0) phia += tpi;
342  else if (phia > tpi) phia -= tpi;
343  if (phib < 0) phib += tpi;
344  else if (phib > tpi) phib -= tpi;
345  dphi = phib - phia;
346  if (dphi < rangeMin) dphi += tpi;
347  else if (dphi > rangeMax) dphi -= tpi;
348 
349  return dphi;
350 }
351 
361 inline void AliAnalysisTaskEmcal::GenerateFixedBinArray(Int_t n, Double_t min, Double_t max, Double_t* array)
362 {
363  Double_t binWidth = (max-min)/n;
364  array[0] = min;
365  for (Int_t i = 1; i <= n; i++) {
366  array[i] = array[i-1]+binWidth;
367  }
368 }
369 
379 inline Double_t* AliAnalysisTaskEmcal::GenerateFixedBinArray(Int_t n, Double_t min, Double_t max)
380 {
381  Double_t *array = new Double_t[n+1];
382  GenerateFixedBinArray(n, min, max, array);
383  return array;
384 }
385 
386 #endif
void SetCentRange(Double_t min, Double_t max)
void SetPythiaInfoName(const char *n)
Bool_t fGeneratePythiaInfoObject
Generate Pythia info object.
TObjArray fClusterCollArray
cluster collection array
Int_t fNVertSPDCont
!event SPD vertex number of contributors
AliMCParticleContainer * GetMCParticleContainer(const char *name) const
Bool_t fIsPythia
trigger, if it is a PYTHIA production
TH1 * fHistTrials
!trials from pyxsec.root
static Double_t DeltaPhi(Double_t phia, Double_t phib, Double_t rMin=-TMath::Pi()/2, Double_t rMax=3 *TMath::Pi()/2)
EMCAL Level1 gamma trigger, low threshold.
AliEmcalPythiaInfo * fPythiaInfo
!event parton info
Bool_t AcceptTrack(AliVParticle *track, Int_t c=0) const
EMCAL Level1 jet trigger, low threshold.
Bool_t HasTriggerType(TriggerType triggersel)
Int_t fNTrials
!event trials
void SetMinNTrack(Int_t min)
UInt_t fOffTrigger
offline trigger for event selection
void AdoptParticleContainer(AliParticleContainer *cont)
Double_t fVertexSPD[3]
!event Svertex
Double_t fMinCent
min centrality for event selection
Double_t fTrackPtCut
cut on track pt in event selection
No special treatment for EMCal triggers.
Recalculated jet trigger patch; does not need to be above trigger threshold.
Base task in the EMCAL framework.
Container with name, TClonesArray and cuts for particles.
void SetUseAliAnaUtils(Bool_t b, Bool_t bRejPilup=kTRUE)
void SetUseSPDTrackletVsClusterBG(Bool_t b)
Double_t fPtHard
!event pt hard
void SetTrackPtCut(Double_t cut, Int_t c=0)
static void GetEtaPhiDiff(const AliVTrack *t, const AliVCluster *v, Double_t &phidiff, Double_t &etadiff)
Double_t fMinBinPt
min pt in histograms
Double_t fEPV0
!event plane V0
void SetUseNewCentralityEstimation(Bool_t b)
Bool_t fGeneralHistograms
whether or not it should fill some general histograms
Bool_t AcceptCluster(AliVCluster *clus, Int_t c=0) const
AliAnalysisTaskEmcal & operator=(const AliAnalysisTaskEmcal &)
Int_t fCentBin
!event centrality bin
TH1 * fHistEventsAfterSel
!total number of events per pt hard bin after selection
void SetTriggerTypeSel(TriggerType t)
Double_t fMinPtTrackInEmcal
min pt track in emcal
void SetVzRange(Double_t min, Double_t max)
TH1 * fHistEventPlane
!event plane distribution
TList * fOutput
!output list
void SetTracksName(const char *n)
AliTrackContainer * GetTrackContainer(const char *name) const
TH1 * fHistEvents
!total number of events per pt hard bin
void SetClusPtCut(Double_t cut, Int_t c=0)
AliClusterContainer * AddClusterContainer(const char *n)
Double_t fEPV0C
!event plane V0C
TH1 * fHistCentrality
!event centrality distribution
void SetCaloTriggerPatchInfoName(const char *n)
Container for particles within the EMCAL framework.
Bool_t fIsEmbedded
trigger, embedded signal
void AdoptClusterContainer(AliClusterContainer *cont)
TObjArray fParticleCollArray
particle/track collection array
BeamType
Switch for the beam type.
void SetTrackEtaLimits(Double_t min, Double_t max, Int_t c=0)
AliParticleContainer * GetParticleContainer(Int_t i=0) const
void SetEMCalTriggerMode(EMCalTriggerMode_t m)
TProfile * fHistXsectionAfterSel
!x section from pythia header
TriggerType
Switch for EMCAL trigger types.
EMCalTriggerMode_t fEMCalTriggerMode
EMCal trigger selection mode.
virtual Bool_t FillHistograms()
void SetForceBeamType(BeamType f)
Int_t GetNParticles(Int_t i=0) const
TClonesArray * fCaloClusters
!clusters
Bool_t fUseNewCentralityEstimation
Use new centrality estimation (for 2015 data)
Bool_t IsTrackInEmcalAcceptance(AliVParticle *part, Double_t edges=0.9) const
TH1 * fHistTriggerClasses
!number of events in each trigger class
Double_t fMaxVz
max vertex for event selection
void GeneratePythiaInfoObject(AliMCEvent *mcEvent)
AliEMCALGeometry * fGeom
!emcal geometry
The overlap between low and high threshold trigger is assigned to the lower threshold only...
kRecalculated gamma trigger patch; does not need to be above trigger threshold
TString fCaloTriggerPatchInfoName
trigger patch info array name
TString fCaloTriggersName
name of calo triggers collection
AliGenPythiaEventHeader * fPythiaHeader
!event Pythia header
void SetTrackPhiLimits(Double_t min, Double_t max, Int_t c=0)
AliParticleContainer * AddParticleContainer(const char *n)
AliAnalysisUtils * fAliAnalysisUtils
!vertex selection (optional)
BeamType fForceBeamType
forced beam type
Int_t fNcentBins
how many centrality bins
AliClusterContainer * GetClusterContainer(Int_t i=0) const
TriggerType fTriggerTypeSel
trigger type to select based on trigger patches
virtual Bool_t FillGeneralHistograms()
TString fTrigClass
trigger class name for event selection
TClonesArray * GetParticleArray(Int_t i=0) const
Double_t fMinVz
min vertex for event selection
BeamType fBeamType
!event beam type
Double_t fCent
!event centrality
Double_t fMinEventPlane
minimum event plane value
TString fCaloCellsName
name of calo cell collection
Int_t fMCLabelShift
if MC label > fMCLabelShift, MC label -= fMCLabelShift
Int_t GetNClusters(Int_t i=0) const
Int_t fNVertCont
!event vertex number of contributors
void SetClusName(const char *n)
void RemoveClusterContainer(Int_t i=0)
Double_t fZvertexDiff
upper limit for distance between primary and SPD vertex
EMCAL Level0 trigger.
EMCAL Level1 jet trigger, high threshold.
void AdoptMCParticleContainer(AliMCParticleContainer *cont)
Int_t fSelectPtHardBin
select one pt hard bin for analysis
AliMCParticleContainer * AddMCParticleContainer(const char *n)
static Double_t GetParallelFraction(AliVParticle *part1, AliVParticle *part2)
virtual Bool_t RetrieveEventObjects()
Bool_t fRejectPileup
Reject pilup using function AliAnalysisUtils::IsPileUpEvent()
void RemoveParticleContainer(Int_t i=0)
TProfile * fHistXsection
!x section from pyxsec.root
Bool_t PythiaInfoFromFile(const char *currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard)
void UserExec(Option_t *option)
Int_t fMinMCLabel
minimum MC label value for the tracks/clusters being considered MC particles
virtual void SetNCentBins(Int_t n)
AliVCaloCells * fCaloCells
!cells
void SetTrigClass(const char *n)
TClonesArray * GetArrayFromEvent(const char *name, const char *clname=0)
Double_t fEventPlaneVsEmcal
select events which have a certain event plane wrt the emcal
virtual Bool_t IsEventSelected()
TH1 * fHistPtHard
!pt hard distribution
static Double_t * GenerateFixedBinArray(Int_t n, Double_t min, Double_t max)
void SetZvertexDiffValue(Double_t cut)
Double_t fMaxBinPt
max pt in histograms
Int_t fPtHardBin
!event pt hard bin
void SetHistoBins(Int_t nbins, Double_t min, Double_t max)
AliMCParticleContainer * GetMCParticleContainer(Int_t i=0) const
TClonesArray * fTracks
!tracks
TH1 * fHistTrialsAfterSel
!total number of trials per pt hard bin after selection
void LoadPythiaInfo(AliVEvent *event)
Bool_t fIsEsd
!whether it's an ESD analysis
Double_t fVertex[3]
!event vertex
AliTrackContainer * AddTrackContainer(const char *n)
AliTrackContainer * GetTrackContainer(Int_t i=0) const
EMCalTriggerMode_t
Handling of the EMCAL trigger thresholds.
Bool_t fCreateHisto
whether or not create histograms
Store some informaion about a Pythia eventThis class is used to store some information about a Pythia...
TH1 * fHistEventRejection
!book keep reasons for rejecting event
void SetMakeGeneralHistograms(Bool_t g)
TClonesArray * fTriggerPatchInfo
!trigger patch info array
TClonesArray * GetClusterArray(Int_t i=0) const
void SetNeedEmcalGeom(Bool_t n)
Double_t fEPV0A
!event plane V0A
void SetEventPlaneVsEmcal(Double_t ep)
void SetCaloTriggersName(const char *n)
const AliEmcalPythiaInfo * GetPythiaInfo() const
TString fCentEst
name of V0 centrality estimator
TString fPythiaInfoName
name of pythia info object
EMCAL Level1 gamma trigger, high threshold.
void AddObjectToEvent(TObject *obj, Bool_t attempt=kFALSE)
AliVCaloTrigger * fCaloTriggers
!calo triggers
void SetRejectionReasonLabels(TAxis *axis)
TH1 * fHistZVertex
!z vertex position
Int_t fMinNTrack
minimum nr of tracks in event with pT>fTrackPtCut
static Byte_t GetTrackType(const AliVTrack *t)
void SetGeneratePythiaInfoObject(Bool_t b)
const Int_t nbins
Bool_t fUseAliAnaUtils
used for LHC13* data: z-vtx, Ncontributors, z-vtx resolution cuts
void SetClusTimeCut(Double_t min, Double_t max, Int_t c=0)
ULong_t fTriggers
list of fired triggers
Double_t fMaxEventPlane
maximum event plane value
Float_t fXsection
!x-section from pythia header
void SetCaloCellsName(const char *n)
Bool_t fInitialized
whether or not the task has been already initialized
void SetCentralityEstimator(const char *c)
TH1 * fHistEventCount
!incoming and selected events
Double_t fMaxCent
max centrality for event selection
void SetMinPtTrackInEmcal(Double_t min)
TriggerCategory
Online trigger categories.
AliVParticle * GetAcceptParticleFromArray(Int_t p, Int_t c=0) const
Container structure for EMCAL clusters.
Container for MC-true particles within the EMCAL framework.
Bool_t fNeedEmcalGeom
whether or not the task needs the emcal geometry
AliVCluster * GetAcceptClusterFromArray(Int_t cl, Int_t c=0) const
const TString & GetPythiaInfoName() const
void AdoptTrackContainer(AliTrackContainer *cont)
Int_t fNbins
no. of pt bins
Bool_t fTklVsClusSPDCut
Apply tracklet-vs-cluster SPD cut to reject background events in pp.
AliEMCALTriggerPatchInfo * GetMainTriggerPatch(TriggerCategory triggersel=kTriggerLevel1Jet, Bool_t doOfflinSimple=kFALSE)
static Double_t fgkEMCalDCalPhiDivide
phi value used to distinguish between DCal and EMCal