]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONRecoCheck.cxx
Adding code to patch St1 HV mapping
[u/mrichter/AliRoot.git] / MUON / AliMUONRecoCheck.cxx
index 7bc85de0e1d525c6d4913488939e4fe21c98aefc..8b5b6c1415020c0ef85d7cdd7b4e6955497e5570 100644 (file)
@@ -31,6 +31,7 @@
 // 25 Jan 2008:
 // Use the new ESDInterface to create MUON objects from ESD data
 // - Philippe Pillot
+// 
 
 #include "AliMUONRecoCheck.h"
 #include "AliMUONTrack.h"
 #include "AliMUONVClusterStore.h"
 #include "AliMUONConstants.h"
 #include "AliMUONESDInterface.h"
-#include "AliMUONRecoParam.h"
+#include "AliMUONTrackParam.h"
+#include "AliMUONTriggerTrack.h"
+#include "AliMUONVTriggerTrackStore.h"
+#include "AliMUONTriggerStoreV1.h"
+#include "AliMUONLocalTrigger.h"
 #include "AliMCEventHandler.h"
 #include "AliMCEvent.h"
 #include "AliStack.h"
 #include "AliLog.h"
 #include "AliESDEvent.h"
 #include "AliESDMuonTrack.h"
+#include "AliMUONDigitStoreV2S.h"
+#include "AliMUONVDigit.h"
+#include "AliMpVSegmentation.h"
+#include "AliMpSegmentation.h"
+#include "AliMpPad.h"
+
+#include "AliGeomManager.h"
+#include "AliCDBManager.h"
+#include "AliMpCDB.h"
+#include "AliMpDDLStore.h"
+#include "AliMUONCDB.h"
+#include "AliMUONGeometryTransformer.h"
+#include "AliMUONTriggerCircuit.h"
+#include "AliMUONVTrackReconstructor.h"
+#include "AliMUONVTriggerStore.h"
+#include "AliMUONCalibrationData.h"
+#include "AliMUONTriggerElectronics.h"
+
+
+#include "TGeoManager.h"
 
 #include <TFile.h>
 #include <TTree.h>
 #include <TParticlePDG.h>
 #include <Riostream.h>
 
+#include "AliMUONRecoCheck.h"
+
 /// \cond CLASSIMP
 ClassImp(AliMUONRecoCheck)
 /// \endcond
 
 //_____________________________________________________________________________
-AliMUONRecoCheck::AliMUONRecoCheck(AliMUONRecoParam* recoParam, Char_t *esdFileName, Char_t *pathSim)
+AliMUONRecoCheck::AliMUONRecoCheck(const Char_t *esdFileName, const Char_t *pathSim)
 : TObject(),
-fRecoParam(recoParam),
 fMCEventHandler(new AliMCEventHandler()),
 fESDEvent(new AliESDEvent()),
 fESDTree (0x0),
@@ -69,7 +95,14 @@ fESDFile (0x0),
 fCurrentEvent(0),
 fTrackRefStore(0x0),
 fRecoTrackRefStore(0x0),
-fRecoTrackStore(0x0)
+fRecoTriggerRefStore(0x0),
+fRecoTrackStore(0x0),
+fRecoTriggerTrackStore(0x0),
+fGeometryTransformer(0x0),
+fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
+fESDEventOwner(kTRUE)
 {
   /// Normal ctor
   
@@ -94,14 +127,48 @@ fRecoTrackStore(0x0)
   fESDEvent->ReadFromTree(fESDTree); // link fESDEvent to the tree
 }
 
