]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONEventReconstructor.cxx
Major revision of the Kalman tracking: works with new data structure,
[u/mrichter/AliRoot.git] / MUON / AliMUONEventReconstructor.cxx
index 0dad50e8eb8ef06085324ae7e75441875fc7ae15..47442b251222bebe595cd3047ce768f7e1abca88 100644 (file)
 //
 ////////////////////////////////////
 
-#include <Riostream.h> // for cout
-#include <stdlib.h> // for exit()
-
-#include <TTree.h>
+#include <stdlib.h>
+#include <Riostream.h>
+#include <TDirectory.h>
+#include <TFile.h>
+#include <TMatrixD.h> //AZ
 
-#include "AliMUON.h"
-//#include "AliMUONChamber.h"
 #include "AliMUONEventReconstructor.h"
+#include "AliMUON.h"
+#include "AliMUONHit.h"
 #include "AliMUONHitForRec.h"
 #include "AliMUONTriggerTrack.h"
-//#include "AliMUONTriggerConstants.h"
 #include "AliMUONTriggerCircuit.h"
 #include "AliMUONRawCluster.h"
 #include "AliMUONLocalTrigger.h"
 #include "AliMUONTrackHit.h"
 #include "AliMagF.h"
 #include "AliRun.h" // for gAlice
-#include "AliConfig.h"
 #include "AliRunLoader.h"
 #include "AliLoader.h"
 #include "AliMUONTrackK.h" //AZ
-#include <TMatrixD.h> //AZ
 #include "AliMC.h"
+#include "AliLog.h"
 
 //************* Defaults parameters for reconstruction
-static const Double_t kDefaultMinBendingMomentum = 3.0;
-static const Double_t kDefaultMaxBendingMomentum = 500.0;
-static const Double_t kDefaultMaxChi2 = 100.0;
-static const Double_t kDefaultMaxSigma2Distance = 16.0;
-static const Double_t kDefaultBendingResolution = 0.01;
-static const Double_t kDefaultNonBendingResolution = 0.144;
-static const Double_t kDefaultChamberThicknessInX0 = 0.03;
+const Double_t AliMUONEventReconstructor::fgkDefaultMinBendingMomentum = 3.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxBendingMomentum = 500.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxChi2 = 100.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultMaxSigma2Distance = 16.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultBendingResolution = 0.01;
+const Double_t AliMUONEventReconstructor::fgkDefaultNonBendingResolution = 0.144;
+const Double_t AliMUONEventReconstructor::fgkDefaultChamberThicknessInX0 = 0.03;
 // Simple magnetic field:
 // Value taken from macro MUONtracking.C: 0.7 T, hence 7 kG
 // Length and Position from reco_muon.F, with opposite sign:
 // Length = ZMAGEND-ZCOIL
 // Position = (ZMAGEND+ZCOIL)/2
 // to be ajusted differently from real magnetic field ????
-static const Double_t kDefaultSimpleBValue = 7.0;
-static const Double_t kDefaultSimpleBLength = 428.0;
-static const Double_t kDefaultSimpleBPosition = 1019.0;
-static const Int_t kDefaultRecGeantHits = 0;
-static const Double_t kDefaultEfficiency = 0.95;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBValue = 7.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBLength = 428.0;
+const Double_t AliMUONEventReconstructor::fgkDefaultSimpleBPosition = 1019.0;
+const Int_t    AliMUONEventReconstructor::fgkDefaultRecGeantHits = 0;
+const Double_t AliMUONEventReconstructor::fgkDefaultEfficiency = 0.95;
 
-static const Int_t kDefaultPrintLevel = -1;
+const Int_t    AliMUONEventReconstructor::fgkDefaultPrintLevel = -1;
 
 ClassImp(AliMUONEventReconstructor) // Class implementation in ROOT context
 
   //__________________________________________________________________________
 AliMUONEventReconstructor::AliMUONEventReconstructor(AliLoader* loader)
+  : TObject()
 {
   // Constructor for class AliMUONEventReconstructor
   SetReconstructionParametersToDefaults();
@@ -96,7 +96,7 @@ AliMUONEventReconstructor::AliMUONEventReconstructor(AliLoader* loader)
   fNHitsForRec = 0; // really needed or GetEntriesFast sufficient ????
   // Memory allocation for the TClonesArray's of segments in stations
   // Is 2000 the right size ????
-  for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+  for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
     fSegmentsPtr[st] = new TClonesArray("AliMUONSegment", 2000);
     fNSegments[st] = 0; // really needed or GetEntriesFast sufficient ????
   }
@@ -109,7 +109,8 @@ AliMUONEventReconstructor::AliMUONEventReconstructor(AliLoader* loader)
   fNRecTriggerTracks = 0; // really needed or GetEntriesFast sufficient ????
   // Memory allocation for the TClonesArray of hits on reconstructed tracks
   // Is 100 the right size ????
-  fRecTrackHitsPtr = new TClonesArray("AliMUONTrack", 100);
+  //AZ fRecTrackHitsPtr = new TClonesArray("AliMUONTrack", 100);
+  fRecTrackHitsPtr = new TClonesArray("AliMUONTrackHit", 100);
   fNRecTrackHits = 0; // really needed or GetEntriesFast sufficient ????
 
   // Sign of fSimpleBValue according to sign of Bx value at (50,50,-950).
@@ -155,15 +156,24 @@ AliMUONEventReconstructor::AliMUONEventReconstructor(AliLoader* loader)
   return;
 }
   //__________________________________________________________________________
-AliMUONEventReconstructor::AliMUONEventReconstructor (const AliMUONEventReconstructor& Reconstructor):TObject(Reconstructor)
+AliMUONEventReconstructor::AliMUONEventReconstructor (const AliMUONEventReconstructor& rhs)
+  : TObject(rhs)
 {
-  // Dummy copy constructor
+// Protected copy constructor
+
+  AliFatal("Not implemented.");
 }
 
-AliMUONEventReconstructor & AliMUONEventReconstructor::operator=(const AliMUONEventReconstructor& /*Reconstructor*/)
+AliMUONEventReconstructor & 
+AliMUONEventReconstructor::operator=(const AliMUONEventReconstructor& rhs)
 {
-  // Dummy assignment operator
-    return *this;
+// Protected assignement operator
+
+  if (this == &rhs) return *this;
+
+  AliFatal("Not implemented.");
+    
+  return *this;  
 }
 
   //__________________________________________________________________________
@@ -177,7 +187,7 @@ AliMUONEventReconstructor::~AliMUONEventReconstructor(void)
 //  if (fEventTree) delete fEventTree;
   if (fRecoEvent) delete fRecoEvent;
   delete fHitsForRecPtr; // Correct destruction of everything ???? or delete [] ????
-  for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
+  for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
     delete fSegmentsPtr[st]; // Correct destruction of everything ????
   return;
 }
