- We now save in MUON ESD both track parameters extrapolated to vertex

[u/mrichter/AliRoot.git] / MUON / AliMUONTrackParam.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

///////////////////////////////////////////////////
//
// Track parameters
// in
// ALICE
// dimuon
// spectrometer
//
///////////////////////////////////////////////////

#include "AliMUONTrackParam.h"
#include "AliMUONHitForRec.h"

#include "AliESDMuonTrack.h"
#include "AliLog.h"

#include <Riostream.h>
#include <TMath.h>
#include <TMatrixD.h>

/// \cond CLASSIMP
ClassImp(AliMUONTrackParam) // Class implementation in ROOT context
/// \endcond

  //_________________________________________________________________________
AliMUONTrackParam::AliMUONTrackParam()
  : TObject(),
    fNonBendingCoor(0.),
    fNonBendingSlope(0.),
    fBendingCoor(0.),
    fBendingSlope(0.),
    fInverseBendingMomentum(0.),
    fZ(0.),
    fCovariances(0x0),
    fHitForRecPtr(0x0)
{
  /// Constructor
}

  //_________________________________________________________________________
AliMUONTrackParam::AliMUONTrackParam(const AliMUONTrackParam& theMUONTrackParam)
  : TObject(theMUONTrackParam),
    fNonBendingCoor(theMUONTrackParam.fNonBendingCoor),
    fNonBendingSlope(theMUONTrackParam.fNonBendingSlope),
    fBendingCoor(theMUONTrackParam.fBendingCoor),
    fBendingSlope(theMUONTrackParam.fBendingSlope),
    fInverseBendingMomentum(theMUONTrackParam.fInverseBendingMomentum), 
    fZ(theMUONTrackParam.fZ),
    fCovariances(0x0),
    fHitForRecPtr(theMUONTrackParam.fHitForRecPtr)
{
  /// Copy constructor
  if (theMUONTrackParam.fCovariances) fCovariances = new TMatrixD(*(theMUONTrackParam.fCovariances));
}

  //_________________________________________________________________________
AliMUONTrackParam& AliMUONTrackParam::operator=(const AliMUONTrackParam& theMUONTrackParam)
{
  /// Asignment operator
  if (this == &theMUONTrackParam)
    return *this;

  // base class assignement
  TObject::operator=(theMUONTrackParam);

  fNonBendingCoor         	=  theMUONTrackParam.fNonBendingCoor;
  fNonBendingSlope        	=  theMUONTrackParam.fNonBendingSlope; 
  fBendingCoor            	=  theMUONTrackParam.fBendingCoor; 
  fBendingSlope           	=  theMUONTrackParam.fBendingSlope; 
  fInverseBendingMomentum 	=  theMUONTrackParam.fInverseBendingMomentum; 
  fZ                      	=  theMUONTrackParam.fZ; 
  
  if (theMUONTrackParam.fCovariances) {
    if (fCovariances) *fCovariances = *(theMUONTrackParam.fCovariances);
    else fCovariances = new TMatrixD(*(theMUONTrackParam.fCovariances));
  } else if (fCovariances) {
    delete fCovariances;
    fCovariances = 0x0;
  }
  
  return *this;
}

  //__________________________________________________________________________
AliMUONTrackParam::~AliMUONTrackParam()
{
/// Destructor
  DeleteCovariances();
}

  //__________________________________________________________________________
void AliMUONTrackParam::SetTrackParam(AliMUONTrackParam& theMUONTrackParam)
{
  /// Set track parameters from "TrackParam" leaving pointer to fHitForRecPtr and parameter covariances unchanged
  fNonBendingCoor         	=  theMUONTrackParam.fNonBendingCoor;
  fNonBendingSlope        	=  theMUONTrackParam.fNonBendingSlope; 
  fBendingCoor            	=  theMUONTrackParam.fBendingCoor; 
  fBendingSlope           	=  theMUONTrackParam.fBendingSlope; 
  fInverseBendingMomentum 	=  theMUONTrackParam.fInverseBendingMomentum; 
  fZ                      	=  theMUONTrackParam.fZ; 
}

  //__________________________________________________________________________
AliMUONHitForRec* AliMUONTrackParam::GetHitForRecPtr(void) const
{
/// return pointer to HitForRec attached to the current TrackParam
/// this method should not be called when fHitForRecPtr == NULL
  if (!fHitForRecPtr) AliWarning("fHitForRecPtr == NULL");
  return fHitForRecPtr;
}

  //_________________________________________________________________________
