]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - MUON/AliMUONTrack.cxx
Updating macro to test GRP preprocessor with new input files.
[u/mrichter/AliRoot.git] / MUON / AliMUONTrack.cxx
index f311a65899ff3b8d512176b6a633e232efdc3a19..da3e83291b471411492e909dae06950db457d7a3 100644 (file)
 ClassImp(AliMUONTrack) // Class implementation in ROOT context
 /// \endcond
 
+
+const Double_t AliMUONTrack::fgkMaxChi2 = 1.e10; ///< maximum chi2 above which the track can be considered as abnormal
+
+
 //__________________________________________________________________________
 AliMUONTrack::AliMUONTrack()
   : TObject(),
@@ -141,24 +145,31 @@ AliMUONTrack::AliMUONTrack(AliMUONObjectPair *segment, Double_t bendingVertexDis
   paramCov(3,2) = paramCov(2,3);
   paramCov(3,3) = ( firstCluster->GetErrY2() + lastCluster->GetErrY2() ) / dZ / dZ;
   // Inverse bending momentum (vertex resolution + bending slope resolution + 10% error on dipole parameters+field)
-  paramCov(4,4) = ( ( bendingVertexDispersion*bendingVertexDispersion +
-                   (z1 * z1 * lastCluster->GetErrY2() + z2 * z2 * firstCluster->GetErrY2()) / dZ / dZ) /
-                  bendingImpact / bendingImpact + 0.1 * 0.1) * inverseBendingMomentum * inverseBendingMomentum ;
-  paramCov(2,4) = - z2 * firstCluster->GetErrY2() * inverseBendingMomentum / bendingImpact / dZ;
-  paramCov(4,2) = paramCov(2,4);
-  paramCov(3,4) = - (z1 * lastCluster->GetErrY2() + z2 * firstCluster->GetErrY2()) * inverseBendingMomentum / bendingImpact / dZ / dZ;
-  paramCov(4,3) = paramCov(3,4);
-  
-  // Set covariances
+  if (AliMUONTrackExtrap::IsFieldON()) {
+    paramCov(4,4) = ( ( bendingVertexDispersion*bendingVertexDispersion +
+                      (z1 * z1 * lastCluster->GetErrY2() + z2 * z2 * firstCluster->GetErrY2()) / dZ / dZ) /
+                    bendingImpact / bendingImpact + 0.1 * 0.1) * inverseBendingMomentum * inverseBendingMomentum ;
+    paramCov(2,4) = - z2 * firstCluster->GetErrY2() * inverseBendingMomentum / bendingImpact / dZ;
+    paramCov(4,2) = paramCov(2,4);
+    paramCov(3,4) = - (z1 * lastCluster->GetErrY2() + z2 * firstCluster->GetErrY2()) * inverseBendingMomentum / bendingImpact / dZ / dZ;
+    paramCov(4,3) = paramCov(3,4);
+  } else paramCov(4,4) = inverseBendingMomentum*inverseBendingMomentum;
   trackParamAtFirstCluster.SetCovariances(paramCov);
   
   // Compute and set track parameters covariances at last cluster
-  paramCov(1,0) = - paramCov(1,0);
-  paramCov(0,1) = - paramCov(0,1);
-  paramCov(3,2) = - paramCov(3,2);
-  paramCov(2,3) = - paramCov(2,3);
-  paramCov(2,4) = z1 * lastCluster->GetErrY2() * inverseBendingMomentum / bendingImpact / dZ;
-  paramCov(4,2) = paramCov(2,4);
+  // Non bending plane
+  paramCov(0,0) = lastCluster->GetErrX2();
+  paramCov(0,1) = - lastCluster->GetErrX2() / dZ;
+  paramCov(1,0) = paramCov(0,1);
+  // Bending plane
+  paramCov(2,2) = lastCluster->GetErrY2();
+  paramCov(2,3) = - lastCluster->GetErrY2() / dZ;
+  paramCov(3,2) = paramCov(2,3);
+  // Inverse bending momentum (vertex resolution + bending slope resolution + 10% error on dipole parameters+field)
+  if (AliMUONTrackExtrap::IsFieldON()) {
+    paramCov(2,4) = z1 * lastCluster->GetErrY2() * inverseBendingMomentum / bendingImpact / dZ;
+    paramCov(4,2) = paramCov(2,4);
+  }
   trackParamAtLastCluster.SetCovariances(paramCov);
   
   // Add track parameters at clusters
@@ -375,15 +386,17 @@ void AliMUONTrack::RemoveTrackParamAtCluster(AliMUONTrackParam *trackParam)
 }
 
   //__________________________________________________________________________
