PWGUD/dNdPt -> PWGLF/SPECTRA/ChargedHadrons/dNdPt
[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / ChargedHadrons / dNdPt / AlidNdPtHelper.h
1 #ifndef ALIDNDPTHELPER_H
2 #define ALIDNDPTHELPER_H
3
4 //
5 // static dNdPt helper functions
6 //
7 // Origin: Jan Fiete Grosse-Oetringhaus
8 // Modified and Extended: Jacek Otwinowski 19/11/2009
9 //
10
11 #include <TObject.h>
12 #include <THnSparse.h>
13 #include "AliTriggerAnalysis.h"
14
15 class TF1;
16 class TH1;
17 class TH2;
18 class THnSparse;
19 class TParticle;
20 class AliHeader;
21 class AliGenEventHeader;
22 class AliStack;
23 class AliESD;
24 class AliESDEvent;
25 class AliESDtrack;
26 class AliMCEvent;
27 class AliESDVertex;
28 class AliESDtrackCuts;
29 class AlidNdPtAcceptanceCuts;
30 class AlidNdPtEventCuts;
31 class AliPWG0Helper;
32
33 class AliGenDPMjetEventHeader;
34 class AliGenCocktailEventHeader;
35 class AliGenPythiaEventHeader;
36 class AliVertexerTracks;
37 class AliLog;
38 class AliHeader;
39
40 class AlidNdPtHelper : public TObject
41 {
42   public:
43     enum AnalysisMode { kInvalid = -1, kSPD = 0, kTPC, kTPCITS, kTPCSPDvtx, kTPCSPDvtxUpdate, kTPCTrackSPDvtx, kTPCTrackSPDvtxUpdate, kTPCITSHybrid, kTPCITSHybridTrackSPDvtx, kTPCITSHybridTrackSPDvtxDCArPt, kITSStandAloneTrackSPDvtx,kITSStandAloneTPCTrackSPDvtx, kMCRec };
44
45     enum ParticleMode { kAllPart = 0, kMCPion, kMCKaon, kMCProton, kPlus, kMinus, kCosmic, kBackgroundTrack, kMCRest, kVZEROCase1, kVZEROCase2};
46
47     enum OutputObject { kInvalidObject = -1, kCutAnalysis = 0, kAnalysis, kAnalysisPbPb, kCorrection, kSystematics, kCutAnalysisPbPb };
48
49     enum TrackObject  { kInvalidTrackObject = -1, kAllTracks = 0, kAccTracks, kRecTracks, kMCTracks };
50     enum EventObject  { kInvalidEventObject = -1, kAllEvents = 0, kTriggeredEvents, kAccEvents, kRecEvents, kMCEvents };
51     enum CutSteps     { kCutSteps = 3 };
52
53     static const AliESDVertex* GetVertex(AliESDEvent* const aEsd, const AlidNdPtEventCuts *const evtCuts,const  AlidNdPtAcceptanceCuts *const accCuts,const  AliESDtrackCuts *const trackCuts,  AnalysisMode analysisMethod, Bool_t debug = kFALSE,Bool_t bRedoTPC = kFALSE, Bool_t bUseMeanVertex = kFALSE);
54
55     static const AliESDVertex* GetTPCVertexZ(const AliESDEvent* const aEsd, const AlidNdPtEventCuts *const evtCuts, const AlidNdPtAcceptanceCuts *const accCuts, const AliESDtrackCuts *const trackCuts, Float_t fraction=0.8, Int_t ntracksMin=2);
56
57     static Bool_t TestRecVertex(const AliESDVertex* vertex, const AliESDVertex* vertexSPD, AnalysisMode analysisMode, Bool_t debug = kFALSE);
58
59     static Bool_t IsPrimaryParticle(AliStack *const stack, Int_t idx, ParticleMode particleMode);
60     static Bool_t IsCosmicTrack(AliESDtrack *const track1, AliESDtrack *const track2);
61     static Bool_t IsGoodImpPar(const AliESDtrack *const track);
62     static Int_t ConvertPdgToPid(const TParticle *const particle);
63
64     static Bool_t SelectEvent(const AliESDEvent* const aEsd, AliESDtrackCuts* const esdTrackCuts);
65     static Bool_t SelectMCEvent(AliMCEvent* const mcEvent);
66
67     static TObjArray *GetAllChargedTracks(AliESDEvent *const esdEvent, AnalysisMode analysisMode);
68
69     static TH1F* MakeResol(TH2F * const his, Int_t integ, Bool_t type, Bool_t drawBins, Int_t minHistEntries);
70     static TH1F* CreateResHisto(TH2F* const hRes2, TH1F **phMean, Int_t integ,  Bool_t drawBinFits, Int_t minHistEntries);
71
72     static Int_t GetTPCMBTrackMult(const AliESDEvent* const esdEvent, const AlidNdPtEventCuts *const evtCuts,const  AlidNdPtAcceptanceCuts *const accCuts, const AliESDtrackCuts *const trackCuts);
73     static Int_t GetTPCMBPrimTrackMult(const AliESDEvent* const esdEvent, AliStack * const stack,const AlidNdPtEventCuts *const evtCuts, const AlidNdPtAcceptanceCuts *const accCuts, const AliESDtrackCuts *const trackCuts);
74
75     static Int_t GetSPDMBTrackMult(const AliESDEvent* const esdEvent, Float_t deltaThetaCut =0.025, Float_t deltaPhiCut = 0.08);
76     static Int_t GetSPDMBPrimTrackMult(const AliESDEvent* const esdEvent, AliStack *const  stack, Float_t deltaThetaCut =0.025, Float_t deltaPhiCut = 0.08);
77     static Int_t GetMCTrueTrackMult(AliMCEvent *const mcEvent, AlidNdPtEventCuts *const evtCuts, AlidNdPtAcceptanceCuts *const accCuts);
78
79     static AliESDtrack* GetTPCOnlyTrackSPDvtx(const AliESDEvent* const esdEvent, Int_t iTrack, Bool_t bUpdate);
80     static AliESDtrack* GetTPCOnlyTrackTrackSPDvtx(const AliESDEvent* const esdEvent, Int_t iTrack, Bool_t bUpdate);
81     static AliESDtrack* GetTrackSPDvtx(const AliESDEvent* const esdEvent, Int_t iTrack, Bool_t bUpdate);
82     static AliESDtrack* GetTrackTrackSPDvtx(const AliESDEvent* const esdEvent, Int_t iTrack, Bool_t bUpdate);
83
84     static void PrintConf(AnalysisMode analysisMode, AliTriggerAnalysis::Trigger trigger);
85     static void PrintMCInfo(AliStack *const pStack, Int_t label);
86
87     static TH1* ScaleByBinWidth(TH1 *const hist=0);
88     static TH1* GetContCorrHisto(TH1 *const hist=0);
89     static TH1* CalcRelativeDifference(const TH1 *const hist1=0, const TH1 *const hist2=0);
90     static TH1* CalcRelativeDifferenceFun(const TH1 *const hist=0, TF1 *const fun=0);
91     static TH1* NormalizeToEvent(const TH2 *const hist1=0, const TH1 *const hist2=0);
92
93     //static THnSparse* GenerateCorrMatrix(THnSparse *const hist1, const THnSparse *const hist2, char *const name);
94     //static TH2* GenerateCorrMatrix(TH2 *const hist1, TH2 *const hist2, char *const name);
95     //static TH1* GenerateCorrMatrix(TH1 *const hist1, TH1 *const hist2, char *const name);
96
97     //static THnSparse* GenerateContCorrMatrix(THnSparse *const hist1, THnSparse *const hist2, char *const name);
98     //static TH2* GenerateContCorrMatrix(TH2 *const hist1, TH2 *const hist2, char *const name);
99     //static TH1* GenerateContCorrMatrix(TH1 *const hist1, TH1 *const hist2, char *const name);
100
101     static THnSparse* GenerateCorrMatrix(THnSparse *const hist1, const THnSparse *const hist2, const char * name);
102     static TH2* GenerateCorrMatrix(TH2 *const hist1, TH2 *const hist2, const char* name);
103     static TH1* GenerateCorrMatrix(TH1 *const hist1, TH1 *const hist2, const char* name);
104
105     static THnSparse* GenerateContCorrMatrix(THnSparse *const hist1, const THnSparse *const hist2, const char* name);
106     static TH2* GenerateContCorrMatrix(TH2 *const hist1, TH2 *const hist2, const char* name);
107     static TH1* GenerateContCorrMatrix(TH1 *const hist1, TH1 *const hist2, const char* name);
108
109     static Double_t GetStrangenessCorrFactor(const Double_t pt);
110     static Double_t GetStrangenessCorrFactorPbPb(const Double_t pt);    
111     static Double_t GetLinearInterpolationValue(const Double_t x1, const Double_t y1, const Double_t x2, const Double_t y2, const Double_t pt);
112
113     // function to rebin THnSparse, the content of hist1 will be rebinned, hist2 serves as a protoype for the binning
114     static THnSparse* RebinTHnSparse(const THnSparse* hist1, THnSparse* hist2, const Char_t* newname = "",  Option_t* option = "");
115
116     ClassDef(AlidNdPtHelper, 1);
117
118   private:
119     AlidNdPtHelper(const AlidNdPtHelper&);
120     AlidNdPtHelper& operator=(const AlidNdPtHelper&);
121 };
122
123 #endif
124