void AliMUONTrackParam::GetParamFrom(const AliESDMuonTrack& esdMuonTrack)
{
  /// assigned value form ESD track.
  fInverseBendingMomentum 	=  esdMuonTrack.GetInverseBendingMomentum();
  fBendingSlope           	=  TMath::Tan(esdMuonTrack.GetThetaY());
  fNonBendingSlope        	=  TMath::Tan(esdMuonTrack.GetThetaX());
  fZ                     	=  esdMuonTrack.GetZ(); 
  fBendingCoor            	=  esdMuonTrack.GetBendingCoor(); 
  fNonBendingCoor         	=  esdMuonTrack.GetNonBendingCoor();
}

  //_________________________________________________________________________
void AliMUONTrackParam::SetParamFor(AliESDMuonTrack& esdMuonTrack)
{
  /// assigned value form ESD track.
  esdMuonTrack.SetInverseBendingMomentum(fInverseBendingMomentum);
  esdMuonTrack.SetThetaX(TMath::ATan(fNonBendingSlope));
  esdMuonTrack.SetThetaY(TMath::ATan(fBendingSlope));
  esdMuonTrack.SetZ(fZ); 
  esdMuonTrack.SetBendingCoor(fBendingCoor); 
  esdMuonTrack.SetNonBendingCoor(fNonBendingCoor);
}

  //_________________________________________________________________________
void AliMUONTrackParam::GetParamFromUncorrected(const AliESDMuonTrack& esdMuonTrack)
{
  /// assigned value form ESD track.
  fInverseBendingMomentum 	=  esdMuonTrack.GetInverseBendingMomentumUncorrected();
  fBendingSlope           	=  TMath::Tan(esdMuonTrack.GetThetaYUncorrected());
  fNonBendingSlope        	=  TMath::Tan(esdMuonTrack.GetThetaXUncorrected());
  fZ                     	=  esdMuonTrack.GetZUncorrected(); 
  fBendingCoor            	=  esdMuonTrack.GetBendingCoorUncorrected(); 
  fNonBendingCoor         	=  esdMuonTrack.GetNonBendingCoorUncorrected();
}

  //_________________________________________________________________________
void AliMUONTrackParam::SetParamForUncorrected(AliESDMuonTrack& esdMuonTrack)
{
  /// assigned value form ESD track.
  esdMuonTrack.SetInverseBendingMomentumUncorrected(fInverseBendingMomentum);
  esdMuonTrack.SetThetaXUncorrected(TMath::ATan(fNonBendingSlope));
  esdMuonTrack.SetThetaYUncorrected(TMath::ATan(fBendingSlope));
  esdMuonTrack.SetZUncorrected(fZ); 
  esdMuonTrack.SetBendingCoorUncorrected(fBendingCoor); 
  esdMuonTrack.SetNonBendingCoorUncorrected(fNonBendingCoor);
}

  //__________________________________________________________________________
Double_t AliMUONTrackParam::Px() const
{
  /// return p_x from track parameters
  Double_t pYZ, pZ, pX;
  pYZ = 0;
  if (  TMath::Abs(fInverseBendingMomentum) > 0 )
    pYZ = TMath::Abs(1.0 / fInverseBendingMomentum);
  pZ = -pYZ / (TMath::Sqrt(1.0 + fBendingSlope * fBendingSlope));  // spectro. (z<0)
  pX = pZ * fNonBendingSlope; 
  return pX;
}

  //__________________________________________________________________________
Double_t AliMUONTrackParam::Py() const
{
  /// return p_y from track parameters
  Double_t pYZ, pZ, pY;
  pYZ = 0;
  if (  TMath::Abs(fInverseBendingMomentum) > 0 )
    pYZ = TMath::Abs(1.0 / fInverseBendingMomentum);
  pZ = -pYZ / (TMath::Sqrt(1.0 + fBendingSlope * fBendingSlope));  // spectro. (z<0)
  pY = pZ * fBendingSlope; 
  return pY;
}

  //__________________________________________________________________________
Double_t AliMUONTrackParam::Pz() const
{
  /// return p_z from track parameters
  Double_t pYZ, pZ;
  pYZ = 0;
  if (  TMath::Abs(fInverseBendingMomentum) > 0 )
    pYZ = TMath::Abs(1.0 / fInverseBendingMomentum);
  pZ = -pYZ / (TMath::Sqrt(1.0 + fBendingSlope * fBendingSlope));  // spectro. (z<0)
  return pZ;
}

  //__________________________________________________________________________
Double_t AliMUONTrackParam::P() const
{
  /// return p from track parameters
  Double_t  pYZ, pZ, p;
  pYZ = 0;
  if (  TMath::Abs(fInverseBendingMomentum) > 0 )
    pYZ = TMath::Abs(1.0 / fInverseBendingMomentum);
  pZ = -pYZ / (TMath::Sqrt(1.0 + fBendingSlope * fBendingSlope));  // spectro. (z<0)
  p = TMath::Abs(pZ) * 
    TMath::Sqrt(1.0 + fBendingSlope * fBendingSlope + fNonBendingSlope * fNonBendingSlope);
  return p;
  
}

  //__________________________________________________________________________