+//_____________________________________________________________________________
+AliMUONRecoCheck::AliMUONRecoCheck(AliESDEvent *esdEvent, AliMCEventHandler *mcEventHandler)
+: TObject(),
+fMCEventHandler(0),
+fESDEvent(0),
+fESDTree (0x0),
+fESDFile (0x0),
+fCurrentEvent(0),
+fTrackRefStore(0x0),
+fRecoTrackRefStore(0x0),
+fRecoTriggerRefStore(0x0),
+fRecoTrackStore(0x0),
+fRecoTriggerTrackStore(0x0),
+fGeometryTransformer(0x0),
+fTriggerCircuit(0x0),
+fCalibrationData(0x0),
+fTriggerElectronics(0x0),
+fESDEventOwner(kFALSE)
+{
+  /// Normal ctor
+  
+  // TrackRefs and Particules
+  fMCEventHandler = mcEventHandler;
+  
+  // ESD MUON Tracks
+  fESDEvent = esdEvent;
+}
+
 //_____________________________________________________________________________
 AliMUONRecoCheck::~AliMUONRecoCheck()
 {
   /// Destructor
-  delete fMCEventHandler;
-  delete fESDEvent;
-  if (fESDFile) fESDFile->Close();
+  if (fESDEventOwner) {
+    delete fMCEventHandler;
+    delete fESDEvent;
+    if (fESDFile) fESDFile->Close();
+  }
   ResetStores();
+  delete fGeometryTransformer;
+  delete fTriggerCircuit;
+  delete fTriggerElectronics;
+  delete fCalibrationData;
 }
 
 //_____________________________________________________________________________
@@ -110,22 +177,64 @@ void AliMUONRecoCheck::ResetStores()
   /// Deletes all the store objects that have been created and resets the pointers to 0x0
   delete fTrackRefStore;      fTrackRefStore = 0x0;
   delete fRecoTrackRefStore;  fRecoTrackRefStore = 0x0;
+  delete fRecoTriggerRefStore;  fRecoTriggerRefStore = 0x0;
   delete fRecoTrackStore;     fRecoTrackStore = 0x0;
+  delete fRecoTriggerTrackStore; fRecoTriggerTrackStore = 0x0;
+}
+
+//_____________________________________________________________________________
+Bool_t AliMUONRecoCheck::InitCircuit()
+{
+  
+  if ( fTriggerCircuit ) return kTRUE;
+  
+  if ( ! InitGeometryTransformer() ) return kFALSE;
+  
+  // reset tracker for local trigger to trigger track conversion
+  if ( ! AliMUONESDInterface::GetTracker() ) AliMUONESDInterface::ResetTracker();
+  
+  fTriggerCircuit = new AliMUONTriggerCircuit(fGeometryTransformer);
+  
+  return kTRUE;
+}
+
+
+//_____________________________________________________________________________
+Int_t AliMUONRecoCheck::GetRunNumber()
+{
+  /// Return the run number of the current ESD event
+  
+  if (fESDEventOwner && fRecoTrackStore == 0x0 && fRecoTriggerTrackStore == 0x0) {
+    if (!fESDTree || fESDTree->GetEvent(fCurrentEvent) <= 0) {
+      AliError(Form("fails to read ESD object for event %d: cannot get the run number",fCurrentEvent));
+      return -1;
+    }
+  }
+  
+  return fESDEvent->GetRunNumber();
 }
 
 //_____________________________________________________________________________
 Int_t AliMUONRecoCheck::NumberOfEvents() const
 {
   /// Return the number of events
-  if (fESDTree) return fESDTree->GetEntries();
+  if (fESDEventOwner && fESDTree) return fESDTree->GetEntries();
   return 0;
 }
 
 //_____________________________________________________________________________
-AliMUONVTrackStore* AliMUONRecoCheck::ReconstructedTracks(Int_t event)
+AliMUONVTrackStore* AliMUONRecoCheck::ReconstructedTracks(Int_t event, Bool_t refit)
 {
   /// Return a track store containing the reconstructed tracks (converted into 
-  /// MUONTrack objects) for a given event
+  /// MUONTrack objects) for a given event.
+  /// Track parameters at each clusters are computed or not depending on the flag "refit".
+  /// If not, only the track parameters at first cluster are valid.
+  
+  if (!fESDEventOwner) {
+    if (fRecoTrackStore == 0x0) MakeReconstructedTracks(refit);
+    return fRecoTrackStore;
+  }
+
   if (event != fCurrentEvent) {
     ResetStores();
     fCurrentEvent = event;
@@ -138,16 +247,52 @@ AliMUONVTrackStore* AliMUONRecoCheck::ReconstructedTracks(Int_t event)
       AliError(Form("fails to read ESD object for event %d", event));
       return 0x0;
     }
-    MakeReconstructedTracks();
+    MakeReconstructedTracks(refit);
     return fRecoTrackStore;
   }
 }
 
