New methods for tagging decay channels
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliVertexingHFUtils.h
1 #ifndef ALIVERTEXINGHFUTILS_H
2 #define ALIVERTEXINGHFUTILS_H
3
4
5 /* $Id$ */
6
7 ///////////////////////////////////////////////////////////////////
8 //                                                               //
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                     //
14 //                                                               //
15 ///////////////////////////////////////////////////////////////////
16
17 #include "TObject.h"
18 #include "AliAODTrack.h"
19 #include "AliAODRecoDecay.h"
20 #include "AliAODRecoDecayHF.h"
21
22 class AliAODMCParticle;
23 class AliAODMCHeader;
24 class AliGenEventHeader;
25 class AliAODEvent;
26 class TProfile;
27 class TClonesArray;
28 class TH1F;
29 class TH2F;
30 class TF1;
31
32 class AliVertexingHFUtils : public TObject{
33  public:
34   AliVertexingHFUtils();
35   AliVertexingHFUtils(Int_t k);
36   virtual ~AliVertexingHFUtils() {};
37
38   // Significance calculator
39   static void ComputeSignificance(Double_t signal, Double_t  errsignal, Double_t  background, Double_t  errbackground, Double_t &significance,Double_t &errsignificance);
40
41   // Functions for Event plane resolution
42   void SetK(Int_t k){fK=k;}
43   void SetSubEvResol(Double_t res){fSubRes=res;}
44   void SetSubEventHisto(const TH1F* hSub){
45     fSubRes=GetSubEvResol(hSub);
46   }
47   Int_t GetK() const  {return fK;}
48   Double_t GetSubEvResol() const  {return fSubRes;}
49   Double_t Pol(Double_t x) const {return Pol(x,fK);}
50   Double_t FindChi() const {return FindChi(fSubRes,fK);}
51   Double_t GetFullEvResol() const {return GetFullEvResol(fSubRes,fK);}
52   static Double_t FindChi(Double_t res,  Int_t k=1);
53   static Double_t Pol(Double_t x, Int_t k);
54   static Double_t ResolK1(Double_t x);
55   static Double_t GetSubEvResol(const TH1F* hSubEvCorr){
56     if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean());
57     else return 1.;
58   }
59   static Double_t GetSubEvResolLowLim(const TH1F* hSubEvCorr){
60     if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean()-hSubEvCorr->GetMeanError());
61     else return 1.;
62   }
63   static Double_t GetSubEvResolHighLim(const TH1F* hSubEvCorr){
64     if(hSubEvCorr) return TMath::Sqrt(hSubEvCorr->GetMean()+hSubEvCorr->GetMeanError());
65     else return 1.;
66   }
67   static Double_t GetFullEvResol(Double_t resSub, Int_t k=1);
68   static Double_t GetFullEvResol(const TH1F* hSubEvCorr, Int_t k=1);
69   static Double_t GetFullEvResolLowLim(const TH1F* hSubEvCorr, Int_t k=1);
70   static Double_t GetFullEvResolHighLim(const TH1F* hSubEvCorr, Int_t k=1);
71   static TString  GetGenerator(Int_t label, AliAODMCHeader* header); 
72   Bool_t IsTrackInjected(AliAODTrack *track,AliAODMCHeader *header,TClonesArray *arrayMC);
73   void GetTrackPrimaryGenerator(AliAODTrack *track,AliAODMCHeader *header,TClonesArray *arrayMC,TString &nameGen);
74   Bool_t IsCandidateInjected(AliAODRecoDecayHF *cand, AliAODMCHeader *header,TClonesArray *arrayMC);
75   // Functions for tracklet multiplcity calculation
76   void SetEtaRangeForTracklets(Double_t mineta, Double_t maxeta){
77     fMinEtaForTracklets=mineta; 
78     fMaxEtaForTracklets=maxeta;
79   }
80   static Int_t GetNumberOfTrackletsInEtaRange(AliAODEvent* ev, Double_t mineta, Double_t maxeta);
81   Int_t GetNumberOfTrackletsInEtaRange(AliAODEvent* ev) const {
82     return GetNumberOfTrackletsInEtaRange(ev,fMinEtaForTracklets,fMaxEtaForTracklets);
83   }
84   static Int_t GetGeneratedMultiplicityInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
85   static Int_t GetGeneratedPrimariesInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
86   static Int_t GetGeneratedPhysicalPrimariesInEtaRange(TClonesArray* arrayMC, Double_t mineta, Double_t maxeta);
87
88   // Functions for computing average pt 
89   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);
90
91   // Functions for computing true impact parameter of D meson
92   static Double_t GetTrueImpactParameterDzero(AliAODMCHeader *mcHeader, TClonesArray* arrayMC, AliAODMCParticle *partDp);
93   static Double_t GetTrueImpactParameterDplus(AliAODMCHeader *mcHeader, TClonesArray* arrayMC, AliAODMCParticle *partDp);
94
95   static Double_t GetCorrectedNtracklets(TProfile* estimatorAvg, Double_t uncorrectedNacc, Double_t vtxZ, Double_t refMult);
96
97   static Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Bool_t searchUpToQuark=kTRUE);
98   static Int_t CheckD0Decay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
99   static Int_t CheckDplusDecay(TClonesArray* arrayMC, AliAODMCParticle *mcPart, Int_t* arrayDauLab);
100
101  private:
102
103   Int_t fK;             // ratio of measured harmonic to event plane harmonic
104   Double_t fSubRes;     // sub-event resolution = sqrt(<cos[n(phiA-phiB)] >)
105   Double_t fMinEtaForTracklets; // min eta for counting tracklets
106   Double_t fMaxEtaForTracklets; // min eta for counting tracklets
107
108   ClassDef(AliVertexingHFUtils,0) 
109 };
110 #endif