* provided "as is" without express or implied warranty. *
**************************************************************************/
-/*
-$Log$
-Revision 1.1.2.3 2000/06/09 21:03:09 morsch
-Make includes consistent with new file structure.
+/* $Id$ */
-Revision 1.1.2.2 2000/06/09 12:58:05 gosset
-Removed comment beginnings in Log sections of .cxx files
-Suppressed most violations of coding rules
+//-----------------------------------------------------------------------------
+// Class AliMUONTrackParam
+//-------------------------
+// Track parameters in ALICE dimuon spectrometer
+//-----------------------------------------------------------------------------
-Revision 1.1.2.1 2000/06/07 14:44:53 gosset
-Addition of files for track reconstruction in C++
-*/
+#include "AliMUONTrackParam.h"
+#include "AliMUONVCluster.h"
-//__________________________________________________________________________
-//
-// Track parameters in ALICE dimuon spectrometer
-//__________________________________________________________________________
+#include "AliLog.h"
-#include <iostream.h>
+#include <TMath.h>
-#include "AliCallf77.h"
-#include "AliMUON.h"
-#include "AliMUONHitForRec.h"
-#include "AliMUONSegment.h"
-#include "AliMUONTrackParam.h"
-#include "AliMUONChamber.h"
-#include "AliRun.h"
+#include <Riostream.h>
+/// \cond CLASSIMP
ClassImp(AliMUONTrackParam) // Class implementation in ROOT context
+/// \endcond
-#ifndef WIN32
-# define reco_ghelix reco_ghelix_
-#else
-# define reco_ghelix RECO_GHELIX
-#endif
-
-extern "C"
-{
-void type_of_call reco_ghelix(Double_t &Charge, Double_t &StepLength, Double_t *VGeant3, Double_t *VGeant3New);
-}
-
-// Inline functions for Get and Set: inline removed because it does not work !!!!
-Double_t AliMUONTrackParam::GetInverseBendingMomentum(void) {
- // Get fInverseBendingMomentum
- return fInverseBendingMomentum;}
-void AliMUONTrackParam::SetInverseBendingMomentum(Double_t InverseBendingMomentum) {
- // Set fInverseBendingMomentum
- fInverseBendingMomentum = InverseBendingMomentum;}
-Double_t AliMUONTrackParam::GetBendingSlope(void) {
- // Get fBendingSlope
- return fBendingSlope;}
-void AliMUONTrackParam::SetBendingSlope(Double_t BendingSlope) {
- // Set fBendingSlope
- fBendingSlope = BendingSlope;}
-Double_t AliMUONTrackParam::GetNonBendingSlope(void) {
- // Get fNonBendingSlope
- return fNonBendingSlope;}
-void AliMUONTrackParam::SetNonBendingSlope(Double_t NonBendingSlope) {
- // Set fNonBendingSlope
- fNonBendingSlope = NonBendingSlope;}
-Double_t AliMUONTrackParam::GetZ(void) {
- // Get fZ
- return fZ;}
-void AliMUONTrackParam::SetZ(Double_t Z) {
- // Set fZ
- fZ = Z;}
-Double_t AliMUONTrackParam::GetBendingCoor(void) {
- // Get fBendingCoor
- return fBendingCoor;}
-void AliMUONTrackParam::SetBendingCoor(Double_t BendingCoor) {
- // Set fBendingCoor
- fBendingCoor = BendingCoor;}
-Double_t AliMUONTrackParam::GetNonBendingCoor(void) {
- // Get fNonBendingCoor
- return fNonBendingCoor;}
-void AliMUONTrackParam::SetNonBendingCoor(Double_t NonBendingCoor) {
- // Set fNonBendingCoor
- fNonBendingCoor = NonBendingCoor;}
-
- //__________________________________________________________________________
-void AliMUONTrackParam::ExtrapToZ(Double_t Z)
-{
- // Track parameter extrapolation to the plane at "Z".
- // On return, the track parameters resulting from the extrapolation
- // replace the current track parameters.
- // Use "reco_ghelix" which should be replaced by something else !!!!
- if (this->fZ == Z) return; // nothing to be done if same Z
- Double_t forwardBackward; // +1 if forward, -1 if backward
- if (Z > this->fZ) forwardBackward = 1.0;
- else forwardBackward = -1.0;
- Double_t temp, vGeant3[7], vGeant3New[7]; // 7 in parameter ????
- Int_t iGeant3, stepNumber;
- Int_t maxStepNumber = 5000; // in parameter ????
- // For safety: return kTRUE or kFALSE ????
- // Parameter vector for calling GHELIX in Geant3
- SetGeant3Parameters(vGeant3, forwardBackward);
- // For use of reco_ghelix...: invert X and Y, PX/PTOT and PY/PTOT !!!!
- temp = vGeant3[0]; vGeant3[0] = vGeant3[1]; vGeant3[1] = temp;
- temp = vGeant3[3]; vGeant3[3] = vGeant3[4]; vGeant3[4] = temp;
- // charge must be changed with momentum for backward motion
- Double_t charge =
- forwardBackward * TMath::Sign(Double_t(1.0), this->fInverseBendingMomentum);
- Double_t stepLength = 6.0; // in parameter ????
- // Extrapolation loop
- stepNumber = 0;
- while (((forwardBackward * (vGeant3[2] - Z)) <= 0.0) &&
- (stepNumber < maxStepNumber)) {
- stepNumber++;
- // call Geant3 "ghelix" subroutine through a copy in "reco_muon.F":
- // the true function should be called, but how ???? and remove prototyping ...
- reco_ghelix(charge, stepLength, vGeant3, vGeant3New);
- if ((forwardBackward * (vGeant3New[2] - Z)) > 0.0) break; // one is beyond Z
- // better use TArray ????
- for (iGeant3 = 0; iGeant3 < 7; iGeant3++)
- {vGeant3[iGeant3] = vGeant3New[iGeant3];}
- }
- // check maxStepNumber ????
- // For use of reco_ghelix...:
- // invert back X and Y, PX/PTOT and PY/PTOT, both for vGeant3 and vGeant3New !!!!
- temp = vGeant3[0]; vGeant3[0] = vGeant3[1]; vGeant3[1] = temp;
- temp = vGeant3New[0]; vGeant3New[0] = vGeant3New[1]; vGeant3New[1] = temp;
- temp = vGeant3[3]; vGeant3[3] = vGeant3[4]; vGeant3[4] = temp;
- temp = vGeant3New[3]; vGeant3New[3] = vGeant3New[4]; vGeant3New[4] = temp;
- // Interpolation back to exact Z (2nd order)
- // should be in function ???? using TArray ????
- Double_t dZ12 = vGeant3New[2] - vGeant3[2]; // 1->2
- Double_t dZ1i = Z - vGeant3[2]; // 1-i
- Double_t dZi2 = vGeant3New[2] - Z; // i->2
- Double_t xPrime = (vGeant3New[0] - vGeant3[0]) / dZ12;
- Double_t xSecond =
- ((vGeant3New[3] / vGeant3New[5]) - (vGeant3[3] / vGeant3[5])) / dZ12;
- Double_t yPrime = (vGeant3New[1] - vGeant3[1]) / dZ12;
- Double_t ySecond =
- ((vGeant3New[4] / vGeant3New[5]) - (vGeant3[4] / vGeant3[5])) / dZ12;
- vGeant3[0] = vGeant3[0] + xPrime * dZ1i - 0.5 * xSecond * dZ1i * dZi2; // X
- vGeant3[1] = vGeant3[1] + yPrime * dZ1i - 0.5 * ySecond * dZ1i * dZi2; // Y
- vGeant3[2] = Z; // Z
- Double_t xPrimeI = xPrime - 0.5 * xSecond * (dZi2 - dZ1i);
- Double_t yPrimeI = yPrime - 0.5 * ySecond * (dZi2 - dZ1i);
- vGeant3[5] =
- 1.0 / TMath::Sqrt(1.0 + xPrimeI * xPrimeI + yPrimeI * yPrimeI); // PZ/PTOT
- vGeant3[3] = xPrimeI * vGeant3[5]; // PX/PTOT
- vGeant3[4] = yPrimeI * vGeant3[5]; // PY/PTOT
- // Track parameters from Geant3 parameters
- GetFromGeant3Parameters(vGeant3, charge);
-}
-
- //__________________________________________________________________________
-void AliMUONTrackParam::SetGeant3Parameters(Double_t *VGeant3, Double_t ForwardBackward)
-{
- // Set vector of Geant3 parameters pointed to by "VGeant3"
- // from track parameters in current AliMUONTrackParam.
- // Since AliMUONTrackParam is only geometry, one uses "ForwardBackward"
- // to know whether the particle is going forward (+1) or backward (-1).
- VGeant3[0] = this->fNonBendingCoor; // X
- VGeant3[1] = this->fBendingCoor; // Y
- VGeant3[2] = this->fZ; // Z
- Double_t pYZ = TMath::Abs(1.0 / this->fInverseBendingMomentum);
- Double_t pZ =
- pYZ / TMath::Sqrt(1.0 + this->fBendingSlope * this->fBendingSlope);
- VGeant3[6] =
- TMath::Sqrt(pYZ * pYZ +
- pZ * pZ * this->fNonBendingSlope * this->fNonBendingSlope); // PTOT
- VGeant3[5] = ForwardBackward * pZ / VGeant3[6]; // PZ/PTOT
- VGeant3[3] = this->fNonBendingSlope * VGeant3[5]; // PX/PTOT
- VGeant3[4] = this->fBendingSlope * VGeant3[5]; // PY/PTOT
-}
-
- //__________________________________________________________________________
-void AliMUONTrackParam::GetFromGeant3Parameters(Double_t *VGeant3, Double_t Charge)
-{
- // Get track parameters in current AliMUONTrackParam
- // from Geant3 parameters pointed to by "VGeant3".
- // "InverseBendingMomentum" is signed with "Charge".
- this->fNonBendingCoor = VGeant3[0]; // X
- this->fBendingCoor = VGeant3[1]; // Y
- this->fZ = VGeant3[2]; // Z
- Double_t pYZ = VGeant3[6] * TMath::Sqrt(1.0 - VGeant3[3] * VGeant3[3]);
- this->fInverseBendingMomentum = Charge / pYZ;
- this->fBendingSlope = VGeant3[4] / VGeant3[5];
- this->fNonBendingSlope = VGeant3[3] / VGeant3[5];
-}
-
- //__________________________________________________________________________
-void AliMUONTrackParam::ExtrapToStation(Int_t Station, AliMUONTrackParam *TrackParam)
-{
- // Track parameters extrapolated from current track parameters ("this")
- // to both chambers of the station(0..) "Station"
- // are returned in the array (dimension 2) of track parameters
- // pointed to by "TrackParam" (index 0 and 1 for first and second chambers).
- Double_t extZ[2], z1, z2;
- Int_t i1, i2;
- AliMUON *pMUON = (AliMUON*) gAlice->GetModule("MUON"); // necessary ????
- // range of Station to be checked ????
- z1 = (&(pMUON->Chamber(2 * Station)))->Z(); // Z of first chamber
- z2 = (&(pMUON->Chamber(2 * Station + 1)))->Z(); // Z of second chamber
- // First and second Z to extrapolate at
- if ((z1 > this->fZ) && (z2 > this->fZ)) {i1 = 0; i2 = 1;}
- else if ((z1 < this->fZ) && (z2 < this->fZ)) {i1 = 1; i2 = 0;}
- else {
- cout << "ERROR in AliMUONTrackParam::CreateExtrapSegmentInStation" << endl;
- cout << "Starting Z (" << this->fZ << ") in between z1 (" << z1 <<
- ") and z2 (" << z2 << ") of station(0..) " << Station << endl;
- }
- extZ[i1] = z1;
- extZ[i2] = z2;
- // copy of track parameters
- TrackParam[i1] = *this;
- // first extrapolation
- (&(TrackParam[i1]))->ExtrapToZ(extZ[0]);
- TrackParam[i2] = TrackParam[i1];
- // second extrapolation
- (&(TrackParam[i2]))->ExtrapToZ(extZ[1]);
- return;
+ //_________________________________________________________________________
+AliMUONTrackParam::AliMUONTrackParam()
+ : TObject(),
+ fZ(0.),
+ fParameters(5,1),
+ fCovariances(0x0),
+ fPropagator(0x0),
+ fExtrapParameters(0x0),
+ fExtrapCovariances(0x0),
+ fSmoothParameters(0x0),
+ fSmoothCovariances(0x0),
+ fClusterPtr(0x0),
+ fOwnCluster(kFALSE),
+ fRemovable(kFALSE),
+ fTrackChi2(0.),
+ fLocalChi2(0.)
+{
+ /// Constructor
+ fParameters.Zero();
+}
+
+ //_________________________________________________________________________
+AliMUONTrackParam::AliMUONTrackParam(const AliMUONTrackParam& theMUONTrackParam)
+ : TObject(theMUONTrackParam),
+ fZ(theMUONTrackParam.fZ),
+ fParameters(theMUONTrackParam.fParameters),
+ fCovariances(0x0),
+ fPropagator(0x0),
+ fExtrapParameters(0x0),
+ fExtrapCovariances(0x0),
+ fSmoothParameters(0x0),
+ fSmoothCovariances(0x0),
+ fClusterPtr(0x0),
+ fOwnCluster(theMUONTrackParam.fOwnCluster),
+ fRemovable(theMUONTrackParam.fRemovable),
+ fTrackChi2(theMUONTrackParam.fTrackChi2),
+ fLocalChi2(theMUONTrackParam.fLocalChi2)
+{
+ /// Copy constructor
+ if (theMUONTrackParam.fCovariances) fCovariances = new TMatrixD(*(theMUONTrackParam.fCovariances));
+ if (theMUONTrackParam.fPropagator) fPropagator = new TMatrixD(*(theMUONTrackParam.fPropagator));
+ if (theMUONTrackParam.fExtrapParameters) fExtrapParameters = new TMatrixD(*(theMUONTrackParam.fExtrapParameters));
+ if (theMUONTrackParam.fExtrapCovariances) fExtrapCovariances = new TMatrixD(*(theMUONTrackParam.fExtrapCovariances));
+ if (theMUONTrackParam.fSmoothParameters) fSmoothParameters = new TMatrixD(*(theMUONTrackParam.fSmoothParameters));
+ if (theMUONTrackParam.fSmoothCovariances) fSmoothCovariances = new TMatrixD(*(theMUONTrackParam.fSmoothCovariances));
+
+ if(fOwnCluster) fClusterPtr = static_cast<AliMUONVCluster*>(theMUONTrackParam.fClusterPtr->Clone());
+ else fClusterPtr = theMUONTrackParam.fClusterPtr;
}
+ //_________________________________________________________________________
+AliMUONTrackParam& AliMUONTrackParam::operator=(const AliMUONTrackParam& theMUONTrackParam)
+{
+ /// Asignment operator
+ if (this == &theMUONTrackParam)
+ return *this;
+
+ // base class assignement
+ TObject::operator=(theMUONTrackParam);
+
+ fZ = theMUONTrackParam.fZ;
+
+ fParameters = theMUONTrackParam.fParameters;
+
+ if (theMUONTrackParam.fCovariances) {
+ if (fCovariances) *fCovariances = *(theMUONTrackParam.fCovariances);
+ else fCovariances = new TMatrixD(*(theMUONTrackParam.fCovariances));
+ } else {
+ delete fCovariances;
+ fCovariances = 0x0;
+ }
+
+ if (theMUONTrackParam.fPropagator) {
+ if (fPropagator) *fPropagator = *(theMUONTrackParam.fPropagator);
+ else fPropagator = new TMatrixD(*(theMUONTrackParam.fPropagator));
+ } else {
+ delete fPropagator;
+ fPropagator = 0x0;
+ }
+
+ if (theMUONTrackParam.fExtrapParameters) {
+ if (fExtrapParameters) *fExtrapParameters = *(theMUONTrackParam.fExtrapParameters);
+ else fExtrapParameters = new TMatrixD(*(theMUONTrackParam.fExtrapParameters));
+ } else {
+ delete fExtrapParameters;
+ fExtrapParameters = 0x0;
+ }
+
+ if (theMUONTrackParam.fExtrapCovariances) {
+ if (fExtrapCovariances) *fExtrapCovariances = *(theMUONTrackParam.fExtrapCovariances);
+ else fExtrapCovariances = new TMatrixD(*(theMUONTrackParam.fExtrapCovariances));
+ } else {
+ delete fExtrapCovariances;
+ fExtrapCovariances = 0x0;
+ }
+
+ if (theMUONTrackParam.fSmoothParameters) {
+ if (fSmoothParameters) *fSmoothParameters = *(theMUONTrackParam.fSmoothParameters);
+ else fSmoothParameters = new TMatrixD(*(theMUONTrackParam.fSmoothParameters));
+ } else {
+ delete fSmoothParameters;
+ fSmoothParameters = 0x0;
+ }
+
+ if (theMUONTrackParam.fSmoothCovariances) {
+ if (fSmoothCovariances) *fSmoothCovariances = *(theMUONTrackParam.fSmoothCovariances);
+ else fSmoothCovariances = new TMatrixD(*(theMUONTrackParam.fSmoothCovariances));
+ } else {
+ delete fSmoothCovariances;
+ fSmoothCovariances = 0x0;
+ }
+
+ if (fOwnCluster) delete fClusterPtr;
+ fOwnCluster = theMUONTrackParam.fOwnCluster;
+ if(fOwnCluster) fClusterPtr = static_cast<AliMUONVCluster*>(theMUONTrackParam.fClusterPtr->Clone());
+ else fClusterPtr = theMUONTrackParam.fClusterPtr;
+
+ fRemovable = theMUONTrackParam.fRemovable;
+
+ fTrackChi2 = theMUONTrackParam.fTrackChi2;
+ fLocalChi2 = theMUONTrackParam.fLocalChi2;
+
+ return *this;
+}
+
+ //__________________________________________________________________________
+AliMUONTrackParam::~AliMUONTrackParam()
+{
+/// Destructor
+ DeleteCovariances();
+ delete fPropagator;
+ delete fExtrapParameters;
+ delete fExtrapCovariances;
+ delete fSmoothParameters;
+ delete fSmoothCovariances;
+ if(fOwnCluster) delete fClusterPtr;
+}
+
+ //__________________________________________________________________________
+void
+AliMUONTrackParam::Clear(Option_t* /*opt*/)
+{
+ /// clear memory
+ DeleteCovariances();
+ delete fPropagator; fPropagator = 0x0;
+ delete fExtrapParameters; fExtrapParameters = 0x0;
+ delete fExtrapCovariances; fExtrapCovariances = 0x0;
+ delete fSmoothParameters; fSmoothParameters = 0x0;
+ delete fSmoothCovariances; fSmoothCovariances = 0x0;
+ if(fOwnCluster) {
+ delete fClusterPtr; fClusterPtr = 0x0;
+ }
+}
+
+ //__________________________________________________________________________
+Double_t AliMUONTrackParam::Px() const
+{
+ /// return p_x from track parameters
+ Double_t pZ;
+ if (TMath::Abs(fParameters(4,0)) > 0) {
+ Double_t pYZ = (TMath::Abs(fParameters(4,0)) > 0) ? TMath::Abs(1.0 / fParameters(4,0)) : FLT_MAX;
+ pZ = - pYZ / (TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0))); // spectro. (z<0)
+ } else {
+ pZ = - FLT_MAX / TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0) + fParameters(1,0) * fParameters(1,0));
+ }
+ return pZ * fParameters(1,0);
+}
+
+ //__________________________________________________________________________
+Double_t AliMUONTrackParam::Py() const
+{
+ /// return p_y from track parameters
+ Double_t pZ;
+ if (TMath::Abs(fParameters(4,0)) > 0) {
+ Double_t pYZ = (TMath::Abs(fParameters(4,0)) > 0) ? TMath::Abs(1.0 / fParameters(4,0)) : FLT_MAX;
+ pZ = - pYZ / (TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0))); // spectro. (z<0)
+ } else {
+ pZ = - FLT_MAX / TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0) + fParameters(1,0) * fParameters(1,0));
+ }
+ return pZ * fParameters(3,0);
+}
+
+ //__________________________________________________________________________
+Double_t AliMUONTrackParam::Pz() const
+{
+ /// return p_z from track parameters
+ if (TMath::Abs(fParameters(4,0)) > 0) {
+ Double_t pYZ = TMath::Abs(1.0 / fParameters(4,0));
+ return - pYZ / (TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0))); // spectro. (z<0)
+ } else return - FLT_MAX / TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0) + fParameters(1,0) * fParameters(1,0));
+}
+
+ //__________________________________________________________________________
+Double_t AliMUONTrackParam::P() const
+{
+ /// return p from track parameters
+ if (TMath::Abs(fParameters(4,0)) > 0) {
+ Double_t pYZ = TMath::Abs(1.0 / fParameters(4,0));
+ Double_t pZ = - pYZ / (TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0))); // spectro. (z<0)
+ return - pZ * TMath::Sqrt(1.0 + fParameters(3,0) * fParameters(3,0) + fParameters(1,0) * fParameters(1,0));
+ } else return FLT_MAX;
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetCovariances() const
+{
+ /// Return the covariance matrix (create it before if needed)
+ if (!fCovariances) {
+ fCovariances = new TMatrixD(5,5);
+ fCovariances->Zero();
+ }
+ return *fCovariances;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetCovariances(const TMatrixD& covariances)
+{
+ /// Set the covariance matrix
+ if (fCovariances) *fCovariances = covariances;
+ else fCovariances = new TMatrixD(covariances);
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetCovariances(const Double_t matrix[5][5])
+{
+ /// Set the covariance matrix
+ if (fCovariances) fCovariances->SetMatrixArray(&(matrix[0][0]));
+ else fCovariances = new TMatrixD(5,5,&(matrix[0][0]));
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetVariances(const Double_t matrix[5][5])
+{
+ /// Set the diagonal terms of the covariance matrix (variances)
+ if (!fCovariances) fCovariances = new TMatrixD(5,5);
+ fCovariances->Zero();
+ for (Int_t i=0; i<5; i++) (*fCovariances)(i,i) = matrix[i][i];
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::DeleteCovariances()
+{
+ /// Delete the covariance matrix
+ delete fCovariances;
+ fCovariances = 0x0;
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetPropagator() const
+{
+ /// Return the propagator (create it before if needed)
+ if (!fPropagator) {
+ fPropagator = new TMatrixD(5,5);
+ fPropagator->UnitMatrix();
+ }
+ return *fPropagator;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::ResetPropagator()
+{
+ /// Reset the propagator
+ if (fPropagator) fPropagator->UnitMatrix();
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::UpdatePropagator(const TMatrixD& propagator)
+{
+ /// Update the propagator
+ if (fPropagator) *fPropagator = TMatrixD(propagator,TMatrixD::kMult,*fPropagator);
+ else fPropagator = new TMatrixD(propagator);
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetExtrapParameters() const
+{
+ /// Return extrapolated parameters (create it before if needed)
+ if (!fExtrapParameters) {
+ fExtrapParameters = new TMatrixD(5,1);
+ fExtrapParameters->Zero();
+ }
+ return *fExtrapParameters;
+ }
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetExtrapParameters(const TMatrixD& extrapParameters)
+{
+ /// Set extrapolated parameters
+ if (fExtrapParameters) *fExtrapParameters = extrapParameters;
+ else fExtrapParameters = new TMatrixD(extrapParameters);
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetExtrapCovariances() const
+{
+ /// Return the extrapolated covariance matrix (create it before if needed)
+ if (!fExtrapCovariances) {
+ fExtrapCovariances = new TMatrixD(5,5);
+ fExtrapCovariances->Zero();
+ }
+ return *fExtrapCovariances;
+ }
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetExtrapCovariances(const TMatrixD& extrapCovariances)
+{
+ /// Set the extrapolated covariance matrix
+ if (fExtrapCovariances) *fExtrapCovariances = extrapCovariances;
+ else fExtrapCovariances = new TMatrixD(extrapCovariances);
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetSmoothParameters() const
+{
+ /// Return the smoothed parameters (create it before if needed)
+ if (!fSmoothParameters) {
+ fSmoothParameters = new TMatrixD(5,1);
+ fSmoothParameters->Zero();
+ }
+ return *fSmoothParameters;
+ }
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetSmoothParameters(const TMatrixD& smoothParameters)
+{
+ /// Set the smoothed parameters
+ if (fSmoothParameters) *fSmoothParameters = smoothParameters;
+ else fSmoothParameters = new TMatrixD(smoothParameters);
+}
+
+ //__________________________________________________________________________
+const TMatrixD& AliMUONTrackParam::GetSmoothCovariances() const
+{
+ /// Return the smoothed covariance matrix (create it before if needed)
+ if (!fSmoothCovariances) {
+ fSmoothCovariances = new TMatrixD(5,5);
+ fSmoothCovariances->Zero();
+ }
+ return *fSmoothCovariances;
+ }
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::SetSmoothCovariances(const TMatrixD& smoothCovariances)
+{
+ /// Set the smoothed covariance matrix
+ if (fSmoothCovariances) *fSmoothCovariances = smoothCovariances;
+ else fSmoothCovariances = new TMatrixD(smoothCovariances);
+}
+
+//__________________________________________________________________________
+void AliMUONTrackParam::SetClusterPtr(AliMUONVCluster* cluster, Bool_t owner)
+{
+ /// set pointeur to associated cluster
+ if (fOwnCluster) delete fClusterPtr;
+ fClusterPtr = cluster;
+ fOwnCluster = owner;
+}
+
+ //__________________________________________________________________________
+Int_t AliMUONTrackParam::Compare(const TObject* trackParam) const
+{
+ /// "Compare" function to sort with decreasing Z (spectro. muon Z <0).
+ /// Returns 1 (0, -1) if the current Z
+ /// is smaller than (equal to, larger than) Z of trackParam
+ if (fZ < ((AliMUONTrackParam*)trackParam)->GetZ()) return(1);
+ else if (fZ == ((AliMUONTrackParam*)trackParam)->GetZ()) return(0);
+ else return(-1);
+}
+
+ //__________________________________________________________________________
+Bool_t AliMUONTrackParam::CompatibleTrackParam(const AliMUONTrackParam &trackParam, Double_t sigma2Cut, Double_t &chi2) const
+{
+ /// Return kTRUE if the two set of track parameters are compatible within sigma2Cut
+ /// Set chi2 to the compatible chi2 value
+ /// Note that parameter covariances must exist for at least one set of parameters
+ /// Note also that if parameters are not given at the same Z, results will be meaningless
+
+ // reset chi2 value
+ chi2 = 0.;
+
+ // ckeck covariance matrices
+ if (!fCovariances && !trackParam.fCovariances) {
+ AliError("Covariance matrix must exist for at least one set of parameters");
+ return kFALSE;
+ }
+
+ Double_t maxChi2 = 5. * sigma2Cut * sigma2Cut; // 5 degrees of freedom
+
+ // check Z parameters
+ if (fZ != trackParam.fZ)
+ AliWarning(Form("Parameters are given at different Z position (%e : %e): results are meaningless", fZ, trackParam.fZ));
+
+ // compute the parameter residuals
+ TMatrixD deltaParam(fParameters, TMatrixD::kMinus, trackParam.fParameters);
+
+ // build the error matrix
+ TMatrixD weight(5,5);
+ if (fCovariances) weight += *fCovariances;
+ if (trackParam.fCovariances) weight += *(trackParam.fCovariances);
+
+ // invert the error matrix to get the parameter weights if possible
+ if (weight.Determinant() == 0) {
+ AliError("Cannot compute the compatibility chi2");
+ return kFALSE;
+ }
+ weight.Invert();
+
+ // compute the compatibility chi2
+ TMatrixD tmp(deltaParam, TMatrixD::kTransposeMult, weight);
+ TMatrixD mChi2(tmp, TMatrixD::kMult, deltaParam);
+
+ // set chi2 value
+ chi2 = mChi2(0,0);
+
+ // check compatibility
+ if (chi2 > maxChi2) return kFALSE;
+
+ return kTRUE;
+}
+
+ //__________________________________________________________________________
+void AliMUONTrackParam::Print(Option_t* opt) const
+{
+ /// Printing TrackParam information
+ /// "full" option for printing all the information about the TrackParam
+ TString sopt(opt);
+ sopt.ToUpper();
+
+ if ( sopt.Contains("FULL") ) {
+ cout << "<AliMUONTrackParam> Bending P=" << setw(5) << setprecision(3) << 1./fParameters(4,0) <<
+ ", NonBendSlope=" << setw(5) << setprecision(3) << fParameters(1,0)*180./TMath::Pi() <<
+ ", BendSlope=" << setw(5) << setprecision(3) << fParameters(3,0)*180./TMath::Pi() <<
+ ", (x,y,z)_IP=(" << setw(5) << setprecision(3) << fParameters(0,0) <<
+ "," << setw(5) << setprecision(3) << fParameters(2,0) <<
+ "," << setw(5) << setprecision(3) << fZ <<
+ ") cm, (px,py,pz)=(" << setw(5) << setprecision(3) << Px() <<
+ "," << setw(5) << setprecision(3) << Py() <<
+ "," << setw(5) << setprecision(3) << Pz() << ") GeV/c"
+ "," << "local chi2=" << GetLocalChi2() << endl;
+ }
+ else {
+ cout << "<AliMUONTrackParam>" << endl;
+ }
+
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff