Adding PiKP-only histograms and eliminating a number of switches where histograms...
[u/mrichter/AliRoot.git] / MUON / AliMUONESDInterface.cxx
index 9c4c3cc..63c73b9 100644 (file)
 #include "AliMUONVClusterStore.h"
 #include "AliMUONVDigit.h"
 #include "AliMUONVDigitStore.h"
-#include "AliMUON1DMapIterator.h"
+#include "AliMUONLocalTrigger.h"
+#include "AliMUONTriggerTrack.h"
+#include "AliMUONVTriggerStore.h"
+#include "AliMUONVTriggerTrackStore.h"
 #include "AliMUON2DMapIterator.h"
 #include "AliMUONTrackParam.h"
 #include "AliMUONTrackExtrap.h"
 #include "AliMUONConstants.h"
+#include "AliMUONTracker.h"
+#include "AliMUONRecoParam.h"
+#include "AliMUONVTrackReconstructor.h"
 
+#include "AliMpExMapIterator.h"
 #include "AliMpVSegmentation.h"
 #include "AliMpSegmentation.h"
-#include "AliMpIntPair.h"
 #include "AliMpPad.h"
 
 #include "AliESDEvent.h"
@@ -44,6 +50,8 @@
 #include <TMath.h>
 #include <TMatrixD.h>
 #include <Riostream.h>
+#include <TGeoGlobalMagField.h>
+#include <TVirtualMagField.h>
 
 //-----------------------------------------------------------------------------
 /// \class AliMUONESDInterface
 /// 
 /// 1) using the static methods converting the objects one by one
 ///
-/// 2) loading a whole ESDEvent and using the getters and/or the iterators
+/// 2) loading a whole ESDEvent and using the finders and/or the iterators
 ///    to access the corresponding MUON objects
 ///
+/// note: You can set the recoParam used to refit the MUON track with ResetTracker(...);
+///       By default we use Kalman filter + Smoother
+///
 /// \author Philippe Pillot
 //-----------------------------------------------------------------------------
 
 ClassImp(AliMUONESDInterface)
 /// \endcond
 
+AliMUONRecoParam* AliMUONESDInterface::fgRecoParam = 0x0;
+AliMUONVTrackReconstructor* AliMUONESDInterface::fgTracker = 0x0;
+
 TString AliMUONESDInterface::fgTrackStoreName = "AliMUONTrackStoreV1";
 TString AliMUONESDInterface::fgClusterStoreName = "AliMUONClusterStoreV2";
 TString AliMUONESDInterface::fgDigitStoreName = "AliMUONDigitStoreV2R";
+TString AliMUONESDInterface::fgTriggerStoreName = "AliMUONTriggerStoreV1";
+TString AliMUONESDInterface::fgTriggerTrackStoreName = "AliMUONTriggerTrackStoreV1";
 
 //_____________________________________________________________________________
 AliMUONESDInterface::AliMUONESDInterface()
 : TObject(),
-  fTracks(0x0),
-  fDigits(0x0),
-  fTrackMap(0x0),
-  fClusterMap(0x0),
-  fDigitMap(0x0)
+fTracks(0x0),
+fDigits(0x0),
+fTriggers(0x0),
+fClusterMap(0x0),
+fDigitMap(0x0)
 {
   /// Default constructor
 }
@@ -85,7 +101,7 @@ AliMUONESDInterface::~AliMUONESDInterface()
   /// Destructor
   delete fTracks;
   delete fDigits;
-  delete fTrackMap;
+  delete fTriggers;
   delete fClusterMap;
   delete fDigitMap;
 }
@@ -100,7 +116,7 @@ void AliMUONESDInterface::Clear(Option_t*)
   /// clear memory
   delete fTracks; fTracks = 0x0;
   delete fDigits; fDigits = 0x0;
-  delete fTrackMap; fTrackMap = 0x0;
+  delete fTriggers; fTriggers = 0x0;
   delete fClusterMap; fClusterMap = 0x0;
   delete fDigitMap; fDigitMap = 0x0;
 }
@@ -116,20 +132,20 @@ void AliMUONESDInterface::Reset()
   if (fDigits) fDigits->Clear("C");
   else fDigits = NewDigitStore();
   
-  if (fTrackMap) fTrackMap->Clear();
-  else fTrackMap = new AliMpExMap(kTRUE);
-  fTrackMap->SetOwner(kFALSE);
+  if (fTriggers) fTriggers->Clear("C");
+  else fTriggers = NewTriggerStore();
   
   if (fClusterMap) fClusterMap->Clear();
-  else fClusterMap = new AliMpExMap(kTRUE);
+  else fClusterMap = new AliMpExMap;
   fClusterMap->SetOwner(kTRUE);
   
-  if (fDigitMap) fDigitMap->Clear("C");
-  else fDigitMap = new TClonesArray("AliMpExMap",10);
+  if (fDigitMap) fDigitMap->Clear();
+  else fDigitMap = new AliMpExMap;
+  fDigitMap->SetOwner(kTRUE);
 }
 
 //_____________________________________________________________________________
-void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent)
+void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent, Bool_t refit)
 {
   /// Extract MUON data from the given ESD event
   
@@ -143,20 +159,24 @@ void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent)
     // get ESD track
     AliESDMuonTrack* esdTrack = esdEvent.GetMuonTrack(iTrack);
     
