]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONVTrackReconstructor.cxx
AliMUONRecoParam:
[u/mrichter/AliRoot.git] / MUON / AliMUONVTrackReconstructor.cxx
index deb0d6946a7e143e091a237ea66ae6203174bed4..4b0e24ea2f84c6a7f9565827ee3612fb6bc41130 100644 (file)
 #include "AliMUONVTrackStore.h"
 #include "AliMUONVClusterStore.h"
 #include "AliMUONVCluster.h"
+#include "AliMUONVClusterServer.h"
 #include "AliMUONVTriggerStore.h"
 #include "AliMUONVTriggerTrackStore.h"
 #include "AliMpDEManager.h"
+#include "AliMpArea.h"
 
 #include "AliLog.h"
 #include "AliCodeTimer.h"
@@ -82,6 +84,7 @@
 #include <TClonesArray.h>
 #include <TMath.h>
 #include <TMatrixD.h>
+#include <TVector2.h>
 
 #include <Riostream.h>
 
@@ -90,10 +93,11 @@ ClassImp(AliMUONVTrackReconstructor) // Class implementation in ROOT context
 /// \endcond
 
   //__________________________________________________________________________
-AliMUONVTrackReconstructor::AliMUONVTrackReconstructor()
+AliMUONVTrackReconstructor::AliMUONVTrackReconstructor(AliMUONVClusterServer& clusterServer)
   : TObject(),
     fRecTracksPtr(0x0),
-    fNRecTracks(0)
+    fNRecTracks(0),
+  fClusterServer(clusterServer)
 {
   /// Constructor for class AliMUONVTrackReconstructor
   
@@ -123,7 +127,7 @@ void AliMUONVTrackReconstructor::ResetTracks()
 }
 
   //__________________________________________________________________________
