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