@@ -187,10 +197,10 @@ void AliMUONEventReconstructor::SetReconstructionParametersToDefaults(void)
 {
   // Set reconstruction parameters to default values
   // Would be much more convenient with a structure (or class) ????
-  fMinBendingMomentum = kDefaultMinBendingMomentum;
-  fMaxBendingMomentum = kDefaultMaxBendingMomentum;
-  fMaxChi2 = kDefaultMaxChi2;
-  fMaxSigma2Distance = kDefaultMaxSigma2Distance;
+  fMinBendingMomentum = fgkDefaultMinBendingMomentum;
+  fMaxBendingMomentum = fgkDefaultMaxBendingMomentum;
+  fMaxChi2 = fgkDefaultMaxChi2;
+  fMaxSigma2Distance = fgkDefaultMaxSigma2Distance;
 
   AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON");
   // ******** Parameters for making HitsForRec
@@ -198,7 +208,7 @@ void AliMUONEventReconstructor::SetReconstructionParametersToDefaults(void)
   // like in TRACKF_STAT:
   // 2 degrees for stations 1 and 2, or ch(0...) from 0 to 3;
   // 30 cm for stations 3 to 5, or ch(0...) from 4 to 9
-  for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+  for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
     if (ch < 4) fRMin[ch] = TMath::Abs((&(pMUON->Chamber(ch)))->Z()) *
                  2.0 * TMath::Pi() / 180.0;
     else fRMin[ch] = 30.0;
@@ -213,31 +223,31 @@ void AliMUONEventReconstructor::SetReconstructionParametersToDefaults(void)
   // Maximum distance in non bending plane
   // 5 * 0.22 just to remember the way it was made in TRACKF_STAT
   // SIGCUT*DYMAX(IZ)
-  for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
+  for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
     fSegmentMaxDistNonBending[st] = 5. * 0.22;
   // Maximum distance in bending plane:
   // values from TRACKF_STAT, corresponding to (J psi 20cm),
   // scaled to the real distance between chambers in a station
   fSegmentMaxDistBending[0] = TMath::Abs( 1.5 *
     ((&(pMUON->Chamber(1)))->Z() - (&(pMUON->Chamber(0)))->Z()) / 20.0);
-  fSegmentMaxDistBending[1] =  TMath::Abs( 1.5 *
+  fSegmentMaxDistBending[1] = TMath::Abs( 1.5 *
     ((&(pMUON->Chamber(3)))->Z() - (&(pMUON->Chamber(2)))->Z()) / 20.0);
-  fSegmentMaxDistBending[2] =  TMath::Abs( 3.0 *
+  fSegmentMaxDistBending[2] = TMath::Abs( 3.0 *
     ((&(pMUON->Chamber(5)))->Z() - (&(pMUON->Chamber(4)))->Z()) / 20.0);
-  fSegmentMaxDistBending[3] =  TMath::Abs( 6.0 *
+  fSegmentMaxDistBending[3] = TMath::Abs( 6.0 *
     ((&(pMUON->Chamber(7)))->Z() - (&(pMUON->Chamber(6)))->Z()) / 20.0);
-  fSegmentMaxDistBending[4] =  TMath::Abs( 6.0 *
+  fSegmentMaxDistBending[4] = TMath::Abs( 6.0 *
     ((&(pMUON->Chamber(9)))->Z() - (&(pMUON->Chamber(8)))->Z()) / 20.0);
   
-  fBendingResolution = kDefaultBendingResolution;
-  fNonBendingResolution = kDefaultNonBendingResolution;
-  fChamberThicknessInX0 = kDefaultChamberThicknessInX0;
-  fSimpleBValue = kDefaultSimpleBValue;
-  fSimpleBLength = kDefaultSimpleBLength;
-  fSimpleBPosition = kDefaultSimpleBPosition;
-  fRecGeantHits = kDefaultRecGeantHits;
-  fEfficiency = kDefaultEfficiency;
-  fPrintLevel = kDefaultPrintLevel;
+  fBendingResolution = fgkDefaultBendingResolution;
+  fNonBendingResolution = fgkDefaultNonBendingResolution;
+  fChamberThicknessInX0 = fgkDefaultChamberThicknessInX0;
+  fSimpleBValue = fgkDefaultSimpleBValue;
+  fSimpleBLength = fgkDefaultSimpleBLength;
+  fSimpleBPosition = fgkDefaultSimpleBPosition;
+  fRecGeantHits = fgkDefaultRecGeantHits;
+  fEfficiency = fgkDefaultEfficiency;
+  fPrintLevel = fgkDefaultPrintLevel;
   return;
 }
 
@@ -336,9 +346,8 @@ void AliMUONEventReconstructor::NextBkgGeantEvent(void)
     sprintf(treeName, "TreeK%d", fBkgGeantEventNumber);
     fBkgGeantTK = (TTree*)gDirectory->Get(treeName);
     if (!fBkgGeantTK) {
-      cout << "ERROR: cannot find Kine Tree for background event: " <<
-       fBkgGeantEventNumber << endl;
-      exit(0);
+               AliError(Form("cannot find Kine Tree for background event: %d",fBkgGeantEventNumber));
+       exit(0);
     }
   }
   if (fBkgGeantTK) 
@@ -348,8 +357,7 @@ void AliMUONEventReconstructor::NextBkgGeantEvent(void)
   sprintf(treeName, "TreeH%d", fBkgGeantEventNumber);
   fBkgGeantTH = (TTree*)gDirectory->Get(treeName);
   if (!fBkgGeantTH) {
-    cout << "ERROR: cannot find Hits Tree for background event: " <<
-      fBkgGeantEventNumber << endl;
+         AliError(Form("cannot find Hits Tree for background event: %d",fBkgGeantEventNumber));
       exit(0);
   }
   if (fBkgGeantTH && fBkgGeantHits) {
@@ -400,7 +408,7 @@ void AliMUONEventReconstructor::ResetHitsForRec(void)
   // and the index of the first HitForRec per chamber
   if (fHitsForRecPtr) fHitsForRecPtr->Clear();
   fNHitsForRec = 0;
-  for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++)
+  for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++)
     fNHitsForRecPerChamber[ch] = fIndexOfFirstHitForRecPerChamber[ch] = 0;
   return;
 }
@@ -410,7 +418,7 @@ void AliMUONEventReconstructor::ResetSegments(void)
 {
   // To reset the TClonesArray of segments and the number of Segments
   // for all stations
-  for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+  for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
     if (fSegmentsPtr[st]) fSegmentsPtr[st]->Clear();
     fNSegments[st] = 0;
   }
@@ -454,7 +462,7 @@ void AliMUONEventReconstructor::MakeEventToBeReconstructed(void)
   // To make the list of hits to be reconstructed,
   // either from the GEANT hits or from the raw clusters
   // according to the parameter set for the reconstructor
-//   TString evfoldname = AliConfig::fgkDefaultEventFolderName;//to be interfaced properly
+//   TString evfoldname = AliConfig::GetDefaultEventFolderName();//to be interfaced properly
   
 //   AliRunLoader* rl = AliRunLoader::GetRunLoader(evfoldname);
 //   if (rl == 0x0)
@@ -482,13 +490,13 @@ void AliMUONEventReconstructor::MakeEventToBeReconstructed(void)
          Int_t retval = fLoader->LoadHits();
          if ( retval)
           {
-            Error("MakeEventToBeReconstructed","Error occured while loading hits.");
+            AliError("Error occured while loading hits.");
             return;
           }
          treeH = fLoader->TreeH();
          if (treeH == 0x0)
           {
-           Error("MakeEventToBeReconstructed","Can not get TreeH");
+           AliError("Can not get TreeH");
            return;
           }
        }
