]> git.uio.no Git - u/mrichter/AliRoot.git/blame - HLT/hough/AliL3Hough.cxx
Moved to the latest version of the HLT code in Bergen.
[u/mrichter/AliRoot.git] / HLT / hough / AliL3Hough.cxx
CommitLineData
3e87ef69 1// @(#) $Id$
b1886074 2
3// Author: Anders Vestbo <mailto:vestbo@fi.uib.no>
3e87ef69 4//*-- Copyright &copy ALICE HLT Group
b1886074 5
e06900d5 6#include "AliL3StandardIncludes.h"
0309a5ee 7#include <sys/time.h>
f000f8a5 8
e06900d5 9#include "AliL3Logging.h"
b1886074 10#include "AliL3HoughMerger.h"
11#include "AliL3HoughIntMerger.h"
12#include "AliL3HoughGlobalMerger.h"
f80b98cb 13#include "AliL3Histogram.h"
f000f8a5 14#include "AliL3Hough.h"
15#include "AliL3HoughTransformer.h"
6c97129d 16#include "AliL3HoughClusterTransformer.h"
636080ea 17#include "AliL3HoughTransformerLUT.h"
b46b53c1 18#include "AliL3HoughTransformerVhdl.h"
f000f8a5 19#include "AliL3HoughMaxFinder.h"
3e87ef69 20#include "AliL3Benchmark.h"
95a00d93 21#ifdef use_aliroot
f80b98cb 22#include "AliL3FileHandler.h"
95a00d93 23#else
24#include "AliL3MemHandler.h"
25#endif
d96f6a4a 26#include "AliL3DataHandler.h"
f80b98cb 27#include "AliL3DigitData.h"
28#include "AliL3HoughEval.h"
29#include "AliL3Transform.h"
f80b98cb 30#include "AliL3TrackArray.h"
31#include "AliL3HoughTrack.h"
b2a02bce 32#include "AliL3DDLDataFileHandler.h"
95a00d93 33
e06900d5 34#if GCCVERSION == 3
35using namespace std;
36#endif
b1886074 37
aa641eb8 38/** /class AliL3Hough
39//<pre>
b1886074 40//_____________________________________________________________
41// AliL3Hough
42//
237d3f5c 43// Interface class for the Hough transform
b1886074 44//
237d3f5c 45// Example how to use:
46//
47// AliL3Hough *hough = new AliL3Hough(path,kTRUE,NumberOfEtaSegments);
48// hough->ReadData(slice);
49// hough->Transform();
50// hough->FindTrackCandidates();
51//
52// AliL3TrackArray *tracks = hough->GetTracks(patch);
aa641eb8 53//</pre>
54*/
f000f8a5 55
56ClassImp(AliL3Hough)
57
58AliL3Hough::AliL3Hough()
59{
b1886074 60 //Constructor
61
6dbc57b4 62 fBinary = kFALSE;
a6008206 63 fAddHistograms = kFALSE;
6dbc57b4 64 fDoIterative = kFALSE;
65 fWriteDigits = kFALSE;
66 fUse8bits = kFALSE;
67
68 fMemHandler = 0;
237d3f5c 69 fHoughTransformer = 0;
6dbc57b4 70 fEval = 0;
71 fPeakFinder = 0;
72 fTracks = 0;
b2a02bce 73 fGlobalTracks = 0;
6dbc57b4 74 fMerger = 0;
75 fInterMerger = 0;
76 fGlobalMerger = 0;
3e87ef69 77 fBenchmark = 0;
78
6dbc57b4 79 fNEtaSegments = 0;
80 fNPatches = 0;
81 fVersion = 0;
82 fCurrentSlice = 0;
b2a02bce 83 fEvent = 0;
84
6dbc57b4 85 SetTransformerParams();
86 SetThreshold();
636080ea 87 SetNSaveIterations();
3e87ef69 88 SetPeakThreshold();
1f1942b8 89#ifdef use_aliroot
90 //just be sure that index is empty for new event
91 AliL3FileHandler::CleanStaticIndex();
92#endif
f000f8a5 93}
94
b2a02bce 95AliL3Hough::AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments,Bool_t bit8,Int_t tv,Char_t *infile)
b46b53c1 96{
b46b53c1 97 fBinary = binary;
98 strcpy(fPath,path);
6dbc57b4 99 fNEtaSegments = n_eta_segments;
b46b53c1 100 fAddHistograms = kFALSE;
6dbc57b4 101 fDoIterative = kFALSE;
102 fWriteDigits = kFALSE;
103 fUse8bits = bit8;
104 fVersion = tv;
b2a02bce 105 if(!fBinary)
106 fInputFile = infile;
107 else
108 fInputFile = 0;
1f1942b8 109
110#ifdef use_aliroot
111 //just be sure that index is empty for new event
112 AliL3FileHandler::CleanStaticIndex();
113#endif
b46b53c1 114}
f000f8a5 115
f000f8a5 116AliL3Hough::~AliL3Hough()
117{
237d3f5c 118 //dtor
119
b1886074 120 CleanUp();
121 if(fMerger)
122 delete fMerger;
1f1942b8 123 //cout << "Cleaned class merger " << endl;
b1886074 124 if(fInterMerger)
125 delete fInterMerger;
1f1942b8 126 //cout << "Cleaned class inter " << endl;
a6008206 127 if(fPeakFinder)
128 delete fPeakFinder;
1f1942b8 129 //cout << "Cleaned class peak " << endl;
1c404dd5 130 if(fGlobalMerger)
131 delete fGlobalMerger;
1f1942b8 132 //cout << "Cleaned class global " << endl;
3e87ef69 133 if(fBenchmark)
134 delete fBenchmark;
1f1942b8 135 //cout << "Cleaned class bench " << endl;
b2a02bce 136 if(fGlobalTracks)
137 delete fGlobalTracks;
1f1942b8 138 //cout << "Cleaned class globaltracks " << endl;
f000f8a5 139}
140
b1886074 141void AliL3Hough::CleanUp()
f000f8a5 142{
b1886074 143 //Cleanup memory
144
1c404dd5 145 for(Int_t i=0; i<fNPatches; i++)
4fc9a6a4 146 {
b1886074 147 if(fTracks[i]) delete fTracks[i];
1f1942b8 148 //cout << "Cleaned tracks " << i << endl;
b1886074 149 if(fEval[i]) delete fEval[i];
1f1942b8 150 //cout << "Cleaned eval " << i << endl;
b1886074 151 if(fHoughTransformer[i]) delete fHoughTransformer[i];
1f1942b8 152 //cout << "Cleaned traf " << i << endl;
b1886074 153 if(fMemHandler[i]) delete fMemHandler[i];
1f1942b8 154 //cout << "Cleaned mem " << i << endl;
4fc9a6a4 155 }
b1886074 156
3e87ef69 157 if(fTracks) delete [] fTracks;
1f1942b8 158 //cout << "Cleaned class tracks " << endl;
3e87ef69 159 if(fEval) delete [] fEval;
1f1942b8 160 //cout << "Cleaned class eval " << endl;
3e87ef69 161 if(fHoughTransformer) delete [] fHoughTransformer;
1f1942b8 162 //cout << "Cleaned cleass trafo " << endl;
3e87ef69 163 if(fMemHandler) delete [] fMemHandler;
1f1942b8 164 //cout << "Cleaned class mem " << endl;
f000f8a5 165}
166
b2a02bce 167void AliL3Hough::Init(Char_t *path,Bool_t binary,Int_t n_eta_segments,Bool_t bit8,Int_t tv,Char_t *infile)
f000f8a5 168{
d96f6a4a 169 fBinary = binary;
170 strcpy(fPath,path);
171 fNEtaSegments = n_eta_segments;
6dbc57b4 172 fWriteDigits = kFALSE;
173 fUse8bits = bit8;
174 fVersion = tv;
b2a02bce 175 if(!fBinary)
176 fInputFile = infile;
177 else
178 fInputFile = 0;
6dbc57b4 179
180 Init(); //do the rest
181}
182
e06900d5 183void AliL3Hough::Init(Bool_t doit, Bool_t addhists)
a4639de2 184{
6dbc57b4 185 fDoIterative = doit;
186 fAddHistograms = addhists;
b46b53c1 187
26abc209 188 fNPatches = AliL3Transform::GetNPatches();
3e87ef69 189
237d3f5c 190 fHoughTransformer = new AliL3HoughBaseTransformer*[fNPatches];
95a00d93 191 fMemHandler = new AliL3MemHandler*[fNPatches];
3e87ef69 192
1c404dd5 193 fTracks = new AliL3TrackArray*[fNPatches];
194 fEval = new AliL3HoughEval*[fNPatches];
b2a02bce 195
196 fGlobalTracks = new AliL3TrackArray("AliL3HoughTrack");
197
1c404dd5 198 for(Int_t i=0; i<fNPatches; i++)
f80b98cb 199 {
6dbc57b4 200 switch (fVersion){ //choose Transformer
b46b53c1 201 case 1:
636080ea 202 fHoughTransformer[i] = new AliL3HoughTransformerLUT(0,i,fNEtaSegments);
b46b53c1 203 break;
6c97129d 204 case 2:
205 fHoughTransformer[i] = new AliL3HoughClusterTransformer(0,i,fNEtaSegments);
206 break;
636080ea 207 case 3:
208 fHoughTransformer[i] = new AliL3HoughTransformerVhdl(0,i,fNEtaSegments,fNSaveIterations);
209 break;
b46b53c1 210 default:
3e87ef69 211 fHoughTransformer[i] = new AliL3HoughTransformer(0,i,fNEtaSegments,kFALSE,kFALSE);
b46b53c1 212 }
aa641eb8 213
b2a02bce 214 fHoughTransformer[i]->CreateHistograms(fNBinX[i],fLowPt[i],fNBinY[i],-fPhi[i],fPhi[i]);
215 //fHoughTransformer[i]->CreateHistograms(fLowPt[i],fUpperPt[i],fPtRes[i],fNBinY[i],fPhi[i]);
1f1942b8 216
b2a02bce 217 fHoughTransformer[i]->SetLowerThreshold(fThreshold[i]);
218 fHoughTransformer[i]->SetUpperThreshold(100);
219
0309a5ee 220 LOG(AliL3Log::kInformational,"AliL3Hough::Init","Version")
6dbc57b4 221 <<"Initializing Hough transformer version "<<fVersion<<ENDLOG;
a4639de2 222
b1886074 223 fEval[i] = new AliL3HoughEval();
224 fTracks[i] = new AliL3TrackArray("AliL3HoughTrack");
d96f6a4a 225 if(fUse8bits)
226 fMemHandler[i] = new AliL3DataHandler();
227 else
95a00d93 228#ifdef use_aliroot
d96f6a4a 229 {
b2a02bce 230 if(!fInputFile) {
231 /* In case of reading digits file */
1f1942b8 232 fMemHandler[i] = new AliL3FileHandler(kTRUE); //use static index
b2a02bce 233 if(!fBinary) {
44c7f8de 234 Char_t filename[1024];
b490510d 235 sprintf(filename,"%s/digitfile.root",fPath);
b2a02bce 236 fMemHandler[i]->SetAliInput(filename);
3bb06991 237 }
b2a02bce 238 }
239 else {
240 /* In case of reading rawdata from ROOT file */
241 fMemHandler[i] = new AliL3DDLDataFileHandler();
242 fMemHandler[i]->SetReaderInput(fInputFile);
243 }
d96f6a4a 244 }
95a00d93 245#else
246 fMemHandler[i] = new AliL3MemHandler();
247#endif
a6008206 248 }
6dbc57b4 249
a4639de2 250 fPeakFinder = new AliL3HoughMaxFinder("KappaPhi",1000);
1c404dd5 251 fMerger = new AliL3HoughMerger(fNPatches);
b1886074 252 fInterMerger = new AliL3HoughIntMerger();
237d3f5c 253 fGlobalMerger = 0;
3e87ef69 254 fBenchmark = new AliL3Benchmark();
b2a02bce 255}
256
257void AliL3Hough::SetTransformerParams(Float_t ptres,Float_t ptmin,Float_t ptmax,Int_t ny,Int_t patch)
258{
259
260 Int_t mrow;
261 Float_t psi=0;
262 if(patch==-1)
263 mrow = 80;
264 else
265 mrow = AliL3Transform::GetLastRow(patch);
266 if(ptmin)
267 {
268 Double_t lineradius = sqrt(pow(AliL3Transform::Row2X(mrow),2) + pow(AliL3Transform::GetMaxY(mrow),2));
269 Double_t kappa = -1*AliL3Transform::GetBField()*AliL3Transform::GetBFact()/ptmin;
270 psi = AliL3Transform::Deg2Rad(10) - asin(lineradius*kappa/2);
271 cout<<"Calculated psi range "<<psi<<" in patch "<<patch<<endl;
272 }
273
274 if(patch==-1)
275 {
276 Int_t i=0;
277 while(i < 6)
278 {
279 fPtRes[i] = ptres;
280 fLowPt[i] = ptmin;
281 fUpperPt[i] = ptmax;
282 fNBinY[i] = ny;
283 fPhi[i] = psi;
284 fNBinX[i]=0;
285 i++;
286 }
287 return;
288 }
289
290 fPtRes[patch] = ptres;
291 fLowPt[patch] = ptmin;
292 fUpperPt[patch] = ptmax;
293 fNBinY[patch] = ny;
294 fPhi[patch] = psi;
295}
296
297void AliL3Hough::SetTransformerParams(Int_t nx,Int_t ny,Float_t ptmin,Int_t patch)
298{
299
300 Int_t mrow=80;
301 Double_t lineradius = sqrt(pow(AliL3Transform::Row2X(mrow),2) + pow(AliL3Transform::GetMaxY(mrow),2));
302 Double_t kappa = -1*AliL3Transform::GetBField()*AliL3Transform::GetBFact()/ptmin;
303 Double_t psi = AliL3Transform::Deg2Rad(10) - asin(lineradius*kappa/2);
304 cout<<"Calculated psi range "<<psi<<" in patch "<<patch<<endl;
305
306 Int_t i=0;
307 while(i < 6)
308 {
309 fLowPt[i] = ptmin;
310 fNBinY[i] = ny;
311 fNBinX[i] = nx;
312 fPhi[i] = psi;
313 i++;
314 }
315}
316
317void AliL3Hough::SetTransformerParams(Int_t nx,Int_t ny,Float_t lpt,Float_t phi)
318{
319 Int_t i=0;
320 while(i < 6)
321 {
322 fLowPt[i] = lpt;
323 fNBinY[i] = ny;
324 fNBinX[i] = nx;
325 fPhi[i] = phi;
326 i++;
327 }
328}
329
330void AliL3Hough::SetThreshold(Int_t t3,Int_t patch)
331{
332 if(patch==-1)
333 {
334 Int_t i=0;
335 while(i < 6)
336 fThreshold[i++]=t3;
337 return;
338 }
339 fThreshold[patch]=t3;
340}
3e87ef69 341
b2a02bce 342void AliL3Hough::SetPeakThreshold(Int_t threshold,Int_t patch)
343{
344 if(patch==-1)
345 {
346 Int_t i=0;
347 while(i < 6)
348 fPeakThreshold[i++]=threshold;
349 return;
350 }
351 fPeakThreshold[patch]=threshold;
3e87ef69 352}
353
354void AliL3Hough::DoBench(Char_t *name)
355{
356 fBenchmark->Analyze(name);
a6008206 357}
358
359void AliL3Hough::Process(Int_t minslice,Int_t maxslice)
360{
361 //Process all slices [minslice,maxslice].
1c404dd5 362 fGlobalMerger = new AliL3HoughGlobalMerger(minslice,maxslice);
b1886074 363
a6008206 364 for(Int_t i=minslice; i<=maxslice; i++)
365 {
b1886074 366 ReadData(i);
367 Transform();
a6008206 368 if(fAddHistograms)
369 AddAllHistograms();
370 FindTrackCandidates();
3e87ef69 371 //Evaluate();
372 //fGlobalMerger->FillTracks(fTracks[0],i);
f80b98cb 373 }
4fc9a6a4 374}
f80b98cb 375
e06900d5 376void AliL3Hough::ReadData(Int_t slice,Int_t eventnr)
4fc9a6a4 377{
b1886074 378 //Read data from files, binary or root.
3bb06991 379
1f1942b8 380#ifdef use_aliroot
381 if(fEvent!=eventnr) //just be sure that index is empty for new event
382 AliL3FileHandler::CleanStaticIndex();
383#endif
b2a02bce 384 fEvent=eventnr;
3bb06991 385 fCurrentSlice = slice;
1f1942b8 386
1c404dd5 387 for(Int_t i=0; i<fNPatches; i++)
f80b98cb 388 {
4fc9a6a4 389 fMemHandler[i]->Free();
390 UInt_t ndigits=0;
391 AliL3DigitRowData *digits =0;
392 Char_t name[256];
3fe49b5b 393 fMemHandler[i]->Init(slice,i);
a6008206 394 if(fBinary)//take input data from binary files
4fc9a6a4 395 {
d96f6a4a 396 if(fUse8bits)
b2a02bce 397 sprintf(name,"%s/binaries/digits_c8_%d_%d_%d.raw",fPath,eventnr,slice,i);
d96f6a4a 398 else
b2a02bce 399 sprintf(name,"%s/binaries/digits_%d_%d_%d.raw",fPath,eventnr,slice,i);
636080ea 400
4fc9a6a4 401 fMemHandler[i]->SetBinaryInput(name);
402 digits = (AliL3DigitRowData *)fMemHandler[i]->CompBinary2Memory(ndigits);
403 fMemHandler[i]->CloseBinaryInput();
404 }
405 else //read data from root file
406 {
95a00d93 407#ifdef use_aliroot
3e87ef69 408 digits=(AliL3DigitRowData *)fMemHandler[i]->AliAltroDigits2Memory(ndigits,eventnr);
95a00d93 409#else
410 cerr<<"You cannot read from rootfile now"<<endl;
411#endif
4fc9a6a4 412 }
aa641eb8 413
636080ea 414 //set input data and init transformer
415 fHoughTransformer[i]->SetInputData(ndigits,digits);
416 fHoughTransformer[i]->Init(slice,i,fNEtaSegments);
b1886074 417 }
418}
419
3e87ef69 420void AliL3Hough::Transform(Int_t *row_range)
b1886074 421{
422 //Transform all data given to the transformer within the given slice
423 //(after ReadData(slice))
0309a5ee 424
425 Double_t initTime,cpuTime;
426 initTime = GetCpuTime();
1c404dd5 427 for(Int_t i=0; i<fNPatches; i++)
b1886074 428 {
429 fHoughTransformer[i]->Reset();//Reset the histograms
3e87ef69 430 fBenchmark->Start("Hough Transform");
431 if(!row_range)
237d3f5c 432 fHoughTransformer[i]->TransformCircle();
433 else
3e87ef69 434 fHoughTransformer[i]->TransformCircleC(row_range,1);
435 fBenchmark->Stop("Hough Transform");
b1886074 436 }
0309a5ee 437 cpuTime = GetCpuTime() - initTime;
438 LOG(AliL3Log::kInformational,"AliL3Hough::Transform()","Timing")
439 <<"Transform done in average per patch of "<<cpuTime*1000/fNPatches<<" ms"<<ENDLOG;
b1886074 440}
441
442void AliL3Hough::MergePatches()
443{
444 if(fAddHistograms) //Nothing to merge here
445 return;
b1886074 446 fMerger->MergePatches(kTRUE);
b1886074 447}
448
449void AliL3Hough::MergeInternally()
450{
451 if(fAddHistograms)
452 fInterMerger->FillTracks(fTracks[0]);
453 else
454 fInterMerger->FillTracks(fMerger->GetOutTracks());
455
456 fInterMerger->MMerge();
457}
458
459void AliL3Hough::ProcessSliceIter()
460{
461 //Process current slice (after ReadData(slice)) iteratively.
462
6c97129d 463 if(!fAddHistograms)
b1886074 464 {
6c97129d 465 for(Int_t i=0; i<fNPatches; i++)
466 {
467 ProcessPatchIter(i);
468 fMerger->FillTracks(fTracks[i],i); //Copy tracks to merger
469 }
470 }
471 else
472 {
473 for(Int_t i=0; i<10; i++)
474 {
475 Transform();
476 AddAllHistograms();
477 InitEvaluate();
478 AliL3HoughBaseTransformer *tr = fHoughTransformer[0];
479 for(Int_t j=0; j<fNEtaSegments; j++)
480 {
481 AliL3Histogram *hist = tr->GetHistogram(j);
482 if(hist->GetNEntries()==0) continue;
483 fPeakFinder->Reset();
484 fPeakFinder->SetHistogram(hist);
485 fPeakFinder->FindAbsMaxima();
486 AliL3HoughTrack *track = (AliL3HoughTrack*)fTracks[0]->NextTrack();
487 track->SetTrackParameters(fPeakFinder->GetXPeak(0),fPeakFinder->GetYPeak(0),fPeakFinder->GetWeight(0));
488 track->SetEtaIndex(j);
489 track->SetEta(tr->GetEta(j,fCurrentSlice));
490 for(Int_t k=0; k<fNPatches; k++)
491 {
492 fEval[i]->SetNumOfPadsToLook(2);
493 fEval[i]->SetNumOfRowsToMiss(2);
494 fEval[i]->RemoveFoundTracks();
3e87ef69 495 /*
6c97129d 496 Int_t nrows=0;
497 if(!fEval[i]->LookInsideRoad(track,nrows))
498 {
499 fTracks[0]->Remove(fTracks[0]->GetNTracks()-1);
500 fTracks[0]->Compress();
501 }
3e87ef69 502 */
6c97129d 503 }
504 }
505
506 }
507
f80b98cb 508 }
4fc9a6a4 509}
510
b1886074 511void AliL3Hough::ProcessPatchIter(Int_t patch)
512{
513 //Process patch in a iterative way.
514 //transform + peakfinding + evaluation + transform +...
515
6c97129d 516 Int_t num_of_tries = 5;
237d3f5c 517 AliL3HoughBaseTransformer *tr = fHoughTransformer[patch];
b1886074 518 AliL3TrackArray *tracks = fTracks[patch];
519 tracks->Reset();
520 AliL3HoughEval *ev = fEval[patch];
521 ev->InitTransformer(tr);
6c97129d 522 //ev->RemoveFoundTracks();
523 ev->SetNumOfRowsToMiss(3);
b5a207b4 524 ev->SetNumOfPadsToLook(2);
b1886074 525 AliL3Histogram *hist;
526 for(Int_t t=0; t<num_of_tries; t++)
527 {
528 tr->Reset();
529 tr->TransformCircle();
530 for(Int_t i=0; i<fNEtaSegments; i++)
531 {
532 hist = tr->GetHistogram(i);
533 if(hist->GetNEntries()==0) continue;
6c97129d 534 fPeakFinder->Reset();
b1886074 535 fPeakFinder->SetHistogram(hist);
6c97129d 536 fPeakFinder->FindAbsMaxima();
537 //fPeakFinder->FindPeak1();
b1886074 538 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->NextTrack();
6c97129d 539 track->SetTrackParameters(fPeakFinder->GetXPeak(0),fPeakFinder->GetYPeak(0),fPeakFinder->GetWeight(0));
540 track->SetEtaIndex(i);
541 track->SetEta(tr->GetEta(i,fCurrentSlice));
3e87ef69 542 /*
6c97129d 543 Int_t nrows=0;
544 if(!ev->LookInsideRoad(track,nrows))
b1886074 545 {
546 tracks->Remove(tracks->GetNTracks()-1);
547 tracks->Compress();
548 }
3e87ef69 549 */
b1886074 550 }
551 }
6c97129d 552 fTracks[0]->QSort();
b1886074 553 LOG(AliL3Log::kInformational,"AliL3Hough::ProcessPatch","NTracks")
554 <<AliL3Log::kDec<<"Found "<<tracks->GetNTracks()<<" tracks in patch "<<patch<<ENDLOG;
555}
556
a6008206 557void AliL3Hough::AddAllHistograms()
4fc9a6a4 558{
a6008206 559 //Add the histograms within one etaslice.
560 //Resulting histogram are in patch=0.
b1886074 561
0309a5ee 562 Double_t initTime,cpuTime;
563 initTime = GetCpuTime();
3e87ef69 564 fBenchmark->Start("Add Histograms");
a6008206 565 for(Int_t i=0; i<fNEtaSegments; i++)
566 {
567 AliL3Histogram *hist0 = fHoughTransformer[0]->GetHistogram(i);
1c404dd5 568 for(Int_t j=1; j<fNPatches; j++)
a6008206 569 {
570 AliL3Histogram *hist = fHoughTransformer[j]->GetHistogram(i);
571 hist0->Add(hist);
572 }
573 }
3e87ef69 574 fBenchmark->Stop("Add Histograms");
b1886074 575 fAddHistograms = kTRUE;
0309a5ee 576 cpuTime = GetCpuTime() - initTime;
577 LOG(AliL3Log::kInformational,"AliL3Hough::AddAllHistograms()","Timing")
578 <<"Adding histograms in "<<cpuTime*1000<<" ms"<<ENDLOG;
a6008206 579}
580
b2a02bce 581void AliL3Hough::AddTracks()
582{
583 if(!fTracks[0])
584 {
585 cerr<<"AliL3Hough::AddTracks : No tracks"<<endl;
586 return;
587 }
588 AliL3TrackArray *tracks = fTracks[0];
589 for(Int_t i=0; i<tracks->GetNTracks(); i++)
590 {
591 AliL3Track *track = tracks->GetCheckedTrack(i);
592 if(!track) continue;
593 if(track->GetNHits()!=1) cerr<<"NHITS "<<track->GetNHits()<<endl;
594 UInt_t *ids = track->GetHitNumbers();
595 ids[0] = (fCurrentSlice&0x7f)<<25;
596 }
597
598 fGlobalTracks->AddTracks(fTracks[0],0,fCurrentSlice);
599}
600
a6008206 601void AliL3Hough::FindTrackCandidates()
602{
603 //Look for peaks in histograms, and find the track candidates
604
a6008206 605 Int_t n_patches;
606 if(fAddHistograms)
b46b53c1 607 n_patches = 1; //Histograms have been added.
a6008206 608 else
1c404dd5 609 n_patches = fNPatches;
b1886074 610
0309a5ee 611 Double_t initTime,cpuTime;
612 initTime = GetCpuTime();
3e87ef69 613 fBenchmark->Start("Find Maxima");
a6008206 614 for(Int_t i=0; i<n_patches; i++)
615 {
237d3f5c 616 AliL3HoughBaseTransformer *tr = fHoughTransformer[i];
b1886074 617 fTracks[i]->Reset();
6c97129d 618
a6008206 619 for(Int_t j=0; j<fNEtaSegments; j++)
620 {
621 AliL3Histogram *hist = tr->GetHistogram(j);
b1886074 622 if(hist->GetNEntries()==0) continue;
3fe49b5b 623 fPeakFinder->Reset();
a6008206 624 fPeakFinder->SetHistogram(hist);
045549b7 625
b2a02bce 626 fPeakFinder->SetThreshold(fPeakThreshold[i]);
045549b7 627 fPeakFinder->FindAdaptedPeaks(6,0.5);
628
b2a02bce 629 //fPeakFinder->FindMaxima(fPeakThreshold[i]); //Simple maxima finder
630
3fe49b5b 631 for(Int_t k=0; k<fPeakFinder->GetEntries(); k++)
a6008206 632 {
b1886074 633 AliL3HoughTrack *track = (AliL3HoughTrack*)fTracks[i]->NextTrack();
3fe49b5b 634 track->SetTrackParameters(fPeakFinder->GetXPeak(k),fPeakFinder->GetYPeak(k),fPeakFinder->GetWeight(k));
a6008206 635 track->SetEtaIndex(j);
afd8fed4 636 track->SetEta(tr->GetEta(j,fCurrentSlice));
26abc209 637 track->SetRowRange(AliL3Transform::GetFirstRow(0),AliL3Transform::GetLastRow(5));
a6008206 638 }
639 }
b2a02bce 640 cout<<"Found "<<fTracks[i]->GetNTracks()<<" tracks in patch "<<i<<endl;
b1886074 641 fTracks[i]->QSort();
a6008206 642 }
3e87ef69 643 fBenchmark->Stop("Find Maxima");
0309a5ee 644 cpuTime = GetCpuTime() - initTime;
645 LOG(AliL3Log::kInformational,"AliL3Hough::FindTrackCandidates()","Timing")
646 <<"Maxima finding done in "<<cpuTime*1000<<" ms"<<ENDLOG;
a6008206 647}
648
3fe49b5b 649void AliL3Hough::InitEvaluate()
650{
651 //Pass the transformer objects to the AliL3HoughEval objects:
652 //This will provide the evaluation objects with all the necessary
208b54c5 653 //data and parameters it needs.
3fe49b5b 654
655 for(Int_t i=0; i<fNPatches; i++)
656 fEval[i]->InitTransformer(fHoughTransformer[i]);
657}
658
0309a5ee 659Int_t AliL3Hough::Evaluate(Int_t road_width,Int_t nrowstomiss)
a6008206 660{
661 //Evaluate the tracks, by looking along the road in the raw data.
3fe49b5b 662 //If track does not cross all padrows - rows2miss, it is removed from the arrray.
0309a5ee 663 //If histograms were not added, the check is done locally in patch,
664 //meaning that nrowstomiss is the number of padrows the road can miss with respect
665 //to the number of rows in the patch.
666 //If the histograms were added, the comparison is done globally in the _slice_,
667 //meaing that nrowstomiss is the number of padrows the road can miss with
668 //respect to the total number of padrows in the slice.
669 //
670 //Return value = number of tracks which were removed (only in case of fAddHistograms)
a6008206 671
b1886074 672 if(!fTracks[0])
a6008206 673 {
b1886074 674 LOG(AliL3Log::kError,"AliL3Hough::Evaluate","Track Array")
675 <<"No tracks to work with..."<<ENDLOG;
0309a5ee 676 return 0;
a6008206 677 }
678
0309a5ee 679 Int_t removed_tracks=0;
680 AliL3TrackArray *tracks=0;
a4639de2 681
0309a5ee 682 if(fAddHistograms)
a4639de2 683 {
0309a5ee 684 tracks = fTracks[0];
0309a5ee 685 for(Int_t i=0; i<tracks->GetNTracks(); i++)
a6008206 686 {
a4639de2 687 AliL3Track *track = tracks->GetCheckedTrack(i);
688 if(!track) continue;
689 track->SetNHits(0);
0309a5ee 690 }
f80b98cb 691 }
f000f8a5 692
a4639de2 693 for(Int_t i=0; i<fNPatches; i++)
694 EvaluatePatch(i,road_width,nrowstomiss);
695
aa641eb8 696 //Here we check the tracks globally;
697 //how many good rows (padrows with signal)
698 //did it cross in the slice
699 if(fAddHistograms)
0309a5ee 700 {
701 for(Int_t j=0; j<tracks->GetNTracks(); j++)
702 {
a4639de2 703 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->GetCheckedTrack(j);
704
705 if(track->GetNHits() < AliL3Transform::GetNRows() - nrowstomiss)
0309a5ee 706 {
707 tracks->Remove(j);
708 removed_tracks++;
709 }
710 }
711 tracks->Compress();
712 tracks->QSort();
713 }
a4639de2 714
715 return removed_tracks;
716}
717
718void AliL3Hough::EvaluatePatch(Int_t i,Int_t road_width,Int_t nrowstomiss)
719{
720 //Evaluate patch i.
0309a5ee 721
a4639de2 722 fEval[i]->InitTransformer(fHoughTransformer[i]);
723 fEval[i]->SetNumOfPadsToLook(road_width);
724 fEval[i]->SetNumOfRowsToMiss(nrowstomiss);
725 //fEval[i]->RemoveFoundTracks();
0309a5ee 726
a4639de2 727 AliL3TrackArray *tracks=0;
728
729 if(!fAddHistograms)
730 tracks = fTracks[i];
731 else
732 tracks = fTracks[0];
733
734 Int_t nrows=0;
735 for(Int_t j=0; j<tracks->GetNTracks(); j++)
736 {
737 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->GetCheckedTrack(j);
738 if(!track)
739 {
740 LOG(AliL3Log::kWarning,"AliL3Hough::EvaluatePatch","Track array")
741 <<"Track object missing!"<<ENDLOG;
742 continue;
743 }
744 nrows=0;
3e87ef69 745 Int_t rowrange[2] = {AliL3Transform::GetFirstRow(i),AliL3Transform::GetLastRow(i)};
746 Bool_t result = fEval[i]->LookInsideRoad(track,nrows,rowrange);
a4639de2 747 if(fAddHistograms)
748 {
749 Int_t pre=track->GetNHits();
750 track->SetNHits(pre+nrows);
751 }
3e87ef69 752 else//the track crossed too few good padrows (padrows with signal) in the patch, so remove it
753 {
754 if(result == kFALSE)
755 tracks->Remove(j);
756 }
a4639de2 757 }
758
759 tracks->Compress();
a4639de2 760
a6008206 761}
f80b98cb 762
44c7f8de 763void AliL3Hough::MergeEtaSlices()
b1886074 764{
44c7f8de 765 //Merge tracks found in neighbouring eta slices.
766 //Removes the track with the lower weight.
767
3e87ef69 768 fBenchmark->Start("Merge Eta-slices");
44c7f8de 769 AliL3TrackArray *tracks = fTracks[0];
770 if(!tracks)
b1886074 771 {
44c7f8de 772 cerr<<"AliL3Hough::MergeEtaSlices : No tracks "<<endl;
b1886074 773 return;
774 }
44c7f8de 775 for(Int_t j=0; j<tracks->GetNTracks(); j++)
b1886074 776 {
44c7f8de 777 AliL3HoughTrack *track1 = (AliL3HoughTrack*)tracks->GetCheckedTrack(j);
778 if(!track1) continue;
779 for(Int_t k=j+1; k<tracks->GetNTracks(); k++)
780 {
781 AliL3HoughTrack *track2 = (AliL3HoughTrack*)tracks->GetCheckedTrack(k);
782 if(!track2) continue;
783 if(abs(track1->GetEtaIndex() - track2->GetEtaIndex()) != 1) continue;
3e87ef69 784 if(fabs(track1->GetKappa()-track2->GetKappa()) < 0.006 &&
785 fabs(track1->GetPsi()- track2->GetPsi()) < 0.1)
44c7f8de 786 {
3e87ef69 787 //cout<<"Merging track in slices "<<track1->GetEtaIndex()<<" "<<track2->GetEtaIndex()<<endl;
788 if(track1->GetWeight() > track2->GetWeight())
789 tracks->Remove(k);
790 else
791 tracks->Remove(j);
44c7f8de 792 }
793 }
b1886074 794 }
3e87ef69 795 fBenchmark->Stop("Merge Eta-slices");
44c7f8de 796 tracks->Compress();
b1886074 797}
798
b2a02bce 799void AliL3Hough::WriteTracks(Char_t *path)
800{
801 //cout<<"AliL3Hough::WriteTracks : Sorting the tracsk"<<endl;
802 //fGlobalTracks->QSort();
803
804 Char_t filename[1024];
805 sprintf(filename,"%s/tracks_%d.raw",path,fEvent);
806 AliL3MemHandler mem;
807 mem.SetBinaryOutput(filename);
808 mem.TrackArray2Binary(fGlobalTracks);
809 mem.CloseBinaryOutput();
810 fGlobalTracks->Reset();
811}
812
3fe49b5b 813void AliL3Hough::WriteTracks(Int_t slice,Char_t *path)
1c404dd5 814{
c2f25cd2 815
3e87ef69 816 AliL3MemHandler mem;
95a00d93 817 Char_t fname[100];
b5a207b4 818 if(fAddHistograms)
819 {
b2a02bce 820 sprintf(fname,"%s/tracks_ho_%d_%d.raw",path,fEvent,slice);
3e87ef69 821 mem.SetBinaryOutput(fname);
822 mem.TrackArray2Binary(fTracks[0]);
823 mem.CloseBinaryOutput();
b5a207b4 824 }
825 else
826 {
827 for(Int_t i=0; i<fNPatches; i++)
828 {
b2a02bce 829 sprintf(fname,"%s/tracks_ho_%d_%d_%d.raw",path,fEvent,slice,i);
3e87ef69 830 mem.SetBinaryOutput(fname);
831 mem.TrackArray2Binary(fTracks[i]);
832 mem.CloseBinaryOutput();
b5a207b4 833 }
834 }
1c404dd5 835}
208b54c5 836
a6008206 837void AliL3Hough::WriteDigits(Char_t *outfile)
838{
208b54c5 839#ifdef use_aliroot
a6008206 840 //Write the current data to a new rootfile.
841
1c404dd5 842 for(Int_t i=0; i<fNPatches; i++)
a6008206 843 {
844 AliL3DigitRowData *tempPt = (AliL3DigitRowData*)fHoughTransformer[i]->GetDataPointer();
845 fMemHandler[i]->AliDigits2RootFile(tempPt,outfile);
846 }
208b54c5 847#else
848 cerr<<"AliL3Hough::WriteDigits : You need to compile with AliROOT!"<<endl;
849 return;
850#endif
4fc9a6a4 851}
208b54c5 852
0309a5ee 853Double_t AliL3Hough::GetCpuTime()
854{
855 //Return the Cputime in seconds.
856 struct timeval tv;
857 gettimeofday( &tv, NULL );
858 return tv.tv_sec+(((Double_t)tv.tv_usec)/1000000.);
0309a5ee 859}
60a3d829 860