CbmStsTrack.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      CbmStsTrack source file                  -----
// -----                  Created 26/01/05  by V. Friese               -----
// -------------------------------------------------------------------------
 
#include "CbmStsTrack.h"
 
#include "FairHit.h"
 
#include <iostream>
#include <map>
 
using std::cout;
using std::endl;
using std::map;
 
// -----   Default constructor   -------------------------------------------
CbmStsTrack::CbmStsTrack()
  : TObject(),
    fStsHits(),
    fMvdHits(),
    fPidHypo(0),
    fParamFirst(),
    fParamLast(),
    fFlag(0),
    fChi2(0.),
    fNDF(0),
    fB(0.),
    fStsHitMap(),
    fMvdHitMap(),
    fnEv(),
    fHitsArr("CbmStsHit", 10)
{
}
// -------------------------------------------------------------------------
 
 
 
// -----   Destructor   ----------------------------------------------------
CbmStsTrack::~CbmStsTrack() {
  fStsHitMap.clear();
  fMvdHitMap.clear();
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method AddStsHit   ---------------------------------------
void CbmStsTrack::AddStsHit(Int_t hitIndex, FairHit* hit) {
  fStsHitMap[hit->GetZ()] = hitIndex;
}
// -------------------------------------------------------------------------
 
// -----  SM   -------------------------------------------------------------
void CbmStsTrack::AddPixelHit(Int_t hitIndex, CbmBaseHit* hit) {
  fStsHitMap[hit->GetZ()] = hitIndex;
}
// -------------------------------------------------------------------------
 
 
// -----   Public method AddMvdHit   ---------------------------------------
void CbmStsTrack::AddMvdHit(Int_t hitIndex, FairHit* hit) {
  fMvdHitMap[hit->GetZ()] = hitIndex;
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method Print   -------------------------------------------
void CbmStsTrack::Print() {
  cout << " Number of attached STS hits : "
       << fStsHits.GetSize()  << endl;
  cout << " Number of attached MVD hits : "
       << fMvdHits.GetSize()  << endl;
  fParamFirst.Print();
  cout << " Chi2: " << fChi2 << ", Quality flag " << fFlag << endl;
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method SortHits   ----------------------------------------
void CbmStsTrack::SortHits() {
 
  Int_t index=0;
  map<Double_t, Int_t>::iterator it;
 
  fStsHits.Reset();
  fStsHits.Set(fStsHitMap.size());
  index = 0;
  for (it = fStsHitMap.begin(); it != fStsHitMap.end(); it++) {
    fStsHits[index] = it->second;
    index++;
  }
 
  fMvdHits.Reset();
  fMvdHits.Set(fMvdHitMap.size());
  index = 0;
  for (it = fMvdHitMap.begin(); it != fMvdHitMap.end(); it++) {
    fMvdHits[index] = it->second;
    index++;
  }
 
}
// -------------------------------------------------------------------------
 
// -----   Public method SortMvdHits   ----------------------------------------
void CbmStsTrack::SortMvdHits() {
 
  Int_t index=0;
  map<Double_t, Int_t>::iterator it;
 
  fMvdHits.Reset();
  fMvdHits.Set(fMvdHitMap.size());
  for (it = fMvdHitMap.begin(); it != fMvdHitMap.end(); it++) {
    fMvdHits[index] = it->second;
    index++;
  }
 
}
// -------------------------------------------------------------------------
 
Double_t CbmStsTrack::GetPt() const
{
   //AZ-260124 - compute Pt at first point
 
   Double_t tx = fParamFirst.GetTx();
   Double_t ty = fParamFirst.GetTy();
   Double_t txy2 = tx*tx + ty*ty;
   Double_t pz = TMath::Abs(1/fParamFirst.GetQp()) / TMath::Sqrt(1+txy2);
   Double_t pt = pz * TMath::Sqrt(txy2);
   return pt;
}
// -------------------------------------------------------------------------
 
Double_t CbmStsTrack::GetEta() const
{
   //AZ-260124 - compute Eta at first point
 
   Double_t tx = fParamFirst.GetTx();
   Double_t ty = fParamFirst.GetTy();
   Double_t txy2 = tx*tx + ty*ty;
   Double_t pz = TMath::Abs(1/fParamFirst.GetQp()) / TMath::Sqrt(1+txy2);
   Double_t pt = pz * TMath::Sqrt(txy2);
   TVector3 mom (pt, 0.0, pz);
   return mom.Eta();
}