New design of tracking classes (Yu.Belikov)
[u/mrichter/AliRoot.git] / STEER / AliTracker.cxx
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 //-------------------------------------------------------------------------
17 //               Implementation of the AliTracker class
18 //
19 //          Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
20 //-------------------------------------------------------------------------
21 #include <TMath.h>
22
23 #include "AliTracker.h"
24 #include "AliCluster.h"
25 #include "AliKalmanTrack.h"
26
27 ClassImp(AliTracker)
28
29 //__________________________________________________________________________
30 void AliTracker::CookLabel(AliKalmanTrack *t, Float_t wrong) const {
31   //--------------------------------------------------------------------
32   //This function "cooks" a track label. If label<0, this track is fake.
33   //--------------------------------------------------------------------
34   Int_t noc=t->GetNumberOfClusters();
35   Int_t *lb=new Int_t[noc];
36   Int_t *mx=new Int_t[noc];
37   AliCluster **clusters=new AliCluster*[noc];
38
39   Int_t i;
40   for (i=0; i<noc; i++) {
41      lb[i]=mx[i]=0;
42      Int_t index=t->GetClusterIndex(i);
43      clusters[i]=GetCluster(index);
44   }
45
46   Int_t lab=123456789;
47   for (i=0; i<noc; i++) {
48     AliCluster *c=clusters[i];
49     lab=TMath::Abs(c->GetLabel(0));
50     Int_t j;
51     for (j=0; j<noc; j++) if (lb[j]==lab || mx[j]==0) break;
52     lb[j]=lab;
53     (mx[j])++;
54   }
55
56   Int_t max=0;
57   for (i=0; i<noc; i++) if (mx[i]>max) {max=mx[i]; lab=lb[i];}
58     
59   for (i=0; i<noc; i++) {
60     AliCluster *c=clusters[i];
61     if (TMath::Abs(c->GetLabel(1)) == lab ||
62         TMath::Abs(c->GetLabel(2)) == lab ) max++;
63   }
64
65   if ((1.- Float_t(max)/noc) > wrong) lab=-lab;
66   t->SetLabel(lab);
67
68   delete[] lb;
69   delete[] mx;
70   delete[] clusters;
71 }
72
73 //____________________________________________________________________________
74 void AliTracker::UseClusters(const AliKalmanTrack *t, Int_t from) const {
75   //------------------------------------------------------------------
76   //This function marks clusters associated with the track.
77   //------------------------------------------------------------------
78   Int_t noc=t->GetNumberOfClusters();
79   for (Int_t i=from; i<noc; i++) {
80      Int_t index=t->GetClusterIndex(i);
81      AliCluster *c=GetCluster(index); 
82      c->Use();   
83   }
84 }
85
86