]>
Commit | Line | Data |
---|---|---|
121a60bd | 1 | /************************************************************************** |
91834b8d | 2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
121a60bd | 15 | |
16 | /* $Id$ */ | |
17 | ||
18 | /////////////////////////////////////////////////////////////////////////////// | |
19 | // // | |
4fad09c9 | 20 | // Class for TRD reconstruction // |
954135b1 | 21 | // // |
22 | // For the special options which can be used during reconstruction and their // | |
23 | // default values pls. see function SetOption(). // | |
121a60bd | 24 | // // |
25 | /////////////////////////////////////////////////////////////////////////////// | |
26 | ||
3a039a31 | 27 | #include <TObjString.h> |
28 | #include <TObjArray.h> | |
d39401ad | 29 | #include <TTreeStream.h> |
30 | #include <TDirectory.h> | |
cb8b99ee | 31 | #include <TRef.h> |
121a60bd | 32 | |
3bc9d03e | 33 | #include "AliRawReader.h" |
cb8b99ee | 34 | #include "AliLog.h" |
3bc9d03e | 35 | |
36 | #include "AliTRDReconstructor.h" | |
3fe61b77 | 37 | #include "AliTRDclusterizer.h" |
a8fbb161 | 38 | #include "AliTRDrawData.h" |
8df1f8f5 | 39 | #include "AliTRDrawStream.h" |
a8fbb161 | 40 | #include "AliTRDdigitsManager.h" |
e4f2f73d | 41 | #include "AliTRDtrackerV1.h" |
cb8b99ee | 42 | #include "AliESDEvent.h" |
43 | #include "AliESDTrdTrack.h" | |
44 | #include "AliESDTrdTracklet.h" | |
c4daee41 | 45 | #include "AliESDTrdTrigger.h" |
cb8b99ee | 46 | #include "AliTRDtrackletWord.h" |
03e2f460 | 47 | #include "AliTRDtrackletMCM.h" |
29f95561 | 48 | |
97b6dfac | 49 | #define SETFLG(n,f) ((n) |= f) |
50 | #define CLRFLG(n,f) ((n) &= ~f) | |
51 | ||
121a60bd | 52 | ClassImp(AliTRDReconstructor) |
53 | ||
c4daee41 | 54 | AliESDTrdTrigger AliTRDReconstructor::fgTriggerFlags; |
828c6f80 | 55 | TClonesArray *AliTRDReconstructor::fgClusters = NULL; |
56 | TClonesArray *AliTRDReconstructor::fgTracklets = NULL; | |
cb8b99ee | 57 | TClonesArray *AliTRDReconstructor::fgTracks = NULL; |
828c6f80 | 58 | Char_t const * AliTRDReconstructor::fgSteerNames[kNsteer] = { |
b72f4eaf | 59 | "DigitsConversion " |
b72f4eaf | 60 | ,"Write Clusters " |
61 | ,"Write Online Tracklets " | |
b72f4eaf | 62 | ,"Stand Alone Tracking " |
8fc736d7 | 63 | ,"HLT Mode " |
64 | ,"Process Online Trklts " | |
65 | ,"Debug Streaming " | |
66 | ,"Cl. Radial Correction " | |
b72f4eaf | 67 | }; |
828c6f80 | 68 | Char_t const * AliTRDReconstructor::fgSteerFlags[kNsteer] = { |
b72f4eaf | 69 | "dc"// digits conversion [false] |
b72f4eaf | 70 | ,"cw"// write clusters [true] |
71 | ,"tw"// write online tracklets [false] | |
b72f4eaf | 72 | ,"sa"// track seeding (stand alone tracking) [true] |
b72f4eaf | 73 | ,"hlt"// HLT reconstruction [false] |
a5b99acd | 74 | ,"tp"// also use online tracklets for reconstruction [false] |
a2fbb6ec | 75 | ,"deb"// Write debug stream [false] |
b30d8c09 | 76 | ,"cc" // Cluster radial correction during reconstruction [false] |
b72f4eaf | 77 | }; |
828c6f80 | 78 | Char_t const * AliTRDReconstructor::fgTaskNames[AliTRDrecoParam::kTRDreconstructionTasks] = { |
a2fbb6ec | 79 | "Clusterizer" |
b72f4eaf | 80 | ,"Tracker" |
81 | ,"PID" | |
82 | }; | |
828c6f80 | 83 | Char_t const * AliTRDReconstructor::fgTaskFlags[AliTRDrecoParam::kTRDreconstructionTasks] = { |
a2fbb6ec | 84 | "cl" |
b72f4eaf | 85 | ,"tr" |
86 | ,"pd" | |
87 | }; | |
c6f7c6cb | 88 | Int_t AliTRDReconstructor::fgNTimeBins = -1; |
952051c5 | 89 | const Float_t AliTRDReconstructor::fgkMinClustersInTrack = 0.5; // |
90 | const Float_t AliTRDReconstructor::fgkLabelFraction = 0.8; // | |
91 | const Double_t AliTRDReconstructor::fgkMaxChi2 = 12.0; // | |
92 | const Double_t AliTRDReconstructor::fgkMaxSnp = 0.95; // Maximum local sine of the azimuthal angle | |
93 | const Double_t AliTRDReconstructor::fgkMaxStep = 2.0; // Maximal step size in propagation | |
94 | const Double_t AliTRDReconstructor::fgkEpsilon = 1.e-5; // Precision of radial coordinate | |
b72f4eaf | 95 | |
eb38ed55 | 96 | //_____________________________________________________________________________ |
3a039a31 | 97 | AliTRDReconstructor::AliTRDReconstructor() |
98 | :AliReconstructor() | |
97b6dfac | 99 | ,fSteerParam(0) |
5127281e | 100 | ,fClusterizer(NULL) |
3a039a31 | 101 | { |
97b6dfac | 102 | // setting default "ON" steering parameters |
89971837 | 103 | // owner of debug streamers |
104 | SETFLG(fSteerParam, kOwner); | |
97b6dfac | 105 | // write clusters [cw] |
106 | SETFLG(fSteerParam, kWriteClusters); | |
107 | // track seeding (stand alone tracking) [sa] | |
9dcc64cc | 108 | //SETFLG(fSteerParam, kSeeding); |
8fc736d7 | 109 | // Cluster radial correction during reconstruction [cc] |
b30d8c09 | 110 | //SETFLG(fSteerParam, kClRadialCorr); |
a2fbb6ec | 111 | memset(fDebugStream, 0, sizeof(TTreeSRedirector *) * AliTRDrecoParam::kTRDreconstructionTasks); |
eb38ed55 | 112 | } |
113 | ||
48f8adf3 | 114 | //_____________________________________________________________________________ |
115 | AliTRDReconstructor::~AliTRDReconstructor() | |
116 | { | |
d39401ad | 117 | // |
118 | // Destructor | |
119 | // | |
120 | ||
48f8adf3 | 121 | if(fgClusters) { |
6d54fcbf | 122 | fgClusters->Delete(); |
123 | delete fgClusters; | |
124 | fgClusters = NULL; | |
48f8adf3 | 125 | } |
a5b99acd | 126 | if(fgTracklets) { |
6d54fcbf | 127 | fgTracklets->Delete(); |
128 | delete fgTracklets; | |
129 | fgTracklets = NULL; | |
a5b99acd | 130 | } |
cb8b99ee | 131 | if(fgTracks) { |
132 | fgTracks->Delete(); | |
133 | delete fgTracks; | |
134 | fgTracks = NULL; | |
135 | } | |
f29f13a6 | 136 | if(fSteerParam&kOwner){ |
a2fbb6ec | 137 | for(Int_t itask = 0; itask < AliTRDrecoParam::kTRDreconstructionTasks; itask++) |
29f95561 | 138 | if(fDebugStream[itask]) delete fDebugStream[itask]; |
139 | } | |
5127281e | 140 | if(fClusterizer){ |
141 | delete fClusterizer; | |
142 | fClusterizer = NULL; | |
143 | } | |
18c05eb3 | 144 | } |
48f8adf3 | 145 | |
91834b8d | 146 | |
acd241e9 | 147 | //_____________________________________________________________________________ |
91834b8d | 148 | void AliTRDReconstructor::Init(){ |
18c05eb3 | 149 | // |
150 | // Init Options | |
151 | // | |
18c05eb3 | 152 | SetOption(GetOption()); |
a2fbb6ec | 153 | Options(fSteerParam); |
154 | ||
5127281e | 155 | if(!fClusterizer){ |
156 | fClusterizer = new AliTRDclusterizer(fgTaskNames[AliTRDrecoParam::kClusterizer], fgTaskNames[AliTRDrecoParam::kClusterizer]); | |
157 | fClusterizer->SetReconstructor(this); | |
158 | } | |
159 | ||
a2fbb6ec | 160 | // Make Debug Streams when Debug Streaming |
161 | if(IsDebugStreaming()){ | |
162 | for(Int_t task = 0; task < AliTRDrecoParam::kTRDreconstructionTasks; task++){ | |
163 | TDirectory *savedir = gDirectory; | |
164 | fDebugStream[task] = new TTreeSRedirector(Form("TRD.Debug%s.root", fgTaskNames[task])); | |
165 | savedir->cd(); | |
166 | SETFLG(fSteerParam, kOwner); | |
167 | } | |
168 | } | |
acd241e9 | 169 | } |
0397ce53 | 170 | |
121a60bd | 171 | //_____________________________________________________________________________ |
a8fbb161 | 172 | void AliTRDReconstructor::ConvertDigits(AliRawReader *rawReader |
91834b8d | 173 | , TTree *digitsTree) const |
121a60bd | 174 | { |
3bc9d03e | 175 | // |
25ca55ce | 176 | // Convert raw data digits into digit objects in a root tree |
3bc9d03e | 177 | // |
121a60bd | 178 | |
ad5dd6ec | 179 | //AliInfo("Convert raw data digits into digit objects [RawReader -> Digit TTree]"); |
91834b8d | 180 | |
a8fbb161 | 181 | AliTRDrawData rawData; |
3a0f6479 | 182 | rawReader->Reset(); |
183 | rawReader->Select("TRD"); | |
a8fbb161 | 184 | AliTRDdigitsManager *manager = rawData.Raw2Digits(rawReader); |
185 | manager->MakeBranch(digitsTree); | |
186 | manager->WriteDigits(); | |
c0ab62ff | 187 | delete manager; |
0ee00e25 | 188 | |
cb8b99ee | 189 | // take over ownership of online tracklets |
190 | fgTracklets = rawData.TrackletsArray(); | |
191 | rawData.SetTrackletsOwner(0x0); | |
192 | ||
193 | // take over GTU tracks | |
194 | fgTracks = rawData.TracksArray(); | |
195 | rawData.SetTracksOwner(0x0); | |
c4daee41 | 196 | |
197 | for (Int_t iSector = 0; iSector < 18; iSector++) { | |
198 | fgTriggerFlags.SetFlags(iSector, rawData.GetTriggerFlags(iSector)); | |
199 | } | |
121a60bd | 200 | } |
201 | ||
25ca55ce | 202 | //_____________________________________________________________________________ |
203 | void AliTRDReconstructor::Reconstruct(AliRawReader *rawReader | |
204 | , TTree *clusterTree) const | |
205 | { | |
206 | // | |
207 | // Reconstruct clusters | |
208 | // | |
209 | ||
91834b8d | 210 | //AliInfo("Reconstruct TRD clusters from RAW data [RawReader -> Cluster TTree]"); |
211 | ||
212 | ||
3a0f6479 | 213 | rawReader->Reset(); |
214 | rawReader->Select("TRD"); | |
215 | ||
5127281e | 216 | if(!fClusterizer){ |
217 | AliFatal("Clusterizer not available!"); | |
218 | return; | |
219 | } | |
220 | ||
da0a4b87 | 221 | fClusterizer->ResetRecPoints(); |
222 | ||
5127281e | 223 | fClusterizer->OpenOutput(clusterTree); |
5127281e | 224 | fClusterizer->SetUseLabels(kFALSE); |
07897df4 | 225 | fClusterizer->SetStoreRawSignals(kTRUE); |
5127281e | 226 | fClusterizer->Raw2ClustersChamber(rawReader); |
66f6bfd9 | 227 | |
3e1f6592 | 228 | fgNTimeBins = fClusterizer->GetNTimeBins(); |
229 | ||
cb8b99ee | 230 | // take over ownership of online tracklets |
231 | fgTracklets = fClusterizer->TrackletsArray(); | |
232 | fClusterizer->SetTrackletsOwner(kFALSE); | |
233 | ||
234 | // take over GTU tracks | |
235 | fgTracks = fClusterizer->TracksArray(); | |
236 | fClusterizer->SetTracksOwner(kFALSE); | |
237 | ||
c4daee41 | 238 | for (Int_t iSector = 0; iSector < 18; iSector++) { |
239 | fgTriggerFlags.SetFlags(iSector, fClusterizer->GetTriggerFlags(iSector)); | |
240 | } | |
241 | ||
66f6bfd9 | 242 | if(IsWritingClusters()) return; |
d6e36038 | 243 | |
66f6bfd9 | 244 | // take over ownership of clusters |
5127281e | 245 | fgClusters = fClusterizer->RecPoints(); |
246 | fClusterizer->SetClustersOwner(kFALSE); | |
25ca55ce | 247 | } |
248 | ||
249 | //_____________________________________________________________________________ | |
a8fbb161 | 250 | void AliTRDReconstructor::Reconstruct(TTree *digitsTree |
251 | , TTree *clusterTree) const | |
25ca55ce | 252 | { |
253 | // | |
254 | // Reconstruct clusters | |
255 | // | |
e4f2f73d | 256 | |
91834b8d | 257 | //AliInfo("Reconstruct TRD clusters from Digits [Digit TTree -> Cluster TTree]"); |
a2fbb6ec | 258 | |
c665eaa7 | 259 | AliTRDclusterizer clusterer(fgTaskNames[AliTRDrecoParam::kClusterizer], fgTaskNames[AliTRDrecoParam::kClusterizer]); |
260 | clusterer.SetReconstructor(this); | |
302bfc4e | 261 | clusterer.SetUseLabels(kTRUE); |
07897df4 | 262 | clusterer.SetStoreRawSignals(kTRUE); |
c665eaa7 | 263 | clusterer.OpenOutput(clusterTree); |
264 | clusterer.ReadDigits(digitsTree); | |
265 | clusterer.MakeClusters(); | |
d6e36038 | 266 | |
cb8b99ee | 267 | // read tracklets and tracks if not done during reading of raw data |
268 | if (!fgTracklets) { | |
269 | clusterer.ReadTracklets(); | |
270 | fgTracklets = clusterer.TrackletsArray(); | |
271 | clusterer.SetTrackletsOwner(kFALSE); | |
272 | } | |
273 | if (!fgTracks) { | |
274 | clusterer.ReadTracks(); | |
275 | fgTracks = clusterer.TracksArray(); | |
276 | clusterer.SetTracksOwner(kFALSE); | |
277 | } | |
278 | ||
3e1f6592 | 279 | fgNTimeBins = clusterer.GetNTimeBins(); |
280 | ||
66f6bfd9 | 281 | if(IsWritingClusters()) return; |
282 | ||
283 | // take over ownership of clusters | |
c665eaa7 | 284 | fgClusters = clusterer.RecPoints(); |
285 | clusterer.SetClustersOwner(kFALSE); | |
25ca55ce | 286 | } |
287 | ||
288 | //_____________________________________________________________________________ | |
44dbae42 | 289 | AliTracker *AliTRDReconstructor::CreateTracker() const |
121a60bd | 290 | { |
3bc9d03e | 291 | // |
292 | // Create a TRD tracker | |
293 | // | |
121a60bd | 294 | |
e4f2f73d | 295 | //return new AliTRDtracker(NULL); |
3a039a31 | 296 | AliTRDtrackerV1 *tracker = new AliTRDtrackerV1(); |
297 | tracker->SetReconstructor(this); | |
298 | return tracker; | |
25ca55ce | 299 | |
25ca55ce | 300 | } |
301 | ||
302 | //_____________________________________________________________________________ | |
303 | void AliTRDReconstructor::FillESD(TTree* /*digitsTree*/ | |
91834b8d | 304 | , TTree* /*clusterTree*/ |
cb8b99ee | 305 | , AliESDEvent* esd) const |
25ca55ce | 306 | { |
307 | // | |
af26ce80 | 308 | // Fill ESD |
25ca55ce | 309 | // |
310 | ||
cb8b99ee | 311 | // ----- filling tracklets ----- |
cb8b99ee | 312 | AliDebug(1, Form("Filling tracklets from %p (%i)", |
313 | fgTracklets, fgTracklets ? fgTracklets->GetEntriesFast() : 0)); | |
5fedeee9 | 314 | Int_t trackletIndex[1080] = { 0 }; |
315 | TList trklList; | |
316 | AliTRDrawStream::SortTracklets(fgTracklets, trklList, trackletIndex); | |
317 | TIter trackletIter(&trklList); | |
03e2f460 | 318 | while (AliTRDtrackletBase* tracklet = (AliTRDtrackletBase*) trackletIter()) { |
319 | Int_t label = -2; // mark raw tracklets with label -2 | |
320 | if (AliTRDtrackletMCM *trklMCM = dynamic_cast<AliTRDtrackletMCM*> (tracklet)) | |
321 | label = trklMCM->GetLabel(); | |
322 | ||
323 | AliESDTrdTracklet *esdTracklet = new AliESDTrdTracklet(tracklet->GetTrackletWord(), tracklet->GetHCId(), label); | |
324 | ||
325 | esd->AddTrdTracklet(esdTracklet); | |
cb8b99ee | 326 | } |
327 | ||
328 | // ----- filling GTU tracks ----- | |
329 | AliDebug(1, Form("Now filling ESD with GTU tracks from %p (%i)", | |
330 | fgTracks, fgTracks ? fgTracks->GetEntriesFast() : 0)); | |
331 | if (fgTracks) { | |
332 | for (Int_t iTrack = 0; iTrack < fgTracks->GetEntriesFast(); iTrack++) { | |
333 | AliESDTrdTrack *trdTrack = (AliESDTrdTrack*) ((*fgTracks)[iTrack]); | |
334 | ||
cb8b99ee | 335 | UInt_t stack = trdTrack->GetStack(); |
336 | ||
03e2f460 | 337 | Int_t refIndex[6]; |
338 | AliTRDrawStream::AssignTracklets(trdTrack, trackletIndex, refIndex); | |
339 | ||
340 | for (Int_t iLayer = 0; iLayer < 6; ++iLayer) { | |
341 | Int_t det = trdTrack->GetSector()*30 + stack*6 + iLayer; | |
342 | ||
343 | AliESDTrdTracklet *trkl = refIndex[iLayer] > -1 ? esd->GetTrdTracklet(refIndex[iLayer]) : 0x0; | |
344 | if (trkl) { | |
345 | AliDebug(5, Form("adding tracklet with index %i: 0x%08x", | |
346 | refIndex[iLayer], trkl->GetTrackletWord())); | |
347 | if (trkl->GetDetector() != det) | |
348 | AliError(Form("inconsistent assignment of tracklet 0x%08x in det %i to track in %i", | |
349 | trkl->GetTrackletWord(), trkl->GetDetector(), det)); | |
350 | trdTrack->AddTrackletReference(trkl, iLayer); | |
cb8b99ee | 351 | } |
352 | } | |
03e2f460 | 353 | |
cb8b99ee | 354 | // only add the track when it's complete (including tracklet references) |
355 | esd->AddTrdTrack(trdTrack); | |
356 | } | |
357 | } | |
358 | ||
c4daee41 | 359 | esd->SetTrdTrigger(&fgTriggerFlags); |
360 | ||
cb8b99ee | 361 | // clearing variables for next event |
362 | fgTracklets = 0x0; | |
363 | fgTracks = 0x0; | |
25ca55ce | 364 | } |
3a039a31 | 365 | |
3a039a31 | 366 | //_____________________________________________________________________________ |
367 | void AliTRDReconstructor::SetOption(Option_t *opt) | |
368 | { | |
d39401ad | 369 | // |
370 | // Read option string into the steer param. | |
371 | // | |
954135b1 | 372 | // The following string options are available during reconstruction. |
373 | // In square brackets the default values are given. | |
374 | // "dc" : digits conversion [false] | |
375 | // "cw" : write clusters [true] | |
376 | // "tw" : write online tracklets [false] | |
377 | // "sa" : track seeding (stand alone tracking) [true] | |
378 | // "hlt" : HLT reconstruction [false] | |
379 | // "tp" : also use online tracklets for reconstruction [false] | |
380 | // "deb" : Write debug stream [false] | |
381 | // "cc" : Cluster radial correction during reconstruction [false] | |
382 | // | |
383 | // To check the actual options used during reconstruction include the following line in your rec.C script | |
384 | // AliLog::SetClassDebugLevel("AliTRDReconstructor", 1); | |
97b6dfac | 385 | |
386 | AliReconstructor::SetOption(opt); | |
3a039a31 | 387 | |
388 | TString s(opt); | |
389 | TObjArray *opar = s.Tokenize(","); | |
390 | for(Int_t ipar=0; ipar<opar->GetEntriesFast(); ipar++){ | |
d39401ad | 391 | Bool_t processed = kFALSE; |
3a039a31 | 392 | TString sopt(((TObjString*)(*opar)[ipar])->String()); |
b72f4eaf | 393 | for(Int_t iopt=0; iopt<kNsteer; iopt++){ |
394 | if(!sopt.Contains(fgSteerFlags[iopt])) continue; | |
395 | SETFLG(fSteerParam, BIT(iopt)); | |
396 | if(sopt.Contains("!")) CLRFLG(fSteerParam, BIT(iopt)); | |
d39401ad | 397 | processed = kTRUE; |
b72f4eaf | 398 | break; |
399 | } | |
d39401ad | 400 | if(processed) continue; |
b72f4eaf | 401 | |
402 | AliWarning(Form("Unknown option flag %s.", sopt.Data())); | |
3a039a31 | 403 | } |
404 | } | |
405 | ||
acd241e9 | 406 | //_____________________________________________________________________________ |
a2fbb6ec | 407 | void AliTRDReconstructor::Options(UInt_t steer) |
b72f4eaf | 408 | { |
d39401ad | 409 | // |
410 | // Print the options | |
411 | // | |
412 | ||
b72f4eaf | 413 | for(Int_t iopt=0; iopt<kNsteer; iopt++){ |
9e8fdff4 | 414 | AliDebugGeneral("AliTRDReconstructor", 1, Form(" %s[%s]%s", fgSteerNames[iopt], fgSteerFlags[iopt], steer ?(((steer>>iopt)&1)?" : ON":" : OFF"):"")); |
b72f4eaf | 415 | } |
b72f4eaf | 416 | } |
417 |