TMatrixD* AliMUONTrackParam::GetCovariances()
{
  /// Return the covariance matrix (create it before if needed)
  if (!fCovariances) {
    fCovariances = new TMatrixD(5,5);
    (*fCovariances) = 0;
  }
  return fCovariances;
  }

  //__________________________________________________________________________
void AliMUONTrackParam::SetCovariances(TMatrixD* covariances)
{
  /// Set the covariance matrix
  if (covariances == fCovariances) return; // nothing to be done
  if (fCovariances) *fCovariances = *covariances;
  else fCovariances = new TMatrixD(*covariances);
}

  //__________________________________________________________________________
void AliMUONTrackParam::SetCovariances(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(Double_t matrix[5][5])
{
  /// Set the diagonal terms of the covariance matrix (variances)
  if (!fCovariances) fCovariances = new TMatrixD(5,5);
  (*fCovariances) = 0;
  for (Int_t i=0; i<5; i++) (*fCovariances)(i,i) = matrix[i][i];
}

  //__________________________________________________________________________
void AliMUONTrackParam::DeleteCovariances()
{
  /// Delete the covariance matrix
  if (fCovariances) delete fCovariances;
  fCovariances = 0x0;
}

  //__________________________________________________________________________
void AliMUONTrackParam::EvalCovariances(AliMUONHitForRec* hit2)
{
  /// Evaluate covariances assuming the track is only a straight line
  /// between the HitForRec attached to the current TrackParam and hit2.
  /// Nothing can be done on fInverseBendingMomentum (-> 50% err).
  
  // Allocate memory if needed
  if (!fCovariances) fCovariances = new TMatrixD(5,5);
  
  // Reset the covariance matrix
  (*fCovariances) = 0;
  
  if (!fHitForRecPtr) {
    AliWarning("fHitForRecPtr == NULL: cannot calculate TrackParam covariances");
    return;
  }
  
  Double_t dz = fHitForRecPtr->GetZ() - hit2->GetZ();
  
  // Non bending plane
  (*fCovariances)(0,0) = fHitForRecPtr->GetNonBendingReso2();
  (*fCovariances)(0,1) = fHitForRecPtr->GetNonBendingReso2() / dz;
  (*fCovariances)(1,0) = (*fCovariances)(0,1);
  (*fCovariances)(1,1) = ( fHitForRecPtr->GetNonBendingReso2() + hit2->GetNonBendingReso2() ) / dz / dz;
  // Bending plane
  (*fCovariances)(2,2) = fHitForRecPtr->GetBendingReso2();
  (*fCovariances)(2,3) = fHitForRecPtr->GetBendingReso2() / dz;
  (*fCovariances)(3,2) = (*fCovariances)(2,3);
  (*fCovariances)(3,3) = ( fHitForRecPtr->GetBendingReso2() + hit2->GetBendingReso2() ) / dz / dz;
  // Inverse bending momentum
  (*fCovariances)(4,4) = 0.5*fInverseBendingMomentum * 0.5*fInverseBendingMomentum; // error 50%
  
}

  //__________________________________________________________________________
Int_t AliMUONTrackParam::Compare(const TObject* trackParam) const
{
  /// "Compare" function to sort with decreasing Z (spectro. muon Z <0).
  /// Returns 1 (0, -1) if Z of current TrackHit
  /// is smaller than (equal to, larger than) Z of TrackHit
  if (fHitForRecPtr) {
    if (fHitForRecPtr->GetZ() != fZ)
      AliWarning("track parameters are given at a different z position than the one of the corresponding hit");
  }
  if (fZ < ((AliMUONTrackParam*)trackParam)->GetZ()) return(1);
  else if (fZ == ((AliMUONTrackParam*)trackParam)->GetZ()) return(0);
  else return(-1);
}

//_____________________________________________-
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./GetInverseBendingMomentum() << 
      ", NonBendSlope=" << setw(5) << setprecision(3)  << GetNonBendingSlope()*180./TMath::Pi() <<
      ", BendSlope=" << setw(5) << setprecision(3)     << GetBendingSlope()*180./TMath::Pi()  << 
      ", (x,y,z)_IP=(" <<  setw(5) << setprecision(3) << GetNonBendingCoor() <<
      "," <<  setw(5) << setprecision(3) << GetBendingCoor() <<
      "," <<  setw(5) << setprecision(3) << GetZ() <<
      ") cm, (px,py,pz)=(" << setw(5) << setprecision(3) << Px() <<
      "," << setw(5) << setprecision(3) << Py() <<
      "," << setw(5) << setprecision(3) << Pz() << ") GeV/c" << endl;
  }
  else {
    cout << "<AliMUONTrackParam>"  << endl;
  }
    
}