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