-    // add it to track store
-    AliMUONTrack* track = Add(*esdTrack, *fTracks);
+    // fill trigger store if related info are availables
+    if (esdTrack->ContainTriggerData()) Add(*esdTrack, *fTriggers);
+    
+    // fill tracker data if availables
+    if (!esdTrack->ContainTrackerData()) continue;
     
-    // fill track map
-    fTrackMap->Add(iTrack, track);
+    // add it to track store
+    AliMUONTrack* track = Add(*esdTrack, *fTracks, refit);
     
     // prepare cluster map
-    fClusterMap->Add(iTrack, new AliMpExMap(kTRUE));
-    AliMpExMap* cMap = (AliMpExMap*) fClusterMap->GetObjectFast(iTrack);
+    AliMpExMap* cMap = new AliMpExMap;
     cMap->SetOwner(kFALSE);
+    fClusterMap->Add(esdTrack->GetUniqueID(), cMap);
     
     // prepare digit maps
-    AliMpExMap* dMaps = new((*fDigitMap)[iTrack]) AliMpExMap(kTRUE);
+    AliMpExMap* dMaps = new AliMpExMap;
     dMaps->SetOwner(kTRUE);
+    fDigitMap->Add(esdTrack->GetUniqueID(), dMaps);
     
     // loop over ESD clusters
     Int_t nClusters = esdTrack->GetNClusters();
@@ -172,13 +192,13 @@ void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent)
       cMap->Add(cluster->GetUniqueID(), cluster);
       
       // prepare digit map
-      dMaps->Add(esdCluster->GetUniqueID(), new AliMpExMap(kTRUE));
-      AliMpExMap* dMap = (AliMpExMap*) dMaps->GetObjectFast(iCluster);
+      AliMpExMap* dMap =new AliMpExMap;
       dMap->SetOwner(kFALSE);
+      dMaps->Add(esdCluster->GetUniqueID(), dMap);
       
       // loop over ESD pads
       Int_t nPads = esdCluster->GetNPads();
-      for (Int_t iPad = 0; iPad <  nPads; iPad++) {
+      for (Int_t iPad = 0; iPad < nPads; iPad++) {
        
        // get ESD pad
        AliESDMuonPad *esdPad = (AliESDMuonPad*) esdCluster->GetPads().UncheckedAt(iPad);
@@ -202,7 +222,7 @@ void AliMUONESDInterface::LoadEvent(AliESDEvent& esdEvent)
 Int_t AliMUONESDInterface::GetNTracks() const
 {
   /// return the number of tracks
-  return fTrackMap ? fTrackMap->GetSize() : 0;
+  return fTracks ? fTracks->GetSize() : 0;
 }
 
 //___________________________________________________________________________
@@ -210,16 +230,17 @@ Int_t AliMUONESDInterface::GetNClusters() const
 {
   /// return the number of clusters
   Int_t nClusters = 0;
-  Int_t nTracks = GetNTracks();
-  for (Int_t i = 0; i < nTracks; i++) nClusters += GetTrackFast(i)->GetNClusters();
+  AliMUONTrack *track;
+  TIter next(CreateTrackIterator());
+  while ((track = static_cast<AliMUONTrack*>(next()))) nClusters += track->GetNClusters();
   return nClusters;
 }
 
 //___________________________________________________________________________
-Int_t AliMUONESDInterface::GetNClusters(Int_t iTrack) const
+Int_t AliMUONESDInterface::GetNClusters(UInt_t trackId) const
 {
-  /// return the number of clusters in track "iTrack"
-  AliMUONTrack* track = GetTrack(iTrack);
+  /// return the number of clusters in track "trackId"
+  AliMUONTrack* track = FindTrack(trackId);
   return track ? track->GetNClusters() : 0;
 }
 
@@ -231,20 +252,21 @@ Int_t AliMUONESDInterface::GetNDigits() const
 }
 
 //___________________________________________________________________________
-Int_t AliMUONESDInterface::GetNDigits(Int_t iTrack) const
+Int_t AliMUONESDInterface::GetNDigits(UInt_t trackId) const
 {
-  /// return the number of digits in all clusters of track "iTrack"
+  /// return the number of digits in all clusters of track "trackId"
   Int_t nDigits = 0;
-  Int_t nClusters = GetNClusters(iTrack);
-  for (Int_t j = 0; j <  nClusters; j++) nDigits += GetClusterFast(iTrack,j)->GetNDigits();
+  AliMUONVCluster *cluster;
+  TIter next(CreateClusterIterator(trackId));
+  while ((cluster = static_cast<AliMUONVCluster*>(next()))) nDigits += cluster->GetNDigits();
   return nDigits;
 }
 
 //___________________________________________________________________________
-Int_t AliMUONESDInterface::GetNDigits(Int_t iTrack, Int_t iCluster) const
+Int_t AliMUONESDInterface::GetNDigits(UInt_t trackId, UInt_t clusterId) const
 {
   /// return the number of digits in cluster numbered "iCluster" of track "iTrack"
-  AliMUONVCluster* cluster = GetCluster(iTrack, iCluster);
+  AliMUONVCluster* cluster = FindCluster(trackId, clusterId);
   return cluster ? cluster->GetNDigits() : 0;
 }
 
@@ -257,50 +279,51 @@ Int_t AliMUONESDInterface::GetNDigitsInCluster(UInt_t clusterId) const
 }
 
 //___________________________________________________________________________
