1 #ifndef ALIANALYSISMULTPBTRACKHISTOMANAGER_H
2 #define ALIANALYSISMULTPBTRACKHISTOMANAGER_H
4 #include "AliHistoListWrapper.h"
13 //#define WEIGHTED_DCA
14 #define TRANSVERSE_DCA
17 //-------------------------------------------------------------------------
18 // AliAnalysisMultPbTrackHistoManager
23 // Author: Michele Floris, CERN
24 //-------------------------------------------------------------------------
27 class AliAnalysisMultPbTrackHistoManager : public AliHistoListWrapper {
31 typedef enum {kHistoGen, kHistoRec, kHistoRecPrim, kHistoRecSecWeak, kHistoRecSecMat, kHistoRecFake, kHistoRecHighestMeanPt, kHistoRecLowestMeanPt, kNHistos} Histo_t;
32 typedef enum {kStatAll, kStatCentr, kStatPhysSel, kStatVtx, kStatZDCCut, kStatVtxRangeCut, kNStatBins} Stat_t;
33 typedef enum {kPartPiPlus, kPartKPlus, kPartP, kPartLPlus, kPartPiMinus, kPartKMinus, kPartPBar, kPartLMinus, kPartOther, kNPart} Part_t;
35 // these bits define the behaviour at merging
36 // if kKeepMax(Mib)Mean is set, the histogram with the highest (lowest) mean is kept
37 enum {kKeepMaxMean = BIT(22), kKeepMinMean = BIT(23)};
41 AliAnalysisMultPbTrackHistoManager();
42 AliAnalysisMultPbTrackHistoManager(const char * name,const char * title);
43 AliAnalysisMultPbTrackHistoManager(const AliAnalysisMultPbTrackHistoManager& obj) ;
44 ~AliAnalysisMultPbTrackHistoManager();
47 void SetSuffix(const char * suffix) { fHNameSuffix = suffix;}
50 TH3D * GetHistoPtEtaVz(Histo_t id, Int_t particle = -1);
51 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);
52 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);
53 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);
54 TH2D * GetHistoPtVz (Histo_t id, Float_t minEta = -22222, Float_t maxEta = -22222, Bool_t scaleWidth = kFALSE);
56 TH1I * GetHistoStats();
57 TH2D * GetHistoDCA(Histo_t id);
58 TH1D * GetHistoMult(Histo_t id);
59 TH2D * GetHistoV0vsNtracks(Histo_t id);
62 TH1D * GetHistoSpecies(Histo_t id);
63 TH1D * GetHistoProcess(Histo_t id);
64 TH1D * GetHistoVzEvent(Histo_t id);
65 TH1D * GetHistoPtEvent(Histo_t id);
66 TH1D * GetHistoMeanPt (Histo_t id);
68 TH2D * GetHistoElectronCutQA();
70 TMap * GetParticleSpecies(){return fParticleSpecies;}
73 void ScaleHistos (Double_t nev, Option_t * option="");
74 Int_t GetLocalParticleID(AliMCParticle * part);
75 void FillParticleID(Histo_t id, AliMCParticle * part) { GetHistoSpecies(id)->Fill(GetLocalParticleID(part));}
76 void FillSpeciesMap(Int_t pdgCode);
80 TH3D * BookHistoPtEtaVz(const char * name, const char * title);
81 TH2D * BookHistoDCA(const char * name, const char * title);
82 TH1I * BookHistoStats();
83 TH1D * BookHistoMult(const char * name, const char * title);
86 TH1 * GetHisto(const char * name);
88 virtual void Print(Option_t* option = "") const {fList->Print(option);}
91 Long64_t Merge(TCollection* list); // need to override this because of the mean pts
95 static const char * kStatStepNames[]; // names of the step hist
96 static const char * kHistoPtEtaVzNames[]; // names of the 3D histograms pt/eta/vz
97 static const char * kHistoDCANames[]; // names of the DCA histograms
98 static const char * kHistoPrefix[]; // prefix for histo names // FIXME: remove the others and keep only this
99 static const char * kSpeciesName[]; // Particle species
100 TString fHNameSuffix; // Suffix added to all histo names. Useful if you have in the same session e.g. MC and data.
102 TMap * fParticleSpecies;// Map assiciating PDG code to number of particles in the physical primary sample.
104 AliAnalysisMultPbTrackHistoManager& operator=(const AliAnalysisMultPbTrackHistoManager& task);
106 ClassDef(AliAnalysisMultPbTrackHistoManager, 4)