+
+//_____________________________________________________________________________
+AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggeredTracks(Int_t event)
+{
+  /// Return a track store containing the reconstructed trigger tracks (converted into 
+  /// MUONTriggerTrack objects) for a given event.
+       
+  if (!fESDEventOwner) {
+    if (fRecoTriggerTrackStore == 0x0) MakeTriggeredTracks();
+    return fRecoTriggerTrackStore;
+  }
+       
+  if (event != fCurrentEvent) {
+    ResetStores();
+    fCurrentEvent = event;
+  }
+       
+  if (fRecoTriggerTrackStore != 0x0) return fRecoTriggerTrackStore;
+  else {
+    if (!fESDTree) return 0x0;
+    if (fESDTree->GetEvent(event) <= 0) {
+      AliError(Form("fails to read ESD object for event %d", event));
+      return 0x0;
+    }
+    MakeTriggeredTracks();
+    return fRecoTriggerTrackStore;
+  }
+}
+
+
 //_____________________________________________________________________________
 AliMUONVTrackStore* AliMUONRecoCheck::TrackRefs(Int_t event)
 {
   /// Return a track store containing the track references (converted into 
   /// MUONTrack objects) for a given event
+  
+  if (!fESDEventOwner) {
+    if (fTrackRefStore == 0x0) MakeTrackRefs();
+    return fTrackRefStore;
+  }
+
   if (event != fCurrentEvent) {
     ResetStores();
     fCurrentEvent = event;
@@ -155,19 +300,60 @@ AliMUONVTrackStore* AliMUONRecoCheck::TrackRefs(Int_t event)
   
   if (fTrackRefStore != 0x0) return fTrackRefStore;
   else {
-    if (!fMCEventHandler->GetEvent(event)) {
+    if (!fMCEventHandler->BeginEvent(event)) {
       AliError(Form("fails to read MC objects for event %d", event));
       return 0x0;
     }
     MakeTrackRefs();
+    fMCEventHandler->FinishEvent();
     return fTrackRefStore;
   }
 }
 
 //_____________________________________________________________________________
-AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event)
+AliMUONVTriggerTrackStore* AliMUONRecoCheck::TriggerableTracks(Int_t event)
 {
-  /// Return a track store containing the reconstructible tracks for a given event
+  /// Return a trigger track store containing the triggerable track references (converted into 
+  /// AliMUONTriggerTrack objects) for a given event
+       
+  if (!fESDEventOwner) {
+    if (fRecoTriggerRefStore == 0x0) MakeTriggerableTracks();
+    return fRecoTriggerRefStore;
+  }
+       
+  if (event != fCurrentEvent) {
+    ResetStores();
+    fCurrentEvent = event;
+  }
+       
+  if (fRecoTriggerRefStore != 0x0) return fRecoTriggerRefStore;
+  else {
+    if (!fMCEventHandler->BeginEvent(event)) {
+      AliError(Form("fails to read MC objects for event %d", event));
+      return 0x0;
+    }
+    MakeTriggerableTracks();
+    fMCEventHandler->FinishEvent();
+    return fRecoTriggerRefStore;
+  }
+}
+
+
+//_____________________________________________________________________________
+AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event, UInt_t requestedStationMask,
+                                                           Bool_t request2ChInSameSt45, Bool_t hitInFrontOfPad)
+{
+  /// Return a track store containing the reconstructible tracks for a given event,
+  /// according to the mask of requested stations and the minimum number of chambers hit in stations 4 & 5.
+
+  if (!fESDEventOwner) {
+    if (fRecoTrackRefStore == 0x0) {
+      if (TrackRefs(event) == 0x0) return 0x0;
+      MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
+    }
+    return fRecoTrackRefStore;
+  }
+
   if (event != fCurrentEvent) {
     ResetStores();
     fCurrentEvent = event;
@@ -176,13 +362,14 @@ AliMUONVTrackStore* AliMUONRecoCheck::ReconstructibleTracks(Int_t event)
   if (fRecoTrackRefStore != 0x0) return fRecoTrackRefStore;
   else {
     if (TrackRefs(event) == 0x0) return 0x0;
-    MakeReconstructibleTracks();
+    MakeReconstructibleTracks(requestedStationMask, request2ChInSameSt45, hitInFrontOfPad);
     return fRecoTrackRefStore;
   }
 }
 
+
 //_____________________________________________________________________________
-void AliMUONRecoCheck::MakeReconstructedTracks()
+void AliMUONRecoCheck::MakeReconstructedTracks(Bool_t refit)
 {
   /// Make reconstructed tracks
   if (!(fRecoTrackStore = AliMUONESDInterface::NewTrackStore())) return;
@@ -191,11 +378,47 @@ void AliMUONRecoCheck::MakeReconstructedTracks()
   Int_t nTracks = (Int_t) fESDEvent->GetNumberOfMuonTracks();
   for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
     AliESDMuonTrack* esdTrack = fESDEvent->GetMuonTrack(iTrack);
-    if (esdTrack->ContainTrackerData()) AliMUONESDInterface::Add(GetRecoParam(),*esdTrack, *fRecoTrackStore);
+    if (esdTrack->ContainTrackerData()) AliMUONESDInterface::Add(*esdTrack, *fRecoTrackStore, refit);
   }
   
 }
 
