/* $Id$ */
-///////////////////////////////////////////////////
-//
-// Reconstructed track
-// in
-// ALICE
-// dimuon
-// spectrometer
-//
-///////////////////////////////////////////////////
-
-#include <stdlib.h> // for exit()
-#include <Riostream.h>
+//-----------------------------------------------------------------------------
+// Class AliMUONTrack
+//-------------------
+// Reconstructed track in ALICE dimuon spectrometer
+//-----------------------------------------------------------------------------
#include "AliMUONTrack.h"
-#include "AliMUONTrackParam.h"
-#include "AliMUONHitForRec.h"
-#include "AliMUONSegment.h"
+#include "AliMUONReconstructor.h"
+#include "AliMUONVCluster.h"
+#include "AliMUONVClusterStore.h"
+#include "AliMUONObjectPair.h"
+#include "AliMUONTrackExtrap.h"
#include "AliMUONConstants.h"
+#include "AliMUONTrackParam.h"
#include "AliLog.h"
#include <TMath.h>
+#include <Riostream.h>
+
+/// \cond CLASSIMP
ClassImp(AliMUONTrack) // Class implementation in ROOT context
+/// \endcond
//__________________________________________________________________________
AliMUONTrack::AliMUONTrack()
: TObject(),
- fTrackParamAtVertex(),
- fTrackParamAtHit(0x0),
- fHitForRecAtHit(0x0),
- fNTrackHits(0),
- fFitFMin(-1.),
- fMatchTrigger(kFALSE),
+ fTrackParamAtCluster(0x0),
+ fFitWithVertex(kFALSE),
+ fVertexErrXY2(),
+ fFitWithMCS(kFALSE),
+ fClusterWeightsNonBending(0x0),
+ fClusterWeightsBending(0x0),
+ fGlobalChi2(-1.),
+ fImproved(kFALSE),
+ fMatchTrigger(-1),
+ floTrgNum(-1),
fChi2MatchTrigger(0.),
- fTrackID(0)
+ fTrackID(-1),
+ fTrackParamAtVertex(0x0),
+ fHitsPatternInTrigCh(0),
+ fLocalTrigger(0)
{
/// Default constructor
+ fVertexErrXY2[0] = 0.;
+ fVertexErrXY2[1] = 0.;
}
//__________________________________________________________________________
-AliMUONTrack::AliMUONTrack(AliMUONSegment* BegSegment, AliMUONSegment* EndSegment)
+AliMUONTrack::AliMUONTrack(AliMUONObjectPair *segment, Double_t bendingVertexDispersion)
: TObject(),
- fTrackParamAtVertex(),
- fTrackParamAtHit(0x0),
- fHitForRecAtHit(0x0),
- fNTrackHits(0),
- fFitFMin(-1.),
- fMatchTrigger(kFALSE),
+ fTrackParamAtCluster(new TClonesArray("AliMUONTrackParam",10)),
+ fFitWithVertex(kFALSE),
+ fVertexErrXY2(),
+ fFitWithMCS(kFALSE),
+ fClusterWeightsNonBending(0x0),
+ fClusterWeightsBending(0x0),
+ fGlobalChi2(0.),
+ fImproved(kFALSE),
+ fMatchTrigger(-1),
+ floTrgNum(-1),
fChi2MatchTrigger(0.),
- fTrackID(0)
+ fTrackID(-1),
+ fTrackParamAtVertex(0x0),
+ fHitsPatternInTrigCh(0),
+ fLocalTrigger(0)
{
- /// Constructor from two Segment's
+ /// Constructor from two clusters
+
+ fVertexErrXY2[0] = 0.;
+ fVertexErrXY2[1] = 0.;
+
+ // Pointers to clusters from the segment
+ AliMUONVCluster* firstCluster = (AliMUONVCluster*) segment->First();
+ AliMUONVCluster* lastCluster = (AliMUONVCluster*) segment->Second();
+
+ // Compute track parameters
+ Double_t z1 = firstCluster->GetZ();
+ Double_t z2 = lastCluster->GetZ();
+ Double_t dZ = z1 - z2;
+ // Non bending plane
+ Double_t nonBendingCoor1 = firstCluster->GetX();
+ Double_t nonBendingCoor2 = lastCluster->GetX();
+ Double_t nonBendingSlope = (nonBendingCoor1 - nonBendingCoor2) / dZ;
+ // Bending plane
+ Double_t bendingCoor1 = firstCluster->GetY();
+ Double_t bendingCoor2 = lastCluster->GetY();
+ Double_t bendingSlope = (bendingCoor1 - bendingCoor2) / dZ;
+ // Inverse bending momentum
+ Double_t bendingImpact = bendingCoor1 - z1 * bendingSlope;
+ Double_t inverseBendingMomentum = 1. / AliMUONTrackExtrap::GetBendingMomentumFromImpactParam(bendingImpact);
+
+ // Set track parameters at first cluster
+ AliMUONTrackParam trackParamAtFirstCluster;
+ trackParamAtFirstCluster.SetZ(z1);
+ trackParamAtFirstCluster.SetNonBendingCoor(nonBendingCoor1);
+ trackParamAtFirstCluster.SetNonBendingSlope(nonBendingSlope);
+ trackParamAtFirstCluster.SetBendingCoor(bendingCoor1);
+ trackParamAtFirstCluster.SetBendingSlope(bendingSlope);
+ trackParamAtFirstCluster.SetInverseBendingMomentum(inverseBendingMomentum);
+
+ // Set track parameters at last cluster
+ AliMUONTrackParam trackParamAtLastCluster;
+ trackParamAtLastCluster.SetZ(z2);
+ trackParamAtLastCluster.SetNonBendingCoor(nonBendingCoor2);
+ trackParamAtLastCluster.SetNonBendingSlope(nonBendingSlope);
+ trackParamAtLastCluster.SetBendingCoor(bendingCoor2);
+ trackParamAtLastCluster.SetBendingSlope(bendingSlope);
+ trackParamAtLastCluster.SetInverseBendingMomentum(inverseBendingMomentum);
+
+ // Compute and set track parameters covariances at first cluster
+ TMatrixD paramCov(5,5);
+ paramCov.Zero();
+ // Non bending plane
+ paramCov(0,0) = firstCluster->GetErrX2();
+ paramCov(0,1) = firstCluster->GetErrX2() / dZ;
+ paramCov(1,0) = paramCov(0,1);
+ paramCov(1,1) = ( firstCluster->GetErrX2() + lastCluster->GetErrX2() ) / dZ / dZ;
+ // Bending plane
+ paramCov(2,2) = firstCluster->GetErrY2();
+ paramCov(2,3) = firstCluster->GetErrY2() / dZ;
+ 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)
+ 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
+ // 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
+ AddTrackParamAtCluster(trackParamAtFirstCluster,*firstCluster);
+ AddTrackParamAtCluster(trackParamAtLastCluster,*lastCluster);
+
+}
- fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
- fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
+//__________________________________________________________________________
+AliMUONTrack::AliMUONTrack(const AliMUONTrack& track)
+ : TObject(track),
+ fTrackParamAtCluster(0x0),
+ fFitWithVertex(track.fFitWithVertex),
+ fVertexErrXY2(),
+ fFitWithMCS(track.fFitWithMCS),
+ fClusterWeightsNonBending(0x0),
+ fClusterWeightsBending(0x0),
+ fGlobalChi2(track.fGlobalChi2),
+ fImproved(track.fImproved),
+ fMatchTrigger(track.fMatchTrigger),
+ floTrgNum(track.floTrgNum),
+ fChi2MatchTrigger(track.fChi2MatchTrigger),
+ fTrackID(track.fTrackID),
+ fTrackParamAtVertex(0x0),
+ fHitsPatternInTrigCh(track.fHitsPatternInTrigCh),
+ fLocalTrigger(track.fLocalTrigger)
+{
+ ///copy constructor
- if (BegSegment) { //AZ
- AddTrackParamAtHit(0,BegSegment->GetHitForRec1());
- AddTrackParamAtHit(0,BegSegment->GetHitForRec2());
- AddTrackParamAtHit(0,EndSegment->GetHitForRec1());
- AddTrackParamAtHit(0,EndSegment->GetHitForRec2());
- fTrackParamAtHit->Sort(); // sort TrackParamAtHit according to increasing Z
+ // necessary to make a copy of the objects and not only the pointers in TClonesArray.
+ if (track.fTrackParamAtCluster) {
+ fTrackParamAtCluster = new TClonesArray("AliMUONTrackParam",10);
+ AliMUONTrackParam *trackParamAtCluster = (AliMUONTrackParam*) track.fTrackParamAtCluster->First();
+ while (trackParamAtCluster) {
+ new ((*fTrackParamAtCluster)[GetNClusters()]) AliMUONTrackParam(*trackParamAtCluster);
+ trackParamAtCluster = (AliMUONTrackParam*) track.fTrackParamAtCluster->After(trackParamAtCluster);
+ }
}
+
+ // copy vertex resolution square used during the tracking procedure
+ fVertexErrXY2[0] = track.fVertexErrXY2[0];
+ fVertexErrXY2[1] = track.fVertexErrXY2[1];
+
+ // copy cluster weights matrices if any
+ if (track.fClusterWeightsNonBending) fClusterWeightsNonBending = new TMatrixD(*(track.fClusterWeightsNonBending));
+ if (track.fClusterWeightsBending) fClusterWeightsBending = new TMatrixD(*(track.fClusterWeightsBending));
+
+ // copy track parameters at vertex if any
+ if (track.fTrackParamAtVertex) fTrackParamAtVertex = new AliMUONTrackParam(*(track.fTrackParamAtVertex));
+
}
//__________________________________________________________________________
-AliMUONTrack::AliMUONTrack(AliMUONSegment* Segment, AliMUONHitForRec* HitForRec)
- : TObject(),
- fTrackParamAtVertex(),
- fTrackParamAtHit(0x0),
- fHitForRecAtHit(0x0),
- fNTrackHits(0),
- fFitFMin(-1.),
- fMatchTrigger(kFALSE),
- fChi2MatchTrigger(0.),
- fTrackID(0)
+AliMUONTrack & AliMUONTrack::operator=(const AliMUONTrack& track)
{
- /// Constructor from one Segment and one HitForRec
+ /// Asignment operator
+ // check assignement to self
+ if (this == &track)
+ return *this;
- fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
- fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
+ // base class assignement
+ TObject::operator=(track);
+
+ // clear memory
+ Clear();
- AddTrackParamAtHit(0,Segment->GetHitForRec1());
- AddTrackParamAtHit(0,Segment->GetHitForRec2());
- AddTrackParamAtHit(0,HitForRec);
- fTrackParamAtHit->Sort(); // sort TrackParamAtHit according to increasing Z
+ // necessary to make a copy of the objects and not only the pointers in TClonesArray
+ if (track.fTrackParamAtCluster) {
+ fTrackParamAtCluster = new TClonesArray("AliMUONTrackParam",10);
+ AliMUONTrackParam *trackParamAtCluster = (AliMUONTrackParam*) track.fTrackParamAtCluster->First();
+ while (trackParamAtCluster) {
+ new ((*fTrackParamAtCluster)[GetNClusters()]) AliMUONTrackParam(*trackParamAtCluster);
+ trackParamAtCluster = (AliMUONTrackParam*) track.fTrackParamAtCluster->After(trackParamAtCluster);
+ }
+ }
+
+ // copy cluster weights matrix if any
+ if (track.fClusterWeightsNonBending) {
+ if (fClusterWeightsNonBending) {
+ fClusterWeightsNonBending->ResizeTo(*(track.fClusterWeightsNonBending));
+ *fClusterWeightsNonBending = *(track.fClusterWeightsNonBending);
+ } else fClusterWeightsNonBending = new TMatrixD(*(track.fClusterWeightsNonBending));
+ }
+
+ // copy cluster weights matrix if any
+ if (track.fClusterWeightsBending) {
+ if (fClusterWeightsBending) {
+ fClusterWeightsBending->ResizeTo(*(track.fClusterWeightsBending));
+ *fClusterWeightsBending = *(track.fClusterWeightsBending);
+ } else fClusterWeightsBending = new TMatrixD(*(track.fClusterWeightsBending));
+ }
+
+ // copy track parameters at vertex if any
+ if (track.fTrackParamAtVertex) {
+ if (fTrackParamAtVertex) *fTrackParamAtVertex = *(track.fTrackParamAtVertex);
+ else fTrackParamAtVertex = new AliMUONTrackParam(*(track.fTrackParamAtVertex));
+ }
+
+ fFitWithVertex = track.fFitWithVertex;
+ fVertexErrXY2[0] = track.fVertexErrXY2[0];
+ fVertexErrXY2[1] = track.fVertexErrXY2[1];
+ fFitWithMCS = track.fFitWithMCS;
+ fGlobalChi2 = track.fGlobalChi2;
+ fImproved = track.fImproved;
+ fMatchTrigger = track.fMatchTrigger;
+ floTrgNum = track.floTrgNum;
+ fChi2MatchTrigger = track.fChi2MatchTrigger;
+ fTrackID = track.fTrackID;
+ fHitsPatternInTrigCh = track.fHitsPatternInTrigCh;
+ fLocalTrigger = track.fLocalTrigger;
+
+ return *this;
}
//__________________________________________________________________________
AliMUONTrack::~AliMUONTrack()
{
/// Destructor
- if (fTrackParamAtHit) {
- // delete the TClonesArray of pointers to TrackParam
- delete fTrackParamAtHit;
- fTrackParamAtHit = NULL;
+ delete fTrackParamAtCluster;
+ delete fClusterWeightsNonBending;
+ delete fClusterWeightsBending;
+ delete fTrackParamAtVertex;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrack::Clear(Option_t* opt)
+{
+ /// Clear arrays
+ if (opt && opt[0] == 'C' && fTrackParamAtCluster) fTrackParamAtCluster->Clear("C");
+ else {
+ delete fTrackParamAtCluster;
+ fTrackParamAtCluster = 0x0;
}
+ delete fClusterWeightsNonBending; fClusterWeightsNonBending = 0x0;
+ delete fClusterWeightsBending; fClusterWeightsBending = 0x0;
+ delete fTrackParamAtVertex; fTrackParamAtVertex = 0x0;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrack::Reset()
+{
+ /// Reset to default values
+ SetUniqueID(0);
+ fFitWithVertex = kFALSE;
+ fVertexErrXY2[0] = 0.;
+ fVertexErrXY2[1] = 0.;
+ fFitWithMCS = kFALSE;
+ fGlobalChi2 = -1.;
+ fImproved = kFALSE;
+ fMatchTrigger = -1;
+ floTrgNum = -1;
+ fChi2MatchTrigger = 0.;
+ fTrackID = -1;
+ fHitsPatternInTrigCh = 0;
+ fLocalTrigger = 0;
+ delete fTrackParamAtCluster; fTrackParamAtCluster = 0x0;
+ delete fClusterWeightsNonBending; fClusterWeightsNonBending = 0x0;
+ delete fClusterWeightsBending; fClusterWeightsBending = 0x0;
+ delete fTrackParamAtVertex; fTrackParamAtVertex = 0x0;
+}
- if (fHitForRecAtHit) {
- // delete the TClonesArray of pointers to HitForRec
- delete fHitForRecAtHit;
- fHitForRecAtHit = NULL;
+ //__________________________________________________________________________
+TClonesArray* AliMUONTrack::GetTrackParamAtCluster() const
+{
+ /// return array of track parameters at cluster (create it if needed)
+ if (!fTrackParamAtCluster) fTrackParamAtCluster = new TClonesArray("AliMUONTrackParam",10);
+ return fTrackParamAtCluster;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrack::AddTrackParamAtCluster(const AliMUONTrackParam &trackParam, AliMUONVCluster &cluster, Bool_t copy)
+{
+ /// Copy given track parameters into a new TrackParamAtCluster
+ /// Link parameters with the associated cluster
+ /// If copy=kTRUE: the cluster is copied then passed the trackParam which become its owner
+ /// otherwise: make sure to do not delete the cluster until it is used by the track
+
+ // check chamber ID of the associated cluster
+ if (cluster.GetChamberId() < 0 || cluster.GetChamberId() > AliMUONConstants::NTrackingCh()) {
+ AliError(Form("Chamber ID of the associated cluster is not valid (ChamberId=%d)",cluster.GetChamberId()));
+ return;
+ }
+
+ // check whether track parameters are given at the correct cluster z position
+ if (cluster.GetZ() != trackParam.GetZ()) {
+ AliError("track parameters are given at a different z position than the one of the associated cluster");
+ return;
}
+
+ // add parameters to the array of track parameters
+ if (!fTrackParamAtCluster) fTrackParamAtCluster = new TClonesArray("AliMUONTrackParam",10);
+ AliMUONTrackParam* trackParamAtCluster = new ((*fTrackParamAtCluster)[GetNClusters()]) AliMUONTrackParam(trackParam);
+
+ // link parameters with the associated cluster or its copy
+ if (copy) {
+ AliMUONVCluster *clusterCopy = static_cast<AliMUONVCluster*>(cluster.Clone());
+ trackParamAtCluster->SetClusterPtr(clusterCopy, kTRUE);
+ } else trackParamAtCluster->SetClusterPtr(&cluster);
+
+ // sort the array of track parameters
+ fTrackParamAtCluster->Sort();
}
//__________________________________________________________________________
-AliMUONTrack::AliMUONTrack (const AliMUONTrack& theMUONTrack)
- : TObject(theMUONTrack),
- fTrackParamAtVertex(theMUONTrack.fTrackParamAtVertex),
- fTrackParamAtHit(0x0),
- fHitForRecAtHit(0x0),
- fNTrackHits(theMUONTrack.fNTrackHits),
- fFitFMin(theMUONTrack.fFitFMin),
- fMatchTrigger(theMUONTrack.fMatchTrigger),
- fChi2MatchTrigger(theMUONTrack.fChi2MatchTrigger),
- fTrackID(theMUONTrack.fTrackID)
+void AliMUONTrack::RemoveTrackParamAtCluster(AliMUONTrackParam *trackParam)
{
- ///copy constructor
- Int_t maxIndex = 0;
+ /// Remove trackParam from the array of TrackParamAtCluster
+ if (!fTrackParamAtCluster || !fTrackParamAtCluster->Remove(trackParam)) {
+ AliWarning("object to remove does not exist in array fTrackParamAtCluster");
+ return;
+ }
- // necessary to make a copy of the objects and not only the pointers in TClonesArray.
- if (theMUONTrack.fTrackParamAtHit) {
- maxIndex = (theMUONTrack.fTrackParamAtHit)->GetEntriesFast();
- fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",maxIndex);
- for (Int_t index = 0; index < maxIndex; index++) {
- new ((*fTrackParamAtHit)[index]) AliMUONTrackParam(*(AliMUONTrackParam*)theMUONTrack.fTrackParamAtHit->At(index));
- }
- }
+ fTrackParamAtCluster->Compress();
+}
- // necessary to make a copy of the objects and not only the pointers in TClonesArray.
- if (theMUONTrack.fHitForRecAtHit) {
- maxIndex = (theMUONTrack.fHitForRecAtHit)->GetEntriesFast();
- fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",maxIndex);
- for (Int_t index = 0; index < maxIndex; index++) {
- new ((*fHitForRecAtHit)[index]) AliMUONHitForRec(*(AliMUONHitForRec*)theMUONTrack.fHitForRecAtHit->At(index));
- }
- }
+ //__________________________________________________________________________
+void AliMUONTrack::UpdateTrackParamAtCluster()
+{
+ /// Update track parameters at each attached cluster
+
+ if (GetNClusters() == 0) {
+ AliWarning("no cluster attached to the track");
+ return;
+ }
+
+ AliMUONTrackParam* startingTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->First();
+ AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(startingTrackParam);
+ while (trackParamAtCluster) {
+
+ // reset track parameters and their covariances
+ trackParamAtCluster->SetParameters(startingTrackParam->GetParameters());
+ trackParamAtCluster->SetZ(startingTrackParam->GetZ());
+
+ // extrapolation to the given z
+ AliMUONTrackExtrap::ExtrapToZ(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ());
+
+ // prepare next step
+ startingTrackParam = trackParamAtCluster;
+ trackParamAtCluster = (AliMUONTrackParam*) (fTrackParamAtCluster->After(trackParamAtCluster));
+ }
}
//__________________________________________________________________________
-AliMUONTrack & AliMUONTrack::operator=(const AliMUONTrack& theMUONTrack)
+void AliMUONTrack::UpdateCovTrackParamAtCluster()
{
- /// Asignment operator
- // check assignement to self
- if (this == &theMUONTrack)
- return *this;
+ /// Update track parameters and their covariances at each attached cluster
+ /// Include effects of multiple scattering in chambers
+
+ if (GetNClusters() == 0) {
+ AliWarning("no cluster attached to the track");
+ return;
+ }
+
+ AliMUONTrackParam* startingTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->First();
+ AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(startingTrackParam);
+ Int_t expectedChamber = startingTrackParam->GetClusterPtr()->GetChamberId() + 1;
+ Int_t currentChamber;
+ while (trackParamAtCluster) {
+
+ // reset track parameters and their covariances
+ trackParamAtCluster->SetParameters(startingTrackParam->GetParameters());
+ trackParamAtCluster->SetZ(startingTrackParam->GetZ());
+ trackParamAtCluster->SetCovariances(startingTrackParam->GetCovariances());
+
+ // add MCS effect
+ AliMUONTrackExtrap::AddMCSEffect(trackParamAtCluster,AliMUONConstants::ChamberThicknessInX0(),1.);
+
+ // add MCS in missing chambers if any
+ currentChamber = trackParamAtCluster->GetClusterPtr()->GetChamberId();
+ while (currentChamber > expectedChamber) {
+ // extrapolation to the missing chamber
+ AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, AliMUONConstants::DefaultChamberZ(expectedChamber));
+ // add MCS effect
+ AliMUONTrackExtrap::AddMCSEffect(trackParamAtCluster,AliMUONConstants::ChamberThicknessInX0(),1.);
+ expectedChamber++;
+ }
+
+ // extrapolation to the z of the current cluster
+ AliMUONTrackExtrap::ExtrapToZCov(trackParamAtCluster, trackParamAtCluster->GetClusterPtr()->GetZ());
+
+ // prepare next step
+ expectedChamber = currentChamber + 1;
+ startingTrackParam = trackParamAtCluster;
+ trackParamAtCluster = (AliMUONTrackParam*) (fTrackParamAtCluster->After(trackParamAtCluster));
+ }
+
+}
- // base class assignement
- TObject::operator=(theMUONTrack);
+ //__________________________________________________________________________
+Bool_t AliMUONTrack::IsValid(UInt_t requestedStationMask)
+{
+ /// check the validity of the current track:
+ /// at least one cluster per requested station
+ /// and at least 2 chambers in stations 4 & 5 that contain cluster(s)
+
+ Int_t nClusters = GetNClusters();
+ AliMUONTrackParam *trackParam;
+ Int_t currentCh, currentSt, previousCh = -1, nChHitInSt45 = 0;
+ UInt_t presentStationMask(0);
+
+ // first loop over clusters
+ for (Int_t i = 0; i < nClusters; i++) {
+ trackParam = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(i);
+
+ currentCh = trackParam->GetClusterPtr()->GetChamberId();
+ currentSt = currentCh/2;
+
+ // build present station mask
+ presentStationMask |= ( 1 << currentSt );
+
+ // count the number of chambers in station 4 & 5 that contain cluster(s)
+ if (currentCh > 5 && currentCh != previousCh) {
+ nChHitInSt45++;
+ previousCh = currentCh;
+ }
+
+ }
+
+ return (((requestedStationMask & presentStationMask) == requestedStationMask) && (nChHitInSt45 >= 2));
+}
- fTrackParamAtVertex = theMUONTrack.fTrackParamAtVertex;
+ //__________________________________________________________________________
+void AliMUONTrack::TagRemovableClusters(UInt_t requestedStationMask) {
+ /// Identify clusters that can be removed from the track,
+ /// with the only requirements to have at least 1 cluster per requested station
+ /// and at least 2 chambers over 4 in stations 4 & 5 that contain cluster(s)
+
+ Int_t nClusters = GetNClusters();
+ AliMUONTrackParam *trackParam, *nextTrackParam;
+ Int_t currentCh, nextCh, currentSt, nextSt, previousCh = -1, nChHitInSt45 = 0;
+
+ // first loop over clusters
+ for (Int_t i = 0; i < nClusters; i++) {
+ trackParam = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(i);
+
+ currentCh = trackParam->GetClusterPtr()->GetChamberId();
+ currentSt = currentCh/2;
+
+ // reset flags to kFALSE for all clusters in required station
+ if ((1 << currentSt) & requestedStationMask) trackParam->SetRemovable(kFALSE);
+ else trackParam->SetRemovable(kTRUE);
+
+ // count the number of chambers in station 4 & 5 that contain cluster(s)
+ if (currentCh > 5 && currentCh != previousCh) {
+ nChHitInSt45++;
+ previousCh = currentCh;
+ }
+
+ }
+
+ // second loop over clusters
+ for (Int_t i = 0; i < nClusters; i++) {
+ trackParam = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(i);
+
+ currentCh = trackParam->GetClusterPtr()->GetChamberId();
+ currentSt = currentCh/2;
+
+ // make sure they are more than 2 clusters in 2 different chambers of stations 4 & 5
+ // but 2 clusters in he same chamber will still be flagged as removable
+ if (nChHitInSt45 < 3 && currentSt > 2) {
+
+ if (i == nClusters-1) {
+
+ trackParam->SetRemovable(kFALSE);
+
+ } else {
+
+ nextTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(i+1);
+ nextCh = nextTrackParam->GetClusterPtr()->GetChamberId();
+
+ // set clusters in the same chamber as being removable
+ if (nextCh == currentCh) {
+ trackParam->SetRemovable(kTRUE);
+ nextTrackParam->SetRemovable(kTRUE);
+ i++; // skip cluster already checked
+ } else {
+ trackParam->SetRemovable(kFALSE);
+ }
+
+ }
+
+ } else {
+
+ // skip clusters already flag as removable
+ if (trackParam->IsRemovable()) continue;
+
+ // loop over next track parameters
+ for (Int_t j = i+1; j < nClusters; j++) {
+ nextTrackParam = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(j);
+
+ nextCh = nextTrackParam->GetClusterPtr()->GetChamberId();
+ nextSt = nextCh/2;
+
+ // set clusters in the same station as being removable
+ if (nextSt == currentSt) {
+ trackParam->SetRemovable(kTRUE);
+ nextTrackParam->SetRemovable(kTRUE);
+ i++; // skip cluster already checked
+ }
+
+ }
+
+ }
+
+ }
+
+}
- Int_t maxIndex = 0;
+ //__________________________________________________________________________
+Bool_t AliMUONTrack::ComputeLocalChi2(Bool_t accountForMCS)
+{
+ /// Compute each cluster contribution to the chi2 of the track
+ /// accounting for multiple scattering or not according to the flag
+ /// - Also recompute the weight matrices of the attached clusters if accountForMCS=kTRUE
+ /// - Assume that track parameters at each cluster are corrects
+ /// - Return kFALSE if computation failed
+ AliDebug(1,"Enter ComputeLocalChi2");
- // necessary to make a copy of the objects and not only the pointers in TClonesArray.
- fTrackParamAtHit = 0;
- if (theMUONTrack.fTrackParamAtHit) {
- fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
- maxIndex = (theMUONTrack.fTrackParamAtHit)->GetEntriesFast();
- for (Int_t index = 0; index < maxIndex; index++) {
- new ((*fTrackParamAtHit)[fTrackParamAtHit->GetEntriesFast()])
- AliMUONTrackParam(*(AliMUONTrackParam*)(theMUONTrack.fTrackParamAtHit)->At(index));
+ if (!fTrackParamAtCluster) {
+ AliWarning("no cluster attached to this track");
+ return kFALSE;
+ }
+
+ if (accountForMCS) { // Compute local chi2 taking into account multiple scattering effects
+
+ // Compute MCS covariance matrix only once
+ Int_t nClusters = GetNClusters();
+ TMatrixD mcsCovariances(nClusters,nClusters);
+ ComputeMCSCovariances(mcsCovariances);
+
+ // Make sure cluster weights are consistent with following calculations
+ if (!ComputeClusterWeights(&mcsCovariances)) {
+ AliWarning("cannot take into account the multiple scattering effects");
+ return ComputeLocalChi2(kFALSE);
+ }
+
+ // Compute chi2 of the track
+ Double_t globalChi2 = ComputeGlobalChi2(kTRUE);
+ if (globalChi2 < 0.) return kFALSE;
+
+ // Loop over removable clusters and compute their local chi2
+ AliMUONTrackParam* trackParamAtCluster1;
+ AliMUONVCluster *cluster, *discardedCluster;
+ Int_t iCluster1, iCluster2, iCurrentCluster1, iCurrentCluster2;
+ TMatrixD clusterWeightsNB(nClusters-1,nClusters-1);
+ TMatrixD clusterWeightsB(nClusters-1,nClusters-1);
+ Double_t *dX = new Double_t[nClusters-1];
+ Double_t *dY = new Double_t[nClusters-1];
+ Double_t globalChi2b;
+ AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->First();
+ while (trackParamAtCluster) {
+
+ discardedCluster = trackParamAtCluster->GetClusterPtr();
+
+ // Recompute cluster weights without the current cluster
+ if (!ComputeClusterWeights(clusterWeightsNB, clusterWeightsB, &mcsCovariances, discardedCluster)) {
+ AliWarning("cannot take into account the multiple scattering effects");
+ delete [] dX;
+ delete [] dY;
+ return ComputeLocalChi2(kFALSE);
+ }
+
+ // Compute track chi2 without the current cluster
+ globalChi2b = 0.;
+ iCurrentCluster1 = 0;
+ for (iCluster1 = 0; iCluster1 < nClusters ; iCluster1++) {
+ trackParamAtCluster1 = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster1);
+ cluster = trackParamAtCluster1->GetClusterPtr();
+
+ if (cluster == discardedCluster) continue;
+
+ // Compute and save residuals
+ dX[iCurrentCluster1] = cluster->GetX() - trackParamAtCluster1->GetNonBendingCoor();
+ dY[iCurrentCluster1] = cluster->GetY() - trackParamAtCluster1->GetBendingCoor();
+
+ iCurrentCluster2 = 0;
+ for (iCluster2 = 0; iCluster2 < iCluster1; iCluster2++) {
+ cluster = ((AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster2))->GetClusterPtr();
+
+ if (cluster == discardedCluster) continue;
+
+ // Add contribution from covariances
+ globalChi2b += (clusterWeightsNB(iCurrentCluster1, iCurrentCluster2) +
+ clusterWeightsNB(iCurrentCluster2, iCurrentCluster1)) * dX[iCurrentCluster1] * dX[iCurrentCluster2] +
+ (clusterWeightsB(iCurrentCluster1, iCurrentCluster2) +
+ clusterWeightsB(iCurrentCluster2, iCurrentCluster1)) * dY[iCurrentCluster1] * dY[iCurrentCluster2];
+
+ iCurrentCluster2++;
+ }
+
+ // Add contribution from variances
+ globalChi2b += clusterWeightsNB(iCurrentCluster1, iCurrentCluster1) * dX[iCurrentCluster1] * dX[iCurrentCluster1] +
+ clusterWeightsB(iCurrentCluster1, iCurrentCluster1) * dY[iCurrentCluster1] * dY[iCurrentCluster1];
+
+ iCurrentCluster1++;
+ }
+
+ // Set local chi2
+ trackParamAtCluster->SetLocalChi2(globalChi2 - globalChi2b);
+
+ trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(trackParamAtCluster);
+ }
+
+ delete [] dX;
+ delete [] dY;
+
+ } else { // without multiple scattering effects
+
+ AliMUONVCluster *discardedCluster;
+ Double_t dX, dY;
+ AliMUONTrackParam* trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->First();
+ while (trackParamAtCluster) {
+
+ discardedCluster = trackParamAtCluster->GetClusterPtr();
+
+ // Compute residuals
+ dX = discardedCluster->GetX() - trackParamAtCluster->GetNonBendingCoor();
+ dY = discardedCluster->GetY() - trackParamAtCluster->GetBendingCoor();
+
+ // Set local chi2
+ trackParamAtCluster->SetLocalChi2(dX * dX / discardedCluster->GetErrX2() + dY * dY / discardedCluster->GetErrY2());
+
+ trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->After(trackParamAtCluster);
}
- }
+
+ }
+
+ return kTRUE;
+
+}
- // necessary to make a copy of the objects and not only the pointers in TClonesArray.
- fHitForRecAtHit = 0;
- if (theMUONTrack.fHitForRecAtHit) {
- fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
- maxIndex = (theMUONTrack.fHitForRecAtHit)->GetEntriesFast();
- for (Int_t index = 0; index < maxIndex; index++) {
- new ((*fHitForRecAtHit)[fHitForRecAtHit->GetEntriesFast()])
- AliMUONHitForRec(*(AliMUONHitForRec*)(theMUONTrack.fHitForRecAtHit)->At(index));
+ //__________________________________________________________________________
+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
+ AliDebug(1,"Enter ComputeGlobalChi2");
+
+ if (!fTrackParamAtCluster) {
+ AliWarning("no cluster attached to this track");
+ return 1.e10;
+ }
+
+ Double_t chi2 = 0.;
+
+ if (accountForMCS) {
+
+ // Check the weight matrices. If weight matrices are not available compute chi2 without MCS
+ if (!fClusterWeightsNonBending || !fClusterWeightsBending) {
+ AliWarning("cluster weights including multiple scattering effects are not available\n\t\t --> compute chi2 WITHOUT multiple scattering");
+ return ComputeGlobalChi2(kFALSE);
+ }
+ Int_t nClusters = GetNClusters();
+ if (fClusterWeightsNonBending->GetNrows() != nClusters || fClusterWeightsBending->GetNcols() != nClusters) {
+ AliWarning("cluster weights including multiple scattering effects are not available\n\t\t --> compute chi2 WITHOUT multiple scattering");
+ return ComputeGlobalChi2(kFALSE);
+ }
+
+ // Compute chi2
+ AliMUONVCluster *cluster;
+ Double_t *dX = new Double_t[nClusters];
+ Double_t *dY = new Double_t[nClusters];
+ AliMUONTrackParam* trackParamAtCluster;
+ for (Int_t iCluster1 = 0; iCluster1 < nClusters; iCluster1++) {
+ trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster1);
+ cluster = trackParamAtCluster->GetClusterPtr();
+ dX[iCluster1] = cluster->GetX() - trackParamAtCluster->GetNonBendingCoor();
+ dY[iCluster1] = cluster->GetY() - trackParamAtCluster->GetBendingCoor();
+ for (Int_t iCluster2 = 0; iCluster2 < iCluster1; iCluster2++) {
+ chi2 += ((*fClusterWeightsNonBending)(iCluster1, iCluster2) + (*fClusterWeightsNonBending)(iCluster2, iCluster1)) * dX[iCluster1] * dX[iCluster2] +
+ ((*fClusterWeightsBending)(iCluster1, iCluster2) + (*fClusterWeightsBending)(iCluster2, iCluster1)) * dY[iCluster1] * dY[iCluster2];
+ }
+ chi2 += ((*fClusterWeightsNonBending)(iCluster1, iCluster1) * dX[iCluster1] * dX[iCluster1]) +
+ ((*fClusterWeightsBending)(iCluster1, iCluster1) * dY[iCluster1] * dY[iCluster1]);
}
- }
+ delete [] dX;
+ delete [] dY;
+
+ } else {
+
+ AliMUONVCluster *cluster;
+ Double_t dX, dY;
+ AliMUONTrackParam* trackParamAtCluster;
+ Int_t nClusters = GetNClusters();
+ for (Int_t iCluster = 0; iCluster < nClusters ; iCluster++) {
+ trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster);
+ cluster = trackParamAtCluster->GetClusterPtr();
+ dX = cluster->GetX() - trackParamAtCluster->GetNonBendingCoor();
+ dY = cluster->GetY() - trackParamAtCluster->GetBendingCoor();
+ chi2 += dX * dX / cluster->GetErrX2() + dY * dY / cluster->GetErrY2();
+ }
+
+ }
+
+ return chi2;
- fNTrackHits = theMUONTrack.fNTrackHits;
- fFitFMin = theMUONTrack.fFitFMin;
- fMatchTrigger = theMUONTrack.fMatchTrigger;
- fChi2MatchTrigger = theMUONTrack.fChi2MatchTrigger;
- fTrackID = theMUONTrack.fTrackID;
+}
- return *this;
+ //__________________________________________________________________________
+Bool_t AliMUONTrack::ComputeClusterWeights(TMatrixD* mcsCovariances)
+{
+ /// Compute the weight matrices of the attached clusters, in non bending and bending direction,
+ /// accounting for multiple scattering correlations and cluster resolution
+ /// - Use the provided MCS covariance matrix if any (otherwise build it temporarily)
+ /// - Assume that track parameters at each cluster are corrects
+ /// - Return kFALSE if computation failed
+ AliDebug(1,"Enter ComputeClusterWeights1");
+
+ if (!fTrackParamAtCluster) {
+ AliWarning("no cluster attached to this track");
+ return kFALSE;
+ }
+
+ // Alocate memory
+ Int_t nClusters = GetNClusters();
+ if (!fClusterWeightsNonBending) fClusterWeightsNonBending = new TMatrixD(nClusters,nClusters);
+ if (!fClusterWeightsBending) fClusterWeightsBending = new TMatrixD(nClusters,nClusters);
+
+ // Compute weights matrices
+ if (!ComputeClusterWeights(*fClusterWeightsNonBending, *fClusterWeightsBending, mcsCovariances)) return kFALSE;
+
+ return kTRUE;
+
}
//__________________________________________________________________________
-void AliMUONTrack::AddTrackParamAtHit(AliMUONTrackParam *trackParam, AliMUONHitForRec *hitForRec)
+Bool_t AliMUONTrack::ComputeClusterWeights(TMatrixD& clusterWeightsNB, TMatrixD& clusterWeightsB,
+ TMatrixD* mcsCovariances, AliMUONVCluster* discardedCluster) const
{
- /// Add TrackParamAtHit if "trackParam" != NULL else create empty TrackParamAtHit
- /// Update link to HitForRec if "hitForRec" != NULL
- if (!fTrackParamAtHit) {
- fTrackParamAtHit = new TClonesArray("AliMUONTrackParam",10);
- fNTrackHits = 0;
- }
- AliMUONTrackParam* trackParamAtHit;
- if (trackParam) trackParamAtHit = new ((*fTrackParamAtHit)[fNTrackHits]) AliMUONTrackParam(*trackParam);
- else trackParamAtHit = new ((*fTrackParamAtHit)[fNTrackHits]) AliMUONTrackParam();
- if (hitForRec) trackParamAtHit->SetHitForRecPtr(hitForRec);
- fNTrackHits++;
+ /// Compute the weight matrices, in non bending and bending direction,
+ /// of the other attached clusters assuming the discarded one does not exist
+ /// accounting for multiple scattering correlations and cluster resolution
+ /// - Use the provided MCS covariance matrix if any (otherwise build it temporarily)
+ /// - Return kFALSE if computation failed
+ AliDebug(1,"Enter ComputeClusterWeights2");
+
+ // Check MCS covariance matrix and recompute it if need
+ Int_t nClusters = GetNClusters();
+ Bool_t deleteMCSCov = kFALSE;
+ if (!mcsCovariances) {
+ mcsCovariances = new TMatrixD(nClusters,nClusters);
+ deleteMCSCov = kTRUE;
+ ComputeMCSCovariances(*mcsCovariances);
+ }
+
+ // Resize the weights matrices; alocate memory
+ if (discardedCluster) {
+ clusterWeightsNB.ResizeTo(nClusters-1,nClusters-1);
+ clusterWeightsB.ResizeTo(nClusters-1,nClusters-1);
+ } else {
+ clusterWeightsNB.ResizeTo(nClusters,nClusters);
+ clusterWeightsB.ResizeTo(nClusters,nClusters);
+ }
+
+ // Define variables
+ AliMUONVCluster *cluster1, *cluster2;
+ Int_t iCurrentCluster1, iCurrentCluster2;
+
+ // Compute the covariance matrices
+ iCurrentCluster1 = 0;
+ for (Int_t iCluster1 = 0; iCluster1 < nClusters; iCluster1++) {
+ cluster1 = ((AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster1))->GetClusterPtr();
+
+ if (cluster1 == discardedCluster) continue;
+
+ // Loop over next clusters
+ iCurrentCluster2 = iCurrentCluster1;
+ for (Int_t iCluster2 = iCluster1; iCluster2 < nClusters; iCluster2++) {
+ cluster2 = ((AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster2))->GetClusterPtr();
+
+ if (cluster2 == discardedCluster) continue;
+
+ // Fill with MCS covariances
+ clusterWeightsNB(iCurrentCluster1, iCurrentCluster2) = (*mcsCovariances)(iCluster1,iCluster2);
+
+ // Equal contribution from multiple scattering in non bending and bending directions
+ clusterWeightsB(iCurrentCluster1, iCurrentCluster2) = clusterWeightsNB(iCurrentCluster1, iCurrentCluster2);
+
+ // Add contribution from cluster resolution to diagonal element and symmetrize the matrix
+ if (iCurrentCluster1 == iCurrentCluster2) {
+
+ // In non bending plane
+ clusterWeightsNB(iCurrentCluster1, iCurrentCluster1) += cluster1->GetErrX2();
+ // In bending plane
+ clusterWeightsB(iCurrentCluster1, iCurrentCluster1) += cluster1->GetErrY2();
+
+ } else {
+
+ // In non bending plane
+ clusterWeightsNB(iCurrentCluster2, iCurrentCluster1) = clusterWeightsNB(iCurrentCluster1, iCurrentCluster2);
+ // In bending plane
+ clusterWeightsB(iCurrentCluster2, iCurrentCluster1) = clusterWeightsB(iCurrentCluster1, iCurrentCluster2);
+
+ }
+
+ iCurrentCluster2++;
+ }
+
+ iCurrentCluster1++;
+ }
+
+ // Inversion of covariance matrices to get the weights
+ if (clusterWeightsNB.Determinant() != 0 && clusterWeightsB.Determinant() != 0) {
+ clusterWeightsNB.Invert();
+ clusterWeightsB.Invert();
+ } else {
+ AliWarning(" Determinant = 0");
+ clusterWeightsNB.ResizeTo(0,0);
+ clusterWeightsB.ResizeTo(0,0);
+ if(deleteMCSCov) delete mcsCovariances;
+ return kFALSE;
+ }
+
+ if(deleteMCSCov) delete mcsCovariances;
+
+ return kTRUE;
+
}
//__________________________________________________________________________
-void AliMUONTrack::AddHitForRecAtHit(const AliMUONHitForRec *hitForRec)
+void AliMUONTrack::ComputeMCSCovariances(TMatrixD& mcsCovariances) const
{
- /// Add hitForRec to the array of hitForRec at hit
- if (!fHitForRecAtHit)
- fHitForRecAtHit = new TClonesArray("AliMUONHitForRec",10);
+ /// Compute the multiple scattering covariance matrix
+ /// (assume that track parameters at each cluster are corrects)
+ AliDebug(1,"Enter ComputeMCSCovariances");
- if (!hitForRec)
- AliFatal("AliMUONTrack::AddHitForRecAtHit: hitForRec == NULL");
+ // Reset the size of the covariance matrix if needed
+ Int_t nClusters = GetNClusters();
+ if (mcsCovariances.GetNrows() != nClusters) mcsCovariances.ResizeTo(nClusters,nClusters);
+
+ // Define variables
+ Int_t nChambers = AliMUONConstants::NTrackingCh();
+ AliMUONTrackParam* trackParamAtCluster;
+ AliMUONTrackParam extrapTrackParam;
+ Int_t currentChamber = 0, expectedChamber = 0, size = 0;
+ Double_t *mcsAngle2 = new Double_t[2*nChambers];
+ Double_t *zMCS = new Double_t[2*nChambers];
+ Int_t *indices = new Int_t[2*nClusters];
+
+ // Compute multiple scattering dispersion angle at each chamber
+ // and save the z position where it is calculated
+ for (Int_t iCluster = 0; iCluster < nClusters; iCluster++) {
+ trackParamAtCluster = (AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster);
+
+ // look for missing chambers if any
+ currentChamber = trackParamAtCluster->GetClusterPtr()->GetChamberId();
+ while (currentChamber > expectedChamber) {
+
+ // Save the z position where MCS dispersion is calculated
+ zMCS[size] = AliMUONConstants::DefaultChamberZ(expectedChamber);
+
+ // Do not take into account MCS in chambers prior the first cluster
+ if (iCluster > 0) {
+
+ // Get track parameters at missing chamber z
+ extrapTrackParam = *trackParamAtCluster;
+ AliMUONTrackExtrap::ExtrapToZ(&extrapTrackParam, zMCS[size]);
+
+ // Save multiple scattering dispersion angle in missing chamber
+ mcsAngle2[size] = AliMUONTrackExtrap::GetMCSAngle2(extrapTrackParam,AliMUONConstants::ChamberThicknessInX0(),1.);
+
+ } else mcsAngle2[size] = 0.;
+
+ expectedChamber++;
+ size++;
+ }
+
+ // Save z position where MCS dispersion is calculated
+ zMCS[size] = trackParamAtCluster->GetZ();
+
+ // Save multiple scattering dispersion angle in current chamber
+ mcsAngle2[size] = AliMUONTrackExtrap::GetMCSAngle2(*trackParamAtCluster,AliMUONConstants::ChamberThicknessInX0(),1.);
+
+ // Save indice in zMCS array corresponding to the current cluster
+ indices[iCluster] = size;
+
+ expectedChamber = currentChamber + 1;
+ size++;
+ }
+
+ // complete array of z if last cluster is on the last but one chamber
+ if (currentChamber != nChambers-1) zMCS[size++] = AliMUONConstants::DefaultChamberZ(nChambers-1);
+
+ // Compute the covariance matrix
+ for (Int_t iCluster1 = 0; iCluster1 < nClusters; iCluster1++) {
+
+ for (Int_t iCluster2 = iCluster1; iCluster2 < nClusters; iCluster2++) {
+
+ // Initialization to 0 (diagonal plus upper triangular part)
+ mcsCovariances(iCluster1,iCluster2) = 0.;
+
+ // Compute contribution from multiple scattering in upstream chambers
+ for (Int_t k = 0; k < indices[iCluster1]; k++) {
+ mcsCovariances(iCluster1,iCluster2) += (zMCS[indices[iCluster1]] - zMCS[k]) * (zMCS[indices[iCluster2]] - zMCS[k]) * mcsAngle2[k];
+ }
+
+ // Symetrize the matrix
+ mcsCovariances(iCluster2,iCluster1) = mcsCovariances(iCluster1,iCluster2);
+ }
+
+ }
+
+ delete [] mcsAngle2;
+ delete [] zMCS;
+ delete [] indices;
- new ((*fHitForRecAtHit)[fHitForRecAtHit->GetEntriesFast()]) AliMUONHitForRec(*hitForRec);
}
//__________________________________________________________________________
-Bool_t* AliMUONTrack::CompatibleTrack(AliMUONTrack * Track, Double_t Sigma2Cut) const
+Int_t AliMUONTrack::ClustersInCommon(AliMUONTrack* track) const
{
- /// Return kTRUE/kFALSE for each chamber if hit is compatible or not
- TClonesArray *hitArray, *thisHitArray;
- AliMUONHitForRec *hit, *thisHit;
- Int_t chamberNumber;
- Float_t deltaZ;
- Float_t deltaZMax = 1.; // 1 cm
- Float_t chi2 = 0;
- Bool_t *nCompHit = new Bool_t[AliMUONConstants::NTrackingCh()];
-
- for ( Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) {
- nCompHit[ch] = kFALSE;
- }
-
- thisHitArray = this->GetHitForRecAtHit();
-
- hitArray = Track->GetHitForRecAtHit();
-
- for (Int_t iHthis = 0; iHthis < thisHitArray->GetEntriesFast(); iHthis++) {
- thisHit = (AliMUONHitForRec*) thisHitArray->At(iHthis);
- chamberNumber = thisHit->GetChamberNumber();
- if (chamberNumber < 0 || chamberNumber > AliMUONConstants::NTrackingCh()) continue;
- nCompHit[chamberNumber] = kFALSE;
- for (Int_t iH = 0; iH < hitArray->GetEntriesFast(); iH++) {
- hit = (AliMUONHitForRec*) hitArray->At(iH);
- deltaZ = TMath::Abs(thisHit->GetZ() - hit->GetZ());
- chi2 = thisHit->NormalizedChi2WithHitForRec(hit,Sigma2Cut); // set cut to 4 sigmas
- if (chi2 < 3. && deltaZ < deltaZMax) {
- nCompHit[chamberNumber] = kTRUE;
+ /// Returns the number of clusters in common 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);
+ // 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;
}
- }
+ }
}
+ 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;
+}
+
+ //__________________________________________________________________________
+Int_t AliMUONTrack::GetNDF() const
+{
+ /// return the number of degrees of freedom
- return nCompHit;
+ Int_t ndf = 2 * GetNClusters() - 5;
+ return (ndf > 0) ? ndf : 0;
}
//__________________________________________________________________________
-Int_t AliMUONTrack::HitsInCommon(AliMUONTrack* Track) const
+Double_t AliMUONTrack::GetNormalizedChi2() const
{
- /// Returns the number of hits in common between the current track ("this")
- /// and the track pointed to by "Track".
- Int_t hitsInCommon = 0;
- AliMUONTrackParam *trackParamAtHit1, *trackParamAtHit2;
- // Loop over hits of first track
- trackParamAtHit1 = (AliMUONTrackParam*) this->fTrackParamAtHit->First();
- while (trackParamAtHit1) {
- // Loop over hits of second track
- trackParamAtHit2 = (AliMUONTrackParam*) Track->fTrackParamAtHit->First();
- while (trackParamAtHit2) {
- // Increment "hitsInCommon" if both TrackParamAtHits point to the same HitForRec
- if ((trackParamAtHit1->GetHitForRecPtr()) == (trackParamAtHit2->GetHitForRecPtr())) hitsInCommon++;
- trackParamAtHit2 = (AliMUONTrackParam*) Track->fTrackParamAtHit->After(trackParamAtHit2);
- } // trackParamAtHit2
- trackParamAtHit1 = (AliMUONTrackParam*) this->fTrackParamAtHit->After(trackParamAtHit1);
- } // trackParamAtHit1
- return hitsInCommon;
+ /// 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;
}
//__________________________________________________________________________
-void AliMUONTrack::RecursiveDump(void) const
+Int_t AliMUONTrack::CompatibleTrack(AliMUONTrack* track, Double_t sigmaCut, Bool_t compatibleCluster[10]) const
+{
+ /// for each chamber: return kTRUE (kFALSE) if clusters are compatible (not compatible).
+ /// nMatchClusters = number of clusters of "this" track matched with one cluster of track "track"
+ AliMUONTrackParam *trackParamAtCluster1, *trackParamAtCluster2;
+ AliMUONVCluster *cluster1, *cluster2;
+ Double_t chi2, dX, dY;
+ Double_t chi2Max = sigmaCut * sigmaCut;
+
+ // initialization
+ Int_t nMatchClusters = 0;
+ for ( Int_t ch = 0; ch < AliMUONConstants::NTrackingCh(); ch++) compatibleCluster[ch] = kFALSE;
+
+ if (!fTrackParamAtCluster || !this->fTrackParamAtCluster) return nMatchClusters;
+
+ // Loop over clusters of first track
+ trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->First();
+ while (trackParamAtCluster1) {
+
+ cluster1 = trackParamAtCluster1->GetClusterPtr();
+
+ // Loop over clusters of second track
+ trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->First();
+ while (trackParamAtCluster2) {
+
+ cluster2 = trackParamAtCluster2->GetClusterPtr();
+
+ //prepare next step
+ trackParamAtCluster2 = (AliMUONTrackParam*) track->fTrackParamAtCluster->After(trackParamAtCluster2);
+
+ // check DE Id
+ if (cluster1->GetDetElemId() != cluster2->GetDetElemId()) continue;
+
+ // check local chi2
+ dX = cluster1->GetX() - cluster2->GetX();
+ dY = cluster1->GetY() - cluster2->GetY();
+ chi2 = dX * dX / (cluster1->GetErrX2() + cluster2->GetErrX2()) + dY * dY / (cluster1->GetErrY2() + cluster2->GetErrY2());
+ if (chi2 > 2. * chi2Max) continue; // 2 because 2 quantities in chi2
+
+ compatibleCluster[cluster1->GetChamberId()] = kTRUE;
+ nMatchClusters++;
+ break;
+ }
+
+ trackParamAtCluster1 = (AliMUONTrackParam*) this->fTrackParamAtCluster->After(trackParamAtCluster1);
+ }
+
+ return nMatchClusters;
+}
+
+//__________________________________________________________________________
+void AliMUONTrack::SetTrackParamAtVertex(const AliMUONTrackParam* trackParam)
{
- /// Recursive dump of AliMUONTrack, i.e. with dump of TrackParamAtHit's and attached HitForRec's
- AliMUONTrackParam *trackParamAtHit;
- AliMUONHitForRec *hitForRec;
+ /// set track parameters at vertex
+ if (trackParam == 0x0) return;
+ if (fTrackParamAtVertex) *fTrackParamAtVertex = *trackParam;
+ else fTrackParamAtVertex = new AliMUONTrackParam(*trackParam);
+}
+
+//__________________________________________________________________________
+void AliMUONTrack::RecursiveDump() const
+{
+ /// Recursive dump of AliMUONTrack, i.e. with dump of trackParamAtCluster and attached clusters
+ AliMUONTrackParam *trackParamAtCluster;
+ AliMUONVCluster *cluster;
cout << "Recursive dump of Track: " << this << endl;
// Track
this->Dump();
- for (Int_t trackHitIndex = 0; trackHitIndex < fNTrackHits; trackHitIndex++) {
- trackParamAtHit = (AliMUONTrackParam*) ((*fTrackParamAtHit)[trackHitIndex]);
- // TrackHit
- cout << "TrackParamAtHit: " << trackParamAtHit << " (index: " << trackHitIndex << ")" << endl;
- trackParamAtHit->Dump();
- hitForRec = trackParamAtHit->GetHitForRecPtr();
- // HitForRec
- cout << "HitForRec: " << hitForRec << endl;
- hitForRec->Dump();
+ for (Int_t iCluster = 0; iCluster < GetNClusters(); iCluster++) {
+ trackParamAtCluster = (AliMUONTrackParam*) ((*fTrackParamAtCluster)[iCluster]);
+ // trackParamAtCluster
+ cout << "trackParamAtCluster: " << trackParamAtCluster << " (index: " << iCluster << ")" << endl;
+ trackParamAtCluster->Dump();
+ cluster = trackParamAtCluster->GetClusterPtr();
+ // cluster
+ cout << "cluster: " << cluster << endl;
+ cluster->Print();
}
return;
}
//_____________________________________________-
-void AliMUONTrack::Print(Option_t* opt) const
+void AliMUONTrack::Print(Option_t*) const
{
/// Printing Track information
- /// "full" option for printing all the information about the track
- TString sopt(opt);
- sopt.ToUpper();
-
- if ( sopt.Contains("FULL") ) {
- cout << "<AliMUONTrack> No.Clusters=" << setw(2) << GetNTrackHits() <<
- // ", Bending P="<< setw(8) << setprecision(5) << 1./GetInverseBendingMomentum() <<
- //", NonBendSlope=" << setw(8) << setprecision(5) << GetNonBendingSlope()*180./TMath::Pi() <<
- //", BendSlope=" << setw(8) << setprecision(5) << GetBendingSlope()*180./TMath::Pi() <<
+
+ cout << "<AliMUONTrack> No.Clusters=" << setw(2) << GetNClusters() <<
", Match2Trig=" << setw(1) << GetMatchTrigger() <<
- ", Chi2-tracking-trigger=" << setw(8) << setprecision(5) << GetChi2MatchTrigger() << endl ;
- GetTrackParamAtHit()->First()->Print("full");
- }
- else {
- cout << "<AliMUONTrack>";
- GetTrackParamAtHit()->First()->Print("");
+ ", LoTrgNum=" << setw(3) << GetLoTrgNum() <<
+ ", Chi2-tracking-trigger=" << setw(8) << setprecision(5) << GetChi2MatchTrigger();
+ cout << Form(" HitTriggerPattern %x",fHitsPatternInTrigCh);
+ cout << Form(" MClabel=%d",fTrackID) << endl;
+ if (fTrackParamAtCluster) fTrackParamAtCluster->First()->Print("FULL");
+}
- }
+//__________________________________________________________________________
+void AliMUONTrack::SetLocalTrigger(Int_t loCirc, Int_t loStripX, Int_t loStripY, Int_t loDev, Int_t loLpt, Int_t loHpt)
+{
+ /// pack the local trigger information and store
+
+ if (loCirc < 0) return;
+
+ fLocalTrigger = 0;
+ fLocalTrigger += loCirc;
+ fLocalTrigger += loStripX << 8;
+ fLocalTrigger += loStripY << 13;
+ fLocalTrigger += loDev << 17;
+ fLocalTrigger += loLpt << 22;
+ fLocalTrigger += loHpt << 24;
+
+}
+
+//__________________________________________________________________________
+void AliMUONTrack::FindMCLabel()
+{
+ /// Determine the MC label from the label of the attached clusters and fill fMCLabel data member:
+ /// More than 50% of clusters, including 1 before and 1 after the dipole, must share the same label
+
+ Int_t nClusters = GetNClusters();
+ Int_t halfCluster = nClusters/2;
+
+ // reset MC label
+ fTrackID = -1;
+
+ // loop over first clusters (if nClusters left < (nClusters-halfCluster) the conditions cannot be fulfilled)
+ for (Int_t iCluster1 = 0; iCluster1 < nClusters-halfCluster; iCluster1++) {
+ AliMUONVCluster* cluster1 = ((AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster1))->GetClusterPtr();
+
+ // if the first cluster is not on station 1 or 2 the conditions cannot be fulfilled
+ if (cluster1->GetChamberId() > 3) return;
+ Int_t label1 = cluster1->GetMCLabel();
+ if (label1 < 0) continue;
+
+ Int_t nIdenticalLabel = 1;
+
+ // Loop over next clusters
+ for (Int_t iCluster2 = iCluster1+1; iCluster2 < nClusters; iCluster2++) {
+ AliMUONVCluster* cluster2 = ((AliMUONTrackParam*) fTrackParamAtCluster->UncheckedAt(iCluster2))->GetClusterPtr();
+
+ if (cluster2->GetMCLabel() != label1) continue;
+
+ nIdenticalLabel++;
+
+ // stop as soon as conditions are fulfilled
+ if (nIdenticalLabel > halfCluster && cluster2->GetChamberId() > 5) {
+ fTrackID = label1;
+ return;
+ }
+
+ }
+
+ }
+
}
+
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff