3 // Author: Anders Vestbo <mailto:vestbo@fi.uib.no>
4 //*-- Copyright © ASV
11 #include "AliL3HoughMerger.h"
12 #include "AliL3HoughIntMerger.h"
13 #include "AliL3HoughGlobalMerger.h"
14 #include "AliL3Logging.h"
15 #include "AliL3Histogram.h"
16 #include "AliL3Hough.h"
17 #include "AliL3HoughTransformer.h"
18 #include "AliL3HoughMaxFinder.h"
19 #include "AliL3FileHandler.h"
20 #include "AliL3DigitData.h"
21 #include "AliL3HoughEval.h"
22 #include "AliL3Transform.h"
23 #include "AliL3Defs.h"
24 #include "AliL3TrackArray.h"
25 #include "AliL3HoughTrack.h"
26 #include "AliL3Benchmark.h"
28 //_____________________________________________________________
31 // Base class for the Hough transform
37 AliL3Hough::AliL3Hough()
43 fAddHistograms = kFALSE;
44 fDoIterative = kFALSE;
50 AliL3Hough::AliL3Hough(Char_t *path,Bool_t binary,Int_t n_eta_segments)
54 fNEtaSegments = n_eta_segments;
55 fAddHistograms = kFALSE;
56 fDoIterative = kFALSE;
57 fWriteDigits = kFALSE;
62 AliL3Hough::~AliL3Hough()
75 void AliL3Hough::CleanUp()
79 for(Int_t i=0; i<fNPatches; i++)
81 if(fTracks[i]) delete fTracks[i];
82 if(fEval[i]) delete fEval[i];
83 if(fHoughTransformer[i]) delete fHoughTransformer[i];
84 if(fMemHandler[i]) delete fMemHandler[i];
87 /*Shitty compiler doesn't allow this:
89 if(fTracks) delete [] fTracks;
90 if(fEval) delete [] fEval;
91 if(fHoughTransformer) delete [] fHoughTransformer;
92 if(fMemHandler) delete [] fMemHandler;
96 void AliL3Hough::Init()
99 fHoughTransformer = new AliL3HoughTransformer*[fNPatches];
100 fMemHandler = new AliL3FileHandler*[fNPatches];
101 fTracks = new AliL3TrackArray*[fNPatches];
102 fEval = new AliL3HoughEval*[fNPatches];
103 for(Int_t i=0; i<fNPatches; i++)
105 fHoughTransformer[i] = new AliL3HoughTransformer(1,i,fNEtaSegments);
106 fHoughTransformer[i]->CreateHistograms(64,-0.003,0.003,64,-0.26,0.26);
107 fHoughTransformer[i]->SetThreshold(3);
108 fEval[i] = new AliL3HoughEval();
109 fTracks[i] = new AliL3TrackArray("AliL3HoughTrack");
110 fMemHandler[i] = new AliL3FileHandler();
112 fMemHandler[i]->SetAliInput(fPath);
114 fPeakFinder = new AliL3HoughMaxFinder("KappaPhi");
115 fMerger = new AliL3HoughMerger(fNPatches);
116 fInterMerger = new AliL3HoughIntMerger();
119 void AliL3Hough::Process(Int_t minslice,Int_t maxslice)
121 //Process all slices [minslice,maxslice].
122 fGlobalMerger = new AliL3HoughGlobalMerger(minslice,maxslice);
124 for(Int_t i=minslice; i<=maxslice; i++)
130 FindTrackCandidates();
134 fGlobalMerger->FillTracks(fTracks[0],i);
140 void AliL3Hough::ReadData(Int_t slice)
142 //Read data from files, binary or root.
144 for(Int_t i=0; i<fNPatches; i++)
146 fMemHandler[i]->Free();
148 AliL3DigitRowData *digits =0;
150 if(fBinary)//take input data from binary files
152 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,i);
153 fMemHandler[i]->SetBinaryInput(name);
154 digits = (AliL3DigitRowData *)fMemHandler[i]->CompBinary2Memory(ndigits);
155 fMemHandler[i]->CloseBinaryInput();
157 else //read data from root file
159 fMemHandler[i]->Init(slice,i,NRows[i]);
160 digits=(AliL3DigitRowData *)fMemHandler[i]->AliDigits2Memory(ndigits);
162 fHoughTransformer[i]->SetInputData(ndigits,digits);
166 void AliL3Hough::Transform()
168 //Transform all data given to the transformer within the given slice
169 //(after ReadData(slice))
171 Double_t initTime,finalTime;
172 for(Int_t i=0; i<fNPatches; i++)
174 fHoughTransformer[i]->Reset();//Reset the histograms
175 initTime = AliL3Benchmark::GetCpuTime();
176 fHoughTransformer[i]->TransformCircle();
177 finalTime = AliL3Benchmark::GetCpuTime();
178 LOG(AliL3Log::kInformational,"AliL3Hough::Transform","Timing")
179 <<AliL3Log::kDec<<"Transform finished in "<<(finalTime-initTime)*1000<<"ms"<<ENDLOG;
183 void AliL3Hough::MergePatches()
185 if(fAddHistograms) //Nothing to merge here
187 AliL3Transform *tr = new AliL3Transform();
188 fMerger->SetTransformer(tr);
189 fMerger->MergePatches(kTRUE);
193 void AliL3Hough::MergeInternally()
196 fInterMerger->FillTracks(fTracks[0]);
198 fInterMerger->FillTracks(fMerger->GetOutTracks());
200 fInterMerger->MMerge();
203 void AliL3Hough::ProcessSliceIter()
205 //Process current slice (after ReadData(slice)) iteratively.
207 for(Int_t i=0; i<fNPatches; i++)
210 fMerger->FillTracks(fTracks[i],i); //Copy tracks to merger
215 void AliL3Hough::ProcessPatchIter(Int_t patch)
217 //Process patch in a iterative way.
218 //transform + peakfinding + evaluation + transform +...
220 Int_t num_of_tries = 10;
221 AliL3HoughTransformer *tr = fHoughTransformer[patch];
222 AliL3TrackArray *tracks = fTracks[patch];
224 AliL3HoughEval *ev = fEval[patch];
225 ev->InitTransformer(tr);
226 ev->RemoveFoundTracks();
227 ev->SetNumOfRowsToMiss(2);
228 AliL3Histogram *hist;
229 for(Int_t t=0; t<num_of_tries; t++)
232 tr->TransformCircle();
233 for(Int_t i=0; i<fNEtaSegments; i++)
235 hist = tr->GetHistogram(i);
236 if(hist->GetNEntries()==0) continue;
237 fPeakFinder->SetHistogram(hist);
240 fPeakFinder->FindAbsMaxima(*x,*y);
241 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->NextTrack();
242 track->SetTrackParameters(hist->GetBinCenterX(*x),hist->GetBinCenterY(*y),1);
243 if(!ev->LookInsideRoad(track,i))
245 tracks->Remove(tracks->GetNTracks()-1);
250 LOG(AliL3Log::kInformational,"AliL3Hough::ProcessPatch","NTracks")
251 <<AliL3Log::kDec<<"Found "<<tracks->GetNTracks()<<" tracks in patch "<<patch<<ENDLOG;
255 void AliL3Hough::AddAllHistograms()
257 //Add the histograms within one etaslice.
258 //Resulting histogram are in patch=0.
260 for(Int_t i=0; i<fNEtaSegments; i++)
262 AliL3Histogram *hist0 = fHoughTransformer[0]->GetHistogram(i);
263 for(Int_t j=1; j<fNPatches; j++)
265 AliL3Histogram *hist = fHoughTransformer[j]->GetHistogram(i);
269 fAddHistograms = kTRUE;
272 void AliL3Hough::FindTrackCandidates()
274 //Look for peaks in histograms, and find the track candidates
278 n_patches = 1; //Histograms has been added.
280 n_patches = fNPatches;
282 for(Int_t i=0; i<n_patches; i++)
284 AliL3HoughTransformer *tr = fHoughTransformer[i];
286 for(Int_t j=0; j<fNEtaSegments; j++)
288 AliL3Histogram *hist = tr->GetHistogram(j);
289 if(hist->GetNEntries()==0) continue;
290 fPeakFinder->SetHistogram(hist);
295 fPeakFinder->FindPeak1(x,y,weight,n,1);
296 for(Int_t k=0; k<n; k++)
298 if(weight[k] == 0) continue;
299 AliL3HoughTrack *track = (AliL3HoughTrack*)fTracks[i]->NextTrack();
300 track->SetTrackParameters(x[k],y[k],weight[k]);
301 track->SetEtaIndex(j);
302 track->SetEta((Double_t)(j*tr->GetEtaSlice()));
303 track->SetRowRange(NRows[0][0],NRows[5][1]);
310 void AliL3Hough::Evaluate(Int_t road_width)
312 //Evaluate the tracks, by looking along the road in the raw data.
317 LOG(AliL3Log::kError,"AliL3Hough::Evaluate","Track Array")
318 <<"No tracks to work with..."<<ENDLOG;
322 printf("Number of tracks before evaluation %d\n",fTracks[0]->GetNTracks());
323 AliL3TrackArray *tracks;
324 for(Int_t i=0; i<fNPatches; i++)
326 fEval[i]->InitTransformer(fHoughTransformer[i]);
327 fEval[i]->SetNumOfRowsToMiss(2);
328 fEval[i]->SetNumOfPadsToLook(road_width);
333 for(Int_t j=0; j<tracks->GetNTracks(); j++)
335 AliL3HoughTrack *track = (AliL3HoughTrack*)tracks->GetCheckedTrack(j);
338 LOG(AliL3Log::kWarning,"AliL3Hough::Evaluate","Track array")
339 <<"Track object missing!"<<ENDLOG;
343 if(!fEval[i]->LookInsideRoad(track,track->GetEtaIndex()))
346 track->SetRowRange(NRows[0][0],NRows[5][1]);//All rows included
349 tracks->QSort(); //Sort the tracks according to weight
352 fMerger->FillTracks(tracks,i); //Copy tracks to the track merger
357 void AliL3Hough::EvaluateWithEta()
361 printf("AliL3Hough::EvaluateWithEta: NO TRACKS\n");
364 printf("Number of tracks before evaluation %d\n",fTracks[0]->GetNTracks());
366 for(Int_t i=0; i<fNPatches; i++)
368 fEval[i]->InitTransformer(fHoughTransformer[i]);
369 fEval[i]->FindEta(fTracks[0]);
371 fMerger->FillTracks(fTracks[0],0);
372 fMerger->MergeEtaSlices(0);
375 void AliL3Hough::WriteTracks()
377 AliL3MemHandler *mem = new AliL3MemHandler();
378 mem->SetBinaryOutput("tracks.raw");
379 mem->TrackArray2Binary(fTracks[0]);
380 mem->CloseBinaryOutput();
385 void AliL3Hough::WriteDigits(Char_t *outfile)
387 //Write the current data to a new rootfile.
389 for(Int_t i=0; i<fNPatches; i++)
391 AliL3DigitRowData *tempPt = (AliL3DigitRowData*)fHoughTransformer[i]->GetDataPointer();
392 fMemHandler[i]->AliDigits2RootFile(tempPt,outfile);