Fix histo title axis name; fix checking of NLM in Pi0EbE
[u/mrichter/AliRoot.git] / TRD / AliTRDCalibTask.h
CommitLineData
6d3c5222 1#ifndef ALITRDCALIBTASK_H
2#define ALITRDCALIBTASK_H
3
4///////////////////////////////////////////////////////////////////////////////
5// //
6// TRD calibration task for offline calibration //
7// //
8///////////////////////////////////////////////////////////////////////////////
9
10
11class TList;
12class TObject;
13class TObjArray;
14class TH2F;
15class TH1F;
16class TH1I;
54f2ff1c 17class TH2S;
6d3c5222 18class TProfile2D;
19class TH2I;
20class TTree;
21class AliESDEvent;
22class AliESDfriend;
23class AliESDtrack;
24class AliESDfriendTrack;
25class AliTRDtrackV1;
54f2ff1c 26class AliTRDseedV1;
6d3c5222 27class AliTRDCalibraFillHisto;
28class AliTRDcluster;
29class AliESDtrackCuts;
30class AliTRDCalDet;
b2277aa2 31class AliTRDCalibChamberStatus;
6d3c5222 32
33#include "TObjString.h"
01239968 34#include "AliAnalysisTaskSE.h"
98937691 35#include "TMath.h"
6d3c5222 36
01239968 37class AliTRDCalibTask : public AliAnalysisTaskSE {
6d3c5222 38 public:
39 AliTRDCalibTask(const char *name = "AliTRDCalibTask");
40 virtual ~AliTRDCalibTask();
41
01239968 42 // virtual void ConnectInputData(Option_t *);
43 virtual void UserCreateOutputObjects();
44 virtual void UserExec(Option_t *);
6d3c5222 45 virtual void Terminate(Option_t *);
46 virtual Bool_t Load(const Char_t *filename);
57185ead 47 virtual Bool_t Load(TList *lister);
57185ead 48 virtual Long64_t Merge(TCollection *li);
49 void AddTask(const AliTRDCalibTask * calibTask);
54f2ff1c 50 Bool_t FindP1TrackPHtrackletV1Test(const AliTRDseedV1 *tracklet, Int_t nbclusters);
57185ead 51 TList *GetList() const {return fListHist;};
6d3c5222 52
54f2ff1c 53 void SetOnInstance(Bool_t onInstance) {fOnInstance=onInstance;};
6d3c5222 54 void SetHisto2d(Bool_t histo2d) {fHisto2d=histo2d;};
55 void SetVector2d(Bool_t vector2d) {fVector2d=vector2d;};
56 void SetVdriftLinear(Bool_t vdriftLinear) {fVdriftLinear = vdriftLinear;};
a0bb5615 57 void SetExbAlt(Bool_t exbalt) {fExbAlt = exbalt;};
6d3c5222 58 void SetNbTimeBins(Int_t nbTimeBins) {fNbTimeBins=nbTimeBins;};
5f689709 59 void SetNumberBinCharge(Short_t nbBinCharge) {fNumberBinCharge=nbBinCharge;};
60 void SetDebugLevelTRDCalibraFillHisto(Short_t debugLevelTRDCalibraFillHisto) {fDebugLevelTRDCalibraFillHisto = debugLevelTRDCalibraFillHisto;};
61
6d3c5222 62
63 void SetNz(Short_t nz, Int_t i) {fNz[i]=nz;};
64 void SetNrphi(Short_t nrphi, Int_t i) {fNrphi[i]=nrphi;};
65
5f689709 66 void SetSelectTrigger(Bool_t selectTrigger) {fSelectTrigger = selectTrigger;};
6d3c5222 67 void AddSelectedTriggerClass(const char*name) {fSelectedTrigger->Add(new TObjString(name));};
68 void SetReject(Bool_t rejected) {fRejected = rejected;};
69
98937691 70 void SetESDtrackCuts(AliESDtrackCuts * const esdtrackCuts) {fEsdTrackCuts = esdtrackCuts;};
6d3c5222 71 void SetRequirePrimaryVertex(Bool_t requirePrimaryVertex) {fRequirePrimaryVertex = requirePrimaryVertex;};
72 void SetUseTPCVertex() {fVtxTPC=kTRUE ; fVtxSPD=kFALSE;}
73 void SetUseSPDVertex() {fVtxTPC=kFALSE; fVtxSPD=kTRUE ;}
74 void SetMinNbOfContributors(Int_t minNbOfContributors) {fMinNbContributors = minNbOfContributors;};
c1105918 75 void SetRangePrimaryVertexZ(Double_t rangePrimaryVertexZ) {fRangePrimaryVertexZ = TMath::Abs(rangePrimaryVertexZ);};
76 void SetMinNbTracks(Int_t minNbTracks) {fMinNbTracks = minNbTracks;};
77 void SetMaxNbTracks(Int_t maxNbTracks) {fMaxNbTracks = maxNbTracks;};
5f689709 78 void SetCutWithVdriftCalib(Bool_t cutWithVdriftCalib) {fCutWithVdriftCalib = cutWithVdriftCalib;};
79 void SetMinNbTRDtracklets(Int_t minNbTRDtracklets) {fMinNbTRDtracklets = minNbTRDtracklets;};
80 void SetMinTRDMometum(Double_t minTRDMomentum) {fMinTRDMomentum = minTRDMomentum;};
532be2d4 81 void SetScaleGainWithTPCSignal(Bool_t scaleGainWithTPCSignal) {fScaleGainWithTPCSignal = scaleGainWithTPCSignal;};
82
4c865c34 83 void SetVersionGainUsed(Int_t versionGainUsed) { fVersionGainUsed = versionGainUsed; }
84 void SetSubVersionGainUsed(Int_t subVersionGainUsed) { fSubVersionGainUsed = subVersionGainUsed; }
a2a4ec8e 85 void SetVersionGainLocalUsed(Int_t versionGainLocalUsed) { fVersionGainLocalUsed = versionGainLocalUsed; }
86 void SetSubVersionGainLocalUsed(Int_t subVersionGainLocalUsed) { fSubVersionGainLocalUsed = subVersionGainLocalUsed; }
4c865c34 87 void SetVersionVdriftUsed(Int_t versionVdriftUsed) { fVersionVdriftUsed = versionVdriftUsed; }
88 void SetSubVersionVdriftUsed(Int_t subVersionVdriftUsed) { fSubVersionVdriftUsed = subVersionVdriftUsed; }
6d3c5222 89
90 void SetLow(Int_t low) {fLow=low;};
91 void SetHigh(Int_t high) {fHigh=high;};
92 void SetFillZero(Bool_t fillZero) {fFillZero = fillZero;};
93 void SetNormalizeNbOfCluster(Bool_t normalizeNbOfCluster = kTRUE) {fNormalizeNbOfCluster = normalizeNbOfCluster;};
94 void SetMaxCluster(Float_t maxCluster) {fMaxCluster = maxCluster; };
95 void SetNbMaxCluster(Short_t nbMaxCluster) {fNbMaxCluster = nbMaxCluster; };
96 void SetOfflineTracks() {fOfflineTracks=kTRUE; fStandaloneTracks=kFALSE; };
97 void SetStandaloneTracks() {fStandaloneTracks=kTRUE; fOfflineTracks=kFALSE; };
4c865c34 98
99 void SetCalDetGain(AliTRDCalDet * const calDetGain) {fCalDetGain = calDetGain;};
6d3c5222 100
101 void SetMaxEvent(Int_t nbevents) { fMaxEvent = nbevents; };
102 void SetDebug(Int_t debug) { fDebug = debug; };
103
9a8b0e85 104 Bool_t IsPHQon() const {return fPHQon;}
105 void SetPHQon(const Bool_t kphq){ fPHQon = kphq; }
106
6d3c5222 107 private:
a2a4ec8e 108 Bool_t SetVersionSubversion();
c1105918 109 Bool_t ParticleGood(int i) const;
a2a4ec8e 110
6d3c5222 111 AliESDEvent *fESD; //! ESD object
6d3c5222 112 const AliESDtrack *fkEsdTrack; //! ESD track
113 AliESDfriendTrack *fFriendTrack; //! ESD friend track
114 TObject *fCalibObject; //! calibration objects attached to the ESD friend
115 AliTRDtrackV1 *fTrdTrack; //! trdtrack
116 AliTRDcluster *fCl; //! cluster
117
118 TList *fListHist; //! list of histograms
119
120 AliTRDCalibraFillHisto *fTRDCalibraFillHisto; //! calibration analyse object
b2277aa2 121 AliTRDCalibChamberStatus *fTRDChamberStatus; //! calibration chamber status
6d3c5222 122
123 TH1I *fNEvents; //! counter
a5dcf618 124 TH1I *fNEventsInput; //! counter
6d3c5222 125
126 TH1F *fNbTRDTrack; //! nb ESD tracks with TRD clusters
127 TH1F *fNbTRDTrackOffline; //! nb ESD tracks with TRD clusters
128 TH1F *fNbTRDTrackStandalone; //! nb ESD tracks with TRD clusters
129 TH2F *fNbTPCTRDtrack; //! nb TPC and TRD tracks when problems
c1105918 130 TH2F *fNbGoodTracks; //! nb of good tracks
6d3c5222 131
132 TH1F *fNbTimeBin; //! nb Time Bin
133 TH1F *fNbTimeBinOffline; //! nb Time Bin offline
134 TH1F *fNbTimeBinStandalone; //! nb Time Bin standalone
135 TH1F *fNbClusters; //! nb Clusters
136 TH1F *fNbClustersOffline; //! nb Clusters offline
137 TH1F *fNbClustersStandalone; //! nb Clusters standalone
138 TH1F *fNbTracklets; //! nb Tracklets
139 TH1F *fNbTrackletsOffline; //! nb Tracklets offline
140 TH1F *fNbTrackletsStandalone; //! nb Tracklets standalone
141
98937691 142 TH2F *fAbsoluteGain; //! Absolute Gain without AliESDfriend
6d3c5222 143 TH2I *fCH2dSum; //! CH2d charge all
144 TProfile2D *fPH2dSum; //! PH2d PH all
145 TH2I *fCH2dSM; //! CH2d per SM
146 TProfile2D *fPH2dSM; //! PH2d per SM
54f2ff1c 147 TH2I *fCH2dTest; //! CH2d for test
148 TProfile2D *fPH2dTest; //! PH2d for test
149 TH2S *fLinearVdriftTest; //! VdriftLinear for test
6d3c5222 150
54f2ff1c 151 Bool_t fOnInstance; // On Instance
98937691 152 Bool_t fHisto2d; // histo
153 Bool_t fVector2d; // vector
154 Bool_t fVdriftLinear; // vdrift Linear
a0bb5615 155 Bool_t fExbAlt; // alternative exb calculation
6d3c5222 156
5f689709 157 Short_t fDebugLevelTRDCalibraFillHisto; // Debug Level Fill Histo
98937691 158 Int_t fNbTimeBins; // number of timebins
5f689709 159 Short_t fNumberBinCharge; // Number of bins for the gain factor
6d3c5222 160
161 Short_t fNz[3]; // Nz mode
162 Short_t fNrphi[3]; // Nrphi mode
163
5f689709 164 Bool_t fSelectTrigger; // Select trigger
6d3c5222 165 TObjArray *fSelectedTrigger; // Trigger class names accepted/rejected
166 Bool_t fRejected; // Reject the selected trigger class
167
168 AliESDtrackCuts *fEsdTrackCuts; // Quality cut on the AliESDtrack
169 Bool_t fRequirePrimaryVertex; // Primary Vertex
170 Bool_t fVtxTPC; // Flag for use of TPC vertex
171 Bool_t fVtxSPD; // Flag for use of SPD vertex
172 Int_t fMinNbContributors; // Min number of contributors
98937691 173 Double_t fRangePrimaryVertexZ; // Were the primary vertex is
c1105918 174 Int_t fMinNbTracks; // Min Nb Tracks
175 Int_t fMaxNbTracks; // Max Nb Tracks
5f689709 176 Bool_t fCutWithVdriftCalib; // CutWithVdriftCalib for the gain and PH
177 Int_t fMinNbTRDtracklets; // Min number of TRD tracklets
178 Float_t fMinTRDMomentum; // Min TRD momentum
532be2d4 179 Bool_t fScaleGainWithTPCSignal; // Scale the TPC gain with the TPC signal
5f689709 180
6d3c5222 181 Int_t fLow; // lower limit of nb of TRD clusters per tracklet
182 Int_t fHigh; // higher limit of nb of TRD clusters per tracklet
183 Bool_t fFillZero; // fill zero
184 Bool_t fNormalizeNbOfCluster; // normalize with number of clusters (per default not)
185 Float_t fRelativeScale; // relative scale for gas gain
186 Float_t fMaxCluster; // Maxcluster
187 Short_t fNbMaxCluster; // Number of tb at the end
188 Bool_t fOfflineTracks; // Only Offline refitted tracks
189 Bool_t fStandaloneTracks; // Take only standalone tracks
190
d085ba91 191 Int_t fFirstRunGain; // FirstRunGainUsed
4c865c34 192 Int_t fVersionGainUsed; // VersionGainUsed
193 Int_t fSubVersionGainUsed; // SubVersionGainUsed
d085ba91 194 Int_t fFirstRunGainLocal; // FirstRunGainLocalUsed
a2a4ec8e 195 Int_t fVersionGainLocalUsed; // VersionGainLocalUsed
196 Int_t fSubVersionGainLocalUsed; // SubVersionGainLocalUsed
d085ba91 197 Int_t fFirstRunVdrift; // FirstRunVdriftUsed
4c865c34 198 Int_t fVersionVdriftUsed; // VersionVdriftUsed
199 Int_t fSubVersionVdriftUsed; // SubVersionVdriftUsed
b2277aa2 200 Int_t fFirstRunExB; // FirstRunExBUsed
201 Int_t fVersionExBUsed; // VersionExBUsed
202 Int_t fSubVersionExBUsed; // SubVersionExBUsed
4c865c34 203
6d3c5222 204 AliTRDCalDet *fCalDetGain; // Calib object gain
205
206 Int_t fMaxEvent; // max events
207 Int_t fCounter; // max events
6d3c5222 208
9a8b0e85 209 Bool_t fPHQon; //switch of phq
210
6d3c5222 211 AliTRDCalibTask(const AliTRDCalibTask&);
212 AliTRDCalibTask& operator=(const AliTRDCalibTask&);
213
532be2d4 214 ClassDef(AliTRDCalibTask, 3);
6d3c5222 215};
216
21185a48 217#endif
01239968 218
c1105918 219