-AliMUONTrack* AliMUONESDInterface::GetTrack(Int_t iTrack) const
-{
-  /// return MUON track "iTrack" (0x0 if not found)
-  AliMUONTrack* track = fTrackMap ? (AliMUONTrack*) fTrackMap->GetObject(iTrack) : 0x0;
-  if (!track) AliWarning(Form("track %d does not exist",iTrack));
-  return track;
-}
-
-//___________________________________________________________________________
-AliMUONVCluster* AliMUONESDInterface::GetCluster(Int_t iTrack, Int_t iCluster) const
+Int_t AliMUONESDInterface::GetNTriggers() const
 {
-  /// return MUON cluster numbered "iCluster" in track "iTrack" (0x0 if not found)
-  AliMpExMap* cMap = fClusterMap ? (AliMpExMap*) fClusterMap->GetObject(iTrack) : 0x0;
-  AliMUONVCluster* cluster = cMap ? (AliMUONVCluster*) cMap->GetObject(iCluster) : 0x0;
-  if (!cluster) AliWarning(Form("cluster #%d in track %d does not exist",iCluster,iTrack));
-  return cluster;
+  /// return the number of triggers
+  return fTriggers ? fTriggers->GetSize() : 0;
 }
 
 //___________________________________________________________________________
-AliMUONVDigit* AliMUONESDInterface::GetDigit(Int_t iTrack, Int_t iCluster, Int_t iDigit) const
+AliMUONTrack* AliMUONESDInterface::FindTrack(UInt_t trackId) const
 {
-  /// return MUON digit numbered "iDigit" in cluster numbered "iCluster" of track "iTrack" (0x0 if not found)
-  AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
-  AliMpExMap* dMap = dMaps ? (AliMpExMap*) dMaps->GetObject(iCluster) : 0x0;
-  AliMUONVDigit* digit = dMap ? (AliMUONVDigit*) dMap->GetObject(iDigit) : 0x0;
-  if (!digit) AliWarning(Form("digit #%d in cluster #%d of track %d does not exist",iDigit,iCluster,iTrack));
-  return digit;
+  /// return track "trackId" (0x0 if not found)
+  AliMUONTrack *track = fTracks ? static_cast<AliMUONTrack*>(fTracks->FindObject(trackId)) : 0x0;
+  if (!track) AliWarning(Form("track %d does not exist",trackId));
+  return track;
 }
 
 //___________________________________________________________________________
 AliMUONVCluster* AliMUONESDInterface::FindCluster(UInt_t clusterId) const
 {
-  /// return cluster "clusterId" (0x0 if not found)
+  /// loop over tracks and return the first cluster "clusterId" found (0x0 if not found)
+  AliMpExMap *cMap;
   AliMUONVCluster* cluster = 0x0;
   
-  Int_t nTracks = GetNTracks();
-  for (Int_t i = 0; i < nTracks; i++) {
+  if (fClusterMap) {
     
-    cluster = (AliMUONVCluster*) ((AliMpExMap*) fClusterMap->GetObjectFast(i))->GetValue(clusterId);
-    if (cluster) break;
+    TIter next(fClusterMap->CreateIterator());
+    while ((cMap = static_cast<AliMpExMap*>(next()))) {
+      
+      cluster = static_cast<AliMUONVCluster*>(cMap->GetValue(clusterId));
+      if (cluster) return cluster;
+      
+    }
     
   }
   
   if (!cluster) AliWarning(Form("cluster %d does not exist",clusterId));
+  return 0x0;
+}
+
+//___________________________________________________________________________
+AliMUONVCluster* AliMUONESDInterface::FindCluster(UInt_t trackId, UInt_t clusterId) const
+{
+  /// return cluster "clusterId" in track "trackId" (0x0 if not found)
+  AliMpExMap *cMap = fClusterMap ? static_cast<AliMpExMap*>(fClusterMap->GetValue(trackId)) : 0x0;
+  AliMUONVCluster* cluster = cMap ? static_cast<AliMUONVCluster*>(cMap->GetValue(clusterId)) : 0x0;
+  if (!cluster) AliWarning(Form("cluster %d does not exist in track %d", clusterId, trackId));
   return cluster;
 }
 
@@ -314,6 +337,13 @@ AliMUONVDigit* AliMUONESDInterface::FindDigit(UInt_t digitId) const
 }
 
 //___________________________________________________________________________
+AliMUONLocalTrigger* AliMUONESDInterface::FindLocalTrigger(Int_t boardNumber) const
+{
+  /// return MUON local trigger "boardNumber"
+  return (fTriggers) ? fTriggers->FindLocal(boardNumber) : 0x0;
+}
+
+//___________________________________________________________________________
 TIterator* AliMUONESDInterface::CreateTrackIterator() const
 {
   /// return iterator over all tracks
@@ -328,11 +358,11 @@ TIterator* AliMUONESDInterface::CreateClusterIterator() const
 }
 
 //___________________________________________________________________________
-TIterator* AliMUONESDInterface::CreateClusterIterator(Int_t iTrack) const
+TIterator* AliMUONESDInterface::CreateClusterIterator(UInt_t trackId) const
 {
-  /// return iterator over clusters of track "iTrack"
-  AliMpExMap* cMap = fClusterMap ? (AliMpExMap*) fClusterMap->GetObject(iTrack) : 0x0;
-  return cMap ? new AliMUON1DMapIterator(*cMap) : 0x0;
+  /// return iterator over clusters of track "trackId"
+  AliMpExMap *cMap = fClusterMap ? static_cast<AliMpExMap*>(fClusterMap->GetValue(trackId)) : 0x0;
+  return cMap ? cMap->CreateIterator() : 0x0;
 }
 
 //___________________________________________________________________________
