]> git.uio.no Git - u/mrichter/AliRoot.git/blob - HLT/src/AliLevel3.cxx
Commented out fInterMerger
[u/mrichter/AliRoot.git] / HLT / src / AliLevel3.cxx
1 //$Id$
2
3 // Author: Anders Vestbo <mailto:vestbo$fi.uib.no>, Uli Frankenfeld <mailto:franken@fi.uib.no>
4 //*-- Copyright &copy ASV
5
6 #include "AliL3StandardIncludes.h"
7
8 #ifndef no_root
9 #include <TFile.h>
10 #include <TDirectory.h>
11 #include <TClonesArray.h>
12 #include <TStopwatch.h>
13 #endif
14
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"
30 #ifdef use_aliroot
31 #include "AliL3FileHandler.h"
32 #endif
33 #include "AliL3Benchmark.h"
34 #include "AliL3DigitData.h"
35 #include "AliL3TrackSegmentData.h"
36 #include "AliL3SpacePointData.h"
37 #include "AliL3VertexData.h"
38
39 //_____________________________________________________________
40 //
41 //  AliLevel3
42 //
43 //  Interface class for Level3 tracker.
44 //  Tracking is done by calling constructor with input,output 
45 //  given as argument. 
46 //  You must always remember to set the tracking parameters. E.g.:
47 // 
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)
51 //Begin_Html
52 /*
53 <img src="tpcsectorsnb.gif">
54 */
55 //End_Html
56
57 ClassImp(AliLevel3)
58
59 AliLevel3::AliLevel3()
60 {
61   fInputFile=0;
62 }
63
64 #ifdef no_root
65 AliLevel3::AliLevel3(Char_t *infile)
66 {
67   //Constructor. Calls constructor of the tracker, vertexfinder and merger classes.
68   
69   fInputFile = fopen(infile,"r");
70   
71   if(!fInputFile)
72     {
73       LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
74         <<"Inputfile "<<infile<<" does not exist!"<<ENDLOG;
75       return;
76     }
77 }
78 #else
79 AliLevel3::AliLevel3(Char_t *infile)
80 {
81   //Constructor. Calls constructor of the tracker, vertexfinder and merger classes.
82   
83   fInputFile = new TFile(infile,"READ");
84   
85   if(!fInputFile->IsOpen())
86     {
87       LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
88         <<"Inputfile "<<infile<<" does not exist!"<<ENDLOG;
89       return;
90     }
91 }
92 #endif
93
94 #ifndef no_root
95 AliLevel3::AliLevel3(TFile *in)
96 {
97   fInputFile  =  in;
98   if(!in){
99     LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
100     <<"Pointer to InFile 0x0!"<<ENDLOG;
101     return;
102   }  
103   
104   if(!fInputFile->IsOpen())
105     {
106     LOG(AliL3Log::kError,"AliLevel3::AliLevel3","File Open")
107     <<"Inputfile does not exist"<<ENDLOG;
108       return;
109     }
110 }
111 #endif
112
113 void AliLevel3::Init(Char_t *path,Bool_t binary,Int_t npatches)
114 {
115 #ifdef ASVVERSION
116   if(npatches != 1 && binary == kFALSE)
117     {
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;
120       npatches = 1;
121     }
122 #endif
123
124   if(!binary && !fInputFile)
125     {
126       LOG(AliL3Log::kError,"AliLevel3::Init","Files")
127         <<"You have not supplied the input rootfile; use the appropriate ctor!"<<ENDLOG;
128       return;
129     }
130   
131   AliL3Transform::Init(path,!binary);//Initialize the detector parameters.
132   fWriteOut = kFALSE;
133   fUseBinary = binary;
134   SetPath(path);
135   fGlobalMerger = 0;
136
137   fDoRoi = kFALSE;
138   fDoNonVertex = kFALSE;
139   fFindVertex = kFALSE;
140   SetClusterFinderParam();
141
142   fEta[0] = 0.;
143   fEta[1] = 0.9;
144
145   fEvent=0;
146
147   switch(npatches){
148   case 1:
149     fNPatch = 1;        //number of patches change row in process
150     fRow[0][0] = 0;
151     fRow[0][1] = AliL3Transform::GetNRows();
152     break;
153   case 2:
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);
159     break;
160   default: 
161     fNPatch = 6;        
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);
174   }
175
176   fVertexFinder = new AliL3VertexFinder();
177   fVertex = new AliL3Vertex();
178   fTracker = new AliL3ConfMapper();
179   fTrackMerger = new AliL3TrackMerger(fNPatch);
180   fInterMerger = new AliL3InterMerger();
181 #ifdef use_aliroot
182   fFileHandler = new AliL3FileHandler();
183   fFileHandler->SetAliInput(fInputFile);
184 #else
185   fFileHandler = new AliL3MemHandler();
186 #endif
187   fBenchmark = new AliL3Benchmark();
188 }
189
190 void AliLevel3::DoBench(char* name){
191   fBenchmark->Analyze(name);
192 }
193
194 void AliLevel3::DoMc(char* file){
195 #ifdef use_aliroot
196   if(!fFileHandler->IsDigit(fEvent))
197     fFileHandler->SetMCOutput(file);
198 #endif
199 }
200
201 AliLevel3::~AliLevel3(){
202   //Destructor
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;
209 }
210
211 void AliLevel3::SetClusterFinderParam(Float_t fXYError, Float_t fZError, Bool_t deconv)
212 {
213   fXYClusterError=fXYError;
214   fZClusterError=fZError;
215   fClusterDeconv=deconv;
216 }
217
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)
225 {
226   //Set parameters input to the tracker
227   //If no arguments are given, default parameters will be used
228   
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);
233   if(vertexconstraint)
234     fTracker->MainVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack,maxphi,maxeta);
235   else
236     fTracker->NonVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack);
237   
238   fTracker->SetParamDone(true);
239   fTracker->InitVolumes();
240 }
241
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);  
248   fEvent=event;
249   for(Int_t i=first; i<=last; i++){
250     ProcessSlice(i);
251     fGlobalMerger->SetVertex(fVertex);
252     fGlobalMerger->InitSlice(i);
253     fGlobalMerger->FillTracks(fNTrackData,fTrackData);
254     fFileHandler->Free();   //free the memory
255     fNTrackData=0;
256     fTrackData=0;
257   }
258   fBenchmark->Start("Global Merger");
259   //fGlobalMerger->AddAllTracks();
260   //fGlobalMerger->Merge();
261   fGlobalMerger->SlowMerge();
262   fBenchmark->Stop("Global Merger");
263
264   if(fWriteOut) WriteResults(); 
265   delete fGlobalMerger; fGlobalMerger = 0;
266 }
267
268 void AliLevel3::ProcessSlice(Int_t slice){
269   char name[256];
270   Bool_t UseCF = kFALSE;
271 #ifdef use_aliroot
272   UseCF = fFileHandler->IsDigit(fEvent);
273 #else
274   if(fUseBinary)
275     UseCF = kTRUE; //In case you are not using aliroot
276 #endif
277   const Int_t maxpoints=100000;
278   const Int_t pointsize = maxpoints * sizeof(AliL3SpacePointData);
279   AliL3MemHandler *memory = new AliL3MemHandler();
280
281   fTrackMerger->Reset();
282   fTrackMerger->SetRows(fRow[0]);
283   for(Int_t patch=fNPatch-1;patch>=0;patch--){
284     fFileHandler->Init(slice,patch,&fRow[patch][0]);
285     UInt_t npoints=0;
286     AliL3SpacePointData *points =0;
287     UInt_t ndigits=0;
288     AliL3DigitRowData *digits =0;
289     if(UseCF){
290       if(fUseBinary){
291         if(!fDoRoi){ 
292           if(1){     //Binary to Memory
293             fFileHandler->Free();
294             sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
295             if(!fFileHandler->SetBinaryInput(name)) return;
296             digits= (AliL3DigitRowData *)fFileHandler->CompBinary2Memory(ndigits);
297             fFileHandler->CloseBinaryInput(); 
298           }
299
300           if(0){     //Binary to Memory with Benchmark 
301             fFileHandler->Free();
302             sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
303             if(!memory->SetBinaryInput(name)) return;
304             UInt_t compsize=memory->GetFileSize();
305             UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
306             memory->CompBinary2CompMemory(ndigits,comp);
307             memory->CloseBinaryInput();
308             UInt_t datasize=memory->GetMemorySize(ndigits,comp);
309             digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
310             fBenchmark->Start("Unpacker"); 
311             fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
312             fBenchmark->Stop("Unpacker");
313             memory->Free();
314           }
315   
316           if(0){     //Binary to Memory with Random
317             fFileHandler->Free();
318             fFileHandler->ResetRandom();
319             fFileHandler->SetRandomCluster(100);
320             fFileHandler->SetNGenerate(100);
321             sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
322             if(!memory->SetBinaryInput(name)) return;
323             UInt_t compsize=memory->GetFileSize();
324             UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
325             memory->CompBinary2CompMemory(ndigits,comp);
326             memory->CloseBinaryInput();
327             UInt_t dsize=memory->GetMemorySize(ndigits,comp);
328             UInt_t rsize=fFileHandler->GetRandomSize();       
329             digits=(AliL3DigitRowData*)fFileHandler->Allocate(dsize+rsize);
330             fBenchmark->Start("Unpacker");
331             fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
332             fBenchmark->Stop("Unpacker");
333             memory->Free();
334           }
335         }
336
337         else{     //Binary to Memory with Roi
338           fFileHandler->Free();
339           Int_t sli[2]={0,0};
340           fFileHandler->SetROI(fEta,sli);
341           sprintf(name,"%sdigits_%d_%d.raw",fPath,slice,patch);
342           if(!memory->SetBinaryInput(name)) return;
343           UInt_t compsize=memory->GetFileSize();
344           UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
345           memory->CompBinary2CompMemory(ndigits,comp);
346           memory->CloseBinaryInput();
347           UInt_t datasize=memory->GetMemorySize(ndigits,comp);
348           digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
349           fBenchmark->Start("Unpacker"); 
350           datasize = fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
351           fBenchmark->Stop("Unpacker"); 
352           memory->Free();
353         }
354       }//end UseBinary
355       else{
356 #ifdef use_aliroot
357         fBenchmark->Start("Dummy Unpacker");
358         sprintf(name,"digits_%d_%d.raw",slice,patch);
359         fBenchmark->Stop("Dummy Unpacker");
360
361         if(0){    //Ali to Binary
362           fFileHandler->SetBinaryOutput(name);
363           fFileHandler->AliDigits2CompBinary();
364           fFileHandler->CloseBinaryOutput();
365         }
366   
367         if(1){     //Ali to Memory
368           digits=(AliL3DigitRowData *)fFileHandler->AliDigits2Memory(ndigits,fEvent);
369           if(0){   //Memory to Binary
370             fFileHandler->SetBinaryOutput(name);
371             fFileHandler->Memory2CompBinary(ndigits,digits);
372             fFileHandler->CloseBinaryOutput();
373           }
374         }
375 #endif
376       }//end else UseBinary
377
378       points = (AliL3SpacePointData *) memory->Allocate(pointsize);
379   
380       fClusterFinder = new AliL3ClustFinderNew();
381       fClusterFinder->InitSlice(slice,patch,fRow[patch][0],fRow[patch][1],maxpoints);
382       fClusterFinder->SetDeconv(fClusterDeconv);
383       fClusterFinder->SetXYError(fXYClusterError);
384       fClusterFinder->SetZError(fZClusterError);
385       if((fXYClusterError>0)&&(fZClusterError>0))
386         fClusterFinder->SetCalcErr(kFALSE);
387       fClusterFinder->SetOutputArray(points);
388       fClusterFinder->Read(ndigits,digits);
389       fBenchmark->Start("Cluster Finder");
390       fClusterFinder->ProcessDigits();
391       fBenchmark->Stop("Cluster Finder");
392       npoints = fClusterFinder->GetNumberOfClusters();
393       delete fClusterFinder;
394       fClusterFinder = 0;
395       fFileHandler->Free();
396       LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Cluster Finder")
397         <<AliL3Log::kDec<<"Found "<<npoints<<" Points"<<ENDLOG;
398     }//end UseCF
399     else{// if not use Clusterfinder
400       if(fUseBinary){//Binary to Memory
401         memory->Free();
402         sprintf(name,"/%spoints_%d_%d.raw",fPath,slice,patch);
403         if(!memory->SetBinaryInput(name)) return;
404         points = (AliL3SpacePointData *) memory->Allocate();
405         memory->Binary2Memory(npoints,points);
406         memory->CloseBinaryInput();
407         LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Read Cluster")
408         <<AliL3Log::kDec<<"Found "<<npoints<<" Points in File"<<ENDLOG;
409       }
410 #ifdef use_aliroot
411       else{
412         points = fFileHandler->AliPoints2Memory(npoints);
413       }
414 #endif
415       fBenchmark->Start("Dummy Unpacker");
416       fBenchmark->Stop("Dummy Unpacker");
417       fBenchmark->Start("Dummy CF");
418       fBenchmark->Stop("Dummy CF");
419     }
420    
421     if(patch == fNPatch-1){
422       // Vertex
423       if(fFindVertex){
424       // Vertex Finder
425       
426         fBenchmark->Start("Vertex Finder Read");
427         fVertexFinder->Reset();
428         fVertexFinder->Read(npoints,points);
429         fBenchmark->Stop("Vertex Finder Read"); 
430         fBenchmark->Start("Vertex Finder");
431         fVertexFinder->Analyze();
432         AliL3VertexData vertex[1];
433         fVertexFinder->Write(vertex);
434         fVertex->Read(vertex);
435         fBenchmark->Stop("Vertex Finder"); 
436       }
437       else{
438         //use 0,0,0 for vertex
439         fVertex->SetZero();
440       }
441       fTrackMerger->SetVertex(fVertex);
442     }
443     fTracker->InitSector(slice,fRow[patch]);//,fEta);
444     fTracker->SetVertex(fVertex);
445     fBenchmark->Start("Tracker Read Hits");
446     fTracker->ReadHits(npoints,points);
447     fBenchmark->Stop("Tracker Read Hits");
448     fBenchmark->Start("MainVertexTracking A"); 
449     fTracker->MainVertexTracking();
450     //fTracker->MainVertexTracking_a();
451     fBenchmark->Stop("MainVertexTracking A");
452     //fBenchmark->Start("MainVertexTracking B"); 
453     //fTracker->MainVertexTracking_b();
454     //fBenchmark->Stop("MainVertexTracking B");
455     fBenchmark->Start("Tracking fit");
456     if(fDoNonVertex)
457       fTracker->NonVertexTracking();//Do a second pass for nonvertex tracks
458     fTracker->FillTracks();
459     fBenchmark->Stop("Tracking fit");
460
461     if(fWriteOut) 
462        WriteSpacePoints(npoints, points, slice, patch); //do after Tracking
463
464     //free memory
465     if(UseCF)
466       memory->Free();
467     else
468       fFileHandler->Free();
469
470     UInt_t ntracks0 =0;
471     AliL3TrackSegmentData *trackdata0  = 
472          (AliL3TrackSegmentData *) memory->Allocate(fTracker->GetTracks());
473     memory->TrackArray2Memory(ntracks0,trackdata0,fTracker->GetTracks());
474     /*
475     //write tracks
476     if(fWriteOut){
477       sprintf(name,"%stracks_tr_%d_%d.raw",fWriteOutPath,slice,patch);
478       memory->SetBinaryOutput(name);
479       memory->Memory2Binary(ntracks0,trackdata0);
480       memory->CloseBinaryOutput();
481     }
482     */
483     fInterMerger->Reset();
484     fInterMerger->Init(fRow[patch],patch);
485
486     fInterMerger->FillTracks(ntracks0,trackdata0);
487     
488     //fBenchmark->Start("Inter Merger");
489     // fInterMerger->Merge();
490     //    fInterMerger->SlowMerge();
491     
492     //fBenchmark->Stop("Inter Merger");
493     /*
494     //write inter merged tracks
495     if(fWriteOut){
496       sprintf(name,"%stracks_im_%d_%d.raw",fWriteOutPath,slice,patch);
497       WriteTracks(name,fInterMerger,'i'); //write output of intermerger
498       }
499     */
500     memory->Free();
501     
502     UInt_t ntracks1 =0;
503     AliL3TrackSegmentData *trackdata1 =
504       (AliL3TrackSegmentData *) memory->Allocate(fInterMerger->GetInTracks(0));
505     memory->TrackArray2Memory(ntracks1,trackdata1,fInterMerger->GetInTracks(0));
506
507     fTrackMerger->InitSector(slice,patch);
508     fTrackMerger->FillTracks(ntracks1,trackdata1);
509
510     memory->Free();
511   }
512   fBenchmark->Start("Patch Merger");
513   //fTrackMerger->SlowMerge();
514   //fTrackMerger->AddAllTracks();
515   fTrackMerger->Merge();
516   fBenchmark->Stop("Patch Merger");
517   /*
518   //write merged tracks
519   if(fWriteOut){
520     sprintf(name,"%stracks_tm_%d.raw",fWriteOutPath,slice);
521     WriteTracks(name,fTrackMerger,'o'); //write output of trackmerger
522   }
523   */
524   fTrackData = (AliL3TrackSegmentData *) 
525                          fFileHandler->Allocate(fTrackMerger->GetOutTracks());
526
527   fFileHandler->TrackArray2Memory(fNTrackData,fTrackData,
528                                                 fTrackMerger->GetOutTracks());
529
530   delete memory;
531 }
532
533 void AliLevel3::WriteSpacePoints(UInt_t npoints,AliL3SpacePointData *points,
534                                                       Int_t slice,Int_t patch){
535   char name[256];
536   sprintf(name,"%spoints_%d_%d.raw",fWriteOutPath,slice,patch);
537   AliL3MemHandler * memory = new AliL3MemHandler();
538   memory->SetBinaryOutput(name);
539   memory->Transform(npoints,points,slice);
540   memory->Memory2Binary(npoints,points);
541   memory->CloseBinaryOutput();
542   delete  memory;
543 }
544
545 Int_t AliLevel3::WriteTracks(char *filename,AliL3Merger *merger,char opt){
546   AliL3MemHandler *memory = new AliL3MemHandler();
547   memory->SetBinaryOutput(filename);
548   if(opt=='a'||opt=='i'){  //add intracks
549     for(Int_t i=0;i<merger->GetNIn();i++){
550       AliL3TrackArray *tr=merger->GetInTracks(i);
551       memory->TrackArray2Binary(tr);
552     }
553   }
554
555   if(opt=='o'||opt=='a'){
556     AliL3TrackArray *tr=merger->GetOutTracks();
557     memory->TrackArray2Binary(tr);
558   }
559
560   memory->CloseBinaryOutput();
561  
562   return 1;
563 }
564
565 void AliLevel3::WriteResults()
566 {
567   //Write the resulting tracks to outputfile
568   Char_t fname[256];
569   sprintf(fname,"%stracks.raw",fWriteOutPath);
570   WriteTracks(fname,fGlobalMerger,'a');
571   //WriteTracks("tracks.raw",fGlobalMerger,'a');
572   sprintf(fname,"%stracks_gl.raw",fWriteOutPath);
573   WriteTracks(fname,fGlobalMerger,'o');
574   //WriteTracks("tracks_gl.raw",fGlobalMerger,'o');
575 }