#include "AliMUONTrackParam.h"
#include "AliMUONTrackExtrap.h"
#include "AliMUONRecoParam.h"
+#include "AliMUONGeometryTransformer.h"
#include "AliMpArea.h"
#include <Riostream.h>
#include <TMath.h>
#include <TMatrixD.h>
+#include <TClonesArray.h>
// Functions to be minimized with Minuit
void TrackChi2(Int_t &nParam, Double_t *gradient, Double_t &chi2, Double_t *param, Int_t flag);
+using std::endl;
+using std::cout;
/// \cond CLASSIMP
ClassImp(AliMUONTrackReconstructor) // Class implementation in ROOT context
/// \endcond
//__________________________________________________________________________
-AliMUONTrackReconstructor::AliMUONTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer)
- : AliMUONVTrackReconstructor(recoParam,clusterServer)
+AliMUONTrackReconstructor::AliMUONTrackReconstructor(const AliMUONRecoParam* recoParam, AliMUONVClusterServer* clusterServer,
+ const AliMUONGeometryTransformer* transformer)
+ : AliMUONVTrackReconstructor(recoParam, clusterServer, transformer)
{
/// Constructor
}
fRecTracksPtr->Remove(track);
fNRecTracks--;
} else if (fNRecTracks > GetRecoParam()->GetMaxTrackCandidates()) {
- AliError(Form("Too many track candidates (%d tracks). Abort tracking.", fNRecTracks));
- delete segments;
+ AliError(Form("Too many track candidates (%d tracks). Stop tracking.", fNRecTracks));
return kFALSE;
}
} else {
if ((fNRecTracks + segments->GetEntriesFast() - iseg - 1) > GetRecoParam()->GetMaxTrackCandidates()) {
- AliError(Form("Too many track candidates (%d tracks). Abort tracking.", fNRecTracks + segments->GetEntriesFast() - iseg - 1));
- delete segments;
+ AliError(Form("Too many track candidates (%d tracks). Stop tracking.", fNRecTracks + segments->GetEntriesFast() - iseg - 1));
return kFALSE;
}
}
}
- // delete the array of segments
- delete segments;
}
// Keep all different tracks or only the best ones as required
// abort tracking if there are too many candidates
if ((fNRecTracks + segments->GetEntriesFast() - iSegment - 1) > GetRecoParam()->GetMaxTrackCandidates()) {
- AliError(Form("Too many track candidates (%d tracks). Abort tracking.", fNRecTracks + segments->GetEntriesFast() - iSegment - 1));
- delete segments;
+ AliError(Form("Too many track candidates (%d tracks). Stop tracking.", fNRecTracks + segments->GetEntriesFast() - iSegment - 1));
return kFALSE;
}
}
- // delete the array of segments
- delete segments;
}
}
if (station==2) { // save track parameters on stations 4 and 5
// extrapolate track parameters and covariances at each cluster
- track->UpdateCovTrackParamAtCluster();
+ // remove the track in case of failure
+ if (!track->UpdateCovTrackParamAtCluster()) {
+ fRecTracksPtr->Remove(track);
+ fNRecTracks--;
+ continue;
+ }
// save them
trackParam = (AliMUONTrackParam*) track->GetTrackParamAtCluster()->First();
nextTrackParam->SetCovariances(trackParam->GetCovariances());
// extrapolate them to the z of the corresponding cluster
- AliMUONTrackExtrap::ExtrapToZCov(nextTrackParam, nextTrackParam->GetClusterPtr()->GetZ());
+ // remove the track in case of failure
+ if (!AliMUONTrackExtrap::ExtrapToZCov(nextTrackParam, nextTrackParam->GetClusterPtr()->GetZ())) {
+ fRecTracksPtr->Remove(track);
+ fNRecTracks--;
+ continue;
+ }
// save them
nextTrackParam->SetSmoothParameters(nextTrackParam->GetParameters());
// abort tracking if there are too many candidates
if (fNRecTracks > GetRecoParam()->GetMaxTrackCandidates()) {
- AliError(Form("Too many track candidates (%d tracks). Abort tracking.", fNRecTracks));
+ AliError(Form("Too many track candidates (%d tracks). Stop tracking.", fNRecTracks));
return kFALSE;
}
nextTrackParam->SetCovariances(trackParam->GetCovariances());
// extrapolate them to the z of the corresponding cluster
- AliMUONTrackExtrap::ExtrapToZCov(nextTrackParam, nextTrackParam->GetClusterPtr()->GetZ());
+ // remove the track in case of failure
+ if (!AliMUONTrackExtrap::ExtrapToZCov(nextTrackParam, nextTrackParam->GetClusterPtr()->GetZ())) {
+ fRecTracksPtr->Remove(track);
+ fNRecTracks--;
+ track = nextTrack;
+ continue;
+ }
// save them
nextTrackParam->SetSmoothParameters(nextTrackParam->GetParameters());
/// kFALSE: add only the best cluster(s) to the "trackCandidate". Try to add a couple of clusters in priority.
AliDebug(1,Form("Enter FollowTrackInChamber(1..) %d", nextChamber+1));
- Double_t chi2WithOneCluster = 1.e10;
+ Double_t chi2WithOneCluster = AliMUONTrack::MaxChi2();
Double_t maxChi2WithOneCluster = 2. * GetRecoParam()->GetSigmaCutForTracking() *
GetRecoParam()->GetSigmaCutForTracking(); // 2 because 2 quantities in chi2
Double_t bestChi2WithOneCluster = maxChi2WithOneCluster;
}
// Add MCS effect
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(),1.);
+ Int_t currentChamber = extrapTrackParamAtCh.GetClusterPtr()->GetChamberId();
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(currentChamber),-1.);
// Add MCS in the missing chamber(s) if any
- Int_t currentChamber = extrapTrackParamAtCh.GetClusterPtr()->GetChamberId();
while (currentChamber > nextChamber + 1) {
// extrapolation to the missing chamber
currentChamber--;
- AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(currentChamber));
+ if (!AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(currentChamber))) return kFALSE;
// add MCS effect
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(),1.);
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(currentChamber),-1.);
}
//Extrapolate trackCandidate to chamber
- AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(nextChamber));
+ if (!AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(nextChamber))) return kFALSE;
// Printout for debuging
if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
ch2 = 2*nextStation+1;
}
- Double_t chi2WithOneCluster = 1.e10;
- Double_t chi2WithTwoClusters = 1.e10;
+ Double_t chi2WithOneCluster = AliMUONTrack::MaxChi2();
+ Double_t chi2WithTwoClusters = AliMUONTrack::MaxChi2();
Double_t maxChi2WithOneCluster = 2. * GetRecoParam()->GetSigmaCutForTracking() *
GetRecoParam()->GetSigmaCutForTracking(); // 2 because 2 quantities in chi2
Double_t maxChi2WithTwoClusters = 4. * GetRecoParam()->GetSigmaCutForTracking() *
AliMUONTrackParam bestTrackParamAtCluster1;
AliMUONTrackParam bestTrackParamAtCluster2;
- Int_t nClusters = clusterStore.GetSize();
- Bool_t *clusterCh1Used = new Bool_t[nClusters];
- for (Int_t i = 0; i < nClusters; i++) clusterCh1Used[i] = kFALSE;
- Int_t iCluster1;
-
// Get track parameters according to the propagation direction
if (nextStation==4) extrapTrackParamAtCh = *(AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->Last();
else extrapTrackParamAtCh = *(AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->First();
}
// Add MCS effect
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(),1.);
+ Int_t currentChamber = extrapTrackParamAtCh.GetClusterPtr()->GetChamberId();
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(currentChamber),-1.);
// Add MCS in the missing chamber(s) if any
- Int_t currentChamber = extrapTrackParamAtCh.GetClusterPtr()->GetChamberId();
while (ch1 < ch2 && currentChamber > ch2 + 1) {
// extrapolation to the missing chamber
currentChamber--;
- AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(currentChamber));
+ if (!AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(currentChamber))) return kFALSE;
// add MCS effect
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(),1.);
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(currentChamber),-1.);
}
//Extrapolate trackCandidate to chamber "ch2"
- AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(ch2));
+ if (!AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(ch2))) return kFALSE;
// Printout for debuging
if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
if (nextStation < 3) AskForNewClustersInStation(extrapTrackParamAtCh, clusterStore, nextStation);
}
+ Int_t nClusters = clusterStore.GetSize();
+ Bool_t *clusterCh1Used = new Bool_t[nClusters];
+ for (Int_t i = 0; i < nClusters; i++) clusterCh1Used[i] = kFALSE;
+ Int_t iCluster1;
+
// Create iterators to loop over clusters in both chambers
TIter nextInCh1(clusterStore.CreateChamberIterator(ch1,ch1));
TIter nextInCh2(clusterStore.CreateChamberIterator(ch2,ch2));
}
// add MCS effect for next step
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCluster2,AliMUONConstants::ChamberThicknessInX0(),1.);
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCluster2,AliMUONConstants::ChamberThicknessInX0(ch2),-1.);
// copy new track parameters for next step
extrapTrackParam = extrapTrackParamAtCluster2;
//Extrapolate track parameters to chamber "ch1"
- AliMUONTrackExtrap::ExtrapToZ(&extrapTrackParam, AliMUONConstants::DefaultChamberZ(ch1));
+ Bool_t normalExtrap = AliMUONTrackExtrap::ExtrapToZ(&extrapTrackParam, AliMUONConstants::DefaultChamberZ(ch1));
// reset cluster iterator of chamber 1
nextInCh1.Reset();
iCluster1 = -1;
// look for second candidates in chamber 1
- while ( ( clusterCh1 = static_cast<AliMUONVCluster*>(nextInCh1()) ) ) {
+
if (normalExtrap) while ( ( clusterCh1 = static_cast<AliMUONVCluster*>(nextInCh1()) ) ) {
iCluster1++;
// try to add the current cluster fast
if (!TryOneClusterFast(extrapTrackParam, clusterCh1)) continue;
- // try to add the current cluster accuratly
+ // try to add the current cluster accurately
chi2WithTwoClusters = TryTwoClusters(extrapTrackParamAtCluster2, clusterCh1, extrapTrackParamAtCluster1);
// if good chi2 then create a new track by adding the 2 clusters to the "trackCandidate"
if (GetRecoParam()->TrackAllTracks() || !foundTwoClusters) {
// add MCS effect for next step
- AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(),1.);
+ AliMUONTrackExtrap::AddMCSEffect(&extrapTrackParamAtCh,AliMUONConstants::ChamberThicknessInX0(ch2),-1.);
//Extrapolate trackCandidate to chamber "ch1"
- AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(ch1));
+ Bool_t normalExtrap = AliMUONTrackExtrap::ExtrapToZCov(&extrapTrackParamAtCh, AliMUONConstants::DefaultChamberZ(ch1));
// Printout for debuging
if ((AliLog::GetDebugLevel("MUON","AliMUONTrackReconstructor") >= 2) || (AliLog::GetGlobalDebugLevel() >= 2)) {
iCluster1 = -1;
// look for second candidates in chamber 1
- while ( ( clusterCh1 = static_cast<AliMUONVCluster*>(nextInCh1()) ) ) {
+
if (normalExtrap) while ( ( clusterCh1 = static_cast<AliMUONVCluster*>(nextInCh1()) ) ) {
iCluster1++;
if (clusterCh1Used[iCluster1]) continue; // Skip cluster already used
/// return trackParamAtCluster1 & 2
// extrapolate track parameters at the z position of the second cluster (no need to extrapolate the covariances)
+ // and set pointer to cluster into trackParamAtCluster
trackParamAtCluster2.SetParameters(trackParamAtCluster1.GetParameters());
trackParamAtCluster2.SetZ(trackParamAtCluster1.GetZ());
- AliMUONTrackExtrap::ExtrapToZ(&trackParamAtCluster2, cluster2->GetZ());
-
- // set pointer to cluster2 into trackParamAtCluster2
trackParamAtCluster2.SetClusterPtr(cluster2);
+ if (!AliMUONTrackExtrap::ExtrapToZ(&trackParamAtCluster2, cluster2->GetZ())) return 2.*AliMUONTrack::MaxChi2();
// Set differences between track and the 2 clusters in the bending and non bending directions
AliMUONVCluster* cluster1 = trackParamAtCluster1.GetClusterPtr();
// copy track parameters at first cluster for jacobian calculation
AliMUONTrackParam trackParam(trackParamAtCluster1);
- // add MCS effect to the covariance matrix at first cluster
- AliMUONTrackExtrap::AddMCSEffect(&trackParam,AliMUONConstants::ChamberThicknessInX0(),1.);
-
// Get the pointer to the parameter covariance matrix at first cluster
const TMatrixD& kParamCov = trackParam.GetCovariances();
trackParam.SetZ(cluster1->GetZ());
// Extrapolate new track parameters to the z position of the second cluster
- AliMUONTrackExtrap::ExtrapToZ(&trackParam,cluster2->GetZ());
+ if (!AliMUONTrackExtrap::ExtrapToZ(&trackParam,cluster2->GetZ())) return 2.*AliMUONTrack::MaxChi2();
// Calculate the jacobian
jacob(2,i) = (trackParam.GetNonBendingCoor() - nonBendingCoor2) / dParam(i,0); // dx2/dParami
error.Invert();
} else {
AliWarning(" Determinant error=0");
- return 1.e10;
+ return 2.*AliMUONTrack::MaxChi2();
}
// Compute the Chi2 value
// Calculate the track parameter covariance matrix
Fit(trackCandidate, kFALSE, kFALSE, kTRUE);
+ // skip track if the normalized chi2 is too high
+ if (trackCandidate.GetNormalizedChi2() > GetRecoParam()->GetSigmaCutForTracking() * GetRecoParam()->GetSigmaCutForTracking()) return kFALSE;
+
// skip track out of limits
if (!IsAcceptable(*((AliMUONTrackParam*)trackCandidate.GetTrackParamAtCluster()->First()))) return kFALSE;
// add multiple scattering effets
AliMUONTrackParam paramAtVertex(*((AliMUONTrackParam*)(trackCandidate.GetTrackParamAtCluster()->First())));
paramAtVertex.DeleteCovariances(); // to be sure to account only for multiple scattering
- AliMUONTrackExtrap::ExtrapToVertexUncorrected(¶mAtVertex,0.);
- const TMatrixD& kParamCov = paramAtVertex.GetCovariances();
- nonBendingReso2 += kParamCov(0,0);
- bendingReso2 += kParamCov(2,2);
+ if (!AliMUONTrackExtrap::ExtrapToZ(¶mAtVertex,AliMUONConstants::AbsZEnd())) {
+ nonBendingReso2 = 0.;
+ bendingReso2 = 0.;
+ } else {
+ AliMUONTrackExtrap::ExtrapToVertexUncorrected(¶mAtVertex,0.);
+ const TMatrixD& kParamCov = paramAtVertex.GetCovariances();
+ nonBendingReso2 += kParamCov(0,0);
+ bendingReso2 += kParamCov(2,2);
+ }
// Set the vertex resolution square
trackCandidate.SetVertexErrXY2(nonBendingReso2,bendingReso2);
track.FitWithMCS(includeMCS);
if (includeMCS) {
// compute cluster weights only once
- if (!track.ComputeClusterWeights()) {
+ if (!track.UpdateTrackParamAtCluster() || !track.ComputeClusterWeights()) {
AliWarning("cannot take into account the multiple scattering effects");
track.FitWithMCS(kFALSE);
}
// Covariance matrix according to HESSE status
// If problem then keep only the diagonal terms (variances)
Double_t matrix[5][5];
+ for (Int_t i=0; i<5; i++) for (Int_t j=0; j<5; j++) matrix[i][j] = 0.;
gMinuit->mnemat(&matrix[0][0],5);
if (covStatus == 3) trackParam->SetCovariances(matrix);
else trackParam->SetVariances(matrix);
trackParamAtCluster->SetBendingCoor(param[2]);
trackParamAtCluster->SetBendingSlope(param[3]);
trackParamAtCluster->SetInverseBendingMomentum(param[4]);
- trackBeingFitted->UpdateTrackParamAtCluster();
+ if (!trackBeingFitted->UpdateTrackParamAtCluster()) {
+ chi2 = 2.*AliMUONTrack::MaxChi2();
+ return;
+ }
// Take the vertex into account in the fit if required
if (trackBeingFitted->FitWithVertex()) {
Double_t nonBendingReso2,bendingReso2;
trackBeingFitted->GetVertexErrXY2(nonBendingReso2,bendingReso2);
- if (nonBendingReso2 == 0. || bendingReso2 == 0.) chi2 += 1.e10;
- else {
- AliMUONTrackParam paramAtVertex(*trackParamAtCluster);
- AliMUONTrackExtrap::ExtrapToZ(¶mAtVertex, 0.); // vextex position = (0,0,0)
+ AliMUONTrackParam paramAtVertex(*trackParamAtCluster);
+ if (nonBendingReso2 != 0. && bendingReso2 != 0. && AliMUONTrackExtrap::ExtrapToZ(¶mAtVertex, 0.)) { // vextex position = (0,0,0)
dX = paramAtVertex.GetNonBendingCoor();
dY = paramAtVertex.GetBendingCoor();
chi2 += dX * dX / nonBendingReso2 + dY * dY / bendingReso2;
+ } else {
+ chi2 = 2.*AliMUONTrack::MaxChi2();
+ return;
}
}
track.TagRemovableClusters(GetRecoParam()->RequestedStationMask());
// Update track parameters and covariances
- track.UpdateCovTrackParamAtCluster();
+ if (!track.UpdateCovTrackParamAtCluster()) {
+ AliWarning("unable to update track parameters and covariances --> stop improvement");
+ break;
+ }
// Compute local chi2 of each clusters
track.ComputeLocalChi2(kTRUE);
}
//__________________________________________________________________________
-void AliMUONTrackReconstructor::FinalizeTrack(AliMUONTrack &track)
+Bool_t AliMUONTrackReconstructor::FinalizeTrack(AliMUONTrack &track)
{
/// Recompute track parameters and covariances at each attached cluster
/// from those at the first one, if not already done
AliDebug(1,"Enter FinalizeTrack");
- if (!track.IsImproved()) track.UpdateCovTrackParamAtCluster();
+ if (!track.IsImproved() && !track.UpdateCovTrackParamAtCluster()) {
+ AliWarning("finalization failed due to extrapolation problem");
+ return kFALSE;
+ }
+ return kTRUE;
}
//__________________________________________________________________________
// compute track parameters at each cluster from parameters at the first one
// necessary to compute multiple scattering effect during refitting
- track.UpdateTrackParamAtCluster();
+ if (!track.UpdateTrackParamAtCluster()) {
+ AliWarning("bad track refitting due to extrapolation failure");
+ return kFALSE;
+ }
// Re-fit the track:
// Take into account the multiple scattering
if (enableImprovement && GetRecoParam()->ImproveTracks()) ImproveTrack(track);
// Fill AliMUONTrack data members
- FinalizeTrack(track);
-
- return kTRUE;
+ if (track.GetGlobalChi2() < AliMUONTrack::MaxChi2()) return FinalizeTrack(track);
+ else {
+ AliWarning("track not finalized due to extrapolation failure");
+ return kFALSE;
+ }
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff