Update for Ds
[u/mrichter/AliRoot.git] / PWG0 / AliPWG0Helper.h
CommitLineData
04a7657f 1/* $Id$ */
2
3#ifndef ALIPWG0HELPER_H
4#define ALIPWG0HELPER_H
5
6#include <TObject.h>
70fdd197 7#include <AliTriggerAnalysis.h>
04a7657f 8
9// static helper functions
10
770a1f1d 11class AliESDEvent;
116083b1 12class AliESDVertex;
04a7657f 13class TParticle;
dd367a14 14class TH1;
15class TH2;
25db2d85 16class TH3;
116083b1 17class AliHeader;
6b62a9c7 18class AliGenEventHeader;
116083b1 19class AliStack;
9cd015f7 20class TTree;
ff8c4f30 21class AliOfflineTrigger;
81be4ee8 22class AliMultiplicity;
04a7657f 23
24class AliPWG0Helper : public TObject
25{
26 public:
2701e5ae 27 // kTPCSPD: tracks + tracklets not using any cluster associated to tracks
28 enum AnalysisMode { kInvalid = -1, kSPD = 0x1, kTPC = 0x2, kTPCITS = 0x4, kFieldOn = 0x8, kSPDOnlyL0 = 0x10, kTPCSPD = 0x20};
6b62a9c7 29 // in case we want to use bitmaps...
81be4ee8 30 enum MCProcessType { kInvalidProcess = -1, kND = 0x1, kDD = 0x2, kSD = 0x4, kOnePart = 0x8 };
31 enum DiffTreatment { kMCFlags = 0, kUA5Cuts = 1, kE710Cuts, kALICEHadronLevel };
e9c3977b 32
81be4ee8 33 static const AliESDVertex* GetVertex(const AliESDEvent* aEsd, AnalysisMode analysisMethod, Bool_t debug = kFALSE);
a6e0ebfe 34 static Bool_t TestVertex(const AliESDVertex* vertex, AnalysisMode analysisMode, Bool_t debug = kFALSE);
ff8c4f30 35
7584d357 36 static Bool_t IsPrimaryCharged(TParticle* aParticle, Int_t aTotalPrimaries, Bool_t adebug = kFALSE);
37
81be4ee8 38 static MCProcessType GetEventProcessType(AliESDEvent* esd, AliHeader* header, AliStack* stack, DiffTreatment diffTreatment);
39 static MCProcessType GetEventProcessType(AliHeader* aHeader, Bool_t adebug = kFALSE);
40 static MCProcessType GetPythiaEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
41 static MCProcessType GetDPMjetEventProcessType(AliGenEventHeader* aHeader, Bool_t adebug = kFALSE);
a67484a6 42 static Int_t GetLastProcessType() { return fgLastProcessType; }
6b62a9c7 43
116083b1 44 static TParticle* FindPrimaryMother(AliStack* stack, Int_t label);
45 static Int_t FindPrimaryMotherLabel(AliStack* stack, Int_t label);
46
29771dc8 47 static void CreateProjections(TH3* hist, Bool_t save = kFALSE);
48 static void CreateDividedProjections(TH3* hist, TH3* hist2, const char* axis = 0, Bool_t putErrors = kFALSE, Bool_t save = kFALSE);
4c351225 49 static const char* GetAxisTitle(TH3* hist, const char axis);
116083b1 50
dd367a14 51 static void NormalizeToBinWidth(TH1* hist);
52 static void NormalizeToBinWidth(TH2* hist);
53
81be4ee8 54 static void PrintConf(AnalysisMode analysisMode, AliTriggerAnalysis::Trigger trigger, DiffTreatment diffTreatment);
55
56 static Double_t Rapidity(Double_t pt, Double_t pz, Double_t m);
57 static Bool_t IsHadronLevelSingleDiffractive(AliStack* stack, Float_t cms, Float_t xiMin, Float_t xiMax);
ff8c4f30 58
04a7657f 59 protected:
d879a92d 60 static Int_t fgLastProcessType; // stores the raw value of the last process type extracted
a67484a6 61
04a7657f 62 ClassDef(AliPWG0Helper, 0)
61385583 63
64 private:
65 AliPWG0Helper(const AliPWG0Helper&);
66 AliPWG0Helper& operator=(const AliPWG0Helper&);
04a7657f 67};
68
69#endif
70