+
+//_____________________________________________________________________________
+void AliMUONRecoCheck::MakeTriggeredTracks()
+{
+  /// Make reconstructed trigger tracks
+  if (!(fRecoTriggerTrackStore = AliMUONESDInterface::NewTriggerTrackStore())) return;
+       
+  AliMUONVTriggerStore* tmpTriggerStore = AliMUONESDInterface::NewTriggerStore();
+  if ( ! tmpTriggerStore ) return;
+  
+  // loop over all reconstructed tracks and add them to the store (include ghosts)
+  Int_t nTracks = (Int_t) fESDEvent->GetNumberOfMuonTracks();
+  for (Int_t iTrack = 0; iTrack < nTracks; iTrack++) {
+    AliESDMuonTrack* esdTrack = fESDEvent->GetMuonTrack(iTrack);
+    if (esdTrack->ContainTriggerData()) AliMUONESDInterface::Add(*esdTrack, *tmpTriggerStore);
+  }
+       
+  if ( InitCircuit() ) {
+  
+    AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
+    tracker->EventReconstructTrigger(*fTriggerCircuit, *tmpTriggerStore, *fRecoTriggerTrackStore);
+  }
+  
+  delete tmpTriggerStore;
+}
+
+//_____________________________________________________________________________
+Bool_t AliMUONRecoCheck::TriggerToTrack(const AliMUONLocalTrigger& locTrg, AliMUONTriggerTrack& triggerTrack)
+{
+  /// Make trigger track from local trigger info
+  if ( ! InitCircuit() ) return kFALSE;
+  AliMUONVTrackReconstructor* tracker = AliMUONESDInterface::GetTracker();
+  return tracker->TriggerToTrack(*fTriggerCircuit, locTrg, triggerTrack);
+}
+
+
 //_____________________________________________________________________________
 void AliMUONRecoCheck::MakeTrackRefs()
 {
@@ -207,9 +430,7 @@ void AliMUONRecoCheck::MakeTrackRefs()
   TParticle* particle;
   TClonesArray* trackRefs;
   Int_t nTrackRef = fMCEventHandler->MCEvent()->GetNumberOfTracks();
-  AliMUONVClusterStore* cStore = AliMUONESDInterface::NewClusterStore();
-  if (!cStore) return;
-  AliMUONVCluster* hit = cStore->CreateCluster(0,0,0);
+  AliMUONVCluster* hit = AliMUONESDInterface::NewCluster();
   
   // loop over simulated tracks
   for (Int_t iTrackRef  = 0; iTrackRef < nTrackRef; ++iTrackRef) {
@@ -240,11 +461,12 @@ void AliMUONRecoCheck::MakeTrackRefs()
       pZ = trackReference->Pz();
       
       // check chamberId of current trackReference
-      Int_t chamberId = AliMUONConstants::ChamberNumber(z);
+      Int_t detElemId = trackReference->UserId();
+      Int_t chamberId = detElemId / 100 - 1;
       if (chamberId < 0 || chamberId >= AliMUONConstants::NTrackingCh()) continue;
       
       // set hit parameters
-      hit->SetUniqueID(AliMUONVCluster::BuildUniqueID(chamberId, 0, 0));
+      hit->SetUniqueID(AliMUONVCluster::BuildUniqueID(chamberId, detElemId, iHit));
       hit->SetXYZ(x,y,z);
       hit->SetErrXY(0.,0.);
       
@@ -309,17 +531,132 @@ void AliMUONRecoCheck::MakeTrackRefs()
     track.SetTrackParamAtVertex(&trackParamAtVertex);
     
     // store the track
-    track.SetTrackID(iTrackRef);
+    track.SetUniqueID(iTrackRef);
     tmpTrackRefStore->Add(track);
   }
   
   CleanMuonTrackRef(tmpTrackRefStore);
   
   delete hit;
-  delete cStore;
   delete tmpTrackRefStore;
 }
 
