]>
Commit | Line | Data |
---|---|---|
3e87ef69 | 1 | // @(#) $Id$ |
2 | ||
f0f02577 | 3 | // Author: Anders Vestbo <mailto:vestbo@fi.uib.no> |
3e87ef69 | 4 | //*-- Copyright © ALICE HLT Group |
f0f02577 | 5 | |
e06900d5 | 6 | #include "AliL3StandardIncludes.h" |
7 | ||
f0f02577 | 8 | #include "AliL3Logging.h" |
9 | #include "AliL3HoughIntMerger.h" | |
10 | #include "AliL3HoughTrack.h" | |
11 | #include "AliL3TrackSegmentData.h" | |
12 | #include "AliL3Transform.h" | |
13 | #include "AliL3TrackArray.h" | |
14 | ||
0bd0c1ef | 15 | #if __GNUC__ == 3 |
e06900d5 | 16 | using namespace std; |
17 | #endif | |
18 | ||
f0f02577 | 19 | //_____________________________________________________________ |
20 | // Hough Inter merger | |
21 | // Merging of multiple reconstructed tracks | |
22 | ||
23 | ClassImp(AliL3HoughIntMerger) | |
24 | ||
0a86fbb7 | 25 | AliL3HoughIntMerger::AliL3HoughIntMerger() |
f0f02577 | 26 | { |
27 | //Default constructor | |
0a86fbb7 | 28 | InitMerger(1,"AliL3HoughTrack"); |
f0f02577 | 29 | fRowMax = fRowMin = 0; |
30 | SetParameters(0.001,0.05,10); | |
31 | Is2Global(kFALSE); | |
32 | } | |
33 | ||
34 | ||
35 | AliL3HoughIntMerger::~AliL3HoughIntMerger() | |
36 | { | |
37 | //Destructor | |
38 | ||
39 | } | |
40 | ||
41 | void AliL3HoughIntMerger::SetParameters(Double_t maxkappa, Double_t maxphi0, Double_t maxtgl) | |
42 | { | |
bd2f8772 | 43 | //Set merger params |
f0f02577 | 44 | fMaxKappa = maxkappa; |
45 | fMaxPhi0 = maxphi0; | |
46 | fMaxTgl = maxtgl; | |
47 | } | |
48 | ||
49 | void AliL3HoughIntMerger::FillTracks(AliL3TrackArray *tracks) | |
50 | { | |
bd2f8772 | 51 | //Fills tracks into merger |
f0f02577 | 52 | if(tracks->GetNTracks()==0) |
53 | LOG(AliL3Log::kWarning,"AliL3HoughIntMerger::FillTracks","Track Array") | |
54 | <<"Adding empty track array"<<ENDLOG; | |
55 | ||
56 | GetInTracks(0)->AddTracks(tracks,kFALSE);//Copy tracks | |
57 | printf("Filling %d tracks to intermerger\n",tracks->GetNTracks()); | |
58 | } | |
59 | ||
60 | Bool_t AliL3HoughIntMerger::IsTrack(AliL3Track *innertrack,AliL3Track *outertrack) | |
61 | { | |
62 | //Check if the tracks can be merged, called by the track merger | |
63 | ||
64 | AliL3HoughTrack *tr1 = (AliL3HoughTrack*)innertrack; | |
65 | AliL3HoughTrack *tr2 = (AliL3HoughTrack*)outertrack; | |
66 | ||
67 | if(abs(tr1->GetEtaIndex() - tr2->GetEtaIndex()) > 1) return kFALSE; | |
68 | if(tr1->GetCharge()!=tr2->GetCharge()) return kFALSE; | |
69 | if(fabs(tr1->GetKappa()-tr2->GetKappa()) >fMaxKappa) return kFALSE; | |
70 | if(fabs(tr1->GetPhi0()-tr2->GetPhi0()) > fMaxPhi0) return kFALSE; | |
71 | ||
72 | return kTRUE;//Tracks could be merged | |
73 | } | |
74 | ||
dd7d3870 | 75 | AliL3Track *AliL3HoughIntMerger::MultiMerge(AliL3TrackArray *mergedtrack,AliL3Track **tracks, Int_t /*ntrack*/) |
f0f02577 | 76 | { |
77 | //Called by the track merger | |
78 | ||
79 | AliL3HoughTrack *newtrack = (AliL3HoughTrack*)mergedtrack->NextTrack(); | |
80 | AliL3HoughTrack **trs = (AliL3HoughTrack**)tracks; | |
81 | ||
82 | AliL3HoughTrack *tpt=trs[0];//this is the "best" track | |
83 | //AliL3HoughTrack *tpl=trs[ntrack-1]; | |
84 | newtrack->Set(tpt); | |
85 | return (AliL3Track*)newtrack; | |
86 | } | |
87 | ||
88 | ||
89 | ||
90 | void AliL3HoughIntMerger::MMerge() | |
91 | { | |
bd2f8772 | 92 | //Track merging?? |
f0f02577 | 93 | GetInTracks(0)->QSort(); |
94 | while(Merge()); | |
95 | GetOutTracks()->AddTracks(GetInTracks(0)); | |
96 | } | |
97 | ||
98 | Int_t AliL3HoughIntMerger::Merge() | |
99 | { | |
bd2f8772 | 100 | //Track merging?? |
f0f02577 | 101 | AliL3TrackArray * tracks = GetInTracks(0); |
102 | const Int_t kNIn =tracks->GetNTracks(); | |
103 | AliL3Track *tr[2]; | |
104 | ||
105 | for(Int_t out=0;out<kNIn;out++) | |
106 | { | |
107 | AliL3HoughTrack *outertrack=(AliL3HoughTrack*)tracks->GetCheckedTrack(out); | |
108 | if(!outertrack) continue; | |
109 | for(Int_t in=0;in<kNIn;in++) | |
110 | { | |
111 | if(in==out) continue; | |
112 | AliL3HoughTrack *innertrack=(AliL3HoughTrack*)tracks->GetCheckedTrack(in); | |
113 | if(!innertrack) continue; | |
114 | if(IsTrack(innertrack,outertrack)) | |
115 | { | |
116 | tr[0]=innertrack; | |
117 | tr[1]=outertrack; | |
118 | SortTracks(tr,2); | |
119 | Print(tr); | |
120 | MultiMerge(tracks,tr,2); | |
121 | tracks->Remove(out); | |
122 | tracks->Remove(in); | |
123 | break; | |
124 | } | |
125 | } | |
126 | } | |
127 | Int_t nmerged = tracks->GetNTracks()-kNIn; | |
128 | LOG(AliL3Log::kInformational,"AliL3HoughIntMerger::Merge","Result") | |
129 | <<AliL3Log::kDec<<"Merged Tracks: "<<nmerged<<ENDLOG; | |
130 | ||
131 | //add in tracks | |
132 | //GetOutTracks()->AddTracks(GetInTracks(0)); | |
133 | ||
134 | return nmerged; | |
135 | } | |
136 | ||
137 | void AliL3HoughIntMerger::Print(AliL3Track **tracks) | |
138 | { | |
bd2f8772 | 139 | //Prints merger results |
f0f02577 | 140 | AliL3HoughTrack *tr1 = (AliL3HoughTrack*)tracks[0]; |
141 | AliL3HoughTrack *tr2 = (AliL3HoughTrack*)tracks[1]; | |
142 | Double_t kappadiff = fabs(tr1->GetKappa()-tr2->GetKappa()); | |
143 | Double_t phi0diff = fabs(tr1->GetPhi0()-tr2->GetPhi0()); | |
144 | cout << "---------Difference in intermerged tracks---------"<<endl; | |
145 | cout << "Kappa: "<<kappadiff<<" Phi0 : "<<phi0diff<<endl; | |
146 | ||
147 | } | |
148 | ||
149 | void AliL3HoughIntMerger::SortTracks(AliL3Track **trs, Int_t ntrack) | |
150 | { | |
151 | //Sort the tracks according to their weight | |
152 | ||
153 | AliL3HoughTrack **tracks = (AliL3HoughTrack**)trs; | |
154 | AliL3HoughTrack **tmp = new AliL3HoughTrack*[ntrack]; | |
155 | for(Int_t i=0;i<ntrack;i++) tmp[i] = (AliL3HoughTrack*)tracks[i]; | |
156 | Int_t *t = new Int_t[ntrack]; | |
157 | for(Int_t i=0;i<ntrack;i++) t[i]=-1; | |
158 | ||
159 | for(Int_t j=0;j<ntrack;j++) | |
160 | { | |
161 | Double_t maxw=0; | |
162 | Int_t maxi=0; | |
163 | for(Int_t i=0;i<ntrack;i++) | |
164 | { | |
165 | if(!tracks[i]) continue; | |
166 | if(tracks[i]->GetWeight() > maxw) | |
167 | { | |
168 | maxw=tracks[i]->GetWeight(); | |
169 | maxi=i; | |
170 | } | |
171 | } | |
172 | t[j]=maxi; | |
173 | tracks[maxi]=0; | |
174 | } | |
175 | for(Int_t i=0;i<ntrack;i++) tracks[i] = tmp[t[i]]; | |
176 | delete[] t; | |
177 | delete[] tmp; | |
178 | } |