Moving PbPb multiplicity in the new directory structure
[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / ChargedHadrons / multPbPb / AliAnalysisMultPbTrackHistoManager.h
CommitLineData
a23f7c97 1#ifndef ALIANALYSISMULTPBTRACKHISTOMANAGER_H
2#define ALIANALYSISMULTPBTRACKHISTOMANAGER_H
3
4#include "AliHistoListWrapper.h"
5#include "TH3D.h"
6
7class TH3D;
8class TH1D;
9class TH1I;
3b8cbf2d 10class AliMCParticle;
a23f7c97 11
bcc49144 12//#define WEIGHTED_DCA
13#define TRANSVERSE_DCA
14
15
a23f7c97 16//-------------------------------------------------------------------------
17// AliAnalysisMultPbTrackHistoManager
18//
19//
20//
21//
22// Author: Michele Floris, CERN
23//-------------------------------------------------------------------------
24
25
a23f7c97 26class AliAnalysisMultPbTrackHistoManager : public AliHistoListWrapper {
27
28public:
29
abd808b9 30 typedef enum {kHistoGen, kHistoRec, kHistoRecPrim, kHistoRecSecWeak, kHistoRecSecMat, kHistoRecFake, kHistoRecHighestMeanPt, kHistoRecLowestMeanPt, kNHistos} Histo_t;
31 typedef enum {kStatAll, kStatCentr, kStatPhysSel, kStatVtx, kStatZDCCut, kStatVtxRangeCut, kNStatBins} Stat_t;
3b8cbf2d 32 typedef enum {kPartPiPlus, kPartKPlus, kPartP, kPartLPlus, kPartPiMinus, kPartKMinus, kPartPBar, kPartLMinus, kPartOther, kNPart} Part_t;
33
abd808b9 34 // these bits define the behaviour at merging
35 // if kKeepMax(Mib)Mean is set, the histogram with the highest (lowest) mean is kept
36 enum {kKeepMaxMean = BIT(22), kKeepMinMean = BIT(23)};
37
38
a23f7c97 39
40 AliAnalysisMultPbTrackHistoManager();
41 AliAnalysisMultPbTrackHistoManager(const char * name,const char * title);
42 AliAnalysisMultPbTrackHistoManager(const AliAnalysisMultPbTrackHistoManager& obj) ;
43 ~AliAnalysisMultPbTrackHistoManager();
44
45 // Setters
46 void SetSuffix(const char * suffix) { fHNameSuffix = suffix;}
47
48 // Histo getters
4d0aa70f 49 TH3D * GetHistoPtEtaVz(Histo_t id, Int_t particle = -1);
a23f7c97 50 TH1D * GetHistoPt (Histo_t id, Float_t minEta = -22222, Float_t maxEta = -22222, Float_t minVz = -22222, Float_t maxVz = -22222, Bool_t scaleWidth = kTRUE);
51 TH1D * GetHistoEta(Histo_t id, Float_t minPt = -22222, Float_t maxPt = -22222, Float_t minVz = -22222, Float_t maxVz = -22222, Bool_t scaleWidth = kTRUE);
52 TH1D * GetHistoVz (Histo_t id, Float_t minPt = -22222, Float_t maxPt = -22222, Float_t minEta = -22222, Float_t maxEta = -22222, Bool_t scaleWidth = kTRUE);
aa6151eb 53 TH2D * GetHistoPtVz (Histo_t id, Float_t minEta = -22222, Float_t maxEta = -22222, Bool_t scaleWidth = kFALSE);
a23f7c97 54
55 TH1I * GetHistoStats();
bcc49144 56 TH2D * GetHistoDCA(Histo_t id);
e0376287 57 TH1D * GetHistoMult(Histo_t id);
bcf2601a 58 TH2D * GetHistoV0vsNtracks(Histo_t id);
59
e0376287 60
61 TH1D * GetHistoSpecies(Histo_t id);
3b8cbf2d 62 TH1D * GetHistoProcess(Histo_t id);
4d0aa70f 63 TH1D * GetHistoVzEvent(Histo_t id);
abd808b9 64 TH1D * GetHistoPtEvent(Histo_t id);
65 TH1D * GetHistoMeanPt (Histo_t id);
4d0aa70f 66
a23f7c97 67
68 // Misch utils
69 void ScaleHistos (Double_t nev, Option_t * option="");
3b8cbf2d 70 Int_t GetLocalParticleID(AliMCParticle * part);
71 void FillParticleID(Histo_t id, AliMCParticle * part) { GetHistoSpecies(id)->Fill(GetLocalParticleID(part));}
a23f7c97 72
73 // Histo bookers
74 TH3D * BookHistoPtEtaVz(const char * name, const char * title);
bcc49144 75 TH2D * BookHistoDCA(const char * name, const char * title);
a23f7c97 76 TH1I * BookHistoStats();
e0376287 77 TH1D * BookHistoMult(const char * name, const char * title);
a23f7c97 78
79 //
80 TH1 * GetHisto(const char * name);
81
abd808b9 82 virtual void Print(Option_t* option = "") const {fList->Print();}
83
84
85 Long64_t Merge(TCollection* list); // need to override this because of the mean pts
86
a23f7c97 87private:
88
89 static const char * kStatStepNames[]; // names of the step hist
90 static const char * kHistoPtEtaVzNames[]; // names of the 3D histograms pt/eta/vz
91 static const char * kHistoDCANames[]; // names of the DCA histograms
e0376287 92 static const char * kHistoPrefix[]; // prefix for histo names // FIXME: remove the others and keep only this
3b8cbf2d 93 static const char * kSpeciesName[]; // Particle species
a23f7c97 94 TString fHNameSuffix; // Suffix added to all histo names. Useful if you have in the same session e.g. MC and data.
95
96 AliAnalysisMultPbTrackHistoManager& operator=(const AliAnalysisMultPbTrackHistoManager& task);
97
3b8cbf2d 98 ClassDef(AliAnalysisMultPbTrackHistoManager, 3)
a23f7c97 99
100
101};
102
103#endif