Update to FindFASTJET.cmake; now accepts also for -DFASTJET= option
[u/mrichter/AliRoot.git] / MFT / AliMFTTrackerMU.cxx
1 /*************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 *                                                                        *
4 * Author: The ALICE Off-line Project.                                    *
5 * Contributors are mentioned in the code where appropriate.              *
6 *                                                                        *
7 * Permission to use, copy, modify and distribute this software and its   *
8 * documentation strictly for non-commercial purposes is hereby granted   *
9 * without fee, provided that the above copyright notice appears in all   *
10 * copies and that both the copyright notice and this permission notice   *
11 * appear in the supporting documentation. The authors make no claims     *
12 * about the suitability of this software for any purpose. It is          *
13 * provided "as is" without express or implied warranty.                  *
14 **************************************************************************/
15
16 //====================================================================================================================================================
17 //
18 //                          MFT tracker
19 //
20 // Class for the creation of the "global muon tracks" built from the tracks reconstructed in the 
21 // muon spectrometer and the clusters of the Muon Forward Tracker
22 //
23 // Contact author: antonio.uras@cern.ch
24 //
25 //====================================================================================================================================================
26
27 #include "TTree.h"
28 #include "AliLog.h"
29 #include "AliGeomManager.h"
30 #include "AliESDEvent.h"
31 #include "AliESDMuonTrack.h"
32 #include "AliESDMuonGlobalTrack.h"
33 #include "AliMFTTrackerMU.h"
34 #include "TMath.h"
35 #include "AliRun.h"
36 #include "AliRunLoader.h"
37 #include "AliHeader.h"
38 #include "AliGenEventHeader.h"
39 #include "TArrayF.h"
40 #include "AliMFT.h"
41 #include "AliMUONTrackExtrap.h"
42 #include "AliMUONTrack.h"
43 #include "AliMUONESDInterface.h"
44 #include "AliMuonForwardTrack.h"
45 #include "AliMUONConstants.h"
46
47 ClassImp(AliMFTTrackerMU)
48
49 const Double_t AliMFTTrackerMU::fRadLengthSi = AliMFTConstants::fRadLengthSi;
50
51 //====================================================================================================================================================
52
53 AliMFTTrackerMU::AliMFTTrackerMU() : 
54   AliTracker(),
55   fESD(0),
56   fMFT(0),
57   fSegmentation(0),
58   fNPlanesMFT(0),
59   fNPlanesMFTAnalyzed(0),
60   fSigmaClusterCut(2),
61   fScaleSigmaClusterCut(1.),
62   fNMaxMissingMFTClusters(0),
63   fGlobalTrackingDiverged(kFALSE),
64   fCandidateTracks(0),
65   fMUONTrack(0),
66   fCurrentTrack(0),
67   fFinalBestCandidate(0),
68   fXExtrapVertex(0),
69   fYExtrapVertex(0),
70   fZExtrapVertex(0),
71   fXExtrapVertexError(0),
72   fYExtrapVertexError(0),
73   fBransonCorrection(kFALSE)
74 {
75
76   //--------------------------------------------------------------------
77   // This is the AliMFTTrackerMU constructor
78   //--------------------------------------------------------------------
79
80   fMFT = (AliMFT*) gAlice->GetDetector("MFT");
81   fSegmentation = fMFT->GetSegmentation();
82   SetNPlanesMFT(fSegmentation->GetNPlanes());
83   AliMUONTrackExtrap::SetField();                 // set the magnetic field for track extrapolations
84
85   AliInfo(Form("fMFT = %p, fSegmentation = %p", fMFT, fSegmentation));
86
87   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
88     fMFTClusterArray[iPlane]      = new TClonesArray("AliMFTCluster");
89     fMFTClusterArrayFront[iPlane] = new TClonesArray("AliMFTCluster");
90     fMFTClusterArrayBack[iPlane]  = new TClonesArray("AliMFTCluster");
91     fMFTClusterArray[iPlane]      -> SetOwner(kTRUE);
92     fMFTClusterArrayFront[iPlane] -> SetOwner(kTRUE);
93     fMFTClusterArrayBack[iPlane]  -> SetOwner(kTRUE);
94     fMinResearchRadiusAtPlane[iPlane] = 0.;
95   }
96
97   fCandidateTracks = new TClonesArray("AliMuonForwardTrack",50000);
98
99 }
100
101 //====================================================================================================================================================
102
103 AliMFTTrackerMU::~AliMFTTrackerMU() {
104
105   // destructor
106
107   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
108     fMFTClusterArray[iPlane] -> Delete();
109     delete fMFTClusterArray[iPlane];
110     delete fMFTClusterArrayFront[iPlane];
111     delete fMFTClusterArrayBack[iPlane];
112   }
113
114   delete fCandidateTracks;
115
116 }
117
118 //====================================================================================================================================================
119
120 Int_t AliMFTTrackerMU::LoadClusters(TTree *cTree) {
121
122   //--------------------------------------------------------------------
123   // This function loads the MFT clusters
124   //--------------------------------------------------------------------
125  
126   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
127     AliDebug(1, Form("Setting Address for Branch Plane_%02d", iPlane)); 
128     cTree->SetBranchAddress(Form("Plane_%02d",iPlane), &fMFTClusterArray[iPlane]);
129   }
130  
131   if (!cTree->GetEvent()) return kFALSE;
132   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
133     AliInfo(Form("plane %02d: nClusters = %d\n", iPlane, fMFTClusterArray[iPlane]->GetEntries()));
134   }
135   SeparateFrontBackClusters();
136
137   return 0;
138
139 }
140
141 //====================================================================================================================================================
142
143 void AliMFTTrackerMU::UnloadClusters() {
144   
145   //--------------------------------------------------------------------
146   // This function unloads MFT clusters
147   //--------------------------------------------------------------------
148
149   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
150     fMFTClusterArray[iPlane]      -> Clear("C");
151     fMFTClusterArrayFront[iPlane] -> Clear("C");
152     fMFTClusterArrayBack[iPlane]  -> Clear("C");
153   }
154
155 }
156
157 //====================================================================================================================================================
158
159 Int_t AliMFTTrackerMU::Clusters2Tracks(AliESDEvent *event) {
160
161   //--------------------------------------------------------------------
162   // This functions reconstructs the Muon Forward Tracks
163   // The clusters must be already loaded !
164   //--------------------------------------------------------------------
165
166   // Tree of AliMuonForwardTrack objects. Created outside the ESD framework for cross-check purposes
167
168   TFile *outputFileMuonGlobalTracks = new TFile("MuonGlobalTracks.root", "update");
169   TTree *outputTreeMuonGlobalTracks = new TTree("AliMuonForwardTracks", "Tree of AliMuonForwardTracks");
170   TClonesArray *muonForwardTracks = new TClonesArray("AliMuonForwardTrack");
171   outputTreeMuonGlobalTracks -> Branch("tracks", &muonForwardTracks);
172  
173   //--------------------------------------------------------------------
174
175   fESD = event;
176
177   fXExtrapVertex = 0;
178   fYExtrapVertex = 0;
179   fZExtrapVertex = 0;
180   fXExtrapVertexError = AliMFTConstants::fXVertexTolerance;
181   fYExtrapVertexError = AliMFTConstants::fYVertexTolerance;
182
183   // Imposing a fixed primary vertex, in order to keep memory of its position. Taking the primary vertex would imply the risk
184   // of loosing memory of its position when passing from ESD to AOD, due to possible refitting
185   const AliESDVertex* esdVert = fESD->GetVertex(); 
186   if (esdVert->GetNContributors() > 0 || !strcmp(esdVert->GetTitle(),"vertexer: smearMC")) {
187     fXExtrapVertex = esdVert->GetX();
188     fYExtrapVertex = esdVert->GetY();
189     fZExtrapVertex = esdVert->GetZ();
190     fXExtrapVertexError = TMath::Max(AliMFTConstants::fXVertexTolerance, esdVert->GetXRes());
191     fYExtrapVertexError = TMath::Max(AliMFTConstants::fYVertexTolerance, esdVert->GetYRes());
192     AliInfo(Form("Found ESD vertex from %d contributors (%f +/- %f,  %f +/- %f,  %f)", 
193                  esdVert->GetNContributors(),fXExtrapVertex,fXExtrapVertexError,fYExtrapVertex,fYExtrapVertexError,fZExtrapVertex));
194   }
195   else GetVertexFromMC();    
196
197   //----------- Read ESD MUON tracks -------------------
198
199   Int_t nTracksMUON = event->GetNumberOfMuonTracks();
200
201   AliInfo(Form("Number of ESD MUON tracks: %d\n", nTracksMUON));
202
203   Int_t iTrack=0;
204   while (iTrack<nTracksMUON) {
205
206     fNPlanesMFTAnalyzed = 0;
207
208     AliInfo("****************************************************************************************");
209     AliInfo(Form("***************************   MUON TRACK %3d/%d   ***************************************", iTrack, nTracksMUON));
210     AliInfo("****************************************************************************************");
211     
212     fCandidateTracks -> Delete();
213     
214     fNPlanesMFTAnalyzed = 0;
215     
216     const AliESDMuonTrack *esdTrack = event->GetMuonTrack(iTrack);
217     if (fMUONTrack) delete fMUONTrack;
218     fMUONTrack = new AliMUONTrack();
219
220     AliMUONESDInterface::ESDToMUON(*esdTrack, *fMUONTrack, kFALSE);
221
222     if (!fMUONTrack->GetTrackParamAtCluster()->First()) {
223       AliInfo("Skipping track, no parameters available!!!");
224       iTrack++;
225       continue;
226     }
227
228     // the track we are going to build, starting from fMUONTrack and adding the MFT clusters
229     AliMuonForwardTrack *track = new ((*fCandidateTracks)[0]) AliMuonForwardTrack();
230     track -> SetMUONTrack(new AliMUONTrack(*fMUONTrack));
231     track -> SetMCLabel(fMUONTrack->GetMCLabel());
232     track -> SetMatchTrigger(fMUONTrack->GetMatchTrigger());
233
234     // track parameters linearly extrapolated from the first tracking station to the end of the absorber
235     AliMUONTrackParam trackParamEndOfAbsorber(*((AliMUONTrackParam*)(fMUONTrack->GetTrackParamAtCluster()->First())));
236     AliMUONTrackExtrap::ExtrapToZCov(&trackParamEndOfAbsorber, AliMUONConstants::AbsZEnd());   // absorber extends from -90 to -503 cm
237     Double_t xEndOfAbsorber = trackParamEndOfAbsorber.GetNonBendingCoor();
238     Double_t yEndOfAbsorber = trackParamEndOfAbsorber.GetBendingCoor();
239     Double_t rAbsorber      = TMath::Sqrt(xEndOfAbsorber*xEndOfAbsorber + yEndOfAbsorber*yEndOfAbsorber);
240     track -> SetRAtAbsorberEnd(rAbsorber);
241
242     //------------------------- NOW THE CYCLE OVER THE MFT PLANES STARTS ---------------------------------------
243
244     for (Int_t iPlane=fNPlanesMFT-1; iPlane>=0; iPlane--) {   /* *** do not reverse the order of this cycle!!! 
245                                                                  *** this reflects the fact that the extrapolation is performed 
246                                                                  *** starting from the last MFT plane back to the origin */
247       
248       // --------- updating the array of candidates according to the clusters available in the i-th plane ---------
249       
250       fNPlanesMFTAnalyzed++;
251       
252       Int_t nCandidates = fCandidateTracks->GetEntriesFast();
253       for (Int_t iCandidate=0; iCandidate<nCandidates; iCandidate++) {
254         fCurrentTrack = (AliMuonForwardTrack*) fCandidateTracks->UncheckedAt(iCandidate);
255
256         // if the old track is compatible with the new cluster, the track is updated and inserted as new track in the array 
257         // (several new tracks can be created for one old track)
258         if (FindClusterInPlane(iPlane) == kDiverged) {
259           fGlobalTrackingDiverged = kTRUE;
260           break;
261         }
262
263         if ((fNPlanesMFTAnalyzed-fCurrentTrack->GetNMFTClusters())>fNMaxMissingMFTClusters || fIsPlaneMandatory[iPlane]) {
264           fCandidateTracks->Remove(fCurrentTrack);     // the old track is removed after the check;
265         }
266       }
267       if (fGlobalTrackingDiverged) {
268         if (fScaleSigmaClusterCut>0) fScaleSigmaClusterCut -= 0.1;
269         continue;
270       }
271
272       fCandidateTracks->Compress();
273       
274     }      
275
276     // -------------------------- END OF THE CYCLE OVER THE MFT PLANES --------------------------------------------
277     
278     fGlobalTrackingDiverged = kFALSE;
279     fScaleSigmaClusterCut = 1.0;
280     
281     AliDebug(1, "Finished cycle over planes");
282     
283     iTrack++;
284
285     // If we have several final tracks, we must find the best candidate:
286     
287     Int_t nFinalTracks = fCandidateTracks->GetEntriesFast();
288     AliInfo(Form("nFinalTracks = %d", nFinalTracks));
289
290     Int_t nGoodClustersBestCandidate =  0;
291     Int_t idBestCandidate            =  0;
292     Double_t bestChi2                = -1.;  // variable defining the best candidate
293
294     for (Int_t iFinalCandidate=0; iFinalCandidate<nFinalTracks; iFinalCandidate++) {
295       
296       AliMuonForwardTrack *finalTrack = (AliMuonForwardTrack*) fCandidateTracks->UncheckedAt(iFinalCandidate);
297       Int_t nMFTClusters  = finalTrack->GetNMFTClusters();
298
299       Double_t chi2 = 0;
300       for (Int_t iCluster=0; iCluster<nMFTClusters; iCluster++) {
301         AliMFTCluster *localCluster = finalTrack->GetMFTCluster(iCluster);
302         chi2 += localCluster->GetLocalChi2();
303       }
304       chi2 /= nMFTClusters;
305
306       // now comparing the tracks in order to find the best one
307       
308       if (chi2<bestChi2 || bestChi2<0) {
309         bestChi2 = chi2;
310         idBestCandidate = iFinalCandidate;
311       }
312       
313     }
314     
315     if (nFinalTracks) {
316
317       AliMuonForwardTrack *newTrack = (AliMuonForwardTrack*) fCandidateTracks->UncheckedAt(idBestCandidate);
318       newTrack -> SetNWrongClustersMC(newTrack->GetNMFTClusters() - nGoodClustersBestCandidate);
319
320       new ((*muonForwardTracks)[muonForwardTracks->GetEntries()]) AliMuonForwardTrack(*newTrack);
321
322       //----------------------- Save the information to the AliESDMuonGlobalTrack object
323
324       newTrack -> EvalKinem(AliMFTConstants::fZEvalKinem);
325
326       AliDebug(2,"Creating a new Muon Global Track");
327       AliESDMuonGlobalTrack *myESDTrack = event->NewMuonGlobalTrack();
328       myESDTrack -> SetPxPyPz(newTrack->Px(), newTrack->Py(), newTrack->Pz());
329       
330       myESDTrack -> SetLabel(newTrack->GetMCLabel());
331       myESDTrack -> SetChi2OverNdf(newTrack->GetChi2OverNdf());
332       myESDTrack -> SetCharge(newTrack->GetCharge());
333       myESDTrack -> SetMatchTrigger(newTrack->GetMatchTrigger());
334       myESDTrack -> SetNMFTClusters(newTrack->GetNMFTClusters());
335       myESDTrack -> SetNWrongMFTClustersMC(newTrack->GetNWrongClustersMC());
336       myESDTrack -> SetFirstTrackingPoint(newTrack->GetMFTCluster(0)->GetX(), newTrack->GetMFTCluster(0)->GetY(), newTrack->GetMFTCluster(0)->GetZ());
337       myESDTrack -> SetXYAtVertex(newTrack->GetOffsetX(0., AliMFTConstants::fZEvalKinem), newTrack->GetOffsetY(0., AliMFTConstants::fZEvalKinem));
338       myESDTrack -> SetRAtAbsorberEnd(newTrack->GetRAtAbsorberEnd());
339       myESDTrack -> SetChi2MatchTrigger(esdTrack->GetChi2MatchTrigger());
340       myESDTrack -> SetMuonClusterMap(esdTrack->GetMuonClusterMap());
341       myESDTrack -> SetHitsPatternInTrigCh(esdTrack->GetHitsPatternInTrigCh());
342       myESDTrack -> SetHitsPatternInTrigChTrk(esdTrack->GetHitsPatternInTrigChTrk());
343       myESDTrack -> Connected(esdTrack->IsConnected());
344
345       ULong_t mftClusterPattern = 0;
346       for (Int_t iCluster=0; iCluster<newTrack->GetNMFTClusters(); iCluster++) {
347         AliMFTCluster *localCluster = newTrack->GetMFTCluster(iCluster);
348         mftClusterPattern |= 1 << localCluster->GetPlane();
349         mftClusterPattern |= IsCorrectMatch(localCluster, newTrack->GetMCLabel()) << fNMaxPlanes+localCluster->GetPlane();
350       }
351       myESDTrack -> SetMFTClusterPattern(mftClusterPattern);
352       
353       //---------------------------------------------------------------------------------
354
355     }
356
357     fFinalBestCandidate = NULL;
358    
359   }
360
361   outputTreeMuonGlobalTracks -> Fill();
362
363   Int_t myEventID = 0;
364   while (outputFileMuonGlobalTracks->cd(Form("Event%d",myEventID))) myEventID++;
365   outputFileMuonGlobalTracks -> mkdir(Form("Event%d",myEventID));
366   outputFileMuonGlobalTracks -> cd(Form("Event%d",myEventID));
367   outputTreeMuonGlobalTracks -> Write();
368   outputFileMuonGlobalTracks -> Close();
369
370   muonForwardTracks -> Delete();
371   delete muonForwardTracks;
372
373   return 0;
374
375 }
376
377 //=========================================================================================================================================
378
379 void AliMFTTrackerMU::SeparateFrontBackClusters() {
380
381   for (Int_t iPlane=0; iPlane<fNPlanesMFT; iPlane++) {
382     fMFTClusterArrayFront[iPlane]->Delete();
383     fMFTClusterArrayBack[iPlane] ->Delete();
384     for (Int_t iCluster=0; iCluster<fMFTClusterArray[iPlane]->GetEntries(); iCluster++) {
385       AliMFTCluster *cluster = (AliMFTCluster*) fMFTClusterArray[iPlane]->At(iCluster);
386       if (TMath::Abs(cluster->GetZ())<TMath::Abs(fSegmentation->GetPlane(iPlane)->GetZCenter())) {
387         new ((*fMFTClusterArrayFront[iPlane])[fMFTClusterArrayFront[iPlane]->GetEntries()]) AliMFTCluster(*cluster);
388       }
389       else {
390         new ((*fMFTClusterArrayBack[iPlane])[fMFTClusterArrayBack[iPlane]->GetEntries()]) AliMFTCluster(*cluster);
391       }
392     }
393   }
394
395 }
396
397 //==========================================================================================================================================
398
399 Int_t AliMFTTrackerMU::FindClusterInPlane(Int_t planeId) { 
400   
401   // !!!!!!!!! coordinates and errors on the interaction vertex should be taken from the event itself (ITS) if available
402
403   // propagate track to plane #planeId (both to front and back active sensors)
404   // look for compatible clusters
405   // update TrackParam at found cluster (if any) using Kalman Filter
406
407   AliMUONTrackParam currentParamFront, currentParamBack, currentParamForResearchFront, currentParamForResearchBack;
408
409   if (planeId == fNPlanesMFT-1) {      // last plane of the telecope
410     currentParamFront = (*((AliMUONTrackParam*)(fMUONTrack->GetTrackParamAtCluster()->First())));
411     currentParamBack  = (*((AliMUONTrackParam*)(fMUONTrack->GetTrackParamAtCluster()->First())));
412     currentParamForResearchFront = currentParamFront;
413     currentParamForResearchBack  = currentParamBack;
414     if (fBransonCorrection) {
415       AliMUONTrackExtrap::ExtrapToVertex(&currentParamFront, fXExtrapVertex, fYExtrapVertex, fZExtrapVertex, fXExtrapVertexError, fYExtrapVertexError); 
416       AliMUONTrackExtrap::ExtrapToVertex(&currentParamBack,  fXExtrapVertex, fYExtrapVertex, fZExtrapVertex, fXExtrapVertexError, fYExtrapVertexError); 
417     }
418     else {
419       AliMUONTrackExtrap::ExtrapToVertexWithoutBranson(&currentParamFront, fZExtrapVertex);
420       AliMUONTrackExtrap::ExtrapToVertexWithoutBranson(&currentParamBack,  fZExtrapVertex);
421     }
422     AliMUONTrackExtrap::ExtrapToVertex(&currentParamForResearchFront, fXExtrapVertex, fYExtrapVertex, fZExtrapVertex, fXExtrapVertexError, fYExtrapVertexError); 
423     AliMUONTrackExtrap::ExtrapToVertex(&currentParamForResearchBack,  fXExtrapVertex, fYExtrapVertex, fZExtrapVertex, fXExtrapVertexError, fYExtrapVertexError); 
424   }
425   else {          // MFT planes others than the last one: mult. scattering correction because of the upstream MFT planes is performed
426     currentParamFront = (*((AliMUONTrackParam*)(fCurrentTrack->GetTrackParamAtCluster()->First())));
427     currentParamBack  = (*((AliMUONTrackParam*)(fCurrentTrack->GetTrackParamAtCluster()->First())));
428     currentParamForResearchFront = currentParamFront;
429     currentParamForResearchBack  = currentParamBack;
430     AliMUONTrackExtrap::AddMCSEffect(&currentParamFront,           (fSegmentation->GetPlane(planeId+1)->GetEquivalentSilicon()+
431                                                                     fSegmentation->GetPlane(planeId)->GetEquivalentSiliconBeforeFront())/fRadLengthSi,-1.);
432     AliMUONTrackExtrap::AddMCSEffect(&currentParamForResearchFront,(fSegmentation->GetPlane(planeId+1)->GetEquivalentSilicon()+
433                                                                     fSegmentation->GetPlane(planeId)->GetEquivalentSiliconBeforeFront())/fRadLengthSi,-1.);
434     AliMUONTrackExtrap::AddMCSEffect(&currentParamBack,            (fSegmentation->GetPlane(planeId+1)->GetEquivalentSilicon()+
435                                                                     fSegmentation->GetPlane(planeId)->GetEquivalentSiliconBeforeBack())/fRadLengthSi,-1.);
436     AliMUONTrackExtrap::AddMCSEffect(&currentParamForResearchBack, (fSegmentation->GetPlane(planeId+1)->GetEquivalentSilicon()+
437                                                                     fSegmentation->GetPlane(planeId)->GetEquivalentSiliconBeforeBack())/fRadLengthSi,-1.);
438   }
439   // for all planes: extrapolation to the Z of the plane
440   AliMUONTrackExtrap::ExtrapToZCov(&currentParamFront,            -1.*fSegmentation->GetPlane(planeId)->GetZCenterActiveFront());   
441   AliMUONTrackExtrap::ExtrapToZCov(&currentParamForResearchFront, -1.*fSegmentation->GetPlane(planeId)->GetZCenterActiveFront());
442   AliMUONTrackExtrap::ExtrapToZCov(&currentParamBack,             -1.*fSegmentation->GetPlane(planeId)->GetZCenterActiveBack());   
443   AliMUONTrackExtrap::ExtrapToZCov(&currentParamForResearchBack,  -1.*fSegmentation->GetPlane(planeId)->GetZCenterActiveBack());
444
445   //---------------------------------------------------------------------------------------
446
447   TMatrixD covFront(5,5); covFront = currentParamForResearchFront.GetCovariances();
448   TMatrixD covBack(5,5);  covBack  = currentParamForResearchBack.GetCovariances();
449   
450   Double_t squaredError_X_Front = covFront(0,0);
451   Double_t squaredError_Y_Front = covFront(2,2);
452   Double_t squaredError_X_Back  = covBack(0,0);
453   Double_t squaredError_Y_Back  = covBack(2,2);
454
455   Double_t corrFact = 1.0;
456
457   Double_t researchRadiusFront = TMath::Sqrt(squaredError_X_Front + squaredError_Y_Front);
458   Double_t researchRadiusBack  = TMath::Sqrt(squaredError_X_Back  + squaredError_Y_Back);
459   if (0.5*(researchRadiusFront+researchRadiusBack)<fMinResearchRadiusAtPlane[planeId]) {
460     corrFact = fMinResearchRadiusAtPlane[planeId]/(0.5*(researchRadiusFront+researchRadiusBack));
461   }
462
463   //---------------------------------------------------------------------------------------
464
465   Double_t chi2cut = 2.*fScaleSigmaClusterCut*fScaleSigmaClusterCut*fSigmaClusterCut*fSigmaClusterCut;     // depends on the number of variables (here, 2)
466   
467   // Analyizing the clusters: FRONT ACTIVE ELEMENTS
468   
469   Int_t nClustersFront = fMFTClusterArrayFront[planeId]->GetEntries();
470   AliDebug(2, Form("There are %3d clusters in plane %02d FRONT\n", nClustersFront, planeId));
471   
472   for (Int_t iCluster=0; iCluster<nClustersFront; iCluster++) {
473
474     Bool_t isGoodChi2 = kFALSE;
475
476     AliMFTCluster *cluster = (AliMFTCluster*) fMFTClusterArrayFront[planeId]->At(iCluster); 
477     Double_t chi2 = (1./(corrFact*corrFact)) * TryOneCluster(currentParamForResearchFront, cluster);     // describes the compatibility between the track and the cluster
478     if (chi2<chi2cut) isGoodChi2 = kTRUE;
479
480     if (isGoodChi2) {
481       AliDebug(3, Form("accepting cluster: chi2=%f (cut = %f)\n", chi2, chi2cut));
482       AliMuonForwardTrack *newTrack = new ((*fCandidateTracks)[fCandidateTracks->GetEntriesFast()]) AliMuonForwardTrack(*fCurrentTrack);
483       if (fCandidateTracks->GetEntriesFast() > fMaxNCandidates) return kDiverged;
484       newTrack->AddTrackParamAtMFTCluster(currentParamFront, *cluster);    // creating new track param and attaching the cluster
485       AliDebug(2, Form("After plane %02d: newTrack->GetNMFTClusters() = %d (fCurrentTrack->GetNMFTClusters() = %d)", 
486                        planeId, newTrack->GetNMFTClusters(), fCurrentTrack->GetNMFTClusters()));
487       newTrack->SetPlaneExists(planeId);
488     }
489     else AliDebug(3, Form("discarding cluster: chi2=%f (cut = %f)\n", chi2, chi2cut));
490
491   }
492
493   // Analyizing the clusters: BACK ACTIVE ELEMENTS
494   
495   Int_t nClustersBack = fMFTClusterArrayBack[planeId]->GetEntries();
496   AliDebug(2, Form("There are %3d clusters in plane %02d BACK\n", nClustersBack, planeId));
497   
498   for (Int_t iCluster=0; iCluster<nClustersBack; iCluster++) {
499
500     Bool_t isGoodChi2 = kFALSE;
501
502     AliMFTCluster *cluster = (AliMFTCluster*) fMFTClusterArrayBack[planeId]->At(iCluster); 
503     Double_t chi2 = (1./(corrFact*corrFact)) * TryOneCluster(currentParamForResearchBack, cluster);     // describes the compatibility between the track and the cluster
504     if (chi2<chi2cut) isGoodChi2 = kTRUE;
505
506     if (isGoodChi2) {
507       AliDebug(3,Form("accepting cluster: chi2=%f (cut = %f)\n", chi2, chi2cut));
508       AliMuonForwardTrack *newTrack = new ((*fCandidateTracks)[fCandidateTracks->GetEntriesFast()]) AliMuonForwardTrack(*fCurrentTrack);
509       if (fCandidateTracks->GetEntriesFast() > fMaxNCandidates) return kDiverged;
510       newTrack->AddTrackParamAtMFTCluster(currentParamBack, *cluster);    // creating new track param and attaching the cluster
511       AliDebug(2, Form("After plane %02d: newTrack->GetNMFTClusters() = %d (fCurrentTrack->GetNMFTClusters() = %d)", 
512                        planeId, newTrack->GetNMFTClusters(), fCurrentTrack->GetNMFTClusters()));
513       newTrack->SetPlaneExists(planeId);
514     }
515     else AliDebug(3,Form("discarding cluster: chi2=%f (cut = %f)\n", chi2, chi2cut));
516
517   }
518
519   //---------------------------------------------------------------------------------------------
520
521   return kConverged;
522   
523 }
524
525 //==========================================================================================================================================
526
527 Double_t AliMFTTrackerMU::TryOneCluster(const AliMUONTrackParam &trackParam, AliMFTCluster *cluster) {
528
529   // Test the compatibility between the track and the cluster (using trackParam's covariance matrix):
530   // return the corresponding Chi2
531   // assume the track parameters are given at the Z of the cluster
532   
533   // Set differences between trackParam and cluster in the bending and non bending directions
534   Double_t dX = cluster->GetX() - trackParam.GetNonBendingCoor();
535   Double_t dY = cluster->GetY() - trackParam.GetBendingCoor();
536   AliDebug(3,Form("dX = %f, dY = %f\n", dX, dY));
537   
538   // Calculate errors and covariances
539   const TMatrixD& kParamCov = trackParam.GetCovariances();
540   Double_t sigmaX2 = kParamCov(0,0) + cluster->GetErrX2();
541   Double_t sigmaY2 = kParamCov(2,2) + cluster->GetErrY2();
542   AliDebug(3, Form("dX2 = %f, dY2 = %f\n", sigmaX2, sigmaY2));
543   Double_t covXY   = kParamCov(0,2);
544   Double_t det     = sigmaX2 * sigmaY2 - covXY * covXY;
545   
546   // Compute chi2
547   if (det==0.) return 1.e10;
548   return (dX*dX*sigmaY2 + dY*dY*sigmaX2 - 2.*dX*dY*covXY) / det;
549   
550 }
551
552 //=========================================================================================================================================
553
554 Bool_t AliMFTTrackerMU::IsCorrectMatch(AliMFTCluster *cluster, Int_t labelMC) {
555
556   Bool_t result = kFALSE;
557
558   // check if the cluster belongs to the correct MC track
559
560   for (Int_t iTrack=0; iTrack<cluster->GetNMCTracks(); iTrack++) {
561     if (cluster->GetMCLabel(iTrack)==labelMC) {
562       result = kTRUE;
563       break;
564     }
565   }
566
567   return result;
568
569 }
570
571 //======================================================================================================================================
572
573 void AliMFTTrackerMU::GetVertexFromMC() {
574
575   AliRunLoader *fRunLoader = AliRunLoader::Open("galice.root");
576   if (!fRunLoader) {
577     AliError("no run loader found in file galice.root");
578     return;
579   }
580
581   fRunLoader->CdGAFile();
582   fRunLoader->LoadgAlice();
583   fRunLoader->LoadHeader();
584   fRunLoader->GetEvent(gAlice->GetEvNumber());
585   
586   TArrayF vtx(3);
587   fRunLoader->GetHeader()->GenEventHeader()->PrimaryVertex(vtx);
588
589   fXExtrapVertex = gRandom->Gaus(vtx[0], AliMFTConstants::fPrimaryVertexResX);
590   fYExtrapVertex = gRandom->Gaus(vtx[1], AliMFTConstants::fPrimaryVertexResY);
591   fZExtrapVertex = gRandom->Gaus(vtx[2], AliMFTConstants::fPrimaryVertexResZ);
592   fXExtrapVertexError = AliMFTConstants::fXVertexTolerance;
593   fYExtrapVertexError = AliMFTConstants::fYVertexTolerance;
594   AliInfo(Form("Set ESD vertex from MC (%f +/- %f,  %f +/- %f,  %f)", 
595                fXExtrapVertex,fXExtrapVertexError,fYExtrapVertex,fYExtrapVertexError,fZExtrapVertex));
596   
597 }
598
599 //======================================================================================================================================
600