PWG0/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     ClassDef(AlidNdPtHelper, 1);
 114
 115   private:
 116     AlidNdPtHelper(const AlidNdPtHelper&);
 117     AlidNdPtHelper& operator=(const AlidNdPtHelper&);
 118 };
 119
 120 #endif
 121