-void AliMUONVTrackReconstructor::EventReconstruct(const AliMUONVClusterStore& clusterStore, AliMUONVTrackStore& trackStore)
+void AliMUONVTrackReconstructor::EventReconstruct(AliMUONVClusterStore& clusterStore, AliMUONVTrackStore& trackStore)
 {
   /// To reconstruct one event
   AliDebug(1,"");
@@ -157,6 +161,7 @@ void AliMUONVTrackReconstructor::EventReconstruct(const AliMUONVClusterStore& cl
   for (Int_t i=0; i<fNRecTracks; ++i) 
   {
     AliMUONTrack * track = (AliMUONTrack*) fRecTracksPtr->At(i);
+    track->SetUniqueID(i+1);
     trackStore.Add(*track);
   }
 }
@@ -181,7 +186,6 @@ TClonesArray* AliMUONVTrackReconstructor::MakeSegmentsInStation(const AliMUONVCl
   
   // list of segments
   TClonesArray *segments = new TClonesArray("AliMUONObjectPair", 100);
-  segments->SetOwner(kTRUE);
   
   // Loop over clusters in the first chamber of the station
   while ( ( cluster1 = static_cast<AliMUONVCluster*>(nextInCh1()) ) ) {
@@ -328,9 +332,43 @@ void AliMUONVTrackReconstructor::RemoveDoubleTracks()
   return;
 }
 
+  //__________________________________________________________________________
+void AliMUONVTrackReconstructor::AskForNewClustersInStation(const AliMUONTrackParam &trackParam,
+                                                           AliMUONVClusterStore& clusterStore, Int_t station)
+{
+  /// Ask the clustering to reconstruct new clusters around the track candidate position
+  /// in the 2 chambers of the given station
+  
+  // maximum shift of the searching area due to distance between detection elements and the track slope
+  static const Double_t kgMaxShift = 2.; // 2 cm
+  
+  // extrapolate track parameters to the second chamber of the station
+  AliMUONTrackParam extrapTrackParam(trackParam);
+  AliMUONTrackExtrap::ExtrapToZ(&extrapTrackParam, AliMUONConstants::DefaultChamberZ(2*station+1));
+  
+  // build the searching area
+  TVector2 position(extrapTrackParam.GetNonBendingCoor(), extrapTrackParam.GetBendingCoor());
+  TVector2 dimensions(AliMUONReconstructor::GetRecoParam()->GetMaxNonBendingDistanceToTrack() + kgMaxShift,
+                     AliMUONReconstructor::GetRecoParam()->GetMaxBendingDistanceToTrack() + kgMaxShift);
+  AliMpArea area2(position, dimensions);
+  
+  // ask to cluterize in the given area of the given chamber
+  fClusterServer.Clusterize(2*station+1, clusterStore, area2);
+  
+  // extrapolate track parameters to the first chamber of the station
+  AliMUONTrackExtrap::ExtrapToZ(&extrapTrackParam, AliMUONConstants::DefaultChamberZ(2*station));
+  
+  // build the searching area
+  position.Set(extrapTrackParam.GetNonBendingCoor(), extrapTrackParam.GetBendingCoor());
+  AliMpArea area1(position, dimensions);
+  
+  // ask to cluterize in the given area of the given chamber
+  fClusterServer.Clusterize(2*station, clusterStore, area1);
+}
+
   //__________________________________________________________________________
 Double_t AliMUONVTrackReconstructor::TryOneCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster* cluster,
-                                                    AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator)
+                                                  AliMUONTrackParam &trackParamAtCluster, Bool_t updatePropagator)
 {
 /// Test the compatibility between the track and the cluster (using trackParam's covariance matrix):
 /// return the corresponding Chi2
@@ -455,6 +493,7 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
   Bool_t foundTwoClusters = kFALSE;
   AliMUONTrack *newTrack = 0x0;
   AliMUONVCluster *clusterCh1, *clusterCh2;
+  AliMUONTrackParam trackParam;
   AliMUONTrackParam extrapTrackParamAtCluster1;
   AliMUONTrackParam extrapTrackParamAtCluster2;
   AliMUONTrackParam bestTrackParamAtCluster1;
@@ -465,8 +504,9 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
   for (Int_t i = 0; i < nClusters; i++) clusterCh1Used[i] = kFALSE;
   Int_t iCluster1;
   
-  // Get track parameters
-  AliMUONTrackParam trackParam(*(AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->First());
+  // Get track parameters according to the propagation direction
+  if (nextStation==4) trackParam = *(AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->Last();
+  else trackParam = *(AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->First();
   
   // Add MCS effect
   AliMUONTrackExtrap::AddMCSEffect(&trackParam,AliMUONConstants::ChamberThicknessInX0(),1.);
@@ -537,7 +577,6 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
            newTrack->AddTrackParamAtCluster(extrapTrackParamAtCluster1,*clusterCh1);
            extrapTrackParamAtCluster2.SetRemovable(kTRUE);
            newTrack->AddTrackParamAtCluster(extrapTrackParamAtCluster2,*clusterCh2);
-           newTrack->GetTrackParamAtCluster()->Sort();
            fNRecTracks++;
            
            // Tag clusterCh1 as used
@@ -569,7 +608,6 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
           newTrack = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(trackCandidate);
          extrapTrackParamAtCluster2.SetRemovable(kFALSE);
          newTrack->AddTrackParamAtCluster(extrapTrackParamAtCluster2,*clusterCh2);
-         newTrack->GetTrackParamAtCluster()->Sort();
          fNRecTracks++;
          
          // Printout for debuging
@@ -639,7 +677,6 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
          newTrack = new ((*fRecTracksPtr)[fRecTracksPtr->GetLast()+1]) AliMUONTrack(trackCandidate);
          extrapTrackParamAtCluster1.SetRemovable(kFALSE);
          newTrack->AddTrackParamAtCluster(extrapTrackParamAtCluster1,*clusterCh1);
-         newTrack->GetTrackParamAtCluster()->Sort();
          fNRecTracks++;
          
          // Printout for debuging
@@ -667,7 +704,6 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
       trackCandidate.AddTrackParamAtCluster(bestTrackParamAtCluster1,*(bestTrackParamAtCluster1.GetClusterPtr()));
       bestTrackParamAtCluster2.SetRemovable(kTRUE);
       trackCandidate.AddTrackParamAtCluster(bestTrackParamAtCluster2,*(bestTrackParamAtCluster2.GetClusterPtr()));
-      trackCandidate.GetTrackParamAtCluster()->Sort();
       
       // Printout for debuging
       if ((AliLog::GetDebugLevel("MUON","AliMUONVTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
@@ -678,7 +714,6 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
     } else if (foundOneCluster) {
       bestTrackParamAtCluster1.SetRemovable(kFALSE);
       trackCandidate.AddTrackParamAtCluster(bestTrackParamAtCluster1,*(bestTrackParamAtCluster1.GetClusterPtr()));
-      trackCandidate.GetTrackParamAtCluster()->Sort();
       
       // Printout for debuging
       if ((AliLog::GetDebugLevel("MUON","AliMUONVTrackReconstructor") >= 1) || (AliLog::GetGlobalDebugLevel() >= 1)) {
@@ -707,173 +742,68 @@ Bool_t AliMUONVTrackReconstructor::FollowLinearTrackInStation(AliMUONTrack &trac
   
 }
 
-  //__________________________________________________________________________
-void AliMUONVTrackReconstructor::ValidateTracksWithTrigger(AliMUONVTrackStore& trackStore,
-                                                           const AliMUONVTriggerTrackStore& triggerTrackStore,
-                                                           const AliMUONVTriggerStore& triggerStore,
-                                                           const AliMUONTrackHitPattern& trackHitPattern)
+//__________________________________________________________________________
+void AliMUONVTrackReconstructor::ImproveTracks()
 {
-  /// Try to match track from tracking system with trigger track
-  AliCodeTimerAuto("");
-
-  const Double_t kDeltaZ = TMath::Abs(AliMUONConstants::DefaultChamberZ(12) - AliMUONConstants::DefaultChamberZ(10));
-
-  //static const Double_t kDistSigma[3]={1,1,0.02}; // sigma of distributions (trigger-track) X,Y,slopeY
-  // sigma of distributions (trigger-track) X,Y,slopeY
-  const Double_t kDistSigma[3]={AliMUONConstants::TriggerNonBendingReso(),
-                               AliMUONConstants::TriggerBendingReso(),
-                               1.414 * AliMUONConstants::TriggerBendingReso()/kDeltaZ};
-
-  const Double_t kTrigNonBendReso = AliMUONConstants::TriggerNonBendingReso();
-  const Double_t kTrigBendReso = AliMUONConstants::TriggerBendingReso();
-  const Double_t kTrigSlopeBendReso = 1.414 * AliMUONConstants::TriggerBendingReso()/kDeltaZ;
-  const Double_t kTrigCovSlopeBend = - kTrigBendReso * kTrigBendReso / kDeltaZ;
-
-  // Covariance matrix 3x3 (X,Y,slopeY) for trigger tracks
-  TMatrixD trigCov(3,3);
-  trigCov.Zero();
-  trigCov(0,0) = kTrigNonBendReso * kTrigNonBendReso;
-  trigCov(1,1) = kTrigBendReso * kTrigBendReso;
-  trigCov(2,2) = kTrigSlopeBendReso * kTrigSlopeBendReso;
-  trigCov(1,2) = trigCov(2,1) = kTrigCovSlopeBend;
-
-  Int_t matchTrigger;
-  Int_t loTrgNum(-1);
-  Double_t distTriggerTrack[3], sigma2[3];
-  Double_t xTrack, yTrack, ySlopeTrack, chi2MatchTrigger, minChi2MatchTrigger, chi2;
-
-  TIter itTrack(trackStore.CreateIterator());
-  AliMUONTrack* track;
-
-  const Float_t kZFilterOut = AliMUONConstants::MuonFilterZEnd();
-  const Float_t kFilterThickness = TMath::Abs(kZFilterOut-AliMUONConstants::MuonFilterZBeg()); // cm
-  const Int_t kFirstTrigCh = AliMUONConstants::NTrackingCh();
+  /// Improve tracks by removing clusters with local chi2 highter than the defined cut
+  /// Recompute track parameters and covariances at the remaining clusters
+  AliDebug(1,"Enter ImproveTracks");
   
-  while ( ( track = static_cast<AliMUONTrack*>(itTrack()) ) )
-  {
-    matchTrigger = 0;
-    chi2MatchTrigger = 0.;
-    loTrgNum = -1;
-    Int_t doubleMatch=-1; // Check if track matches 2 trigger tracks
-    Double_t doubleChi2 = -1.;
+  AliMUONTrack *track, *nextTrack;
+  
+  // Remove double track to improve only "good" tracks
+  RemoveDoubleTracks();
+  
+  track = (AliMUONTrack*) fRecTracksPtr->First();
+  while (track) {
     
-    AliMUONTrackParam trackParam(*((AliMUONTrackParam*) (track->GetTrackParamAtCluster()->Last())));
-
-    AliMUONTrackExtrap::ExtrapToZCov(&trackParam, kZFilterOut); // Extrap to muon filter end
-    AliMUONTrackExtrap::AddMCSEffect(&trackParam, kFilterThickness, AliMUONConstants::MuonFilterX0()); // Add MCS effects
-    AliMUONTrackExtrap::ExtrapToZCov(&trackParam, AliMUONConstants::DefaultChamberZ(kFirstTrigCh)); // extrap to 1st trigger chamber
-
-    const TMatrixD& kParamCov = trackParam.GetCovariances();
+    // prepare next track in case the actual track is suppressed
+    nextTrack = (AliMUONTrack*) fRecTracksPtr->After(track);
     
-    xTrack = trackParam.GetNonBendingCoor();
-    yTrack = trackParam.GetBendingCoor();
-    ySlopeTrack = trackParam.GetBendingSlope();
-
-    // Covariance matrix 3x3 (X,Y,slopeY) for tracker tracks
-    TMatrixD trackCov(3,3);
-    trackCov.Zero();
-    trackCov(0,0) = kParamCov(0,0);
-    trackCov(1,1) = kParamCov(2,2);
-    trackCov(2,2) = kParamCov(3,3);
-    trackCov(1,2) = kParamCov(2,3);
-    trackCov(2,1) = kParamCov(3,2);
-
-    TMatrixD sumCov(trackCov,TMatrixD::kPlus,trigCov);
-
-    Bool_t isCovOK = kTRUE;
-
-    if (sumCov.Determinant() != 0) {
-      sumCov.Invert();
-    } else {
-      AliWarning(" Determinant = 0");
-      isCovOK = kFALSE;
-      sigma2[0] = kParamCov(0,0);
-      sigma2[1] = kParamCov(2,2);
-      sigma2[2] = kParamCov(3,3);
-      for (Int_t iVar = 0; iVar < 3; iVar++) sigma2[iVar] += kDistSigma[iVar] * kDistSigma[iVar];
+    ImproveTrack(*track);
+    
+    // remove track if improvement failed
+    if (!track->IsImproved()) {
+      fRecTracksPtr->Remove(track);
+      fNRecTracks--;
     }
+    
+    track = nextTrack;
+  }
+  
+  // compress the array in case of some tracks have been removed
+  fRecTracksPtr->Compress();
+  
+}
 
-    minChi2MatchTrigger = 999.;
-
-    AliMUONTriggerTrack *triggerTrack;
-    TIter itTriggerTrack(triggerTrackStore.CreateIterator());
-    while ( ( triggerTrack = static_cast<AliMUONTriggerTrack*>(itTriggerTrack() ) ) )
-    {
-      distTriggerTrack[0] = triggerTrack->GetX11()-xTrack;
-      distTriggerTrack[1] = triggerTrack->GetY11()-yTrack;
-      distTriggerTrack[2] = TMath::Tan(triggerTrack->GetThetay())-ySlopeTrack;
-
-      if(isCovOK){
-       TMatrixD paramDiff(3,1);
-       for(Int_t iVar = 0; iVar < 3; iVar++)
-         paramDiff(iVar,0) = distTriggerTrack[iVar];
-       
-       TMatrixD tmp(sumCov,TMatrixD::kMult,paramDiff);
-       TMatrixD chi2M(paramDiff,TMatrixD::kTransposeMult,tmp);
-       chi2 = chi2M(0,0);
-      }
-      else {
-       chi2 = 0.;
-       for (Int_t iVar = 0; iVar < 3; iVar++) chi2 += distTriggerTrack[iVar]*distTriggerTrack[iVar]/sigma2[iVar];
-      }
-
-      chi2 /= 3.; // Normalized Chi2: 3 degrees of freedom (X,Y,slopeY)
-      if (chi2 < AliMUONReconstructor::GetRecoParam()->GetMaxNormChi2MatchTrigger()) 
-      {
-        Bool_t isDoubleTrack = (TMath::Abs(chi2 - minChi2MatchTrigger)<1.);
-        if (chi2 < minChi2MatchTrigger && chi2 < AliMUONReconstructor::GetRecoParam()->GetMaxNormChi2MatchTrigger()) 
-        {
-          if(isDoubleTrack)
-          {
-            doubleMatch = loTrgNum;
-            doubleChi2 = chi2MatchTrigger;
-          }
-          minChi2MatchTrigger = chi2;
-          chi2MatchTrigger = chi2;
-          loTrgNum = triggerTrack->GetLoTrgNum();
-          AliMUONLocalTrigger* locTrg = triggerStore.FindLocal(loTrgNum);
-          matchTrigger=1;
-          if(locTrg->LoLpt()>0)matchTrigger=2;
-          if(locTrg->LoHpt()>0)matchTrigger=3;
-        }
-        else if(isDoubleTrack) 
-        {
-          doubleMatch = triggerTrack->GetLoTrgNum();
-          doubleChi2 = chi2;
-        }
-      }
-    }
-    if(doubleMatch>=0)
-    { // If two trigger tracks match, select the one passing more trigger cuts
-      AliDebug(1, Form("Two candidates found: %i and %i",loTrgNum,doubleMatch));
-      AliMUONLocalTrigger* locTrg1 = triggerStore.FindLocal(doubleMatch);
-      if((locTrg1->LoLpt()>0 && matchTrigger<2) || (locTrg1->LoHpt() && matchTrigger<3))
-      {
-        if(locTrg1->LoHpt()>0)matchTrigger=3;
-        else matchTrigger=2;
-        loTrgNum = doubleMatch;
-        chi2MatchTrigger=doubleChi2;
-      }
-    }
+//__________________________________________________________________________
+void AliMUONVTrackReconstructor::Finalize()
+{
+  /// Recompute track parameters and covariances at each attached cluster from those at the first one
+  
+  AliMUONTrack *track;
+  
+  track = (AliMUONTrack*) fRecTracksPtr->First();
+  while (track) {
     
-    track->SetMatchTrigger(matchTrigger);
-    track->SetLoTrgNum(loTrgNum);
-    track->SetChi2MatchTrigger(chi2MatchTrigger);
+    FinalizeTrack(*track);
     
-    AliMUONLocalTrigger* locTrg = static_cast<AliMUONLocalTrigger*>(triggerStore.FindLocal(loTrgNum));
+    track = (AliMUONTrack*) fRecTracksPtr->After(track);
     
-    if (locTrg)
-    {    
-      track->SetLocalTrigger(locTrg->LoCircuit(),
-                             locTrg->LoStripX(),
-                             locTrg->LoStripY(),
-                             locTrg->LoDev(),
-                             locTrg->LoLpt(),
-                             locTrg->LoHpt());
-    }    
-  }  
-  
-  trackHitPattern.GetHitPattern(trackStore,triggerStore);
+  }
+  
+}
+
+//__________________________________________________________________________
+void AliMUONVTrackReconstructor::ValidateTracksWithTrigger(AliMUONVTrackStore& trackStore,
+                                                           const AliMUONVTriggerTrackStore& triggerTrackStore,
+                                                           const AliMUONVTriggerStore& triggerStore,
+                                                           const AliMUONTrackHitPattern& trackHitPattern)
+{
+  /// Try to match track from tracking system with trigger track
+  AliCodeTimerAuto("");
+
+  trackHitPattern.ExecuteValidation(trackStore, triggerTrackStore, triggerStore);
 }
 
   //__________________________________________________________________________