]>
Commit | Line | Data |
---|---|---|
a9017e49 | 1 | /* $Id$ */ |
2 | ||
3 | #ifndef AliMultiplicityTask_H | |
4 | #define AliMultiplicityTask_H | |
5 | ||
6 | #include "AliAnalysisTask.h" | |
745d6088 | 7 | |
a9017e49 | 8 | #include <TString.h> |
745d6088 | 9 | #include "AliPWG0Helper.h" |
a9017e49 | 10 | |
11 | class AliESDtrackCuts; | |
12 | class AliMultiplicityCorrection; | |
13 | class TNtuple; | |
14 | class AliCorrection; | |
15 | class TH1; | |
16 | class AliESDEvent; | |
17 | ||
18 | class AliMultiplicityTask : public AliAnalysisTask { | |
19 | public: | |
20 | AliMultiplicityTask(const char* opt = ""); | |
21 | virtual ~AliMultiplicityTask(); | |
22 | ||
23 | virtual void ConnectInputData(Option_t *); | |
24 | virtual void CreateOutputObjects(); | |
25 | virtual void Exec(Option_t*); | |
26 | virtual void Terminate(Option_t *); | |
27 | ||
28 | void SetTrackCuts(AliESDtrackCuts* cuts) { fEsdTrackCuts = cuts; } | |
29 | void SetPtSpectrum(TH1* hist) { fPtSpectrum = hist; } | |
30 | ||
745d6088 | 31 | void SetAnalysisMode(AliPWG0Helper::AnalysisMode mode) { fAnalysisMode = mode; } |
2fa65f52 | 32 | void SetReadMC(Bool_t flag = kTRUE) { fReadMC = flag; } |
f3eb27f6 | 33 | void SetUseMCVertex(Bool_t flag = kTRUE) { fUseMCVertex = flag; } |
2fa65f52 | 34 | |
a9017e49 | 35 | protected: |
36 | AliESDEvent *fESD; //! ESD object | |
37 | ||
38 | TString fOption; // option string | |
745d6088 | 39 | AliPWG0Helper::AnalysisMode fAnalysisMode; // detector that is used for analysis |
2fa65f52 | 40 | Bool_t fReadMC; // if true reads MC data (to build correlation maps) |
f3eb27f6 | 41 | Bool_t fUseMCVertex; // the MC vtx is used instead of the ESD vertex (for syst. check) |
a9017e49 | 42 | |
43 | AliMultiplicityCorrection* fMultiplicity; //! object containing the extracted data | |
44 | AliESDtrackCuts* fEsdTrackCuts; // Object containing the parameters of the esd track cuts | |
45 | ||
2fa65f52 | 46 | Bool_t fSystSkipParticles; //! if true skips particles (systematic study) |
a9017e49 | 47 | AliCorrection* fParticleCorrection[4]; //! correction from measured to generated particles for trigger, vertex sample in |eta| < 2; |
48 | // for each of the species: pi, k, p, other; for systematic study of pt cut off | |
2fa65f52 | 49 | Int_t fSelectProcessType; //! 0 = all (default), 1 = ND, 2 = SD, 3 = DD (for systematic study) |
a9017e49 | 50 | TNtuple *fParticleSpecies; //! per event: vtx_mc, (pi, k, p, rest (in |eta| < 2)) X (true, recon) + (nolabel, |
51 | // doubleTracks, doublePrimaries) [doubleTracks + doublePrimaries are already part of | |
52 | // rec. particles!) | |
53 | ||
54 | TH1* fPtSpectrum; //! function that modifies the pt spectrum (syst. study) | |
55 | ||
56 | TList* fOutput; //! list send on output slot 0 | |
57 | ||
58 | private: | |
59 | AliMultiplicityTask(const AliMultiplicityTask&); | |
60 | AliMultiplicityTask& operator=(const AliMultiplicityTask&); | |
61 | ||
62 | ClassDef(AliMultiplicityTask, 1); | |
63 | }; | |
64 | ||
65 | #endif |