@@ -343,37 +373,49 @@ TIterator* AliMUONESDInterface::CreateDigitIterator() const
 }
 
 //___________________________________________________________________________
-TIterator* AliMUONESDInterface::CreateDigitIterator(Int_t iTrack) const
+TIterator* AliMUONESDInterface::CreateDigitIterator(UInt_t trackId) const
 {
-  /// return iterator over all digits of track "iTrack"
-  AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
+  /// return iterator over all digits of track "trackId"
+  AliMpExMap* dMaps = fDigitMap ? static_cast<AliMpExMap*>(fDigitMap->GetValue(trackId)) : 0x0;
   return dMaps ? new AliMUON2DMapIterator(*dMaps) : 0x0;
 }
 
 //___________________________________________________________________________
-TIterator* AliMUONESDInterface::CreateDigitIterator(Int_t iTrack, Int_t iCluster) const
+TIterator* AliMUONESDInterface::CreateDigitIterator(UInt_t trackId, UInt_t clusterId) const
 {
-  /// return iterator over digits of cluster numbered "iCluster" in track "iTrack"
-  AliMpExMap* dMaps = fDigitMap ? (AliMpExMap*) fDigitMap->At(iTrack) : 0x0;
-  AliMpExMap* dMap = dMaps ? (AliMpExMap*) dMaps->GetObject(iCluster) : 0x0;
-  return dMap ? new AliMUON1DMapIterator(*dMap) : 0x0;
+  /// return iterator over digits of cluster "clusterId" in track "trackId"
+  AliMpExMap* dMaps = fDigitMap ? static_cast<AliMpExMap*>(fDigitMap->GetValue(trackId)) : 0x0;
+  AliMpExMap* dMap = dMaps ? static_cast<AliMpExMap*>(dMaps->GetValue(clusterId)) : 0x0;
+  return dMap ? dMap->CreateIterator() : 0x0;
 }
 
 //___________________________________________________________________________
 TIterator* AliMUONESDInterface::CreateDigitIteratorInCluster(UInt_t clusterId) const
 {
-  /// return iterator over digits of cluster "clusterId"
+  /// return iterator over digits of the first cluster "clusterId" found by looping over all tracks
+  AliMpExMap *dMaps;
   AliMpExMap* dMap = 0x0;
   
-  Int_t nTracks = GetNTracks();
-  for (Int_t i = 0; i < nTracks; i++) {
+  if (fDigitMap) {
     
-    dMap = (AliMpExMap*) ((AliMpExMap*) fDigitMap->UncheckedAt(i))->GetValue(clusterId);
-    if (dMap) break;
+    TIter next(fDigitMap->CreateIterator());
+    while ((dMaps = static_cast<AliMpExMap*>(next()))) {
+      
+      dMap = static_cast<AliMpExMap*>(dMaps->GetValue(clusterId));
+      if (dMap) return dMap->CreateIterator();
+      
+    }
     
   }
   
-  return dMap ? new AliMUON1DMapIterator(*dMap) : 0x0;
+  return 0x0;
+}
+
+//___________________________________________________________________________
+TIterator* AliMUONESDInterface::CreateLocalTriggerIterator() const
+{
+  /// return iterator over all local trigger
+  return fTriggers ? fTriggers->CreateLocalIterator() : 0x0;
 }
 
 //___________________________________________________________________________
@@ -397,6 +439,45 @@ AliMUONVCluster* AliMUONESDInterface::FindClusterInTrack(const AliMUONTrack& tra
 //+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++//
 
 //_____________________________________________________________________________
+void AliMUONESDInterface::ResetTracker(const AliMUONRecoParam* recoParam, Bool_t info)
+{
+  /// Reset the MUON tracker using "recoParam" if provided.
+  /// If not provided, will use Kalman filter + Smoother
+  
+  delete fgTracker;
+  delete fgRecoParam;
+  
+  if (recoParam) {
+    
+    fgRecoParam = new AliMUONRecoParam(*recoParam);
+    
+    if (info) cout<<"I-AliMUONESDInterface::ResetTracker: will refit tracks with provided RecoParam:"<<endl;
+    
+  } else {
+    
+    fgRecoParam = AliMUONRecoParam::GetLowFluxParam();
+    
+    cout<<"W-AliMUONESDInterface::ResetTracker: RecoParam not provided. Will use default LowFlux parametrization:"<<endl;
+    
+  }
+  
+  // print useful parameters for refitting
+  if (info) {
+    cout<<"                                     --> Tracking mode = "<<fgRecoParam->GetTrackingMode()<<endl;
+    if (fgRecoParam->UseSmoother()) cout<<"                                     --> Use Smoother"<<endl;
+    else cout<<"                                     --> Do not use smoother"<<endl;
+    cout<<"                                     --> Vertex dispersion in bending direction = "
+    <<fgRecoParam->GetBendingVertexDispersion()<<" cm"<<endl;
+  }
+  
+  if (!TGeoGlobalMagField::Instance()->GetField())
+    cout<<"W-AliMUONESDInterface::ResetTracker: Magnetic field has not been set --> assume field is OFF"<<endl;
+  
+  fgTracker = AliMUONTracker::CreateTrackReconstructor(fgRecoParam,0x0);
+  
+}
+
+//_____________________________________________________________________________
 AliMUONVTrackStore* AliMUONESDInterface::NewTrackStore()
 {
   /// Create an empty track store of type fgTrackStoreName
@@ -432,6 +513,30 @@ AliMUONVDigitStore* AliMUONESDInterface::NewDigitStore()
   return reinterpret_cast<AliMUONVDigitStore*>(classPtr->New());
 }
 
+//_____________________________________________________________________________
+AliMUONVTriggerStore* AliMUONESDInterface::NewTriggerStore()
+{
+  /// Create an empty trigger store of type fgTriggerStoreName
+  TClass* classPtr = TClass::GetClass(fgTriggerStoreName);
+  if (!classPtr || !classPtr->InheritsFrom("AliMUONVTriggerStore")) {
+    cout<<"E-AliMUONESDInterface::NewTriggerStore: Unable to create store of type "<<fgTriggerStoreName.Data()<<endl;
+    return 0x0;
+  }
+  return reinterpret_cast<AliMUONVTriggerStore*>(classPtr->New());
+}
+
+//_____________________________________________________________________________
+AliMUONVTriggerTrackStore* AliMUONESDInterface::NewTriggerTrackStore()
+{
+  /// Create an empty trigger track store of type fgTriggerTrackStoreName
+  TClass* classPtr = TClass::GetClass(fgTriggerTrackStoreName);
+  if (!classPtr || !classPtr->InheritsFrom("AliMUONVTriggerTrackStore")) {
+    cout<<"E-AliMUONESDInterface::NewTriggerTrackStore: Unable to create store of type "<<fgTriggerTrackStoreName.Data()<<endl;
+    return 0x0;
+  }
+  return reinterpret_cast<AliMUONVTriggerTrackStore*>(classPtr->New());
+}
+
 //_________________________________________________________________________
 void AliMUONESDInterface::GetParamAtVertex(const AliESDMuonTrack& esdTrack, AliMUONTrackParam& trackParam)
 {
@@ -562,9 +667,21 @@ void AliMUONESDInterface::SetParamCov(const AliMUONTrackParam& trackParam, AliES
 }
 
 //_____________________________________________________________________________
-void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrack& track)
+void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrack& track, Bool_t refit)
 {
-  /// Transfert data from ESDMuon track to MUON track
+  /// Transfert data from ESDMuon track to MUON track.
+  /// If refit = kTRUE, the track parameters at each cluster are obtained by refitting the track
+  /// or by extrapolating the parameters at the first one if the refit failed.
+  /// If refit = kFALSE, only the track parameters at first cluster are valid.
+  /// note: You can set the recoParam used to refit the MUON track with ResetTracker(...);
+  ///       By default we use Kalman filter + Smoother
+  
+  // if the ESDMuon track is a ghost then return an empty MUON track
+  if (!esdTrack.ContainTrackerData()) {
+    track.Reset();
+    track.SetUniqueID(esdTrack.GetUniqueID());
+    return;
+  }
   
   track.Clear("C");
   
@@ -576,12 +693,12 @@ void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrac
   track.SetVertexErrXY2(0.,0.);
   track.SetGlobalChi2(esdTrack.GetChi2());
   track.SetMatchTrigger(esdTrack.GetMatchTrigger());
-  track.SetLoTrgNum(-1);
   track.SetChi2MatchTrigger(esdTrack.GetChi2MatchTrigger());
-  track.SetTrackID(0);
   track.SetHitsPatternInTrigCh(esdTrack.GetHitsPatternInTrigCh());
   track.SetLocalTrigger(esdTrack.LoCircuit(), esdTrack.LoStripX(), esdTrack.LoStripY(),
-                       esdTrack.LoDev(), esdTrack.LoLpt(), esdTrack.LoHpt());
+                       esdTrack.LoDev(), esdTrack.LoLpt(), esdTrack.LoHpt(),
+                       esdTrack.GetTriggerWithoutChamber());
+  track.Connected(esdTrack.IsConnected());
   
   // track parameters at vertex
   AliMUONTrackParam paramAtVertex;
@@ -618,21 +735,26 @@ void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrac
       esdCluster = (AliESDMuonCluster*) esdTrack.GetClusters().After(esdCluster);
     }
     
