1 #ifndef ALIVERTEXINGHFUTILS_H
2 #define ALIVERTEXINGHFUTILS_H
7 ///////////////////////////////////////////////////////////////////
9 // Class with functions useful for different D2H analyses //
10 // - event plane resolution //
11 // - <pt> calculation with side band subtraction //
12 // - tracklet multiplicity calculation //
13 // Origin: F.Prino, Torino, prino@to.infn.it //
15 ///////////////////////////////////////////////////////////////////
18 #include "AliAODTrack.h"
19 #include "AliAODRecoDecay.h"
20 #include "AliAODRecoDecayHF.h"
21 #include "AliAODRecoCascadeHF.h"
24 class AliAODMCParticle;
26 class AliGenEventHeader;
35 class AliVertexingHFUtils : public TObject{
37 AliVertexingHFUtils();
38 AliVertexingHFUtils(Int_t k);
39 virtual ~AliVertexingHFUtils() {};
41 // Significance calculator
42 static void ComputeSignificance(Double_t signal, Double_t errsignal, Double_t background, Double_t errbackground, Double_t &significance,Double_t &errsignificance);
44 // Functions for Event plane resolution
45 void SetK(Int_t k){fK=k;}
46 void SetSubEvResol(Double_t res){fSubRes=res;}
47 void SetSubEventHisto(const TH1F* hSub){
48 fSubRes=GetSubEvResol(hSub);
50 Int_t GetK() const {return fK;}
51 Double_t GetSubEvResol() const {return fSubRes;}
52 Double_t Pol(Double_t x) const {return Pol(x,fK);}
53 Double_t FindChi() const {return FindChi(fSubRes,fK);}
54 Double_t GetFullEvResol() const {return GetFullEvResol(fSubRes,fK);}
55 static Double_t FindChi(Double_t res, Int_t k=1);
56 static Double_t Pol(Double_t x, Int_t k);
57 static Double_t ResolK1(Double_t x);
58 static Double_t GetSubEvResol(const TH1F* hSubEvCorr){
59 if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean());
62 static Double_t GetSubEvResolLowLim(const TH1F* hSubEvCorr){
63 if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean()-hSubEvCorr->GetMeanError());
66 static Double_t GetSubEvResolHighLim(const TH1F* hSubEvCorr){
67 if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean()+hSubEvCorr->GetMeanError());
70 static Double_t GetFullEvResol(Double_t resSub, Int_t k=1);
71 static Double_t GetFullEvResol(const TH1F* hSubEvCorr, Int_t k=1);
72 static Double_t GetFullEvResolLowLim(const TH1F* hSubEvCorr, Int_t k=1);
73 static Double_t GetFullEvResolHighLim(const TH1F* hSubEvCorr, Int_t k=1);
74 static TString GetGenerator(Int_t label, AliAODMCHeader* header);
75 Bool_t IsTrackInjected(AliAODTrack *track,AliAODMCHeader *header,TClonesArray *arrayMC);
76 void GetTrackPrimaryGenerator(AliAODTrack *track,AliAODMCHeader *header,TClonesArray *arrayMC,TString &nameGen);
77 Bool_t IsCandidateInjected(AliAODRecoDecayHF *cand, AliAODMCHeader *header,TClonesArray *arrayMC);
78 Bool_t HasCascadeCandidateAnyDaughInjected(AliAODRecoCascadeHF *cand, AliAODMCHeader *header,TClonesArray *arrayMC);
79 // Functions for tracklet multiplcity calculation
80 void SetEtaRangeForTracklets(Double_t mineta, Double_t maxeta){
81 fMinEtaForTracklets=mineta;
82 fMaxEtaForTracklets=maxeta;
84 static Int_t GetNumberOfTrackletsInEtaRange(AliAODEvent* ev, Double_t mineta, Double_t maxeta);
85 Int_t GetNumberOfTrackletsInEtaRange(AliAODEvent* ev) const {
86 return GetNumberOfTrackletsInEtaRange(ev,fMinEtaForTracklets,fMaxEtaForTracklets);
88 static Int_t GetGeneratedMultiplicityInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
89 static Int_t GetGeneratedPrimariesInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
90 static Int_t GetGeneratedPhysicalPrimariesInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
92 // Functions for computing average pt
93 static void AveragePt(Float_t& averagePt, Float_t& errorPt, Float_t ptmin, Float_t ptmax, TH2F* hMassD, Float_t massFromFit, Float_t sigmaFromFit, TF1* funcB2, Float_t sigmaRangeForSig=2.5, Float_t sigmaRangeForBkg=4.5, Float_t minMass=0., Float_t maxMass=3., Int_t rebin=1);
95 // Functions for processing trigger information
96 static Bool_t CheckT0TriggerFired(AliAODEvent* aodEv);
98 // Functions for computing true impact parameter of D meson
99 static Double_t GetTrueImpactParameterDzero(AliAODMCHeader *mcHeader, TClonesArray* arrayMC, AliAODMCParticle *partDp);
100 static Double_t GetTrueImpactParameterDplus(AliAODMCHeader *mcHeader, TClonesArray* arrayMC, AliAODMCParticle *partDp);
102 static Double_t GetCorrectedNtracklets(TProfile* estimatorAvg, Double_t uncorrectedNacc, Double_t vtxZ, Double_t refMult);
104 static Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Bool_t searchUpToQuark=kTRUE);
105 static Int_t CheckOrigin(AliStack* stack, TParticle *mcPart, Bool_t searchUpToQuark=kTRUE);
106 static Int_t CheckD0Decay(AliStack* stack, Int_t label, Int_t* arrayDauLab);
107 static Int_t CheckD0Decay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
108 static Int_t CheckDplusDecay(AliStack* stack, Int_t label, Int_t* arrayDauLab);
109 static Int_t CheckDplusDecay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
110 static Int_t CheckDsDecay(AliStack* stack, Int_t label, Int_t* arrayDauLab);
111 static Int_t CheckDsDecay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
112 static Int_t CheckDstarDecay(AliStack* stack, Int_t label, Int_t* arrayDauLab);
113 static Int_t CheckDstarDecay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
117 Int_t fK; // ratio of measured harmonic to event plane harmonic
118 Double_t fSubRes; // sub-event resolution = sqrt(<cos[n(phiA-phiB)] >)
119 Double_t fMinEtaForTracklets; // min eta for counting tracklets
120 Double_t fMaxEtaForTracklets; // min eta for counting tracklets
122 ClassDef(AliVertexingHFUtils,0)