@@ -514,7 +522,7 @@ void AliMUONEventReconstructor::MakeEventToBeReconstructed(void)
   if (fPrintLevel >= 10) {
     cout << "end of MakeEventToBeReconstructed" << endl;
     cout << "NHitsForRec: " << fNHitsForRec << endl;
-    for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+    for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
       cout << "chamber(0...): " << ch
           << "  NHitsForRec: " << fNHitsForRecPerChamber[ch]
           << "  index(first HitForRec): " << fIndexOfFirstHitForRecPerChamber[ch]
@@ -622,7 +630,7 @@ AliMUONHitForRec* AliMUONEventReconstructor::NewHitForRecFromGEANT(AliMUONHit* H
   Double_t bendCoor, nonBendCoor, radius;
   Int_t chamber = Hit->Chamber() - 1; // chamber(0...)
   // only in tracking chambers (fChamber starts at 1)
-  if (chamber >= kMaxMuonTrackingChambers) return NULL;
+  if (chamber >= fgkMaxMuonTrackingChambers) return NULL;
   // only if hit is efficient (keep track for checking ????)
   if (gRandom->Rndm() > fEfficiency) return NULL;
   // only if radius between RMin and RMax
@@ -665,7 +673,7 @@ void AliMUONEventReconstructor::SortHitsForRecWithIncreasingChamber()
   // Update the information for HitsForRec per chamber too.
   Int_t ch, nhits, prevch;
   fHitsForRecPtr->Sort();
-  for (ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+  for (ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
     fNHitsForRecPerChamber[ch] = 0;
     fIndexOfFirstHitForRecPerChamber[ch] = 0;
   }
@@ -700,7 +708,7 @@ void AliMUONEventReconstructor::SortHitsForRecWithIncreasingChamber()
 //   AliMUON *MUON  = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
 //   // Security on MUON ????
 //   // Loop over tracking chambers
-//   for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+//   for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
 //     // number of HitsForRec to 0 for the chamber
 //     fNHitsForRecPerChamber[ch] = 0;
 //     // index of first HitForRec for the chamber
@@ -748,7 +756,7 @@ void AliMUONEventReconstructor::AddHitsForRecFromRawClusters(TTree* TR)
   TClonesArray *rawclusters;
   if (fPrintLevel >= 1) cout << "enter AddHitsForRecFromRawClusters" << endl;
 
-//   TString evfoldname = AliConfig::fgkDefaultEventFolderName;//to be interfaced properly
+//   TString evfoldname = AliConfig::GetDefaultEventFolderName();//to be interfaced properly
 //   AliRunLoader* rl = AliRunLoader::GetRunLoader(evfoldname);
 //   if (rl == 0x0)
 //    {
@@ -772,15 +780,13 @@ void AliMUONEventReconstructor::AddHitsForRecFromRawClusters(TTree* TR)
 
   nTRentries = Int_t(TR->GetEntries());
   if (nTRentries != 1) {
-    cout << "Error in AliMUONEventReconstructor::AddHitsForRecFromRawClusters"
-        << endl;
-    cout << "nTRentries = " << nTRentries << " not equal to 1" << endl;
+    AliError(Form("nTRentries = %d not equal to 1 ",nTRentries));
     exit(0);
   }
   fLoader->TreeR()->GetEvent(0); // only one entry  
 
   // Loop over tracking chambers
-  for (Int_t ch = 0; ch < kMaxMuonTrackingChambers; ch++) {
+  for (Int_t ch = 0; ch < fgkMaxMuonTrackingChambers; ch++) {
     // number of HitsForRec to 0 for the chamber
     fNHitsForRecPerChamber[ch] = 0;
     // index of first HitForRec for the chamber
@@ -806,7 +812,7 @@ void AliMUONEventReconstructor::AddHitsForRecFromRawClusters(TTree* TR)
       hitForRec->SetChamberNumber(ch);
       hitForRec->SetHitNumber(iclus);
       // Z coordinate of the raw cluster (cm)
-      hitForRec->SetZ(clus->fZ[0]);
+      hitForRec->SetZ(clus->GetZ(0));
       if (fPrintLevel >= 10) {
        cout << "chamber (0...): " << ch <<
          " raw cluster (0...): " << iclus << endl;
@@ -828,12 +834,12 @@ void AliMUONEventReconstructor::MakeSegments(void)
   ResetSegments();
   // Loop over stations
   Int_t nb = (fTrackMethod == 2) ? 3 : 0; //AZ
-  //AZ for (Int_t st = 0; st < kMaxMuonTrackingStations; st++)
-  for (Int_t st = nb; st < kMaxMuonTrackingStations; st++) //AZ
+  //AZ for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++)
+  for (Int_t st = nb; st < fgkMaxMuonTrackingStations; st++) //AZ
     MakeSegmentsPerStation(st); 
   if (fPrintLevel >= 10) {
     cout << "end of MakeSegments" << endl;
-    for (Int_t st = 0; st < kMaxMuonTrackingStations; st++) {
+    for (Int_t st = 0; st < fgkMaxMuonTrackingStations; st++) {
       cout << "station(0...): " << st
           << "  Segments: " << fNSegments[st]
           << endl;
@@ -861,7 +867,6 @@ void AliMUONEventReconstructor::MakeSegmentsPerStation(Int_t Station)
   Bool_t last2st;
   Double_t bendingSlope, distBend, distNonBend, extBendCoor, extNonBendCoor,
       impactParam = 0., maxImpactParam = 0., minImpactParam = 0.; // =0 to avoid compilation warnings.
-  AliMUON *pMUON  = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
   if (fPrintLevel >= 1)
     cout << "enter MakeSegmentsPerStation (0...) " << Station << endl;
   // first and second chambers (0...) in the station
@@ -881,8 +886,7 @@ void AliMUONEventReconstructor::MakeSegmentsPerStation(Int_t Station)
   else last2st = kFALSE;
   // extrapolation factor from Z of first chamber to Z of second chamber
   // dZ to be changed to take into account fine structure of chambers ????
-  Double_t extrapFact =
-    (&(pMUON->Chamber(ch2)))->Z() / (&(pMUON->Chamber(ch1)))->Z();
+  Double_t extrapFact;
   // index for current segment
   Int_t segmentIndex = 0;
   // Loop over HitsForRec in the first chamber of the station
@@ -894,8 +898,6 @@ void AliMUONEventReconstructor::MakeSegmentsPerStation(Int_t Station)
     // extrapolation,
     // on the straight line joining the HitForRec to the vertex (0,0,0),
     // to the Z of the second chamber of the station
-    extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
-    extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
     // Loop over HitsForRec in the second chamber of the station
     for (Int_t hit2 = fIndexOfFirstHitForRecPerChamber[ch2];
         hit2 < fIndexOfFirstHitForRecPerChamber[ch2] + fNHitsForRecPerChamber[ch2];
@@ -905,6 +907,9 @@ void AliMUONEventReconstructor::MakeSegmentsPerStation(Int_t Station)
       // absolute values of distances, in bending and non bending planes,
       // between the HitForRec in the second chamber
       // and the previous extrapolation
+      extrapFact = hit2Ptr->GetZ()/ hit1Ptr->GetZ();
+      extBendCoor = extrapFact * hit1Ptr->GetBendingCoor();
+      extNonBendCoor = extrapFact * hit1Ptr->GetNonBendingCoor();
       distBend = TMath::Abs(hit2Ptr->GetBendingCoor() - extBendCoor);
       distNonBend = TMath::Abs(hit2Ptr->GetNonBendingCoor() - extNonBendCoor);
       if (last2st) {
@@ -969,12 +974,15 @@ void AliMUONEventReconstructor::MakeTracks(void)
   ResetTrackHits();
   if (fTrackMethod == 2) { //AZ - Kalman filter
     MakeTrackCandidatesK();
+    if (fRecTracksPtr->GetEntriesFast() == 0) return;
     // Follow tracks in stations(1..) 3, 2 and 1
     FollowTracksK();
     // Remove double tracks
     RemoveDoubleTracksK();
     // Propagate tracks to the vertex thru absorber
     GoToVertex();
+    // Fill AliMUONTrack data members
+    FillMUONTrack();
   } else { 
     // Look for candidates from at least 3 aligned points in stations(1..) 4 and 5
     MakeTrackCandidates();
@@ -983,6 +991,7 @@ void AliMUONEventReconstructor::MakeTracks(void)
     // Remove double tracks
     RemoveDoubleTracks();
     UpdateTrackParamAtHit();
+    UpdateHitForRecAtHit();
   }
   return;
 }
@@ -990,18 +999,21 @@ void AliMUONEventReconstructor::MakeTracks(void)
   //__________________________________________________________________________
 void AliMUONEventReconstructor::ValidateTracksWithTrigger(void)
 {
+  // Try to match track from tracking system with trigger track
   AliMUONTrack *track;
-  TClonesArray *recTriggerTracks = fMUONData->RecTriggerTracks();
-  if (recTriggerTracks == 0x0) {
-    fMUONData->SetTreeAddress("RL");
-    fMUONData->GetRecTriggerTracks();
-    recTriggerTracks = fMUONData->RecTriggerTracks();
-  } 
+  TClonesArray *recTriggerTracks;
+  
+  fMUONData->ResetRecTriggerTracks();
+  fMUONData->SetTreeAddress("RL");
+  fMUONData->GetRecTriggerTracks();
+  recTriggerTracks = fMUONData->RecTriggerTracks();
+
   track = (AliMUONTrack*) fRecTracksPtr->First();
   while (track) {
     track->MatchTriggerTrack(recTriggerTracks);
     track = (AliMUONTrack*) fRecTracksPtr->After(track);
-  } 
+  }
+
 }
 
   //__________________________________________________________________________
@@ -1020,48 +1032,48 @@ Bool_t AliMUONEventReconstructor::MakeTriggerTracks(void)
     AliMUONTriggerCircuit *circuit;
     AliMUONTriggerTrack *recTriggerTrack = 0;
 
-    TTree* TR = fLoader->TreeR();
+    TTree* treeR = fLoader->TreeR();
     
     // Loading MUON subsystem
     AliMUON * pMUON = (AliMUON *) gAlice->GetDetector("MUON");
     
-    nTRentries = Int_t(TR->GetEntries());
+    nTRentries = Int_t(treeR->GetEntries());
      
-    TR->GetEvent(0); // only one entry  
+    treeR->GetEvent(0); // only one entry  
 
     if (!(fMUONData->IsTriggerBranchesInTree())) {
-      cout << "Warning in AliMUONEventReconstructor::MakeTriggerTracks"
-          << endl;
-      cout << "Trigger information is not avalaible, nTRentries = " << nTRentries << " not equal to 1" << endl;
+      AliWarning(Form("Trigger information is not avalaible, nTRentries = %d not equal to 1",nTRentries));
       return kFALSE;
     }
 
-    fMUONData->SetTreeAddress("GLT");
+    fMUONData->SetTreeAddress("TC");
     fMUONData->GetTrigger();
 
     // global trigger for trigger pattern
     gloTrigPat = 0;
     globalTrigger = fMUONData->GlobalTrigger(); 
     gloTrg = (AliMUONGlobalTrigger*)globalTrigger->UncheckedAt(0);     
-    if (gloTrg->SinglePlusLpt())  gloTrigPat|= 0x1;
-    if (gloTrg->SinglePlusHpt())  gloTrigPat|= 0x2;
-    if (gloTrg->SinglePlusApt())  gloTrigPat|= 0x4;
-    if (gloTrg->SingleMinusLpt()) gloTrigPat|= 0x8;
-    if (gloTrg->SingleMinusHpt()) gloTrigPat|= 0x10;
-    if (gloTrg->SingleMinusApt()) gloTrigPat|= 0x20;
-    if (gloTrg->SingleUndefLpt()) gloTrigPat|= 0x40;
-    if (gloTrg->SingleUndefHpt()) gloTrigPat|= 0x80;
-    if (gloTrg->SingleUndefApt()) gloTrigPat|= 0x100;
-    if (gloTrg->PairUnlikeLpt())  gloTrigPat|= 0x200;
-    if (gloTrg->PairUnlikeHpt())  gloTrigPat|= 0x400;
-    if (gloTrg->PairUnlikeApt())  gloTrigPat|= 0x800;
-
-    if (gloTrg->PairLikeLpt())    gloTrigPat|= 0x1000;
-    if (gloTrg->PairLikeHpt())    gloTrigPat|= 0x2000;
-    if (gloTrg->PairLikeApt())    gloTrigPat|= 0x4000;
+    if (gloTrg) {
+      if (gloTrg->SinglePlusLpt())  gloTrigPat|= 0x1;
+      if (gloTrg->SinglePlusHpt())  gloTrigPat|= 0x2;
+      if (gloTrg->SinglePlusApt())  gloTrigPat|= 0x4;
+      
+      if (gloTrg->SingleMinusLpt()) gloTrigPat|= 0x8;
+      if (gloTrg->SingleMinusHpt()) gloTrigPat|= 0x10;
+      if (gloTrg->SingleMinusApt()) gloTrigPat|= 0x20;
+      
+      if (gloTrg->SingleUndefLpt()) gloTrigPat|= 0x40;
+      if (gloTrg->SingleUndefHpt()) gloTrigPat|= 0x80;
+      if (gloTrg->SingleUndefApt()) gloTrigPat|= 0x100;
+      
+      if (gloTrg->PairUnlikeLpt())  gloTrigPat|= 0x200;
+      if (gloTrg->PairUnlikeHpt())  gloTrigPat|= 0x400;
+      if (gloTrg->PairUnlikeApt())  gloTrigPat|= 0x800;
+      
+      if (gloTrg->PairLikeLpt())    gloTrigPat|= 0x1000;
+      if (gloTrg->PairLikeHpt())    gloTrigPat|= 0x2000;
+      if (gloTrg->PairLikeApt())    gloTrigPat|= 0x4000;
+    }
 
  
 
@@ -1097,7 +1109,8 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithTwoSegments(AliMUONSegme
   // the first segment being pointed to by "BegSegment".
   // Returns the number of such track candidates.
   Int_t endStation, iEndSegment, nbCan2Seg;
-  AliMUONSegment *endSegment, *extrapSegment;
+  AliMUONSegment *endSegment;
+  AliMUONSegment *extrapSegment = NULL;
   AliMUONTrack *recTrack;
   Double_t mcsFactor;
   if (fPrintLevel >= 1) cout << "enter MakeTrackCandidatesWithTwoSegments" << endl;
@@ -1108,8 +1121,6 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithTwoSegments(AliMUONSegme
     GetBendingMomentumFromImpactParam(BegSegment->GetBendingImpact());
   mcsFactor    = fChamberThicknessInX0 * mcsFactor * mcsFactor;
   // linear extrapolation to end station
-  extrapSegment =
-    BegSegment->CreateSegmentFromLinearExtrapToStation(endStation, mcsFactor);
   // number of candidates with 2 segments to 0
   nbCan2Seg = 0;
   // Loop over segments in the end station
@@ -1118,6 +1129,8 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithTwoSegments(AliMUONSegme
     endSegment = (AliMUONSegment*) ((*fSegmentsPtr[endStation])[iEndSegment]);
     // test compatibility between current segment and "extrapSegment"
     // 4 because 4 quantities in chi2
+    extrapSegment =
+      BegSegment->CreateSegmentFromLinearExtrapToStation(endSegment->GetZ(), mcsFactor);
     if ((endSegment->
         NormalizedChi2WithSegment(extrapSegment,
                                   fMaxSigma2Distance)) <= 4.0) {
@@ -1149,7 +1162,8 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithOneSegmentAndOnePoint(Al
   // in stations(1..) 4 and 5,
   // the segment being pointed to by "BegSegment".
   Int_t ch, ch1, ch2, endStation, iHit, iHitMax, iHitMin, nbCan1Seg1Hit;
-  AliMUONHitForRec *extrapHitForRec, *hit;
+  AliMUONHitForRec *extrapHitForRec= NULL;
+  AliMUONHitForRec *hit;
   AliMUONTrack *recTrack;
   Double_t mcsFactor;
   if (fPrintLevel >= 1)
@@ -1167,9 +1181,6 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithOneSegmentAndOnePoint(Al
   nbCan1Seg1Hit = 0;
   // Loop over chambers of the end station
   for (ch = ch2; ch >= ch1; ch--) {
-    // linear extrapolation to chamber
-    extrapHitForRec =
-      BegSegment->CreateHitForRecFromLinearExtrapToChamber(ch, mcsFactor);
     // limits for the hit index in the loop
     iHitMin = fIndexOfFirstHitForRecPerChamber[ch];
     iHitMax = iHitMin + fNHitsForRecPerChamber[ch];
@@ -1179,6 +1190,9 @@ Int_t AliMUONEventReconstructor::MakeTrackCandidatesWithOneSegmentAndOnePoint(Al
       hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[iHit]);
       // test compatibility between current HitForRec and "extrapHitForRec"
       // 2 because 2 quantities in chi2
+      // linear extrapolation to chamber
+      extrapHitForRec =
+       BegSegment->CreateHitForRecFromLinearExtrapToChamber( hit->GetZ(), mcsFactor);
       if ((hit->
           NormalizedChi2WithHitForRec(extrapHitForRec,
                                       fMaxSigma2Distance)) <= 2.0) {
@@ -1244,8 +1258,8 @@ void AliMUONEventReconstructor::FollowTracks(void)
 {
   // Follow tracks in stations(1..) 3, 2 and 1
   // too long: should be made more modular !!!!
-  AliMUONHitForRec *bestHit, *extrapHit, *extrapCorrHit, *hit;
-  AliMUONSegment *bestSegment, *extrapSegment, *extrapCorrSegment, *segment;
+  AliMUONHitForRec *bestHit, *extrapHit, *hit;
+  AliMUONSegment *bestSegment, *extrapSegment, *segment;
   AliMUONTrack *track, *nextTrack;
   AliMUONTrackParam *trackParam1, trackParam[2], trackParamVertex;
   // -1 to avoid compilation warnings
@@ -1286,7 +1300,6 @@ void AliMUONEventReconstructor::FollowTracks(void)
       // extrapolation to station
       trackParam1->ExtrapToStation(station, trackParam);
       extrapSegment = new AliMUONSegment(); //  empty segment
-      extrapCorrSegment = new AliMUONSegment(); //  empty corrected segment
       // multiple scattering factor corresponding to one chamber
       // and momentum in bending plane (not total)
       mcsFactor = 0.0136 * trackParam1->GetInverseBendingMomentum();
@@ -1306,14 +1319,6 @@ void AliMUONEventReconstructor::FollowTracks(void)
       extrapSegment->UpdateFromStationTrackParam
        (trackParam, mcsFactor, dZ1, dZ2, dZ3, station,
         trackParam1->GetInverseBendingMomentum());
-      // same thing for corrected segment
-      // better to use copy constructor, after checking that it works properly !!!!
-      extrapCorrSegment->SetBendingCoorReso2(fBendingResolution * fBendingResolution);
-      extrapCorrSegment->
-       SetNonBendingCoorReso2(fNonBendingResolution * fNonBendingResolution);
-      extrapCorrSegment->UpdateFromStationTrackParam
-       (trackParam, mcsFactor, dZ1, dZ2, dZ3, station,
-        trackParam1->GetInverseBendingMomentum());
       bestChi2 = 5.0;
       bestSegment = NULL;
       if (fPrintLevel >= 10) {
@@ -1329,18 +1334,14 @@ void AliMUONEventReconstructor::FollowTracks(void)
        segment = (AliMUONSegment*) ((*fSegmentsPtr[station])[iSegment]);
        // correction of corrected segment (fBendingCoor and fNonBendingCoor)
        // according to real Z value of "segment" and slopes of "extrapSegment"
-       extrapCorrSegment->
-         SetBendingCoor(extrapSegment->GetBendingCoor() +
-                        extrapSegment->GetBendingSlope() *
-                        (segment->GetHitForRec1()->GetZ() -
-                         (&(pMUON->Chamber(2 * station)))->Z()));
-       extrapCorrSegment->
-         SetNonBendingCoor(extrapSegment->GetNonBendingCoor() +
-                           extrapSegment->GetNonBendingSlope() *
-                           (segment->GetHitForRec1()->GetZ() -
-                            (&(pMUON->Chamber(2 * station)))->Z()));
+       (&(trackParam[0]))->ExtrapToZ(segment->GetZ());
+       (&(trackParam[1]))->ExtrapToZ(segment->GetZ());
+       extrapSegment->SetBendingCoor((&(trackParam[0]))->GetBendingCoor());
+       extrapSegment->SetNonBendingCoor((&(trackParam[0]))->GetNonBendingCoor());
+       extrapSegment->SetBendingSlope((&(trackParam[0]))->GetBendingSlope());
+       extrapSegment->SetNonBendingSlope((&(trackParam[0]))->GetNonBendingSlope());
        chi2 = segment->
-         NormalizedChi2WithSegment(extrapCorrSegment, maxSigma2Distance);
+         NormalizedChi2WithSegment(extrapSegment, maxSigma2Distance);
        if (chi2 < bestChi2) {
          // update best Chi2 and Segment if better found
          bestSegment = segment;
@@ -1349,6 +1350,8 @@ void AliMUONEventReconstructor::FollowTracks(void)
       }
       if (bestSegment) {
        // best segment found: add it to track candidate
+       (&(trackParam[0]))->ExtrapToZ(bestSegment->GetZ());
+       (&(trackParam[1]))->ExtrapToZ(bestSegment->GetZ());
        track->AddSegment(bestSegment);
        // set track parameters at these two TrakHit's
        track->SetTrackParamAtHit(track->GetNTrackHits() - 2, &(trackParam[0]));
@@ -1365,7 +1368,6 @@ void AliMUONEventReconstructor::FollowTracks(void)
        // should consider all possibilities ????
        // multiple scattering ???? do about like for extrapSegment !!!!
        extrapHit = new AliMUONHitForRec(); //  empty hit
-       extrapCorrHit = new AliMUONHitForRec(); //  empty corrected hit
        bestChi2 = 3.0;
        bestHit = NULL;
        if (fPrintLevel >= 10) {
@@ -1375,41 +1377,22 @@ void AliMUONEventReconstructor::FollowTracks(void)
        }
        // Loop over chambers of the station
        for (chInStation = 0; chInStation < 2; chInStation++) {
-         // coordinates of extrapolated hit
-         extrapHit->
-           SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
-         extrapHit->
-           SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
-         // resolutions from "extrapSegment"
-         extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
-         extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
-         // same things for corrected hit
-         // better to use copy constructor, after checking that it works properly !!!!
-         extrapCorrHit->
-           SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
-         extrapCorrHit->
-           SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
-         extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
-         extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
-         // Loop over hits in the chamber
          ch = 2 * station + chInStation;
          for (iHit = fIndexOfFirstHitForRecPerChamber[ch];
               iHit < fIndexOfFirstHitForRecPerChamber[ch] +
                 fNHitsForRecPerChamber[ch];
               iHit++) {
            hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[iHit]);
-           // correction of corrected hit (fBendingCoor and fNonBendingCoor)
-           // according to real Z value of "hit" and slopes of right "trackParam"
-           extrapCorrHit->
-             SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor() +
-                            (&(trackParam[chInStation]))->GetBendingSlope() *
-                            (hit->GetZ() -
-                             (&(trackParam[chInStation]))->GetZ()));
-           extrapCorrHit->
-             SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor() +
-                               (&(trackParam[chInStation]))->GetNonBendingSlope() *
-                               (hit->GetZ() -
-                                (&(trackParam[chInStation]))->GetZ()));
+           // coordinates of extrapolated hit
+           (&(trackParam[chInStation]))->ExtrapToZ(hit->GetZ());
+           extrapHit->
+             SetBendingCoor((&(trackParam[chInStation]))->GetBendingCoor());
+           extrapHit->
+             SetNonBendingCoor((&(trackParam[chInStation]))->GetNonBendingCoor());
+           // resolutions from "extrapSegment"
+           extrapHit->SetBendingReso2(extrapSegment->GetBendingCoorReso2());
+           extrapHit->SetNonBendingReso2(extrapSegment->GetNonBendingCoorReso2());
+           // Loop over hits in the chamber
            // condition for hit not already in segment ????
            chi2 = hit->NormalizedChi2WithHitForRec(extrapHit, maxSigma2Distance);
            if (chi2 < bestChi2) {
@@ -1422,6 +1405,7 @@ void AliMUONEventReconstructor::FollowTracks(void)
        }
        if (bestHit) {
          // best hit found: add it to track candidate
+         (&(trackParam[chBestHit]))->ExtrapToZ(bestHit->GetZ());
          track->AddHitForRec(bestHit);
          // set track parameters at this TrackHit
          track->SetTrackParamAtHit(track->GetNTrackHits() - 1,
@@ -1437,17 +1421,13 @@ void AliMUONEventReconstructor::FollowTracks(void)
          // and corresponding TrackHit's, ...
          track->Remove();
          delete extrapSegment;
-         delete extrapCorrSegment;
          delete extrapHit;
-         delete extrapCorrHit;
          break; // stop the search for this candidate:
          // exit from the loop over station
        }
        delete extrapHit;
-       delete extrapCorrHit;
       }
       delete extrapSegment;
-      delete extrapCorrSegment;
       // Sort track hits according to increasing Z
       track->GetTrackHitsPtr()->Sort();
       // Update track parameters at first track hit (smallest Z)
@@ -1491,7 +1471,7 @@ void AliMUONEventReconstructor::FollowTracks(void)
       // after going through the first station
       if (station == 0) {
        trackParamVertex = *trackParam1;
-       (&trackParamVertex)->ExtrapToVertex();
+       (&trackParamVertex)->ExtrapToVertex(0.,0.,0.);
        track->SetTrackParamAtVertex(&trackParamVertex);
        if (fPrintLevel >= 1) {
          cout << "FollowTracks: track candidate(0..): " << trackIndex
@@ -1572,6 +1552,39 @@ void AliMUONEventReconstructor::UpdateTrackParamAtHit()
   return;
 }
 
+  //__________________________________________________________________________
+void AliMUONEventReconstructor::UpdateHitForRecAtHit()
+{
+  // Set cluster parameterss after track fitting. Fill fHitForRecAtHit of AliMUONTrack's
+  AliMUONTrack *track;
+  AliMUONTrackHit *trackHit;
+  AliMUONHitForRec *hitForRec;
+  track = (AliMUONTrack*) fRecTracksPtr->First();
+  while (track) {
+    trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->First();
+    while (trackHit) {
+      hitForRec = trackHit->GetHitForRecPtr();
+      track->AddHitForRecAtHit(hitForRec);
+      trackHit = (AliMUONTrackHit*) (track->GetTrackHitsPtr())->After(trackHit); 
+    } // trackHit    
+    track = (AliMUONTrack*) fRecTracksPtr->After(track);
+  } // track
+  return;
+}
+
+  //__________________________________________________________________________
+void AliMUONEventReconstructor::FillMUONTrack()
+{
+  // Set track parameters at hits for Kalman track. Fill fTrackParamAtHit of AliMUONTrack's
+  AliMUONTrackK *track;
+  track = (AliMUONTrackK*) fRecTracksPtr->First();
+  while (track) {
+    track->FillMUONTrack();
+    track = (AliMUONTrackK*) fRecTracksPtr->After(track);
+  } 
+  return;
+}
+
   //__________________________________________________________________________
 void AliMUONEventReconstructor::EventDump(void)
 {
@@ -1731,83 +1744,86 @@ void AliMUONEventReconstructor::MakeTrackCandidatesK(void)
 void AliMUONEventReconstructor::FollowTracksK(void)
 {
   // Follow tracks using Kalman filter
-  Bool_t Ok;
-  Int_t icand, ichamBeg, ichamEnd, chamBits;
+  Bool_t ok;
+  Int_t icand, ichamBeg = 0, ichamEnd, chamBits;
   Double_t zDipole1, zDipole2;
   AliMUONTrackK *trackK;
   AliMUONHitForRec *hit;
   AliMUONRawCluster *clus;
   TClonesArray *rawclusters;
-  AliMUON *pMUON;
   clus = 0; rawclusters = 0;
 
-  zDipole1 = GetSimpleBPosition() - GetSimpleBLength()/2;
-  zDipole2 = zDipole1 + GetSimpleBLength();
+  //AZ(z->-z) zDipole1 = GetSimpleBPosition() - GetSimpleBLength()/2;
+  //AZ(z->-z) zDipole2 = zDipole1 + GetSimpleBLength();
+  zDipole1 = GetSimpleBPosition() + GetSimpleBLength()/2;
+  zDipole2 = zDipole1 - GetSimpleBLength();
 
   // Print hits
-  pMUON  = (AliMUON*) gAlice->GetModule("MUON");
-  for (Int_t i1=0; i1<fNHitsForRec; i1++) {
-    hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[i1]);
-    //if (hit->GetTHTrack() > 1 || hit->GetGeantSignal() == 0) continue;
-    /*
-    cout << " Hit #" << hit->GetChamberNumber() << " ";
-    cout << hit->GetBendingCoor() << " ";
-    cout << hit->GetNonBendingCoor() << " ";
-    cout << hit->GetZ() << " ";
-    cout << hit->GetGeantSignal() << " ";
-    cout << hit->GetTHTrack() << endl;
-    */
-    /*
-    printf(" Hit # %d %10.4f %10.4f %10.4f",
-           hit->GetChamberNumber(), hit->GetBendingCoor(),
-           hit->GetNonBendingCoor(), hit->GetZ());
-    if (fRecGeantHits) {
-      // from GEANT hits
-      printf(" %3d %3d \n", hit->GetGeantSignal(), hit->GetTHTrack());
-    } else {
-      // from raw clusters
-      rawclusters = pMUON->RawClustAddress(hit->GetChamberNumber());
-      clus = (AliMUONRawCluster*) rawclusters->UncheckedAt(hit->
-                                               GetHitNumber());
-      printf("%3d", clus->fTracks[1]-1);
-      if (clus->fTracks[2] != 0) printf("%3d \n", clus->fTracks[2]-1);
-      else printf("\n");
+  trackK = (AliMUONTrackK*) ((*fRecTracksPtr)[0]);
+  if (trackK->DebugLevel() > 0) {
+    for (Int_t i1=0; i1<fNHitsForRec; i1++) {
+      hit = (AliMUONHitForRec*) ((*fHitsForRecPtr)[i1]);
+      //if (hit->GetTHTrack() > 1 || hit->GetGeantSignal() == 0) continue;
+      printf(" Hit # %d %10.4f %10.4f %10.4f",
+             hit->GetChamberNumber(), hit->GetBendingCoor(),
+             hit->GetNonBendingCoor(), hit->GetZ());
+      if (fRecGeantHits) {
+        // from GEANT hits
+       printf(" %3d %3d \n", hit->GetGeantSignal(), hit->GetTHTrack());
+      } else {
+       // from raw clusters
+       rawclusters = fMUONData->RawClusters(hit->GetChamberNumber());
+       clus = (AliMUONRawCluster*) rawclusters->UncheckedAt(hit->
+                                                 GetHitNumber());
+       printf("%3d", clus->GetTrack(1)-1);
+       if (clus->GetTrack(2) != 0) printf("%3d \n", clus->GetTrack(2)-1);
+       else printf("\n");
+      } // if (fRecGeantHits)
     }
-    */
-  }
+  } // if (trackK->DebugLevel() > 0)
 
   icand = -1;
-  Int_t nSeeds = fNRecTracks; // starting number of seeds
+  Int_t nSeeds;
+  nSeeds = fNRecTracks; // starting number of seeds
   // Loop over track candidates
   while (icand < fNRecTracks-1) {
     icand ++;
+    if (trackK->DebugLevel()>0) cout << " *** Kalman track candidate No. " << icand << endl;
     trackK = (AliMUONTrackK*) ((*fRecTracksPtr)[icand]);
+    if (trackK->GetRecover() < 0) continue; // failed track
 
     // Discard candidate which will produce the double track
+    /*
     if (icand > 0) {
-      Ok = CheckCandidateK(icand,nSeeds);
-      if (!Ok) {
-        //trackK->SetRecover(-1); // mark candidate to be removed
-        //continue;
+      ok = CheckCandidateK(icand,nSeeds);
+      if (!ok) {
+        trackK->SetRecover(-1); // mark candidate to be removed
+        continue;
       }
     }
+    */
 
-    Ok = kTRUE;
+    ok = kTRUE;
     if (trackK->GetRecover() == 0) hit = (AliMUONHitForRec*) 
                                    trackK->GetHitOnTrack()->Last(); // last hit
-    else hit = (AliMUONHitForRec*) (*trackK->GetHitOnTrack())[1]; // 2'nd hit
-    ichamBeg = hit->GetChamberNumber();
+    //else hit = (AliMUONHitForRec*) (*trackK->GetHitOnTrack())[1]; // 2'nd hit
+    else hit = trackK->GetHitLastOk(); // hit where track stopped
+    if (hit) ichamBeg = hit->GetChamberNumber();
     ichamEnd = 0;
     // Check propagation direction
-    if (trackK->GetTrackDir() > 0) {
+    if (hit == NULL) ichamBeg = ichamEnd;
+    //AZ(z->-z) else if (trackK->GetTrackDir() > 0) {
+    else if (trackK->GetTrackDir() < 0) {
       ichamEnd = 9; // forward propagation
-      Ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
-      if (Ok) {
+      ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
+      if (ok) {
         ichamBeg = ichamEnd;
         ichamEnd = 6; // backward propagation
        // Change weight matrix and zero fChi2 for backpropagation
         trackK->StartBack();
-        Ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
+       //AZ trackK->SetTrackDir(-1);
+       trackK->SetTrackDir(1);
+        ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
         ichamBeg = ichamEnd;
         ichamEnd = 0;
       }
@@ -1817,28 +1833,41 @@ void AliMUONEventReconstructor::FollowTracksK(void)
         ichamEnd = 6; // backward propagation
        // Change weight matrix and zero fChi2 for backpropagation
         trackK->StartBack();
-        Ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
+        ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kTRUE,zDipole1,zDipole2);
         ichamBeg = ichamEnd;
         ichamEnd = 0;
       }
     }
 
-    if (Ok) {
-      trackK->SetTrackDir(-1);
+    if (ok) {
+      //AZ trackK->SetTrackDir(-1);
+      trackK->SetTrackDir(1);
       trackK->SetBPFlag(kFALSE);
-      Ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
+      ok = trackK->KalmanFilter(ichamBeg,ichamEnd,kFALSE,zDipole1,zDipole2);
+    }
+    if (!ok) { trackK->SetRecover(-1); continue; } // mark candidate to be removed
+
+    // Apply smoother
+    if (trackK->GetRecover() >= 0) {
+      ok = trackK->Smooth();
+      if (!ok) trackK->SetRecover(-1); // mark candidate to be removed
     }
-    if (Ok) trackK->SetTrackQuality(0); // compute "track quality"
-    else trackK->SetRecover(-1); // mark candidate to be removed
 
     // Majority 3 of 4 in first 2 stations
+    if (!ok) continue;
     chamBits = 0;
+    Double_t chi2_max = 0;
     for (Int_t i=0; i<trackK->GetNTrackHits(); i++) {
       hit = (AliMUONHitForRec*) (*trackK->GetHitOnTrack())[i];
-      chamBits |= BIT(hit->GetChamberNumber()-1);
+      chamBits |= BIT(hit->GetChamberNumber());
+      if (trackK->GetChi2PerPoint(i) > chi2_max) chi2_max = trackK->GetChi2PerPoint(i);
+    }
+    if (!((chamBits&3)==3 || (chamBits>>2&3)==3) && chi2_max > 25) {
+      //trackK->Recover();
+      trackK->SetRecover(-1); //mark candidate to be removed
+      continue;
     }
-    //if (!((chamBits&3)==3 || (chamBits>>2&3)==3)) trackK->SetRecover(-1); 
-                                 //mark candidate to be removed
+    if (ok) trackK->SetTrackQuality(0); // compute "track quality"
   } // while
 
   for (Int_t i=0; i<fNRecTracks; i++) {
@@ -1853,7 +1882,7 @@ void AliMUONEventReconstructor::FollowTracksK(void)
 }
 
 //__________________________________________________________________________
-Bool_t AliMUONEventReconstructor::CheckCandidateK(Int_t icand, Int_t nSeeds)
+Bool_t AliMUONEventReconstructor::CheckCandidateK(Int_t icand, Int_t nSeeds) const
 {
   // Discards track candidate if it will produce the double track (having
   // the same seed segment hits as hits of a good track found before)
@@ -1897,13 +1926,14 @@ void AliMUONEventReconstructor::RemoveDoubleTracksK(void)
 
   // Loop over first track of the pair
   track1 = (AliMUONTrackK*) fRecTracksPtr->First();
+  Int_t debug = track1->DebugLevel();
   while (track1) {
     // Loop over second track of the pair
     track2 = (AliMUONTrackK*) fRecTracksPtr->After(track1);
     while (track2) {
       // Check whether or not to keep track2
       if (!track2->KeepTrack(track1)) {
-        cout << " Killed track: " << 1/(*track2->GetTrackParameters())(4,0) <<
+        if (debug >= 0) cout << " Killed track: " << 1/(*track2->GetTrackParameters())(4,0) <<
          " " << track2->GetTrackQuality() << endl;
         trackToKill = track2;
         track2 = (AliMUONTrackK*) fRecTracksPtr->After(track2);
@@ -1915,7 +1945,7 @@ void AliMUONEventReconstructor::RemoveDoubleTracksK(void)
   } // track1
 
   fNRecTracks = fRecTracksPtr->GetEntriesFast();
-  cout << " Number of Kalman tracks: " << fNRecTracks << endl;
+  if (debug >= 0) cout << " Number of Kalman tracks: " << fNRecTracks << endl;
 }
 
 //__________________________________________________________________________
@@ -1928,8 +1958,22 @@ void AliMUONEventReconstructor::GoToVertex(void)
   zVertex = 0;
   for (Int_t i=0; i<fNRecTracks; i++) {
     //((AliMUONTrackK*)(*fRecTracksPtr)[i])->Branson();
-    //((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToZ(zVertex); // w/out absorber
     ((AliMUONTrackK*)(*fRecTracksPtr)[i])->SetTrackQuality(1); // compute Chi2
-    ((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToVertex(); // with absorber
+    //((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToZ(zVertex); // w/out absorber
+    ((AliMUONTrackK*)(*fRecTracksPtr)[i])->GoToVertex(1); // with absorber
   }
 }
+
+//__________________________________________________________________________
+void AliMUONEventReconstructor::SetTrackMethod(Int_t iTrackMethod)
+{
+  // Set track method and recreate track container if necessary
+  
+  fTrackMethod = iTrackMethod == 2 ? 2 : 1;
+  if (fTrackMethod == 2) {
+    if (fRecTracksPtr) delete fRecTracksPtr;
+    fRecTracksPtr = new TClonesArray("AliMUONTrackK", 10);
+    cout << " *** Tracking with the Kalman filter *** " << endl;
+  } else cout << " *** Traditional tracking *** " << endl;
+
+}