-    // recompute parameters at first cluster in case of those stored
-    // in ESD are not related to the most upstream cluster
-    AliMUONTrackParam *firstTrackParam = (AliMUONTrackParam*) track.GetTrackParamAtCluster()->First();
-    firstTrackParam->SetZ(esdTrack.GetZUncorrected()); // reset the z to the one stored in ESD
-    AliMUONTrackExtrap::ExtrapToZCov(firstTrackParam,firstTrackParam->GetClusterPtr()->GetZ());
-    
-    // Compute track parameters and covariances at each cluster from those at the first one
-    track.UpdateCovTrackParamAtCluster();
+    // refit the track to get better parameters and covariances at each cluster (temporary disable track improvement)
+    if (refit) {
+      
+      AliMUONTrackParam *firstTrackParam = (AliMUONTrackParam*) track.GetTrackParamAtCluster()->First();
+      AliMUONTrackParam paramSave(*firstTrackParam);
+      
+      if (!fgTracker) ResetTracker();
+      
+      if (!fgTracker->RefitTrack(track, kFALSE) && track.GetGlobalChi2() < AliMUONTrack::MaxChi2()) {
+       *firstTrackParam = paramSave;
+       track.UpdateCovTrackParamAtCluster();
+      }
+      
+    }
     
   } else {
     
-    // get number of the first hit chamber (according to the MUONClusterMap if not empty)
+    // get number of the first hit chamber according to the MUONClusterMap
     Int_t firstCh = 0;
-    if (esdTrack.GetMuonClusterMap() != 0) while (!esdTrack.IsInMuonClusterMap(firstCh)) firstCh++;
-    else firstCh = AliMUONConstants::ChamberNumber(param.GetZ());
+    while (firstCh < 10 && !esdTrack.IsInMuonClusterMap(firstCh)) firstCh++;
     
     // produce fake cluster at this chamber
     cluster->SetUniqueID(AliMUONVCluster::BuildUniqueID(firstCh, 0, 0));
@@ -644,12 +766,47 @@ void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONTrac
     
   }
   
