]> git.uio.no Git - u/mrichter/AliRoot.git/blame - TRD/AliTRDgtuTMU.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
ndex: TRD/AliTRDdEdxReconUtils.h
[u/mrichter/AliRoot.git] / TRD / AliTRDgtuTMU.cxx
CommitLineData
52c19022 1/**************************************************************************
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 **************************************************************************/
15
16/* $Id: AliTRDgtuTMU.cxx 28397 2008-09-02 09:33:00Z cblume $ */
17
18////////////////////////////////////////////////////////////////////////////
19// //
20// Track Matching Unit (TMU) simulation //
21// //
22// Author: J. Klein (Jochen.Klein@cern.ch) //
23// //
24////////////////////////////////////////////////////////////////////////////
25
26#include "TTree.h"
27#include "TList.h"
28#include "TVectorD.h"
29#include "TMath.h"
30
31#include "AliESDEvent.h"
32#include "AliESDTrdTrack.h"
33
34#include "AliLog.h"
35#include "AliTRDgeometry.h"
36#include "AliTRDpadPlane.h"
37
38#include "AliTRDgtuParam.h"
39#include "AliTRDgtuTMU.h"
40#include "AliTRDtrackGTU.h"
41
42ClassImp(AliTRDgtuTMU)
43
44AliTRDgtuTMU::AliTRDgtuTMU(Int_t stack, Int_t sector) :
45 TObject(),
46 fTracklets(0x0),
44eafcf2 47 fTrackletsPostInput(0x0),
52c19022 48 fZChannelTracklets(0x0),
49 fTracks(0x0),
50 fGtuParam(0x0),
51 fStack(-1),
52 fSector(-1)
53{
36dc3337 54 // constructor which initializes the position information of the TMU
55
52c19022 56 fGtuParam = AliTRDgtuParam::Instance();
44eafcf2 57
58 // store tracklets per link
59 fTracklets = new TObjArray*[fGtuParam->GetNLinks()];
60 for (Int_t iLink = 0; iLink < fGtuParam->GetNLinks(); iLink++) {
61 fTracklets[iLink] = new TObjArray();
62 }
63
64 // tracklets after input units per layer
65 fTrackletsPostInput = new TObjArray*[fGtuParam->GetNLayers()];
52c19022 66 fZChannelTracklets = new TList*[fGtuParam->GetNLayers()];
67 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
52c19022 68 fZChannelTracklets[layer] = new TList[fGtuParam->GetNZChannels()];
44eafcf2 69 fTrackletsPostInput[layer] = new TObjArray();
52c19022 70 }
44eafcf2 71
52c19022 72 fTracks = new TList*[fGtuParam->GetNZChannels()];
73 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
74 fTracks[zch] = new TList[fGtuParam->GetNRefLayers()];
75 }
44eafcf2 76
5f006bd7 77 if (stack > -1)
52c19022 78 SetStack(stack);
79 if (sector > -1)
80 SetSector(sector);
81}
82
5f006bd7 83AliTRDgtuTMU::~AliTRDgtuTMU()
52c19022 84{
36dc3337 85 // destructor
86
52c19022 87 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
88 delete [] fTracks[zch];
89 }
90 delete [] fTracks;
91 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
52c19022 92 delete [] fZChannelTracklets[layer];
44eafcf2 93 // delete [] fTrackletsPostInput[layer];
52c19022 94 }
95 delete [] fZChannelTracklets;
44eafcf2 96 // delete [] fTrackletsPostInput;
97
98 for (Int_t iLink = 0; iLink < fGtuParam->GetNLinks(); iLink++) {
99 delete fTracklets[iLink];
100 }
52c19022 101 delete [] fTracklets;
102}
103
104Bool_t AliTRDgtuTMU::SetSector(Int_t sector)
105{
106 // set the sector
107
108 if (sector > -1 && sector < fGtuParam->GetGeo()->Nsector() ) {
5f006bd7 109 fSector = sector;
52c19022 110 return kTRUE;
111 }
112
113 AliError(Form("Invalid sector given: %i", sector));
114 return kFALSE;
115}
116
5f006bd7 117Bool_t AliTRDgtuTMU::SetStack(Int_t stack)
52c19022 118{
119 // Set the stack (necessary for tracking)
120
121 if (stack > -1 && stack < fGtuParam->GetGeo()->Nstack() ) {
122 fStack = stack;
123 return kTRUE;
124 }
125
126 AliError(Form("Invalid stack given: %i", stack));
127 return kFALSE;
128}
129
5f006bd7 130Bool_t AliTRDgtuTMU::Reset()
52c19022 131{
132 // delete all tracks
133
134 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
135 for (Int_t reflayeridx = 0; reflayeridx < fGtuParam->GetNRefLayers(); reflayeridx++) {
136 fTracks[zch][reflayeridx].Clear();
137 }
138 }
139
140 // delete all added tracklets
44eafcf2 141 for (Int_t iLink = 0; iLink < fGtuParam->GetNLinks(); iLink++) {
142 fTracklets[iLink]->Clear();
143 }
52c19022 144 for (Int_t layer = 0; layer < fGtuParam->GetNLinks()/2; layer++) {
44eafcf2 145 fTrackletsPostInput[layer]->Clear();
52c19022 146 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++)
147 fZChannelTracklets[layer][zch].Clear();
52c19022 148 }
149
150 fSector = -1;
151 fStack = -1;
152
153 return kTRUE;
154}
155
44eafcf2 156Bool_t AliTRDgtuTMU::AddTracklet(AliTRDtrackletGTU *tracklet, Int_t link)
52c19022 157{
158 // add a tracklet on the given link
159
44eafcf2 160 fTracklets[link]->Add(tracklet);
52c19022 161 return kTRUE;
162}
163
164
5f006bd7 165Bool_t AliTRDgtuTMU::RunTMU(TList *ListOfTracks, AliESDEvent *esd)
52c19022 166{
167 // performs the analysis as in a TMU module of the GTU, i. e.
168 // track matching
169 // calculation of track parameteres (pt, deflection, ???)
170
171 if (fStack < 0 || fSector < 0) {
36dc3337 172 AliError("No valid stack/sector set for this TMU! No tracking!");
52c19022 173 return kFALSE;
174 }
175
176 // ----- Input units -----
c8b1590d 177 AliDebug(1,"--------- Running Input units ----------");
52c19022 178 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
179 if (!RunInputUnit(layer)) {
180 AliError(Form("Input unit in layer %i failed", layer));
181 return kFALSE;
182 }
183 }
184
185 // ----- Z-channel units -----
c8b1590d 186 AliDebug(1,"--------- Running Z-channel units ----------");
52c19022 187 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
188 fZChannelTracklets[layer] = new TList[fGtuParam->GetNZChannels()];
189 if (!RunZChannelUnit(layer)) {
190 AliError(Form("Z-Channel unit in layer %i failed", layer));
191 return kFALSE;
192 }
193 }
194
195 // ----- track finding -----
c8b1590d 196 AliDebug(1,"--------- Running tracking units ----------");
52c19022 197 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
52c19022 198 if (!RunTrackFinder(zch, ListOfTracks)) {
199 AliError(Form("Track Finder in z-channel %i failed", zch));
200 return kFALSE;
201 }
5f006bd7 202 }
52c19022 203
204 // ----- Track Merging -----
205 if (!RunTrackMerging(ListOfTracks)) {
206 AliError("Track merging failed");
207 return kFALSE;
208 }
209
210 // ----- track reconstruction -----
211 if (!RunTrackReconstruction(ListOfTracks)) {
212 AliError("Track reconstruction failed");
213 return kFALSE;
214 }
215
2f1bde31 216 // ----- label calculation and ESD storage -----
217 TIter next(ListOfTracks);
218 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) next()) {
219 trk->CookLabel();
220 if (esd) {
221 AliESDTrdTrack *trdtrack = trk->CreateTrdTrack();
222 esd->AddTrdTrack(trdtrack);
223 delete trdtrack;
224 }
52c19022 225 }
226
227 return kTRUE;
228}
229
5f006bd7 230Bool_t AliTRDgtuTMU::RunInputUnit(Int_t layer)
52c19022 231{
232 // precalculations for the tracking and reconstruction
233
44eafcf2 234 Int_t iTrkl0 = 0; // A-side tracklet
235 Int_t iTrkl1 = 0; // B-side tracklet
236
237 while ((iTrkl0 < fTracklets[2*layer + 0]->GetEntriesFast()) || (iTrkl1 < fTracklets[2*layer + 1]->GetEntriesFast())) {
238 if (iTrkl1 >= fTracklets[2*layer + 1]->GetEntriesFast()) {
239 fTrackletsPostInput[layer]->AddLast(fTracklets[2*layer + 0]->At(iTrkl0));
240 iTrkl0++;
241 }
242 else {
243 if (iTrkl0 >= fTracklets[2*layer + 0]->GetEntriesFast()) {
244 fTrackletsPostInput[layer]->AddLast(fTracklets[2*layer + 1]->At(iTrkl1));
245 iTrkl1++;
246 }
247 else {
248 if (((AliTRDtrackletGTU*) fTracklets[2*layer + 1]->At(iTrkl1))->GetZbin() <
249 ((AliTRDtrackletGTU*) fTracklets[2*layer + 0]->At(iTrkl0))->GetZbin()) {
250 fTrackletsPostInput[layer]->AddLast(fTracklets[2*layer + 1]->At(iTrkl1));
251 iTrkl1++;
252
253 }
254 else {
255 fTrackletsPostInput[layer]->AddLast(fTracklets[2*layer + 0]->At(iTrkl0));
256 iTrkl0++;
257 }
258 }
259 }
260 }
261
262 TIter next(fTrackletsPostInput[layer]);
52c19022 263
264 while ( AliTRDtrackletGTU *trk = (AliTRDtrackletGTU*) next() ) {
44eafcf2 265 trk->SetIndex(fTrackletsPostInput[layer]->IndexOf(trk));
52c19022 266
5f006bd7 267 Int_t alpha = (trk->GetYbin() >> fGtuParam->GetBitExcessY()) * fGtuParam->GetCiAlpha(layer);
52c19022 268 alpha = ( 2 * trk->GetdY() - (alpha >> fGtuParam->GetBitExcessAlpha()) + 1 ) >> 1;
a10c6981 269 // wrapping projected alpha as in hardware
270 if ((alpha < -64) || (alpha > 63))
271 AliError(Form("alpha out of range: %i", alpha));
272 alpha = alpha & 0x7f;
273 if (alpha & 0x40)
274 trk->SetAlpha(0xffffffc0 | alpha);
275 else
276 trk->SetAlpha(alpha);
52c19022 277
4ff7ed2b 278 Int_t yproj = trk->GetdY() * (fGtuParam->GetCiYProj(layer)); //??? sign?
52c19022 279 yproj = ( ( ( (yproj >> fGtuParam->GetBitExcessYProj()) + trk->GetYbin() ) >> 2) + 1) >> 1;
280 trk->SetYProj(yproj);
281
282 trk->SetYPrime(trk->GetYbin() + fGtuParam->GetYt(fStack, layer, trk->GetZbin()));
283
44eafcf2 284 AliDebug(10, Form("0x%08x: idx: %3i, z: %2i, y: %5i, dy: %3i, y': %5i, y_proj: %5i, alpha: %3i, pid: %3i",
285 trk->GetTrackletWord(), trk->GetIndex(), trk->GetZbin(), trk->GetYbin(), trk->GetdY(), trk->GetYPrime(),
286 trk->GetYProj(), trk->GetAlpha(), trk->GetPID() ));
52c19022 287 }
288 return kTRUE;
289}
290
5f006bd7 291Bool_t AliTRDgtuTMU::RunZChannelUnit(Int_t layer)
52c19022 292{
293 // run the z-channel unit
294
44eafcf2 295 TIter next(fTrackletsPostInput[layer]);
52c19022 296
297 while (AliTRDtrackletGTU *trk = (AliTRDtrackletGTU*) next()) {
52c19022 298 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
299 if (fGtuParam->IsInZChannel(fStack, layer, zch, trk->GetZbin()) ) {
300 trk->SetSubChannel(zch, fGtuParam->GetZSubchannel(fStack, layer, zch, trk->GetZbin()) );
52c19022 301
302 TIter nexttrkl(&fZChannelTracklets[layer][zch], kIterBackward);
303 AliTRDtrackletGTU *t = 0x0;
4cc89512 304 while ((t = (AliTRDtrackletGTU*) nexttrkl.Next())) {
5f006bd7 305 if (t->GetSubChannel(zch) < trk->GetSubChannel(zch) ||
44eafcf2 306 ((t->GetSubChannel(zch) == trk->GetSubChannel(zch)) && (t->GetYProj() < trk->GetYProj())) ) {
52c19022 307 break;
44eafcf2 308 }
52c19022 309 }
44eafcf2 310 if (t)
311 fZChannelTracklets[layer][zch].AddAfter(t, trk);
312 else
313 fZChannelTracklets[layer][zch].AddFirst(trk);
52c19022 314 }
52c19022 315 }
44eafcf2 316 AliDebug(10, Form("stack %d, layer %2d: 0x%08x Z(2..0)=%i/%i/%i",
317 fStack, layer, trk->GetTrackletWord(),
318 fGtuParam->IsInZChannel(fStack, layer, 2, trk->GetZbin()) ? trk->GetSubChannel(2) : -1,
319 fGtuParam->IsInZChannel(fStack, layer, 1, trk->GetZbin()) ? trk->GetSubChannel(1) : -1,
320 fGtuParam->IsInZChannel(fStack, layer, 0, trk->GetZbin()) ? trk->GetSubChannel(0) : -1
321 ));
52c19022 322 }
323 return kTRUE;
324}
325
5f006bd7 326Bool_t AliTRDgtuTMU::RunTrackFinder(Int_t zch, TList* /* ListOfTracks */)
52c19022 327{
328 // run the track finding
329
330 Int_t *notr = new Int_t[fGtuParam->GetNLayers()];
331 Int_t *ptrA = new Int_t[fGtuParam->GetNLayers()];
332 Int_t *ptrB = new Int_t[fGtuParam->GetNLayers()];
333
5f006bd7 334 Bool_t *bHitA = new Bool_t[fGtuParam->GetNLayers()];
52c19022 335 Bool_t *bHitB = new Bool_t[fGtuParam->GetNLayers()];
336 Bool_t *bAligned = new Bool_t[fGtuParam->GetNLayers()];
337 Bool_t *bAlignedA = new Bool_t[fGtuParam->GetNLayers()];
338 Bool_t *bAlignedB = new Bool_t[fGtuParam->GetNLayers()];
339 Bool_t *bDone = new Bool_t[fGtuParam->GetNLayers()];
340 Bool_t ready;
341
342 Int_t *inc = new Int_t[fGtuParam->GetNLayers()];
343 Int_t *incprime = new Int_t[fGtuParam->GetNLayers()];
344
345// ----- signals within current layer -----
36dc3337 346 Int_t yPlus;
5f006bd7 347 Int_t yMinus;
348 Int_t ybPlus;
36dc3337 349 Int_t ybMinus;
350 Int_t alphaPlus;
5f006bd7 351 Int_t alphaMinus;
36dc3337 352 Int_t nHits;
5f006bd7 353 Int_t nUnc;
36dc3337 354 Int_t nWayBeyond;
52c19022 355
356 AliTRDtrackletGTU *trkRA = 0x0; // reference tracklet A
357 AliTRDtrackletGTU *trkRB = 0x0; // reference tracklet B
358 AliTRDtrackletGTU *trkA = 0x0; // tracklet A in current layer
359 AliTRDtrackletGTU *trkB = 0x0; // tracklet B in current layer
360/*
361 AliTRDtrackletGTU *trkEnd = new AliTRDtrackletGTU();
5f006bd7 362 for (Int_t i = 0; i < fGtuParam->GetNZChannels(); i++)
52c19022 363 trkEnd->SetSubChannel(i, 7);
364 trkEnd->SetYProj(0);
365 trkEnd->SetAlpha(0);
366*/
367
368 for (Int_t refLayerIdx = 0; refLayerIdx < fGtuParam->GetNRefLayers(); refLayerIdx++) {
369 Int_t reflayer = fGtuParam->GetRefLayer(refLayerIdx);
d2c8b010 370 AliDebug(5,Form("Tracking for z-channel: %i, reflayer: %i", zch, reflayer));
52c19022 371
372 ready = kFALSE; // ready if all channels done
373
44eafcf2 374// for (Int_t iLayer = 0; iLayer < fGtuParam->GetNLayers(); iLayer++) {
375// for (Int_t iTrkl = 0; iTrkl < fZChannelTracklets[iLayer][zch].GetEntries(); iTrkl++) {
376// printf("%4i/%4i(%i,%i) ",
377// ((AliTRDtrackletGTU*) fZChannelTracklets[iLayer][zch].At(iTrkl))->GetYProj(),
378// ((AliTRDtrackletGTU*) fZChannelTracklets[iLayer][zch].At(iTrkl))->GetAlpha(),
379// ((AliTRDtrackletGTU*) fZChannelTracklets[iLayer][zch].At(iTrkl))->GetIndex(),
380// ((AliTRDtrackletGTU*) fZChannelTracklets[iLayer][zch].At(iTrkl))->GetSubChannel(zch)
381// );
382// }
383// printf("\n");
384// }
385
52c19022 386 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
387 notr[layer] = fZChannelTracklets[layer][zch].GetEntries();
388 ptrA[layer] = 0; // notr[layer] > 0 ? 0 : -1;
389 ptrB[layer] = 1; // notr[layer] > 1 ? 1 : -1;
390
391// not necessary here
392// bDone[layer] = (ptrB[layer] >= notr[layer] - 1); // trkB is last one
393// bDone[layer] = (notr[layer] <= 0);
394// ready = ready && bDone[layer];
395
396 if (reflayer == 1)
c8b1590d 397 AliDebug(5,Form("in layer: %i (zchannel = %i) there are: %i tracklets", layer, zch, notr[layer]));
52c19022 398 }
5f006bd7 399
400 if (ptrA[reflayer] < 0 && ptrB[reflayer] < 0)
52c19022 401 continue;
402
403 while (!ready) {
404 // ----- reference tracklets -----
405 trkRA = 0x0;
406 trkRB = 0x0;
407 if (0 <= ptrA[reflayer] && ptrA[reflayer] < notr[reflayer])
408 trkRA = (AliTRDtrackletGTU*) fZChannelTracklets[reflayer][zch].At(ptrA[reflayer]);
409 else {
a8518fd2 410 AliDebug(10,Form("No valid tracklet in the reference at ptr: %i! Nothing done!", ptrA[reflayer]));
5f006bd7 411 break;
52c19022 412 }
413
414 if (0 <= ptrB[reflayer] && ptrB[reflayer] < notr[reflayer])
415 trkRB = (AliTRDtrackletGTU*) fZChannelTracklets[reflayer][zch].At(ptrB[reflayer]);
416
36dc3337 417 yPlus = trkRA->GetYProj() + fGtuParam->GetDeltaY();
418 yMinus = trkRA->GetYProj() - fGtuParam->GetDeltaY();
419 alphaPlus = trkRA->GetAlpha() + fGtuParam->GetDeltaAlpha();
420 alphaMinus = trkRA->GetAlpha() - fGtuParam->GetDeltaAlpha();
52c19022 421 if (trkRB) {
36dc3337 422 ybPlus = trkRB->GetYProj() + fGtuParam->GetDeltaY();
423 ybMinus = trkRB->GetYProj() - fGtuParam->GetDeltaY();
52c19022 424 }
425 else { // irrelevant (should be, is it?)
36dc3337 426 ybPlus = trkRA->GetYProj() + fGtuParam->GetDeltaY();
427 ybMinus = trkRA->GetYProj() - fGtuParam->GetDeltaY();
52c19022 428 }
429
36dc3337 430 nHits = 0;
431 nUnc = 0;
432 nWayBeyond = 0;
5f006bd7 433
52c19022 434 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
435 bHitA[layer] = bHitB[layer] = bAligned[layer] = kFALSE;
436 inc[layer] = incprime[layer] = 0;
437
438 if (layer == reflayer) {
439 bHitA[layer] = kTRUE;
440 bAligned[layer] = kTRUE;
36dc3337 441 nHits++;
5f006bd7 442 continue;
52c19022 443 }
444
445 trkA = 0x0;
446 trkB = 0x0;
447 if (0 <= ptrA[layer] && ptrA[layer] < notr[layer])
448 trkA = (AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]);
449 if (0 <= ptrB[layer] && ptrB[layer] < notr[layer])
450 trkB = (AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]);
451
452 bAlignedA[layer] = kFALSE;
453 bAlignedB[layer] = kFALSE;
454
5f006bd7 455 if (trkA) {
36dc3337 456 bHitA[layer] = ( !(trkA->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() < yMinus) ) &&
457 !(trkA->GetSubChannel(zch) > trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() > yPlus) ) &&
458 !(trkA->GetAlpha() < alphaMinus) &&
459 !(trkA->GetAlpha() > alphaPlus) );
5f006bd7 460 bAlignedA[layer] = !(trkA->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() < yMinus) );
52c19022 461 }
462 else {
463 bHitA[layer] = 0;
464 bAlignedA[layer] = kTRUE;
465 }
466
467 if (trkB) {
36dc3337 468 bHitB[layer] = ( !(trkB->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkB->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkB->GetYProj() < yMinus) ) &&
469 !(trkB->GetSubChannel(zch) > trkRA->GetSubChannel(zch) || (trkB->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkB->GetYProj() > yPlus) ) &&
44eafcf2 470 !(trkB->GetAlpha() < alphaMinus) &&
471 !(trkB->GetAlpha() > alphaPlus) );
36dc3337 472 bAlignedB[layer] = (trkB->GetSubChannel(zch) > trkRA->GetSubChannel(zch) || (trkB->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkB->GetYProj() > yPlus) );
5f006bd7 473 }
52c19022 474 else {
475 bHitB[layer] = 0;
476 bAlignedB[layer] = kTRUE;
477 }
5f006bd7 478
52c19022 479 bAligned[layer] = bAlignedA[layer] || bAlignedB[layer]; //???
480// bAligned[layer] = bAlignedA[layer]; //???
5f006bd7 481
52c19022 482 if (bAligned[layer] && (bHitA[layer] || bHitB[layer]) )
36dc3337 483 nHits++;
52c19022 484 else if (!bAligned[layer] )
36dc3337 485 nUnc++;
52c19022 486 if (trkRB) {
487 if (trkA) {
36dc3337 488 if ((trkA->GetSubChannel(zch) > trkRB->GetSubChannel(zch)) || (trkA->GetSubChannel(zch) == trkRB->GetSubChannel(zch) && trkA->GetYProj() > ybPlus) )
489 nWayBeyond++;
52c19022 490 }
5f006bd7 491 else
36dc3337 492 nWayBeyond++;
52c19022 493 }
494
495 // pre-calculation for the layer shifting (alignment w. r. t. trkRB)
496 if (trkA) {
497 if(trkRB) {
36dc3337 498 if ((trkA->GetSubChannel(zch) < trkRB->GetSubChannel(zch)) || (trkA->GetSubChannel(zch) == trkRB->GetSubChannel(zch) && trkA->GetYProj() < ybMinus )) // could trkA be aligned for trkRB
52c19022 499 incprime[layer] = 1;
5f006bd7 500 else
52c19022 501 incprime[layer] = 0;
502 }
5f006bd7 503 else
52c19022 504 incprime[layer] = 1;
505
5f006bd7 506 if (trkB) {
52c19022 507 if (trkRB) {
36dc3337 508 if ((trkB->GetSubChannel(zch) < trkRB->GetSubChannel(zch)) || (trkB->GetSubChannel(zch) == trkRB->GetSubChannel(zch) && trkB->GetYProj() < ybMinus )) // could trkB be aligned for trkRB
52c19022 509 incprime[layer] = 2;
510 }
5f006bd7 511 else
52c19022 512 incprime[layer] = 2;
513 }
514 }
515 } // end of loop over layers
516
d2c8b010 517 AliDebug(5,Form("logic calculation finished, Nhits: %i %s",
518 nHits, (nHits >= 4) ? "-> track found" : ""));
52c19022 519
36dc3337 520 if (nHits >= 4) {
52c19022 521 // ----- track registration -----
52c19022 522 AliTRDtrackGTU *track = new AliTRDtrackGTU();
523 track->SetSector(fSector);
524 track->SetStack(fStack);
525 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++ ) {
5f006bd7 526 if (bHitA[layer] || layer == reflayer)
52c19022 527 track->AddTracklet((AliTRDtrackletGTU* ) fZChannelTracklets[layer][zch].At(ptrA[layer]), layer );
5f006bd7 528 else if (bHitB[layer])
52c19022 529 track->AddTracklet((AliTRDtrackletGTU* ) fZChannelTracklets[layer][zch].At(ptrB[layer]), layer );
530 }
531
532 Bool_t registerTrack = kTRUE;
44eafcf2 533 for (Int_t layerIdx = refLayerIdx-1; layerIdx >= 0; layerIdx--) {
80f93426 534 if (track->IsTrackletInLayer(fGtuParam->GetRefLayer(layerIdx))) {
535 if ((track->GetTracklet(fGtuParam->GetRefLayer(layerIdx)))->GetSubChannel(zch) > 0) {
c8b1590d 536 AliDebug(1,"Not registered");
52c19022 537 registerTrack = kFALSE;
80f93426 538 }
539 }
52c19022 540 }
541 if (registerTrack) {
542 track->SetZChannel(zch);
543 track->SetRefLayerIdx(refLayerIdx);
544 fTracks[zch][refLayerIdx].Add(track);
545 }
546 }
5f006bd7 547
36dc3337 548 if ( (nUnc != 0) && (nUnc + nHits >= 4) ) // could this position of the reference layer give some track //??? special check in case of hit?
52c19022 549 inc[reflayer] = 0;
36dc3337 550 else if (nWayBeyond > 2) // no track possible for both reference tracklets
52c19022 551 inc[reflayer] = 2;
5f006bd7 552 else
52c19022 553 inc[reflayer] = 1;
5f006bd7 554
52c19022 555 if (inc[reflayer] != 0) // reflayer is shifted
556 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
557 if (layer == reflayer)
558 continue;
5f006bd7 559 inc[layer] = incprime[layer];
52c19022 560 }
561 else { // reflayer is not shifted
562 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
563 if (layer == reflayer || notr[layer] == 0)
564 continue;
565 inc[layer] = 0;
566 trkA = 0x0;
567 trkB = 0x0;
568 if (0 <= ptrA[layer] && ptrA[layer] < notr[layer])
569 trkA = (AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]);
570
571 if (0 <= ptrB[layer] && ptrB[layer] < notr[layer])
572 trkB = (AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]);
573
574 if (trkA) {
44eafcf2 575 if ( !(trkA->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() < yMinus) ) &&
576 !(trkA->GetSubChannel(zch) > trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() > yPlus ) ) ) // trkA could hit trkRA
577 // if ( !(trkA->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkA->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkA->GetYProj() < yMinus) ) )
52c19022 578 inc[layer] = 0;
579 else if (trkB) {
36dc3337 580 if ( trkB->GetSubChannel(zch) < trkRA->GetSubChannel(zch) || (trkB->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkB->GetYProj() < yMinus) )
52c19022 581 inc[layer] = 2;
36dc3337 582 else if ( !(trkB->GetSubChannel(zch) > trkRA->GetSubChannel(zch) || (trkB->GetSubChannel(zch) == trkRA->GetSubChannel(zch) && trkB->GetYProj() > yPlus) ) )
52c19022 583 inc[layer] = 1;
5f006bd7 584 else
52c19022 585 inc[layer] = incprime[layer];
586 }
5f006bd7 587 else
52c19022 588 inc[layer] = incprime[layer];
589 }
590 }
591 }
5f006bd7 592
52c19022 593 ready = kTRUE;
d2c8b010 594 for (Int_t layer = fGtuParam->GetNLayers()-1; layer >= 0; layer--) {
52c19022 595
596 bDone[layer] = ptrB[layer] < 0 || ptrB[layer] >= notr[layer];
597 ready = ready && bDone[layer];
598
599 if (inc[layer] != 0 && ptrA[layer] >= notr[layer])
600 AliError(Form("Invalid increment: %i at ptrA: %i, notr: %i", inc[layer], ptrA[layer], notr[layer]));
601
602// AliInfo(Form("Shifting layer: %i, notr: %i, ptrA: %i, ptrB: %i, inc: %i", layer, notr[layer], ptrA[layer], ptrB[layer], inc[layer]));
d2c8b010 603 AliDebug(10,Form(" Layer: %i %2i(%2i, %2i, %4i, %3i)%s%s %2i(%2i, %2i, %4i, %3i)%s%s +%i %s (no: %i)",
44eafcf2 604 layer,
605 ptrA[layer],
606 (0 <= ptrA[layer] && ptrA[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]))->GetIndex() : -1,
d2c8b010 607 (0 <= ptrA[layer] && ptrA[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]))->GetSubChannel(zch) : -1,
608 (0 <= ptrA[layer] && ptrA[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]))->GetYProj() : -1,
609 (0 <= ptrA[layer] && ptrA[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrA[layer]))->GetAlpha() : -1,
44eafcf2 610 bHitA[layer] ? "*" : " ", bAlignedA[layer] ? "+" : " ",
611 ptrB[layer],
612 (0 <= ptrB[layer] && ptrB[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]))->GetIndex() : -1,
d2c8b010 613 (0 <= ptrB[layer] && ptrB[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]))->GetSubChannel(zch) : -1,
614 (0 <= ptrB[layer] && ptrB[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]))->GetYProj() : -1,
615 (0 <= ptrB[layer] && ptrB[layer] < notr[layer]) ? ((AliTRDtrackletGTU*) fZChannelTracklets[layer][zch].At(ptrB[layer]))->GetAlpha() : -1,
44eafcf2 616 bHitB[layer] ? "*" : " ", bAlignedB[layer] ? "+" : " ",
617 inc[layer], bDone[layer] ? "done" : " ", notr[layer]));
52c19022 618 ptrA[layer] += inc[layer];
619 ptrB[layer] += inc[layer];
620 }
621
622 } // end of while
623 } // end of loop over reflayer
624
625 delete [] bHitA;
626 delete [] bHitB;
627 delete [] bAligned;
628 delete [] bDone;
629 delete [] inc;
630 delete [] incprime;
631 delete [] bAlignedA;
632 delete [] bAlignedB;
633 delete [] notr;
634 delete [] ptrA;
635 delete [] ptrB;
636
637 return kTRUE;
638}
639
5f006bd7 640Bool_t AliTRDgtuTMU::RunTrackMerging(TList* ListOfTracks)
52c19022 641{
642 TList **tracksRefMerged = new TList*[fGtuParam->GetNZChannels()];
643 TList **tracksRefUnique = new TList*[fGtuParam->GetNZChannels()];
644
645 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
646 tracksRefMerged[zch] = new TList;
647 tracksRefUnique[zch] = new TList;
648 }
649
650 TList *tracksZMergedStage0 = new TList;
651 TList *tracksZUniqueStage0 = new TList;
652
653 TList **tracksZSplitted = new TList*[2];
654 for (Int_t i = 0; i < 2; i++)
655 tracksZSplitted[i] = new TList;
656
657 TList *tracksZMergedStage1 = new TList;
658
4cc89512 659 AliTRDtrackGTU **trkInRefLayer = new AliTRDtrackGTU*[fGtuParam->GetNRefLayers()];
52c19022 660
661 Bool_t done = kFALSE;
662 Int_t minIdx = 0;
663 AliTRDtrackGTU *trkStage0 = 0x0;
664
665 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
44eafcf2 666 // ----- Merging and Unification in Reflayers (seed_merger) -----
52c19022 667 do {
668 done = kTRUE;
669 trkStage0 = 0x0;
670 for (Int_t refLayerIdx = 0; refLayerIdx < fGtuParam->GetNRefLayers(); refLayerIdx++) {
4cc89512 671 trkInRefLayer[refLayerIdx] = (AliTRDtrackGTU*) fTracks[zch][refLayerIdx].First();
672 if (trkInRefLayer[refLayerIdx] == 0) {
52c19022 673 continue;
674 }
675 else if (trkStage0 == 0x0 ) {
4cc89512 676 trkStage0 = trkInRefLayer[refLayerIdx];
52c19022 677 minIdx = refLayerIdx;
678 done = kFALSE;
679 }
5f006bd7 680 else if (trkInRefLayer[refLayerIdx]->GetZSubChannel() < trkStage0->GetZSubChannel() ||
4cc89512 681 (trkInRefLayer[refLayerIdx]->GetZSubChannel() == trkStage0->GetZSubChannel() && trkInRefLayer[refLayerIdx]->GetYapprox() < trkStage0->GetYapprox()) ) {
52c19022 682 minIdx = refLayerIdx;
4cc89512 683 trkStage0 = trkInRefLayer[refLayerIdx];
52c19022 684 done = kFALSE;
685 }
686 }
c8b1590d 687 if (!trkStage0)
688 break;
52c19022 689 tracksRefMerged[zch]->Add(trkStage0);
690 fTracks[zch][minIdx].RemoveFirst();
691 } while (trkStage0 != 0);
692
693 Uniquifier(tracksRefMerged[zch], tracksRefUnique[zch]);
44eafcf2 694
695 AliDebug(2, Form("zchannel %i", zch));
696 TIter trackRefMerged(tracksRefMerged[zch]);
697 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackRefMerged())
698 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, z_idx=%i",
d2c8b010 699 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
44eafcf2 700 trk->GetTrackletIndex(5),
701 trk->GetTrackletIndex(4),
702 trk->GetTrackletIndex(3),
703 trk->GetTrackletIndex(2),
704 trk->GetTrackletIndex(1),
705 trk->GetTrackletIndex(0),
706 trk->GetYapprox() >> 3,
707 trk->GetZSubChannel()));
708 AliDebug(2, "uniquified:");
709 TIter trackRefUnique(tracksRefUnique[zch]);
710 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackRefUnique())
711 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, z_idx=%i",
d2c8b010 712 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
44eafcf2 713 trk->GetTrackletIndex(5),
714 trk->GetTrackletIndex(4),
715 trk->GetTrackletIndex(3),
716 trk->GetTrackletIndex(2),
717 trk->GetTrackletIndex(1),
718 trk->GetTrackletIndex(0),
719 trk->GetYapprox() >> 3,
720 trk->GetZSubChannel()));
52c19022 721 }
722
723// ----- Merging in zchannels - 1st stage -----
5f006bd7 724
52c19022 725 do {
726 done = kTRUE;
727 trkStage0 = 0x0;
a10c6981 728 for (Int_t zch = fGtuParam->GetNZChannels() - 1; zch > -1; zch--) {
52c19022 729 AliTRDtrackGTU *trk = (AliTRDtrackGTU*) tracksRefUnique[zch]->First();
730 if (trk == 0) {
731 continue;
732 }
733 else if (trkStage0 == 0x0 ) {
734 trkStage0 = trk;
735 minIdx = zch;
736 done = kFALSE;
737 }
d2c8b010 738 else if ( ((trk->GetZChannel() + 3 * trk->GetZSubChannel()) / 2 - 1) < ((trkStage0->GetZChannel() + 3 * trkStage0->GetZSubChannel()) / 2 -1 ) ||
739 (((trk->GetZChannel() + 3 * trk->GetZSubChannel()) / 2 - 1) == ((trkStage0->GetZChannel() + 3 * trkStage0->GetZSubChannel()) / 2 -1 ) && (trk->GetYapprox() < trkStage0->GetYapprox()) ) ) {
52c19022 740 minIdx = zch;
741 trkStage0 = trk;
742 done = kFALSE;
743 }
744 }
5f006bd7 745
c8b1590d 746 if (!trkStage0)
747 break;
52c19022 748 tracksZMergedStage0->Add(trkStage0);
749 tracksRefUnique[minIdx]->RemoveFirst();
750 } while (trkStage0 != 0);
5f006bd7 751
52c19022 752 Uniquifier(tracksZMergedStage0, tracksZUniqueStage0);
5f006bd7 753
d2c8b010 754 AliDebug(2, "stage 0:");
755 TIter trackZMergedStage0(tracksZMergedStage0);
756 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackZMergedStage0())
757 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, zch=%i idx=%i",
758 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
759 trk->GetTrackletIndex(5),
760 trk->GetTrackletIndex(4),
761 trk->GetTrackletIndex(3),
762 trk->GetTrackletIndex(2),
763 trk->GetTrackletIndex(1),
764 trk->GetTrackletIndex(0),
765 trk->GetYapprox() >> 3,
766 trk->GetZChannel(),
767 trk->GetZSubChannel()));
768
769 AliDebug(2, "uniquified:");
770 TIter trackZUniqueStage0(tracksZUniqueStage0);
771 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackZUniqueStage0())
772 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, zch=%i idx=%i",
773 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
774 trk->GetTrackletIndex(5),
775 trk->GetTrackletIndex(4),
776 trk->GetTrackletIndex(3),
777 trk->GetTrackletIndex(2),
778 trk->GetTrackletIndex(1),
779 trk->GetTrackletIndex(0),
780 trk->GetYapprox() >> 3,
781 trk->GetZChannel(),
782 trk->GetZSubChannel()));
783
52c19022 784// ----- Splitting in z -----
5f006bd7 785
52c19022 786 TIter next(tracksZUniqueStage0);
787 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) next()) {
788 tracksZSplitted[(trk->GetZChannel() + 3 * (trk->GetZSubChannel() - 1)) % 2]->Add(trk);
789 }
5f006bd7 790
d2c8b010 791 for (Int_t i = 0; i < 2; i++) {
792 AliDebug(2, Form("split %i:", i));
793 TIter trackZSplit(tracksZSplitted[i]);
794 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackZSplit())
795 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, zch=%i idx=%i",
796 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
797 trk->GetTrackletIndex(5),
798 trk->GetTrackletIndex(4),
799 trk->GetTrackletIndex(3),
800 trk->GetTrackletIndex(2),
801 trk->GetTrackletIndex(1),
802 trk->GetTrackletIndex(0),
803 trk->GetYapprox() >> 3,
804 trk->GetZChannel(),
805 trk->GetZSubChannel()));
806 }
807
52c19022 808// ----- Merging in zchanels - 2nd stage -----
5f006bd7 809
52c19022 810 do {
811 done = kTRUE;
812 trkStage0 = 0x0;
44eafcf2 813 for (Int_t i = 1; i >= 0; i--) {
52c19022 814 AliTRDtrackGTU *trk = (AliTRDtrackGTU*) tracksZSplitted[i]->First();
815 if (trk == 0) {
816 continue;
817 }
818 else if (trkStage0 == 0x0 ) {
819 trkStage0 = trk;
820 minIdx = i;
821 done = kFALSE;
822 }
d2c8b010 823 else if ( (((trk->GetZChannel() + 3 * (trk->GetZSubChannel() - 1)) / 2) < ((trkStage0->GetZChannel() + 3 * (trkStage0->GetZSubChannel() - 1)) / 2)) ||
824 ((((trk->GetZChannel() + 3 * (trk->GetZSubChannel() - 1)) / 2) == ((trkStage0->GetZChannel() + 3 * (trkStage0->GetZSubChannel() - 1)) / 2)) && (trk->GetYapprox() < trkStage0->GetYapprox()) ) ) {
52c19022 825 minIdx = i;
826 trkStage0 = trk;
827 done = kFALSE;
828 }
829 }
5f006bd7 830
c8b1590d 831 if (!trkStage0)
832 break;
52c19022 833 tracksZMergedStage1->Add(trkStage0);
834 tracksZSplitted[minIdx]->RemoveFirst();
835 } while (trkStage0 != 0);
5f006bd7 836
52c19022 837 Uniquifier(tracksZMergedStage1, ListOfTracks);
5f006bd7 838
d2c8b010 839 AliDebug(2, "merged:");
840 TIter trackZMergedStage1(tracksZMergedStage1);
841 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) trackZMergedStage1())
842 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, zch=%i idx=%i",
843 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
844 trk->GetTrackletIndex(5),
845 trk->GetTrackletIndex(4),
846 trk->GetTrackletIndex(3),
847 trk->GetTrackletIndex(2),
848 trk->GetTrackletIndex(1),
849 trk->GetTrackletIndex(0),
850 trk->GetYapprox() >> 3,
851 trk->GetZChannel(),
852 trk->GetZSubChannel()));
853
854 AliDebug(2, "uniquified:");
855 TIter track(ListOfTracks);
856 while (AliTRDtrackGTU *trk = (AliTRDtrackGTU*) track())
857 AliDebug(2, Form("track ref layer %i : %i %i %i %i %i %i, y=%i, zch=%i idx=%i",
858 AliTRDgtuParam::GetRefLayer(trk->GetRefLayerIdx()),
859 trk->GetTrackletIndex(5),
860 trk->GetTrackletIndex(4),
861 trk->GetTrackletIndex(3),
862 trk->GetTrackletIndex(2),
863 trk->GetTrackletIndex(1),
864 trk->GetTrackletIndex(0),
865 trk->GetYapprox() >> 3,
866 trk->GetZChannel(),
867 trk->GetZSubChannel()));
868
5f006bd7 869 // cleaning up
637666cd 870 for (Int_t zch = 0; zch < fGtuParam->GetNZChannels(); zch++) {
871 delete tracksRefMerged[zch];
872 delete tracksRefUnique[zch];
873 }
874 delete [] tracksRefMerged;
875 delete [] tracksRefUnique;
876
877
878 delete tracksZMergedStage0;
879 delete tracksZUniqueStage0;
880
881 for (Int_t i = 0; i < 2; i++)
882 delete tracksZSplitted[i];
54d34aac 883 delete [] tracksZSplitted;
637666cd 884
885 delete tracksZMergedStage1;
886
887 delete [] trkInRefLayer;
888
52c19022 889 return kTRUE;
890}
891
5f006bd7 892Bool_t AliTRDgtuTMU::RunTrackReconstruction(TList* ListOfTracks)
52c19022 893{
36dc3337 894 // run the track reconstruction for all tracks in the list
895
52c19022 896 TIter next(ListOfTracks);
897 while (AliTRDtrackGTU *track = (AliTRDtrackGTU*) next()) {
898 CalculateTrackParams(track);
4ff7ed2b 899 CalculatePID(track);
44eafcf2 900 AliDebug(1, Form("track with pid = %i", track->GetPID()));
52c19022 901 }
902 return kTRUE;
903}
904
4ff7ed2b 905Bool_t AliTRDgtuTMU::CalculatePID(AliTRDtrackGTU *track)
906{
36dc3337 907 // calculate PID for the given track
4ff7ed2b 908 if (!track) {
909 AliError("No track to calculate!");
910 return kFALSE;
911 }
912
4b43eeaf 913 if (AliTRDgtuParam::GetUseGTUconst()) {
44eafcf2 914 // averaging as in GTU
915 ULong64_t coeff;
916
917 // selection of coefficient for averaging
918 switch(track->GetTrackletMask()){
919 case 0x3f:
920 // 6 tracklets
921 coeff=0x5558; // approx. 1/6
922 break;
923
924 case 0x3e:
925 case 0x3d:
926 case 0x3b:
927 case 0x37:
928 case 0x2f:
929 case 0x1f:
930 // 5 tracklets
931 coeff=0x6666; // approx. 1/5
932 break;
933
934 default:
935 // 4 tracklets
936 coeff=0x8000; // = 0.25
4ff7ed2b 937 }
44eafcf2 938 coeff &= 0x1ffff; // 17-bit constant
939
940 ULong64_t sum = 0;
941 for (Int_t iLayer = 0; iLayer < fGtuParam->GetNLayers(); iLayer++) {
942 if ((track->GetTrackletMask() >> iLayer) & 1) {
943 sum += track->GetTracklet(iLayer)->GetPID();
944 }
945 }
946
a10c6981 947 sum = sum & 0x7ff;
948 ULong64_t prod = (sum * coeff) & 0xfffffffffull;
949 ULong64_t prodFinal = ((prod >> 17) + ((prod >> 16) & 1)) & 0xff;
44eafcf2 950
951 track->SetPID(prodFinal & 0xff);
952
953 return kTRUE;
954 }
955 else {
956
957 // simple averaging
958 Int_t nTracklets = 0;
959 Int_t pidSum = 0;
960 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
961 if (!track->IsTrackletInLayer(layer)) {
962 continue;
963 }
964 AliTRDtrackletGTU *trk = track->GetTracklet(layer);
965 pidSum += trk->GetPID();
966 nTracklets++;
967 }
ee373611 968
eaf6dbb0 969 if (nTracklets>0)
970 track->SetPID(pidSum/nTracklets);
ee373611 971 else
972 AliError("Track without contributing tracklets, no PID assigned");
44eafcf2 973
974 return kTRUE;
4ff7ed2b 975 }
4ff7ed2b 976}
977
5f006bd7 978Bool_t AliTRDgtuTMU::CalculateTrackParams(AliTRDtrackGTU *track)
52c19022 979{
980 // calculate the track parameters
981
982 if (!track) {
983 AliError("No track to calculate!");
984 return kFALSE;
985 }
986
987 Int_t a = 0;
988 Float_t b = 0;
989 Float_t c = 0;
990 Float_t x1;
991 Float_t x2;
992
c8b1590d 993 AliDebug(5,Form("There are %i tracklets in this track.", track->GetNTracklets()));
52c19022 994
995 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
996 if (!track->IsTrackletInLayer(layer)) {
997 continue;
998 }
999 AliTRDtrackletGTU *trk = track->GetTracklet(layer);
1000 if (!trk) {
1001 AliError(Form("Could not get tracklet in layer %i\n", layer));
1002 continue;
1003 }
a10c6981 1004 AliDebug(10,Form(" layer %i trk yprime: %6i, aki: %6i", layer, trk->GetYPrime(), (Int_t) (2048 * fGtuParam->GetAki(track->GetTrackletMask(), layer))));
5f006bd7 1005 a += (((Int_t) (2048 * fGtuParam->GetAki(track->GetTrackletMask(), layer))) * trk->GetYPrime() + 1) >> 8;
52c19022 1006 b += fGtuParam->GetBki(track->GetTrackletMask(), layer) * trk->GetYPrime() * fGtuParam->GetBinWidthY();
1007 c += fGtuParam->GetCki(track->GetTrackletMask(), layer) * trk->GetYPrime() * fGtuParam->GetBinWidthY();
1008 }
1009 if (a < 0)
1010 a += 3;
1011 a = a >> 2;
1012
52c19022 1013 track->SetFitParams(a, b, c);
1014
d2c8b010 1015 fGtuParam->GetIntersectionPoints(track->GetTrackletMask(), x1, x2);
1016
1017 AliDebug(5,Form(" Track parameters: a = %i, b = %f, c = %f, x1 = %f, x2 = %f, pt = %f (trkl mask: %i)", a, b, c, x1, x2, track->GetPt(), track->GetTrackletMask()));
1018
52c19022 1019 return kTRUE;
1020}
1021
52c19022 1022
6419bebb 1023Bool_t AliTRDgtuTMU::Uniquifier(const TList *inlist, TList *outlist)
52c19022 1024{
36dc3337 1025 // remove multiple occurences of the same track
1026
52c19022 1027 TIter next(inlist);
1028 AliTRDtrackGTU *trkStage0 = 0x0;
1029 AliTRDtrackGTU *trkStage1 = 0x0;
1030
1031 do {
1032 trkStage0 = (AliTRDtrackGTU*) next();
1033
36dc3337 1034 Bool_t tracksEqual = kFALSE;
52c19022 1035 if (trkStage0 != 0 && trkStage1 != 0) {
1036 for (Int_t layer = 0; layer < fGtuParam->GetNLayers(); layer++) {
1037 if (trkStage0->GetTrackletIndex(layer) != -1 && trkStage0->GetTrackletIndex(layer) == trkStage1->GetTrackletIndex(layer)) {
36dc3337 1038 tracksEqual = kTRUE;
52c19022 1039 break;
1040 }
1041 }
1042 }
1043
36dc3337 1044 if (tracksEqual) {
44eafcf2 1045 if (trkStage0->GetNTracklets() >= trkStage1->GetNTracklets())
52c19022 1046 trkStage1 = trkStage0;
5f006bd7 1047 }
52c19022 1048 else {
5f006bd7 1049 if (trkStage1 != 0x0)
52c19022 1050 outlist->Add(trkStage1);
1051 trkStage1 = trkStage0;
5f006bd7 1052 }
1053
52c19022 1054 } while (trkStage1 != 0x0);
1055 return kTRUE;
1056}