3 // Author: Anders Vestbo <mailto:vestbo$fi.uib.no>, Uli Frankenfeld <mailto:franken@fi.uib.no>
4 //*-- Copyright © ASV
6 #include "AliL3StandardIncludes.h"
10 #include <TDirectory.h>
11 #include <TClonesArray.h>
12 #include <TStopwatch.h>
15 #include "AliL3Logging.h"
16 #include "AliLevel3.h"
17 #include "AliL3ConfMapper.h"
18 #include "AliL3Vertex.h"
19 #include "AliL3VertexFinder.h"
20 #include "AliL3TrackMerger.h"
21 #include "AliL3GlobalMerger.h"
22 #include "AliL3InterMerger.h"
23 #include "AliL3ConfMapPoint.h"
24 #include "AliL3ConfMapTrack.h"
25 #include "AliL3Transform.h"
26 #include "AliL3ClustFinderNew.h"
27 #include "AliL3DigitData.h"
28 #include "AliL3TrackArray.h"
29 #include "AliL3MemHandler.h"
31 #include "AliL3FileHandler.h"
33 #include "AliL3Benchmark.h"
34 #include "AliL3DigitData.h"
35 #include "AliL3TrackSegmentData.h"
36 #include "AliL3SpacePointData.h"
37 #include "AliL3VertexData.h"
39 //_____________________________________________________________
43 // Interface class for Level3 tracker.
44 // Tracking is done by calling constructor with input,output
46 // You must always remember to set the tracking parameters. E.g.:
48 // AliLevel3 *level3 = new AliLevel3(inputfile,outputfile);
49 // level3->SetTrackerParam(); //Sets default tracking parameters
50 // level3->ProcessSector(2,2); //Does tracking on sector 2 (actually 2+38)
53 <img src="tpcsectorsnb.gif">
59 AliLevel3::AliLevel3()
65 AliLevel3::AliLevel3(Char_t *infile)
67 //Constructor. Calls constructor of the tracker, vertexfinder and merger classes.
69 fInputFile = fopen(infile,"r");
73 LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
74 <<"Inputfile "<<infile<<" does not exist!"<<ENDLOG;
79 AliLevel3::AliLevel3(Char_t *infile)
81 //Constructor. Calls constructor of the tracker, vertexfinder and merger classes.
83 fInputFile = new TFile(infile,"READ");
85 if(!fInputFile->IsOpen())
87 LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
88 <<"Inputfile "<<infile<<" does not exist!"<<ENDLOG;
95 AliLevel3::AliLevel3(TFile *in)
99 LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
100 <<"Pointer to InFile 0x0!"<<ENDLOG;
104 if(!fInputFile->IsOpen())
106 LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
107 <<"Inputfile does not exist"<<ENDLOG;
113 void AliLevel3::Init(Char_t *path,Bool_t binary,Int_t npatches)
116 if(npatches != 1 && binary == kFALSE)
118 LOG(AliL3Log::kWarning,"AliLevel3::Init","NPatches")
119 <<"You can only read data slice by slice (1 patch) when reading from a rootfile and using the patched (ASV) AliTPC version."<<ENDLOG;
124 if(!binary && !fInputFile)
126 LOG(AliL3Log::kError,"AliLevel3::Init","Files")
127 <<"You have not supplied the input rootfile; use the appropriate ctor!"<<ENDLOG;
131 AliL3Transform::Init(path,!binary);//Initialize the detector parameters.
138 fDoNonVertex = kFALSE;
139 fFindVertex = kFALSE;
140 SetClusterFinderParam();
149 fNPatch = 1; //number of patches change row in process
151 fRow[0][1] = AliL3Transform::GetNRows();
154 fNPatch = 2; //number of patches change row in process
155 fRow[0][0] = 0; // first row
156 fRow[0][1] = AliL3Transform::GetLastRow(1);
157 fRow[1][0] = AliL3Transform::GetFirstRow(2);
158 fRow[1][1] = AliL3Transform::GetLastRow(5);
162 fRow[0][0] = AliL3Transform::GetFirstRow(0);
163 fRow[0][1] = AliL3Transform::GetLastRow(0);
164 fRow[1][0] = AliL3Transform::GetFirstRow(1);
165 fRow[1][1] = AliL3Transform::GetLastRow(1);
166 fRow[2][0] = AliL3Transform::GetFirstRow(2);
167 fRow[2][1] = AliL3Transform::GetLastRow(2);
168 fRow[3][0] = AliL3Transform::GetFirstRow(3);
169 fRow[3][1] = AliL3Transform::GetLastRow(3);
170 fRow[4][0] = AliL3Transform::GetFirstRow(4);
171 fRow[4][1] = AliL3Transform::GetLastRow(4);
172 fRow[5][0] = AliL3Transform::GetFirstRow(5);
173 fRow[5][1] = AliL3Transform::GetLastRow(5);
176 fVertexFinder = new AliL3VertexFinder();
177 fVertex = new AliL3Vertex();
178 fTracker = new AliL3ConfMapper();
179 fTrackMerger = new AliL3TrackMerger(fNPatch);
180 fInterMerger = new AliL3InterMerger();
182 fFileHandler = new AliL3FileHandler();
183 fFileHandler->SetAliInput(fInputFile);
185 fFileHandler = new AliL3MemHandler();
187 fBenchmark = new AliL3Benchmark();
190 void AliLevel3::DoBench(char* name){
191 fBenchmark->Analyze(name);
194 void AliLevel3::DoMc(char* file){
196 if(!fFileHandler->IsDigit(fEvent))
197 fFileHandler->SetMCOutput(file);
201 AliLevel3::~AliLevel3(){
203 if(fVertexFinder) delete fVertexFinder;
204 if(fVertex) delete fVertex;
205 if(fTracker) delete fTracker;
206 if(fTrackMerger) delete fTrackMerger;
207 if(fInterMerger) delete fInterMerger;
208 if(fFileHandler) delete fFileHandler;
211 void AliLevel3::SetClusterFinderParam(Float_t fXYError, Float_t fZError, Bool_t deconv)
213 fXYClusterError=fXYError;
214 fZClusterError=fZError;
215 fClusterDeconv=deconv;
218 void AliLevel3::SetTrackerParam(Int_t phi_segments, Int_t eta_segments,
219 Int_t trackletlength, Int_t tracklength,
220 Int_t rowscopetracklet, Int_t rowscopetrack,
221 Double_t min_pt_fit, Double_t maxangle,
222 Double_t goodDist, Double_t hitChi2Cut,
223 Double_t goodHitChi2, Double_t trackChi2Cut,
224 Int_t maxdist,Double_t maxphi,Double_t maxeta,Bool_t vertexconstraint)
226 //Set parameters input to the tracker
227 //If no arguments are given, default parameters will be used
229 fTracker->SetNSegments(phi_segments,eta_segments);
230 fTracker->SetMaxDca(min_pt_fit);
231 fTracker->SetTrackCuts(hitChi2Cut,goodHitChi2,trackChi2Cut,maxdist,vertexconstraint);
232 fTracker->SetTrackletCuts(maxangle,goodDist,vertexconstraint);
234 fTracker->MainVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack,maxphi,maxeta);
236 fTracker->NonVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack);
238 fTracker->SetParamDone(true);
239 fTracker->InitVolumes();
242 void AliLevel3::ProcessEvent(Int_t first,Int_t last,Int_t event){
243 //Do tracking on all slices in region [first,last]
244 //Slices numbering in TPC goes from 0-35, which means that one slice
245 //corresponds to inner+outer sector.E.g. slice 2 corresponds to
246 //inner=2 + outer=38.
247 fGlobalMerger= new AliL3GlobalMerger(first,last);
249 for(Int_t i=first; i<=last; i++){
251 fGlobalMerger->SetVertex(fVertex);
252 fGlobalMerger->InitSlice(i);
253 fGlobalMerger->FillTracks(fNTrackData,fTrackData);
254 fFileHandler->Free(); //free the memory
258 fBenchmark->Start("Global Merger");
259 fGlobalMerger->AddAllTracks();
260 //fGlobalMerger->Merge();
261 fGlobalMerger->SlowMerge();
262 fBenchmark->Stop("Global Merger");
264 if(fWriteOut) WriteResults();
265 delete fGlobalMerger; fGlobalMerger = 0;
268 void AliLevel3::ProcessSlice(Int_t slice){
270 Bool_t UseCF = kFALSE;
272 UseCF = fFileHandler->IsDigit(fEvent);
275 UseCF = kTRUE; //In case you are not using aliroot
277 const Int_t maxpoints=100000;
278 const Int_t pointsize = maxpoints * sizeof(AliL3SpacePointData);
279 AliL3MemHandler *memory = new AliL3MemHandler();
281 fTrackMerger->Reset();
282 fTrackMerger->SetRows(fRow[0]);
285 for(Int_t patch=fNPatch-1;patch>=0;patch--){
286 fFileHandler->Init(slice,patch,&fRow[patch][0]);
288 AliL3SpacePointData *points =0;
290 AliL3DigitRowData *digits =0;
294 if(1){ //Binary to Memory
295 fFileHandler->Free();
297 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,-1);
299 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
300 if(!fFileHandler->SetBinaryInput(name)) return;
301 digits= (AliL3DigitRowData *)fFileHandler->CompBinary2Memory(ndigits);
302 fFileHandler->CloseBinaryInput();
305 if(0){ //Binary to Memory with Benchmark
306 fFileHandler->Free();
307 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
308 if(!memory->SetBinaryInput(name)) return;
309 UInt_t compsize=memory->GetFileSize();
310 UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
311 memory->CompBinary2CompMemory(ndigits,comp);
312 memory->CloseBinaryInput();
313 UInt_t datasize=memory->GetMemorySize(ndigits,comp);
314 digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
315 fBenchmark->Start("Unpacker");
316 fFileHandler->CompMemory2Memory(ndigits,digits,comp);
317 fBenchmark->Stop("Unpacker");
321 if(0){ //Binary to Memory with Random
322 fFileHandler->Free();
323 fFileHandler->ResetRandom();
324 fFileHandler->SetRandomCluster(100);
325 fFileHandler->SetNGenerate(100);
326 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
327 if(!memory->SetBinaryInput(name)) return;
328 UInt_t compsize=memory->GetFileSize();
329 UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
330 memory->CompBinary2CompMemory(ndigits,comp);
331 memory->CloseBinaryInput();
332 UInt_t dsize=memory->GetMemorySize(ndigits,comp);
333 UInt_t rsize=fFileHandler->GetRandomSize();
334 digits=(AliL3DigitRowData*)fFileHandler->Allocate(dsize+rsize);
335 fBenchmark->Start("Unpacker");
336 fFileHandler->CompMemory2Memory(ndigits,digits,comp);
337 fBenchmark->Stop("Unpacker");
342 else{ //Binary to Memory with Roi
343 fFileHandler->Free();
345 fFileHandler->SetROI(fEta,sli);
346 sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
347 if(!memory->SetBinaryInput(name)) return;
348 UInt_t compsize=memory->GetFileSize();
349 UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
350 memory->CompBinary2CompMemory(ndigits,comp);
351 memory->CloseBinaryInput();
352 UInt_t datasize=memory->GetMemorySize(ndigits,comp);
353 digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
354 fBenchmark->Start("Unpacker");
355 datasize = fFileHandler->CompMemory2Memory(ndigits,digits,comp);
356 fBenchmark->Stop("Unpacker");
362 fBenchmark->Start("Dummy Unpacker");
363 sprintf(name,"digits_%d_%d.raw",slice,patch);
364 fBenchmark->Stop("Dummy Unpacker");
366 if(0){ //Ali to Binary
367 fFileHandler->SetBinaryOutput(name);
368 fFileHandler->AliDigits2CompBinary();
369 fFileHandler->CloseBinaryOutput();
372 if(1){ //Ali to Memory
373 digits=(AliL3DigitRowData *)fFileHandler->AliDigits2Memory(ndigits,fEvent);
374 if(0){ //Memory to Binary
375 fFileHandler->SetBinaryOutput(name);
376 fFileHandler->Memory2CompBinary(ndigits,digits);
377 fFileHandler->CloseBinaryOutput();
381 }//end else UseBinary
383 points = (AliL3SpacePointData *) memory->Allocate(pointsize);
385 fClusterFinder = new AliL3ClustFinderNew();
386 fClusterFinder->InitSlice(slice,patch,fRow[patch][0],fRow[patch][1],maxpoints);
387 fClusterFinder->SetDeconv(fClusterDeconv);
388 fClusterFinder->SetXYError(fXYClusterError);
389 fClusterFinder->SetZError(fZClusterError);
390 if((fXYClusterError>0)&&(fZClusterError>0))
391 fClusterFinder->SetCalcErr(kFALSE);
392 fClusterFinder->SetOutputArray(points);
393 fBenchmark->Start("Cluster Finder");
394 fClusterFinder->Read(ndigits,digits);
395 fClusterFinder->ProcessDigits();
396 fBenchmark->Stop("Cluster Finder");
397 npoints = fClusterFinder->GetNumberOfClusters();
398 delete fClusterFinder;
400 fFileHandler->Free();
401 LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Cluster Finder")
402 <<AliL3Log::kDec<<"Found "<<npoints<<" Points"<<ENDLOG;
404 else{// if not use Clusterfinder
405 if(fUseBinary){//Binary to Memory
407 sprintf(name,"/%spoints_%d_%d.raw",fPath,slice,patch);
408 if(!memory->SetBinaryInput(name)) return;
409 points = (AliL3SpacePointData *) memory->Allocate();
410 memory->Binary2Memory(npoints,points);
411 memory->CloseBinaryInput();
412 LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Read Cluster")
413 <<AliL3Log::kDec<<"Found "<<npoints<<" Points in File"<<ENDLOG;
417 points = fFileHandler->AliPoints2Memory(npoints);
420 fBenchmark->Start("Dummy Unpacker");
421 fBenchmark->Stop("Dummy Unpacker");
422 fBenchmark->Start("Dummy CF");
423 fBenchmark->Stop("Dummy CF");
426 if(patch == fNPatch-1){
431 fBenchmark->Start("Vertex Finder Read");
432 fVertexFinder->Reset();
433 fVertexFinder->Read(npoints,points);
434 fBenchmark->Stop("Vertex Finder Read");
435 fBenchmark->Start("Vertex Finder");
436 fVertexFinder->Analyze();
437 AliL3VertexData vertex[1];
438 fVertexFinder->Write(vertex);
439 fVertex->Read(vertex);
440 fBenchmark->Stop("Vertex Finder");
443 //use 0,0,0 for vertex
446 fTrackMerger->SetVertex(fVertex);
448 fTracker->InitSector(slice,fRow[patch],fEta);
449 fTracker->SetVertex(fVertex);
450 fBenchmark->Start("Tracker Read Hits");
451 fTracker->ReadHits(npoints,points);
452 fBenchmark->Stop("Tracker Read Hits");
453 fBenchmark->Start("MainVertexTracking A");
454 fTracker->MainVertexTracking();
455 //fTracker->MainVertexTracking_a();
456 fBenchmark->Stop("MainVertexTracking A");
457 //fBenchmark->Start("MainVertexTracking B");
458 //fTracker->MainVertexTracking_b();
459 //fBenchmark->Stop("MainVertexTracking B");
460 fBenchmark->Start("Tracking fit");
462 fTracker->NonVertexTracking();//Do a second pass for nonvertex tracks
463 fTracker->FillTracks();
464 fBenchmark->Stop("Tracking fit");
467 WriteSpacePoints(npoints, points, slice, patch); //do after Tracking
473 fFileHandler->Free();
476 AliL3TrackSegmentData *trackdata0 =
477 (AliL3TrackSegmentData *) memory->Allocate(fTracker->GetTracks());
478 memory->TrackArray2Memory(ntracks0,trackdata0,fTracker->GetTracks());
482 sprintf(name,"%stracks_tr_%d_%d.raw",fWriteOutPath,slice,patch);
483 memory->SetBinaryOutput(name);
484 memory->Memory2Binary(ntracks0,trackdata0);
485 memory->CloseBinaryOutput();
488 fInterMerger->Reset();
489 fInterMerger->Init(fRow[patch],patch);
491 fInterMerger->FillTracks(ntracks0,trackdata0);
493 //fBenchmark->Start("Inter Merger");
494 // fInterMerger->Merge();
495 // fInterMerger->SlowMerge();
497 //fBenchmark->Stop("Inter Merger");
499 //write inter merged tracks
501 sprintf(name,"%stracks_im_%d_%d.raw",fWriteOutPath,slice,patch);
502 WriteTracks(name,fInterMerger,'i'); //write output of intermerger
508 AliL3TrackSegmentData *trackdata1 =
509 (AliL3TrackSegmentData *) memory->Allocate(fInterMerger->GetInTracks(0));
510 memory->TrackArray2Memory(ntracks1,trackdata1,fInterMerger->GetInTracks(0));
512 fTrackMerger->InitSector(slice,patch);
513 fTrackMerger->FillTracks(ntracks1,trackdata1);
517 fBenchmark->Start("Patch Merger");
518 //fTrackMerger->SlowMerge();
519 fTrackMerger->AddAllTracks();
520 //fTrackMerger->Merge();
521 fBenchmark->Stop("Patch Merger");
523 //write merged tracks
525 sprintf(name,"%stracks_tm_%d.raw",fWriteOutPath,slice);
526 WriteTracks(name,fTrackMerger,'o'); //write output of trackmerger
529 fTrackData = (AliL3TrackSegmentData *)
530 fFileHandler->Allocate(fTrackMerger->GetOutTracks());
532 fFileHandler->TrackArray2Memory(fNTrackData,fTrackData,
533 fTrackMerger->GetOutTracks());
538 void AliLevel3::WriteSpacePoints(UInt_t npoints,AliL3SpacePointData *points,
539 Int_t slice,Int_t patch){
541 sprintf(name,"%spoints_%d_%d.raw",fWriteOutPath,slice,patch);
542 AliL3MemHandler * memory = new AliL3MemHandler();
543 memory->SetBinaryOutput(name);
544 memory->Transform(npoints,points,slice);
545 memory->Memory2Binary(npoints,points);
546 memory->CloseBinaryOutput();
550 Int_t AliLevel3::WriteTracks(char *filename,AliL3Merger *merger,char opt){
551 AliL3MemHandler *memory = new AliL3MemHandler();
552 memory->SetBinaryOutput(filename);
553 if(opt=='a'||opt=='i'){ //add intracks
554 for(Int_t i=0;i<merger->GetNIn();i++){
555 AliL3TrackArray *tr=merger->GetInTracks(i);
556 memory->TrackArray2Binary(tr);
560 if(opt=='o'||opt=='a'){
561 AliL3TrackArray *tr=merger->GetOutTracks();
562 memory->TrackArray2Binary(tr);
565 memory->CloseBinaryOutput();
570 void AliLevel3::WriteResults()
572 //Write the resulting tracks to outputfile
574 sprintf(fname,"%stracks.raw",fWriteOutPath);
575 WriteTracks(fname,fGlobalMerger,'a');
576 //WriteTracks("tracks.raw",fGlobalMerger,'a');
577 sprintf(fname,"%stracks_gl.raw",fWriteOutPath);
578 WriteTracks(fname,fGlobalMerger,'o');
579 //WriteTracks("tracks_gl.raw",fGlobalMerger,'o');