Update
[u/mrichter/AliRoot.git] / PWG0 / dNdEta / AlidNdEtaTask.h
CommitLineData
0f67a57c 1/* $Id$ */
2
3#ifndef AlidNdEtaTask_H
4#define AlidNdEtaTask_H
5
76532b17 6#include "AliAnalysisTaskSE.h"
770a1f1d 7#include "AliPWG0Helper.h"
70fdd197 8#include "AliTriggerAnalysis.h"
0f67a57c 9#include <TString.h>
10
11class AliESDtrackCuts;
12class dNdEtaAnalysis;
13class TH1F;
ea441adf 14class TH2F;
0f67a57c 15class TH3F;
16class AliESDEvent;
12bb57f1 17class AliTriggerAnalysis;
68fa248f 18class TH1D;
0f67a57c 19
76532b17 20class AlidNdEtaTask : public AliAnalysisTaskSE {
0f67a57c 21 public:
0f67a57c 22 AlidNdEtaTask(const char* opt = "");
23 virtual ~AlidNdEtaTask();
24
76532b17 25 virtual void ConnectInputData(Option_t *opt);
26 virtual void UserCreateOutputObjects();
27 virtual void UserExec(Option_t*);
0f67a57c 28 virtual void Terminate(Option_t*);
76532b17 29 virtual Bool_t UserNotify();
30
31 Bool_t IsEventInBinZero();
0f67a57c 32
33 void SetTrackCuts(AliESDtrackCuts* cuts) { fEsdTrackCuts = cuts; }
770a1f1d 34 void SetAnalysisMode(AliPWG0Helper::AnalysisMode mode) { fAnalysisMode = mode; }
0f67a57c 35 void SetReadMC(Bool_t flag = kTRUE) { fReadMC = flag; }
54b096ef 36 void SetUseMCVertex(Bool_t flag = kTRUE) { fUseMCVertex = flag; }
c17301f3 37 void SetOnlyPrimaries(Bool_t flag = kTRUE) { fOnlyPrimaries = flag; }
3d7758c1 38 void SetUseMCKine(Bool_t flag = kTRUE) { fUseMCKine = flag; }
70fdd197 39 void SetTrigger(AliTriggerAnalysis::Trigger trigger) { fTrigger = trigger; }
69b09e3b 40 void SetFillPhi(Bool_t flag = kTRUE) { fFillPhi = flag; }
41 void SetDeltaPhiCut(Float_t cut) { fDeltaPhiCut = cut; }
1d107532 42 void SetSymmetrize(Bool_t flag = kTRUE) { fSymmetrize = flag; }
81be4ee8 43 void SetMultAxisEta1(Bool_t flag = kTRUE) { fMultAxisEta1 = flag; }
44 void SetDiffTreatment(AliPWG0Helper::DiffTreatment diffTreatment) { fDiffTreatment = diffTreatment; }
a7f69e56 45
0fc41645 46 void SetOption(const char* opt) { fOption = opt; }
68fa248f 47 void SetPtMin(Float_t ptMin) { fPtMin = ptMin;}
0f67a57c 48
49 protected:
12bb57f1 50 AliESDEvent *fESD; //! ESD object
51 TList* fOutput; //! list send on output slot 0
0f67a57c 52
12bb57f1 53 TString fOption; // option string
770a1f1d 54 AliPWG0Helper::AnalysisMode fAnalysisMode; // detector that is used for analysis
70fdd197 55 AliTriggerAnalysis::Trigger fTrigger; // trigger that is used
69b09e3b 56 Bool_t fFillPhi; // if true phi is filled as 3rd coordinate in all maps
57 Float_t fDeltaPhiCut; // cut in delta phi (only SPD)
0f67a57c 58
12bb57f1 59 Bool_t fReadMC; // if true reads MC data (to build correlation maps)
60 Bool_t fUseMCVertex; // the MC vtx is used instead of the ESD vertex (for syst. check)
61 Bool_t fOnlyPrimaries; // Process only primaries by using the MC information (for syst. check)
62 Bool_t fUseMCKine; // use the MC values for each found track/tracklet (for syst. check)
12bb57f1 63 Bool_t fSymmetrize; // move all negative to positive eta
81be4ee8 64 Bool_t fMultAxisEta1; // restrict multiplicity count to |eta| < 1
65 AliPWG0Helper::DiffTreatment fDiffTreatment; // how to identify SD events (see AliPWG0Helper::GetEventProcessType)
0f67a57c 66
67 AliESDtrackCuts* fEsdTrackCuts; // Object containing the parameters of the esd track cuts
68
69 // Gathered from ESD
70 dNdEtaAnalysis* fdNdEtaAnalysisESD; //! contains the dndeta from the ESD
71 // control hists
81be4ee8 72 TH1F* fMult; //! raw multiplicity histogram
0f67a57c 73 TH1F* fMultVtx; //! raw multiplicity histogram of evts with vtx (control histogram)
74 TH1F* fPartEta[3]; //! counted particles as function of eta (full vertex range, below 0 range, above 0 range)
75 TH1F* fEvents; //! events counted as function of vtx
81be4ee8 76 TH2F* fVertexResolution; //! z resolution of the vertex
0f67a57c 77
78 // Gathered from MC (when fReadMC is set)
79 dNdEtaAnalysis* fdNdEtaAnalysis; //! contains the dndeta from the full sample
69b09e3b 80 dNdEtaAnalysis* fdNdEtaAnalysisND; //! contains the dndeta for the ND sample
81 dNdEtaAnalysis* fdNdEtaAnalysisNSD; //! contains the dndeta for the NSD sample
81be4ee8 82 dNdEtaAnalysis* fdNdEtaAnalysisOnePart; //! contains the dndeta for the one particle sample
0f67a57c 83 dNdEtaAnalysis* fdNdEtaAnalysisTr; //! contains the dndeta from the triggered events
84 dNdEtaAnalysis* fdNdEtaAnalysisTrVtx; //! contains the dndeta from the triggered events with vertex
85 dNdEtaAnalysis* fdNdEtaAnalysisTracks; //! contains the dndeta from the triggered events with vertex counted from the mc particles associated to the tracks (comparing this to the raw values from the esd shows the effect of the detector resolution)
ea441adf 86
87 // control histograms (MC)
0f67a57c 88 TH1F* fPartPt; //! counted particles as function of pt
ea441adf 89
90 // control histograms (ESD)
51f6de65 91 TH3F* fVertex; //! 3d vertex distribution
1d107532 92 TH3F* fVertexVsMult; //! x-vtx vs y-vtx vs multiplicity
ea441adf 93 TH1F* fPhi; //! raw phi distribution
69b09e3b 94 TH1F* fRawPt; //! raw pt distribution
ea441adf 95 TH2F* fEtaPhi; //! raw eta - phi distribution
69b09e3b 96 TH2F* fZPhi[2]; //! raw z - phi distribution from tracklets per layer (only SPD)
12bb57f1 97 TH1F* fModuleMap; //! count clusters as function of module number (only SPD)
54b096ef 98 TH1F* fDeltaPhi; //! histogram of delta_phi values for tracklets (only for SPD analysis)
a7f69e56 99 TH1F* fDeltaTheta; //! histogram of delta_theta values for tracklets (only for SPD analysis)
69b09e3b 100 TH2F* fFiredChips; //! fired chips l1+l2 vs. number of tracklets (only for SPD analysis)
1d107532 101 TH2F* fTrackletsVsClusters; //! number of tracklets vs. clusters in all ITS detectors (only for SPD analysis)
102 TH2F* fTrackletsVsUnassigned; //! number of tracklets vs. number of unassigned clusters in L1 (only for SPD analysis)
1d107532 103 TH1F* fStats; //! further statistics : bin 1 = vertexer 3d, bin 2 = vertexer z, etc (see CreateOutputObjects)
104 TH2F* fStats2; //! V0 vs SPD statistics
68fa248f 105 Float_t fPtMin; // pt min, to be used in kOneTrack case
106 TH1D* fEta; //! eta distribution from ESD
107 TH1D* fEtaMC; //! eta distribution from MC
108 TH1D* fHistEvents; //! histo for n. of selected ESD events
109 TH1D* fHistEventsMC; //! histo for n. of selected MC events
110 TH1D* fTrigEffNum; //!
111 TH1D* fTrigEffDen; //!
112 TH1D* fVtxEffNum; //!
113 TH1D* fVtxEffDen; //!
114 TH1D* fVtxTrigEffNum; //!
115 TH1D* fVtxTrigEffDen; //!
0f67a57c 116 private:
117 AlidNdEtaTask(const AlidNdEtaTask&);
118 AlidNdEtaTask& operator=(const AlidNdEtaTask&);
119
68fa248f 120 ClassDef(AlidNdEtaTask, 2);
0f67a57c 121};
122
123#endif