-void AliMUONTrack::UpdateTrackParamAtCluster()
+Bool_t AliMUONTrack::UpdateTrackParamAtCluster()
 {
   /// Update track parameters at each attached cluster
+  /// Return kFALSE in case of failure (i.e. extrapolation problem)
   
   if (GetNClusters() == 0) {
     AliWarning("no cluster attached to the track");
-    return;
+    return kFALSE;
   }
   
+  Bool_t extrapStatus = kTRUE;
   AliMUONTrackParam* startingTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->First();
   AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(startingTrackParam);
   while (trackParamAtCluster) {
@@ -393,26 +406,32 @@ void AliMUONTrack::UpdateTrackParamAtCluster()
     trackParamAtCluster->SetZ(startingTrackParam->GetZ());
     
     // extrapolation to the given z
-    AliMUONTrackExtrap::ExtrapToZ(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ());
+    if (!AliMUONTrackExtrap::ExtrapToZ(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ())) extrapStatus = kFALSE;
     
     // prepare next step
     startingTrackParam = trackParamAtCluster;
     trackParamAtCluster = (AliMUONTrackParam*) (fTrackParamAtCluster->After(trackParamAtCluster));
   }
 
+  // set global chi2 to max value in case of problem during track extrapolation
+  if (!extrapStatus) SetGlobalChi2(2.*MaxChi2());
+  return extrapStatus;
+  
 }
 
   //__________________________________________________________________________
-void AliMUONTrack::UpdateCovTrackParamAtCluster()
+Bool_t AliMUONTrack::UpdateCovTrackParamAtCluster()
 {
   /// Update track parameters and their covariances at each attached cluster
   /// Include effects of multiple scattering in chambers
+  /// Return kFALSE in case of failure (i.e. extrapolation problem)
   
   if (GetNClusters() == 0) {
     AliWarning("no cluster attached to the track");
-    return;
+    return kFALSE;
   }
   
+  Bool_t extrapStatus = kTRUE;
   AliMUONTrackParam* startingTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->First();
   AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(startingTrackParam);
   Int_t expectedChamber = startingTrackParam->GetClusterPtr()->GetChamberId() + 1;
@@ -431,14 +450,14 @@ void AliMUONTrack::UpdateCovTrackParamAtCluster()
     currentChamber = trackParamAtCluster->GetClusterPtr()->GetChamberId();
     while (currentChamber > expectedChamber) {
       // extrapolation to the missing chamber
-      AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, AliMUONConstants::DefaultChamberZ(expectedChamber));
+      if (!AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, AliMUONConstants::DefaultChamberZ(expectedChamber))) extrapStatus = kFALSE;
       // add MCS effect
       AliMUONTrackExtrap::AddMCSEffect(trackParamAtCluster,AliMUONConstants::ChamberThicknessInX0(),1.);
       expectedChamber++;
     }
     
     // extrapolation to the z of the current cluster
-    AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ());
+    if (!AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ())) extrapStatus = kFALSE;
     
     // prepare next step
     expectedChamber = currentChamber + 1;
@@ -446,6 +465,10 @@ void AliMUONTrack::UpdateCovTrackParamAtCluster()
     trackParamAtCluster = (AliMUONTrackParam*) (fTrackParamAtCluster->After(trackParamAtCluster));
   }
   
+  // set global chi2 to max value in case of problem during track extrapolation
+  if (!extrapStatus) SetGlobalChi2(2.*MaxChi2());
+  return extrapStatus;
+  
 }
 
   //__________________________________________________________________________
@@ -697,12 +720,12 @@ Double_t AliMUONTrack::ComputeGlobalChi2(Bool_t accountForMCS)
   /// Compute the chi2 of the track accounting for multiple scattering or not according to the flag
   /// - Assume that track parameters at each cluster are corrects
   /// - Assume the cluster weights matrices are corrects
-  /// - Return negative value if chi2 computation failed
+  /// - Return a value of chi2 higher than the maximum allowed if computation failed
   AliDebug(1,"Enter ComputeGlobalChi2");
   
   if (!fTrackParamAtCluster) {
     AliWarning("no cluster attached to this track");
-    return 1.e10;
+    return 2.*MaxChi2();
   }
   
   Double_t chi2 = 0.;