+  // set the MC label from ESD track
+  track.SetMCLabel(esdTrack.GetLabel());
+  
   delete cluster;
   delete cStore;
   
 }
 
 //_____________________________________________________________________________
+void AliMUONESDInterface::ESDToMUON(const AliESDMuonTrack& esdTrack, AliMUONLocalTrigger& locTrg)
+{
+  /// Transfert trigger data from ESDMuon track to the MUONLocalTtrigger object
+  
+  // if the ESDMuon track is a ghost then return an empty MUON track
+  if (!esdTrack.ContainTriggerData()) {
+    AliMUONLocalTrigger emptyLocTrg;
+    locTrg = emptyLocTrg;
+    return;
+  }
+  
+  locTrg.SetUniqueID(esdTrack.GetUniqueID());
+  locTrg.SetLoCircuit(esdTrack.LoCircuit());
+  locTrg.SetLoStripX(esdTrack.LoStripX());
+  locTrg.SetLoStripY(esdTrack.LoStripY());
+  locTrg.SetDeviation(esdTrack.LoDev());
+  locTrg.SetLoLpt(esdTrack.LoLpt());
+  locTrg.SetLoHpt(esdTrack.LoHpt());
+  locTrg.SetTriggerWithoutChamber(esdTrack.GetTriggerWithoutChamber());
+  locTrg.SetLoTrigY(1);
+  locTrg.SetX1Pattern(esdTrack.GetTriggerX1Pattern());
+  locTrg.SetX2Pattern(esdTrack.GetTriggerX2Pattern());
+  locTrg.SetX3Pattern(esdTrack.GetTriggerX3Pattern());
+  locTrg.SetX4Pattern(esdTrack.GetTriggerX4Pattern());
+  locTrg.SetY1Pattern(esdTrack.GetTriggerY1Pattern());
+  locTrg.SetY2Pattern(esdTrack.GetTriggerY2Pattern());
+  locTrg.SetY3Pattern(esdTrack.GetTriggerY3Pattern());
+  locTrg.SetY4Pattern(esdTrack.GetTriggerY4Pattern());
+  
+}
+
+//_____________________________________________________________________________
 void AliMUONESDInterface::ESDToMUON(const AliESDMuonCluster& esdCluster, AliMUONVCluster& cluster)
 {
   /// Transfert data from ESDMuon cluster to MUON cluster
@@ -661,6 +818,7 @@ void AliMUONESDInterface::ESDToMUON(const AliESDMuonCluster& esdCluster, AliMUON
   cluster.SetErrXY(esdCluster.GetErrX(),esdCluster.GetErrY());
   cluster.SetCharge(esdCluster.GetCharge());
   cluster.SetChi2(esdCluster.GetChi2());
+  cluster.SetMCLabel(esdCluster.GetLabel());
   
   if (esdCluster.PadsStored()) {
     Int_t nPads = esdCluster.GetNPads();
@@ -675,32 +833,63 @@ void AliMUONESDInterface::ESDToMUON(const AliESDMuonPad& esdPad, AliMUONVDigit&
 {
   /// Transfert data from ESDMuon pad to MUON digit
   
+  if (!AliMpSegmentation::Instance(kFALSE)) {
+    cout<<"E-AliMUONESDInterface::ESDToMUON: need mapping segmentation to convert ESD pad to MUON digit"<<endl;
+    return;
+  }
+  
   const AliMpVSegmentation* seg = AliMpSegmentation::Instance()->GetMpSegmentationByElectronics(esdPad.GetDetElemId(), esdPad.GetManuId());  
-  AliMpPad pad = seg->PadByLocation(AliMpIntPair(esdPad.GetManuId(), esdPad.GetManuChannel()), kFALSE);
+  AliMpPad pad = seg->PadByLocation(esdPad.GetManuId(), esdPad.GetManuChannel(), kFALSE);
   
-  digit.Saturated(kFALSE);
+  digit.Saturated(esdPad.IsSaturated());
   digit.Used(kFALSE);
-  digit.Calibrated(kTRUE);
+  digit.Calibrated(esdPad.IsCalibrated());
   digit.SetUniqueID(esdPad.GetUniqueID());
   digit.SetCharge(esdPad.GetCharge());
   digit.SetADC(esdPad.GetADC());
-  digit.SetPadXY(pad.GetIndices().GetFirst(), pad.GetIndices().GetSecond());
+  digit.SetPadXY(pad.GetIx(), pad.GetIy());
   
 }
 
 //_____________________________________________________________________________
-void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack& esdTrack, const Double_t vertex[3], const AliMUONVDigitStore* digits)
+void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack& esdTrack, const Double_t vertex[3],
+                                   const AliMUONVDigitStore* digits, const AliMUONLocalTrigger* locTrg)
 {
   /// Transfert data from MUON track to ESDMuon track
   /// Incorporate the ESDPads if the digits are provided
+  /// Add trigger info if the MUON track is matched with a trigger track
+  
+  // empty MUON track -> produce a ghost ESDMuon track if trigger info are available otherwise produce an empty track
+  if (track.GetNClusters() == 0) {
+    if (locTrg) MUONToESD(*locTrg, esdTrack, track.GetUniqueID());
+    else {
+      cout<<"W-AliMUONESDInterface::MUONToESD: will produce an empty ESDMuon track"<<endl;
+      esdTrack.Reset();
+      esdTrack.SetUniqueID(0xFFFFFFFF);
+    }
+    return;
+  }
   
   esdTrack.Clear("C");
   
+  // set global info
+  esdTrack.SetUniqueID(track.GetUniqueID());
+  esdTrack.SetChi2(track.GetGlobalChi2());
+  esdTrack.SetNHit(track.GetNClusters());
+  esdTrack.SetLabel(track.GetMCLabel());
+  
   // set param at first cluster
   AliMUONTrackParam* trackParam = static_cast<AliMUONTrackParam*>((track.GetTrackParamAtCluster())->First());
   SetParamAtFirstCluster(*trackParam, esdTrack);
   SetParamCov(*trackParam, esdTrack);
   
+  // set transverse position at the end of the absorber
+  AliMUONTrackParam trackParamAtAbsEnd(*trackParam);
+  AliMUONTrackExtrap::ExtrapToZ(&trackParamAtAbsEnd, AliMUONConstants::AbsZEnd());
+  Double_t xAbs = trackParamAtAbsEnd.GetNonBendingCoor();
+  Double_t yAbs = trackParamAtAbsEnd.GetBendingCoor();
+  esdTrack.SetRAtAbsorberEnd(TMath::Sqrt(xAbs*xAbs + yAbs*yAbs));
+  
   // set param at vertex
   AliMUONTrackParam trackParamAtVtx(*trackParam);
   AliMUONTrackExtrap::ExtrapToVertex(&trackParamAtVtx, vertex[0], vertex[1], vertex[2], 0., 0.);
@@ -711,14 +900,6 @@ void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack&
   AliMUONTrackExtrap::ExtrapToVertexWithoutBranson(&trackParamAtDCA, vertex[2]);
   SetParamAtDCA(trackParamAtDCA, esdTrack);
   
-  // set global info
-  esdTrack.SetUniqueID(track.GetUniqueID());
-  esdTrack.SetChi2(track.GetGlobalChi2());
-  esdTrack.SetNHit(track.GetNClusters());
-  esdTrack.SetLocalTrigger(track.GetLocalTrigger());
-  esdTrack.SetChi2MatchTrigger(track.GetChi2MatchTrigger());
-  esdTrack.SetHitsPatternInTrigCh(track.GetHitsPatternInTrigCh());
-  
   // set muon cluster info
   AliESDMuonCluster esdCluster;
   esdTrack.SetMuonClusterMap(0);
@@ -729,6 +910,73 @@ void AliMUONESDInterface::MUONToESD(const AliMUONTrack& track, AliESDMuonTrack&
     trackParam = static_cast<AliMUONTrackParam*>(track.GetTrackParamAtCluster()->After(trackParam));
   }
   
+  // set connected flag
+  esdTrack.Connected(track.IsConnected());
+  
+  // set trigger info
+  esdTrack.SetLocalTrigger(track.GetLocalTrigger());
+  esdTrack.SetChi2MatchTrigger(track.GetChi2MatchTrigger());
+  esdTrack.SetHitsPatternInTrigCh(track.GetHitsPatternInTrigCh());
+  if (locTrg) {
+    esdTrack.SetTriggerX1Pattern(locTrg->GetX1Pattern());
+    esdTrack.SetTriggerY1Pattern(locTrg->GetY1Pattern());
+    esdTrack.SetTriggerX2Pattern(locTrg->GetX2Pattern());
+    esdTrack.SetTriggerY2Pattern(locTrg->GetY2Pattern());
+    esdTrack.SetTriggerX3Pattern(locTrg->GetX3Pattern());
+    esdTrack.SetTriggerY3Pattern(locTrg->GetY3Pattern());
+    esdTrack.SetTriggerX4Pattern(locTrg->GetX4Pattern());
+    esdTrack.SetTriggerY4Pattern(locTrg->GetY4Pattern());
+  } else {
+    esdTrack.SetTriggerX1Pattern(0);
+    esdTrack.SetTriggerY1Pattern(0);
+    esdTrack.SetTriggerX2Pattern(0);
+    esdTrack.SetTriggerY2Pattern(0);
+    esdTrack.SetTriggerX3Pattern(0);
+    esdTrack.SetTriggerY3Pattern(0);
+    esdTrack.SetTriggerX4Pattern(0);
+    esdTrack.SetTriggerY4Pattern(0);
+  }
+  
+}
+
+//_____________________________________________________________________________
+void AliMUONESDInterface::MUONToESD(const AliMUONLocalTrigger& locTrg, AliESDMuonTrack& esdTrack,
+                                   UInt_t trackId, const AliMUONTriggerTrack* triggerTrack)
+{
+  /// Build ghost ESDMuon track containing only informations about trigger track
+  
+  esdTrack.Reset();
+  esdTrack.SetUniqueID(trackId);
+  
+  // set trigger info
+  AliMUONTrack muonTrack;
+  muonTrack.SetLocalTrigger(locTrg.LoCircuit(),
+                           locTrg.LoStripX(),
+                           locTrg.LoStripY(),
+                           locTrg.GetDeviation(),
+                           locTrg.LoLpt(),
+                           locTrg.LoHpt(),
+                           locTrg.GetTriggerWithoutChamber());
+  esdTrack.SetLocalTrigger(muonTrack.GetLocalTrigger());
+  esdTrack.SetChi2MatchTrigger(0.);
+  esdTrack.SetTriggerX1Pattern(locTrg.GetX1Pattern());
+  esdTrack.SetTriggerY1Pattern(locTrg.GetY1Pattern());
+  esdTrack.SetTriggerX2Pattern(locTrg.GetX2Pattern());
+  esdTrack.SetTriggerY2Pattern(locTrg.GetY2Pattern());
+  esdTrack.SetTriggerX3Pattern(locTrg.GetX3Pattern());
+  esdTrack.SetTriggerY3Pattern(locTrg.GetY3Pattern());
+  esdTrack.SetTriggerX4Pattern(locTrg.GetX4Pattern());
+  esdTrack.SetTriggerY4Pattern(locTrg.GetY4Pattern());
+  UShort_t hitPattern = 0;
+  if(triggerTrack){
+    hitPattern = triggerTrack->GetHitsPatternInTrigCh();
+    esdTrack.SetHitsPatternInTrigCh(hitPattern);
+    esdTrack.SetThetaXUncorrected(triggerTrack->GetThetax());
+    esdTrack.SetThetaYUncorrected(triggerTrack->GetThetay());
+    esdTrack.SetNonBendingCoorUncorrected(triggerTrack->GetX11());
+    esdTrack.SetBendingCoorUncorrected(triggerTrack->GetY11());
+    esdTrack.SetZUncorrected(triggerTrack->GetZ11());
+  }
 }
 
 //_____________________________________________________________________________
@@ -742,11 +990,12 @@ void AliMUONESDInterface::MUONToESD(const AliMUONVCluster& cluster, AliESDMuonCl
   esdCluster.SetUniqueID(cluster.GetUniqueID());
   esdCluster.SetXYZ(cluster.GetX(), cluster.GetY(), cluster.GetZ());
   esdCluster.SetErrXY(cluster.GetErrX(), cluster.GetErrY());
+  esdCluster.SetCharge(cluster.GetCharge());
+  esdCluster.SetChi2(cluster.GetChi2());
+  esdCluster.SetLabel(cluster.GetMCLabel());
   
-  if (digits) { // transfert all data...
+  if (digits) { // transfert all data if required
     
-    esdCluster.SetCharge(cluster.GetCharge());
-    esdCluster.SetChi2(cluster.GetChi2());
     AliESDMuonPad esdPad;
     for (Int_t i=0; i<cluster.GetNDigits(); i++) {
       AliMUONVDigit* digit = digits->FindObject(cluster.GetDigitId(i));
@@ -758,11 +1007,6 @@ void AliMUONESDInterface::MUONToESD(const AliMUONVCluster& cluster, AliESDMuonCl
       esdCluster.AddPad(esdPad);
     }
     
-  } else { // ...or not
-    
-    esdCluster.SetCharge(0.);
-    esdCluster.SetChi2(0.);
-    
   }
   
 }
@@ -774,19 +1018,33 @@ void AliMUONESDInterface::MUONToESD(const AliMUONVDigit& digit, AliESDMuonPad& e
   esdPad.SetUniqueID(digit.GetUniqueID());
   esdPad.SetADC(digit.ADC());
   esdPad.SetCharge(digit.Charge());
+  esdPad.SetCalibrated(digit.IsCalibrated());
+  esdPad.SetSaturated(digit.IsSaturated());
 }
 
 //___________________________________________________________________________
-AliMUONTrack* AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTrackStore& trackStore)
+AliMUONTrack* AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTrackStore& trackStore, Bool_t refit)
 {
   /// Create MUON track from ESDMuon track and add it to the store
-  /// return a pointer to the track into the store
+  /// Track parameters at each clusters are recomputed or not depending on the flag "refit"
+  /// return a pointer to the track into the store (0x0 if the track already exist)
+  if(trackStore.FindObject(esdTrack.GetUniqueID())) return 0x0;
   AliMUONTrack* track = trackStore.Add(AliMUONTrack());
-  ESDToMUON(esdTrack, *track);
+  ESDToMUON(esdTrack, *track, refit);
   return track;
 }
 
 //___________________________________________________________________________
+void AliMUONESDInterface::Add(const AliESDMuonTrack& esdTrack, AliMUONVTriggerStore& triggerStore)
+{
+  /// Create MUON local trigger from ESDMuon track and add it to the store if not already there
+  if (!triggerStore.FindLocal(esdTrack.LoCircuit())->IsNull()) return;
+  AliMUONLocalTrigger locTrg;
+  ESDToMUON(esdTrack, locTrg);
+  triggerStore.Add(locTrg);
+}
+
+//___________________________________________________________________________
 AliMUONVCluster* AliMUONESDInterface::Add(const AliESDMuonCluster& esdCluster, AliMUONVClusterStore& clusterStore)
 {
   /// Create MUON cluster from ESDMuon cluster and add it to the store