updated dndeta analysis
[u/mrichter/AliRoot.git] / PWG0 / dNdEta / AlidNdEtaCorrectionTask.h
CommitLineData
0f67a57c 1/* $Id$ */
2
3#ifndef AlidNdEtaCorrectionTask_H
4#define AlidNdEtaCorrectionTask_H
5
6#include "AliAnalysisTask.h"
7#include <TString.h>
770a1f1d 8#include "AliPWG0Helper.h"
0f67a57c 9
10class AliESDtrackCuts;
11class dNdEtaAnalysis;
12class AlidNdEtaCorrection;
1c15d51a 13class TH1;
0f67a57c 14class TH1F;
15class AliESDEvent;
16class TParticlePDG;
54b096ef 17class TH2F;
a7f69e56 18class TH3F;
54b096ef 19class TProfile;
0f67a57c 20
21class AlidNdEtaCorrectionTask : public AliAnalysisTask {
22 public:
1c15d51a 23 AlidNdEtaCorrectionTask();
24 AlidNdEtaCorrectionTask(const char* opt);
0f67a57c 25 virtual ~AlidNdEtaCorrectionTask();
26
27 virtual void ConnectInputData(Option_t *);
28 virtual void CreateOutputObjects();
29 virtual void Exec(Option_t*);
30 virtual void Terminate(Option_t *);
31
32 void SetTrackCuts(AliESDtrackCuts* cuts) { fEsdTrackCuts = cuts; }
770a1f1d 33 void SetAnalysisMode(AliPWG0Helper::AnalysisMode mode) { fAnalysisMode = mode; }
3d7758c1 34 void SetOnlyPrimaries(Bool_t flag = kTRUE) { fOnlyPrimaries = flag; }
70fdd197 35 void SetTrigger(AliTriggerAnalysis::Trigger trigger) { fTrigger = trigger; }
69b09e3b 36 void SetFillPhi(Bool_t flag = kTRUE) { fFillPhi = flag; }
37 void SetDeltaPhiCut(Float_t cut) { fDeltaPhiCut = cut; }
1d107532 38 void SetSymmetrize(Bool_t flag = kTRUE) { fSymmetrize = flag; }
81be4ee8 39 void SetMultAxisEta1(Bool_t flag = kTRUE) { fMultAxisEta1 = flag; }
40 void SetDiffTreatment(AliPWG0Helper::DiffTreatment diffTreatment) { fDiffTreatment = diffTreatment; }
41 void SetSkipParticles(Bool_t flag = kTRUE) { fSystSkipParticles = flag; }
0fc41645 42
43 void SetOption(const char* opt) { fOption = opt; }
0f67a57c 44
45 protected:
46 Bool_t SignOK(TParticlePDG* particle);
47
48 AliESDEvent *fESD; //! ESD object
49 TList* fOutput; //! list send on output slot 0
50
51 TString fOption; // option string
770a1f1d 52 AliPWG0Helper::AnalysisMode fAnalysisMode; // detector that is used for analysis
70fdd197 53 AliTriggerAnalysis::Trigger fTrigger; // trigger used in the analysis
69b09e3b 54 Bool_t fFillPhi; // if true phi is filled as 3rd coordinate in all maps
55 Float_t fDeltaPhiCut; // cut in delta phi (only SPD)
1d107532 56 Bool_t fSymmetrize; // move all negative to positive eta
81be4ee8 57 Bool_t fMultAxisEta1; // restrict multiplicity count to |eta| < 1
58 AliPWG0Helper::DiffTreatment fDiffTreatment; // how to identify SD events (see AliPWG0Helper::GetEventProcessType)
69b09e3b 59
0f67a57c 60 Int_t fSignMode; // if 0 process all particles, if +-1 process only particles with that sign
3d7758c1 61 Bool_t fOnlyPrimaries; // only process primaries (syst. studies)
69b09e3b 62 Int_t fStatError; // statistical error evaluation: if set to 1 we only count unique primaries (binomial errors are valid), for 2 all the rest
81be4ee8 63 Bool_t fSystSkipParticles; // if true skips particles (systematic study)
0f67a57c 64
65 AliESDtrackCuts* fEsdTrackCuts; // Object containing the parameters of the esd track cuts
66
67 AlidNdEtaCorrection* fdNdEtaCorrection; //! contains the intermediate histograms (on each slave)
68
69 dNdEtaAnalysis* fdNdEtaAnalysisMC; //! analysis from MC (only triggered, vertex events)
70 dNdEtaAnalysis* fdNdEtaAnalysisESD; //! analysis from ESD (not yet corrected!)
71
72 // control histograms
73 TH1F* fPIDParticles; //! pid of primary particles
74 TH1F* fPIDTracks; //! pid of reconstructed tracks
1c15d51a 75
54b096ef 76 TH2F* fVertexCorrelation; //! ESD z-vtx vs MC z-vtx
a7f69e56 77 TH3F* fVertexCorrelationShift; //! (MC z-vtx - ESD z-vtx) vs MC z-vtx vs n# rec tracks
54b096ef 78 TProfile* fVertexProfile; //! Profile of MC z-vtx - ESD z-vtx vs. MC z-vtx
1c15d51a 79 TH1F* fVertexShift; //! (MC z-vtx - ESD z-vtx) in +- 10 cm
a7f69e56 80 TH2F* fVertexShiftNorm; //! (MC z-vtx - ESD z-vtx) / (sigma_ESD-z-vtx) vs. no. rec tracks
0f67a57c 81
3d7758c1 82 TH2F* fEtaCorrelation; //! ESD eta vs MC eta
51f6de65 83 TH2F* fEtaCorrelationShift; //! (MC eta - ESD eta) vs MC eta
3d7758c1 84 TProfile* fEtaProfile; //! Profile of MC eta - ESD eta vs. MC eta
1c15d51a 85 TH1F* fEtaResolution; //! MC eta - ESD eta in |eta| < 1
69b09e3b 86 TH2F* fDeltaPhiCorrelation; //! delta phi ESD vs. MC
1c15d51a 87
69b09e3b 88 TH2F* fpTResolution; //! (MC pT - ESD pT) / MC pT vs. MC pT in |eta| < 0.9
1c15d51a 89
90 AliESDtrackCuts* fEsdTrackCutsPrim; //! control histograms for primaries
91 AliESDtrackCuts* fEsdTrackCutsSec; //! control histograms for secondaries
3d7758c1 92
96fcc8a7 93 // histograms for systematic studies (must be enabled with option)
94
1c15d51a 95 TH1* fTemp1; //! temp histogram for quick study of variables
96 TH1* fTemp2; //! temp histogram for quick study of variables
96fcc8a7 97
4ef701a0 98 TH1F* fMultAll; //! primary particles in |eta| < 1 and pT > 0.2 in all events
99 TH1F* fMultTr; //! primary particles in |eta| < 1 and pT > 0.2 in triggered events
100 TH1F* fMultVtx; //! primary particles in |eta| < 1 and pT > 0.2 in triggered events with vertex
101
69b09e3b 102 TH2* fDeltaPhi[8]; //! delta phi of primaries, secondaries, other (= unclear cases)
4ef701a0 103
69b09e3b 104 TH2F* fEventStats; //! some stats on number of events, see CreateOutputObjects for a detailed definition
50ec344d 105
69b09e3b 106 AlidNdEtaCorrection* fdNdEtaCorrectionSpecial[4]; //! correction maps used for systematic studies, may contain:
107 // for specific process type (ND, SD, DD), enable with option: process-types
108 // for particle species (pi, K, p, rest), enable with: particle-species
96fcc8a7 109
0f67a57c 110 private:
111 AlidNdEtaCorrectionTask(const AlidNdEtaCorrectionTask&);
112 AlidNdEtaCorrectionTask& operator=(const AlidNdEtaCorrectionTask&);
113
114 ClassDef(AlidNdEtaCorrectionTask, 1);
115};
116
117#endif