@@ -971,33 +994,56 @@ void AliMUONTrack::ComputeMCSCovariances(TMatrixD& mcsCovariances) const
 }
 
   //__________________________________________________________________________
-Int_t AliMUONTrack::ClustersInCommon(AliMUONTrack* track, Bool_t inSt345) const
+Int_t AliMUONTrack::ClustersInCommon(AliMUONTrack* track) const
 {
   /// Returns the number of clusters in common between the current track ("this")
   /// and the track pointed to by "track".
-  /// If inSt345=kTRUE only stations 3, 4 and 5 are considered.
   if (!fTrackParamAtCluster || !this->fTrackParamAtCluster) return 0;
+  Int_t nCluster1 = this->GetNClusters();
+  Int_t nCluster2 = track->GetNClusters();
   Int_t clustersInCommon = 0;
   AliMUONTrackParam *trackParamAtCluster1, *trackParamAtCluster2;
   // Loop over clusters of first track
-  trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->First();
-  while (trackParamAtCluster1) {
-    if ((!inSt345) || (trackParamAtCluster1->GetClusterPtr()->GetChamberId() > 3)) {
-      // Loop over clusters of second track
-      trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->First();
-      while (trackParamAtCluster2) {
-       if ((!inSt345) || (trackParamAtCluster2->GetClusterPtr()->GetChamberId() > 3)) {
-         // Increment "clustersInCommon" if both trackParamAtCluster1 & 2 point to the same cluster
-         if ((trackParamAtCluster1->GetClusterPtr()) == (trackParamAtCluster2->GetClusterPtr())) {
-           clustersInCommon++;
-           break;
-         }
-       }
-       trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->After(trackParamAtCluster2);
-      } // trackParamAtCluster2
+  for(Int_t iCluster1 = 0; iCluster1 < nCluster1; iCluster1++) {
+    trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->UncheckedAt(iCluster1);
+    // Loop over clusters of second track
+    for(Int_t iCluster2 = 0; iCluster2 < nCluster2; iCluster2++) {
+      trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->UncheckedAt(iCluster2);
+      // Increment "clustersInCommon" if both trackParamAtCluster1 & 2 point to the same cluster
+      if ((trackParamAtCluster1->GetClusterPtr()) == (trackParamAtCluster2->GetClusterPtr())) {
+       clustersInCommon++;
+       break;
+      }
     }
-    trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->After(trackParamAtCluster1);
-  } // trackParamAtCluster1
+  }
+  return clustersInCommon;
+}
+
+  //__________________________________________________________________________
+Int_t AliMUONTrack::ClustersInCommonInSt345(AliMUONTrack* track) const
+{
+  /// Returns the number of clusters in common on stations 3, 4 and 5
+  /// between the current track ("this") and the track pointed to by "track".
+  if (!fTrackParamAtCluster || !this->fTrackParamAtCluster) return 0;
+  Int_t nCluster1 = this->GetNClusters();
+  Int_t nCluster2 = track->GetNClusters();
+  Int_t clustersInCommon = 0;
+  AliMUONTrackParam *trackParamAtCluster1, *trackParamAtCluster2;
+  // Loop over clusters of first track
+  for(Int_t iCluster1 = 0; iCluster1 < nCluster1; iCluster1++) {
+    trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->UncheckedAt(iCluster1);
+    if (trackParamAtCluster1->GetClusterPtr()->GetChamberId() < 4) continue;
+    // Loop over clusters of second track
+    for(Int_t iCluster2 = 0; iCluster2 < nCluster2; iCluster2++) {
+      trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->UncheckedAt(iCluster2);
+      if (trackParamAtCluster2->GetClusterPtr()->GetChamberId() < 4) continue;
+      // Increment "clustersInCommon" if both trackParamAtCluster1 & 2 point to the same cluster
+      if ((trackParamAtCluster1->GetClusterPtr()) == (trackParamAtCluster2->GetClusterPtr())) {
+       clustersInCommon++;
+       break;
+      }
+    }
+  }
   return clustersInCommon;
 }
 
@@ -1016,7 +1062,7 @@ Double_t AliMUONTrack::GetNormalizedChi2() const
   /// return the chi2 value divided by the number of degrees of freedom (or FLT_MAX if ndf <= 0)
   
   Double_t ndf = (Double_t) GetNDF();
-  return (ndf > 0.) ? fGlobalChi2 / ndf : FLT_MAX;
+  return (ndf > 0.) ? fGlobalChi2 / ndf : 2.*MaxChi2();
 }
 
   //__________________________________________________________________________