+//_____________________________________________________________________________
+void AliMUONRecoCheck::MakeTriggerableTracks()
+{
+  /// Make triggerable tracks
+ if (!(fRecoTriggerRefStore = AliMUONESDInterface::NewTriggerTrackStore()))
+   return;
+
+  Double_t x, y, z, slopeX, slopeY, pZ, xLoc, yLoc, zLoc;
+  TParticle* particle;
+  TClonesArray* trackRefs;
+  Int_t nTrackRef = fMCEventHandler->MCEvent()->GetNumberOfTracks();
+       
+  // loop over simulated tracks
+  for (Int_t iTrackRef  = 0; iTrackRef < nTrackRef; ++iTrackRef) {
+    Int_t nHits = fMCEventHandler->GetParticleAndTR(iTrackRef, particle, trackRefs);
+               
+    // skip empty trackRefs
+    if (nHits < 1) continue;
+                               
+    AliMUONTriggerTrack track;
+    AliMUONDigitStoreV2S digitStore;
+    Int_t hitsOnTrigger = 0;
+    Int_t currCh = -1;
+               
+    // loop over simulated track hits
+    for (Int_t iHit = 0; iHit < nHits; ++iHit) {        
+      AliTrackReference* trackReference = static_cast<AliTrackReference*>(trackRefs->UncheckedAt(iHit));
+                       
+      // skip trackRefs not in MUON
+      if (trackReference->DetectorId() != AliTrackReference::kMUON) continue;
+
+      // check chamberId of current trackReference
+      Int_t detElemId = trackReference->UserId();
+      Int_t chamberId = detElemId / 100 - 1;
+      if (chamberId < AliMUONConstants::NTrackingCh() || chamberId >= AliMUONConstants::NCh() ) continue;
+
+      // Get track parameters of current hit
+      x = trackReference->X();
+      y = trackReference->Y();
+      z = trackReference->Z();
+      
+      if ( InitTriggerResponse() ) {
+        fGeometryTransformer->Global2Local(detElemId, x, y, z, xLoc, yLoc, zLoc);
+      
+        Int_t nboard = 0;
+        for ( Int_t cath = AliMp::kCath0; cath <= AliMp::kCath1; ++cath )
+        {
+          const AliMpVSegmentation* seg 
+          = AliMpSegmentation::Instance()
+          ->GetMpSegmentation(detElemId,AliMp::GetCathodType(cath));
+        
+          AliMpPad pad = seg->PadByPosition(xLoc,yLoc,kFALSE);
+          Int_t ix = pad.GetIx();
+          Int_t iy = pad.GetIy();
+        
+          if ( !pad.IsValid() ) continue;
+        
+          if ( cath == AliMp::kCath0 ) nboard = pad.GetLocalBoardId(0);
+        
+          AliMUONVDigit* digit = digitStore.Add(detElemId, nboard, pad.GetLocalBoardChannel(0),
+                                               cath, AliMUONVDigitStore::kDeny);
+          if (digit) {
+           digit->SetPadXY(ix,iy);
+           digit->SetCharge(1.);
+         }
+        }
+      }
+    
+      if ( hitsOnTrigger == 0 ) {
+        pZ = trackReference->Pz();
+        slopeX = ( pZ == 0. ) ? 99999. : trackReference->Px() / pZ;
+        slopeY = ( pZ == 0. ) ? 99999. : trackReference->Py() / pZ;
+
+        track.SetX11(x);
+        track.SetY11(y);
+        track.SetZ11(z);
+        track.SetSlopeX(slopeX);
+        track.SetSlopeY(slopeY);
+      }
+
+      if ( currCh != chamberId ) {
+        hitsOnTrigger++;
+        currCh = chamberId;
+      }
+
+    } // loop on hits
+    
+    if ( hitsOnTrigger < 3 ) continue;
+
+    // Check if the track passes the trigger algorithm
+    if ( InitTriggerResponse() ) {
+      AliMUONTriggerStoreV1 triggerStore;
+      fTriggerElectronics->Digits2Trigger(digitStore,triggerStore);
+    
+      TIter next(triggerStore.CreateIterator());
+      AliMUONLocalTrigger* locTrg(0x0);
+    
+      Int_t ptCutLevel = 0;
+    
+      while ( ( locTrg = static_cast<AliMUONLocalTrigger*>(next()) ) )
+      {
+        if ( locTrg->IsTrigX() && locTrg->IsTrigY() ) 
+        {
+          ptCutLevel = TMath::Max(ptCutLevel, 1);
+          if ( locTrg->LoHpt() ) ptCutLevel = TMath::Max(ptCutLevel, 3);
+          else if ( locTrg->LoLpt() ) ptCutLevel = TMath::Max(ptCutLevel, 2);
+        } // board is fired 
+      } // end of loop on Local Trigger
+      track.SetPtCutLevel(ptCutLevel);
+    }
+    track.SetUniqueID(iTrackRef);
+    fRecoTriggerRefStore->Add(track);
+  }
+}
+
+
 //_____________________________________________________________________________
 void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefStore)
 {
@@ -331,9 +668,7 @@ void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefSt
   Double_t maxGasGap = 1.; // cm 
   Double_t x, y, z, pX, pY, pZ, x1, y1, z1, pX1, pY1, pZ1, z2;
   Double_t bendingSlope,nonBendingSlope,inverseBendingMomentum;
-  AliMUONVClusterStore* cStore = AliMUONESDInterface::NewClusterStore();
-  if (!cStore) return;
-  AliMUONVCluster* hit = cStore->CreateCluster(0,0,0);
+  AliMUONVCluster* hit = AliMUONESDInterface::NewCluster();
   
   // create iterator
   TIter next(tmpTrackRefStore->CreateIterator());
@@ -407,7 +742,7 @@ void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefSt
       inverseBendingMomentum *= trackParam1->GetCharge();
       
       // set hit parameters
-      hit->SetUniqueID(AliMUONVCluster::BuildUniqueID(trackParam1->GetClusterPtr()->GetChamberId(), 0, 0));
+      hit->SetUniqueID(trackParam1->GetClusterPtr()->GetUniqueID());
       hit->SetXYZ(x,y,z);
       hit->SetErrXY(0.,0.);
       
@@ -426,55 +761,211 @@ void AliMUONRecoCheck::CleanMuonTrackRef(const AliMUONVTrackStore *tmpTrackRefSt
       iHit1++;
     }
     
-    newTrack.SetTrackID(track->GetTrackID());
+    newTrack.SetUniqueID(track->GetUniqueID());
     newTrack.SetTrackParamAtVertex(track->GetTrackParamAtVertex());
     fTrackRefStore->Add(newTrack);
     
   }
   
   delete hit;
-  delete cStore;
 }
 
 //_____________________________________________________________________________
-void AliMUONRecoCheck::MakeReconstructibleTracks()
+void AliMUONRecoCheck::MakeReconstructibleTracks(UInt_t requestedStationMask, Bool_t request2ChInSameSt45,
+                                                Bool_t hitInFrontOfPad)
 {
   /// Isolate the reconstructible tracks
   if (!(fRecoTrackRefStore = AliMUONESDInterface::NewTrackStore())) return;
   
+  // need geometry transformer and segmentation to check if trackrefs are located in front of pad(s)
+  if (hitInFrontOfPad) {
+    if (!InitGeometryTransformer()) return;
+    if (!AliMpSegmentation::Instance(kFALSE) && !AliMUONCDB::LoadMapping(kTRUE)) return;
+  }
+  
   // create iterator on trackRef
   TIter next(fTrackRefStore->CreateIterator());
   
-  // loop over trackRef
+  // loop over trackRef and add reconstructible tracks to fRecoTrackRefStore
   AliMUONTrack* track;
   while ( ( track = static_cast<AliMUONTrack*>(next()) ) ) {
     
-    Bool_t* chamberInTrack = new Bool_t(AliMUONConstants::NTrackingCh());
-    for (Int_t iCh = 0; iCh < AliMUONConstants::NTrackingCh(); iCh++) chamberInTrack[iCh] = kFALSE;
+    // check if the track is valid as it is
+    if (track->IsValid(requestedStationMask, request2ChInSameSt45)) {
+      
+      if (hitInFrontOfPad) {
+       
+       AliMUONTrack trackTmp(*track);
+       
+       // loop over clusters and remove those which are not in front of pad(s)
+       AliMUONTrackParam *param = static_cast<AliMUONTrackParam*>(trackTmp.GetTrackParamAtCluster()->First());
+       while (param) {
+         AliMUONTrackParam *nextParam = static_cast<AliMUONTrackParam*>(trackTmp.GetTrackParamAtCluster()->After(param));
+         if (!IsHitInFrontOfPad(param)) trackTmp.RemoveTrackParamAtCluster(param);
+         param = nextParam;
+       }
+       
+       // add the track if it is still valid
+       if (trackTmp.IsValid(requestedStationMask, request2ChInSameSt45)) fRecoTrackRefStore->Add(trackTmp);
+       
+      } else {
+       
+       // add the track
+       fRecoTrackRefStore->Add(*track);
+       
+      }
+      
+    }
+      
+  }
+
+}
+
+//_____________________________________________________________________________
+AliMUONTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTrack &track, AliMUONVTrackStore &trackStore,
+                                                   Int_t &nMatchClusters, Bool_t useLabel, Double_t sigmaCut)
+{
+  /// Return the track from the store matched with the given track (or 0x0) and the number of matched clusters.
+  /// Matching is done by using the MC label of by comparing cluster/TrackRef positions according to the flag "useLabel".
+  /// WARNING: Who match who matters since the matching algorithm uses the *fraction* of matched clusters of the given track
+  
+  AliMUONTrack *matchedTrack = 0x0;
+  nMatchClusters = 0;
+  
+  if (useLabel) { // by using the MC label
     
-    // loop over trackRef's hits to get hit chambers
-    Int_t nTrackHits = track->GetNClusters();
-    for (Int_t iHit = 0; iHit < nTrackHits; iHit++) {
-      AliMUONVCluster* hit = ((AliMUONTrackParam*) track->GetTrackParamAtCluster()->UncheckedAt(iHit))->GetClusterPtr(); 
-      chamberInTrack[hit->GetChamberId()] = kTRUE;
-    } 
+    // get the corresponding simulated track if any
+    Int_t label = track.GetMCLabel();
+    matchedTrack = (AliMUONTrack*) trackStore.FindObject(label);
+    
+    // get the fraction of matched clusters
+    if (matchedTrack) {
+      Int_t nClusters = track.GetNClusters();
+      for (Int_t iCl = 0; iCl < nClusters; iCl++)
+       if (((AliMUONTrackParam*) track.GetTrackParamAtCluster()->UncheckedAt(iCl))->GetClusterPtr()->GetMCLabel() == label)
+         nMatchClusters++;
+    }
+    
+  } else { // by comparing cluster/TrackRef positions
+    
+    // look for the corresponding simulated track if any
+    TIter next(trackStore.CreateIterator());
+    AliMUONTrack* track2;
+    while ( ( track2 = static_cast<AliMUONTrack*>(next()) ) ) {
+      
+      // check compatibility
+      Int_t n = 0;
+      if (track.Match(*track2, sigmaCut, n)) {
+       matchedTrack = track2;
+       nMatchClusters = n;
+       break;
+      }
+      
+    }
+    
+  }
+  
+  return matchedTrack;
+  
+}
+
+
+//_____________________________________________________________________________
+AliMUONTriggerTrack* AliMUONRecoCheck::FindCompatibleTrack(AliMUONTriggerTrack &track, const AliMUONVTriggerTrackStore &triggerTrackStore,
+                                                           Double_t sigmaCut)
+{
+  /// Return the trigger track from the store matched with the given track (or 0x0).
+  /// Matching is done by comparing cluster/TrackRef positions.
+  
+  AliMUONTriggerTrack *matchedTrack = 0x0;
+  
+  // look for the corresponding simulated track if any
+  TIter next(triggerTrackStore.CreateIterator());
+  AliMUONTriggerTrack* track2;
+  while ( ( track2 = static_cast<AliMUONTriggerTrack*>(next()) ) ) {
+    // check compatibility
+    if (track.Match(*track2, sigmaCut)) {
+      matchedTrack = track2;
+      break;
+    }
+  }
+  
+  return matchedTrack;
+  
+}
+
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitTriggerResponse()
+{
+  /// Initialize trigger electronics
+  /// for building of triggerable tracks from MC
+  
+  if ( fTriggerElectronics ) return kTRUE;
+  
+  if ( ! InitGeometryTransformer() ) return kFALSE;
+  
+  if ( ! AliMpDDLStore::Instance(false) && ! AliMpCDB::LoadDDLStore()) return kFALSE;
+  
+  if ( ! InitCalibrationData() ) return kFALSE;
+  
+  fTriggerElectronics = new AliMUONTriggerElectronics(fCalibrationData);
+  
+  return kTRUE;
+}
+
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitCalibrationData()
+{
+  /// Initialize calibration data
+  if ( ! fCalibrationData ) {
+    if ( !AliMUONCDB::CheckOCDB() ) return kFALSE;
+    fCalibrationData = new AliMUONCalibrationData(AliCDBManager::Instance()->GetRun());
+  }
+  return kTRUE;
+}
+
+
+//____________________________________________________________________________ 
+Bool_t AliMUONRecoCheck::InitGeometryTransformer()
+{
+  /// Return geometry transformer
+  /// (create it if necessary)
+  if ( ! fGeometryTransformer ) {
     
-    // track is reconstructible if the particle is depositing a hit
-    // in the following chamber combinations:
-    Bool_t trackOK = kTRUE;
-    if (!chamberInTrack[0] && !chamberInTrack[1]) trackOK = kFALSE;
-    if (!chamberInTrack[2] && !chamberInTrack[3]) trackOK = kFALSE;
-    if (!chamberInTrack[4] && !chamberInTrack[5]) trackOK = kFALSE;
-    Int_t nHitsInLastStations = 0;
-    for (Int_t iCh = 6; iCh < AliMUONConstants::NTrackingCh(); iCh++)
-      if (chamberInTrack[iCh]) nHitsInLastStations++; 
-    if(nHitsInLastStations < 3) trackOK = kFALSE;
+    if ( !AliMUONCDB::CheckOCDB() ) return kFALSE;
     
-    // Add reconstructible tracks to fRecoTrackRefStore
-    if (trackOK) fRecoTrackRefStore->Add(*track);
+    if (!AliGeomManager::GetGeometry()) {
+      AliGeomManager::LoadGeometry();
+      if (!AliGeomManager::GetGeometry()) return kFALSE;  
+      if (!AliGeomManager::ApplyAlignObjsFromCDB("MUON")) return kFALSE;
+    }
     
-    delete [] chamberInTrack;
+    fGeometryTransformer = new AliMUONGeometryTransformer();
+    fGeometryTransformer->LoadGeometryData();
   }
+  
+  return kTRUE;
+}
 
+//________________________________________________________________________
+Bool_t AliMUONRecoCheck::IsHitInFrontOfPad(AliMUONTrackParam *param) const
+{
+  /// Return kTRUE if the hit is located in front of at least 1 pad
+  
+  Int_t deId = param->GetClusterPtr()->GetDetElemId();
+  
+  Double_t xL, yL, zL;
+  fGeometryTransformer->Global2Local(deId, param->GetNonBendingCoor(), param->GetBendingCoor(), param->GetZ(), xL, yL, zL);
+  
+  const AliMpVSegmentation* seg1 = AliMpSegmentation::Instance()->GetMpSegmentation(deId, AliMp::kCath0);
+  const AliMpVSegmentation* seg2 = AliMpSegmentation::Instance()->GetMpSegmentation(deId, AliMp::kCath1);
+  if (!seg1 || !seg2) return kFALSE;
+  
+  AliMpPad pad1 = seg1->PadByPosition(xL, yL, kFALSE);
+  AliMpPad pad2 = seg2->PadByPosition(xL, yL, kFALSE);
+  
+  return (pad1.IsValid() || pad2.IsValid());
 }