CbmStsTrack.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                      CbmStsTrack header file                  -----
// -----                  Created 26/01/05  by V. Friese               -----
// -------------------------------------------------------------------------
 
#ifndef CBMSTSTRACK_H
#define CBMSTSTRACK_H 1
 
#include "CbmBaseHit.h"
#include "FairTrackParam.h"
#include "TArrayI.h"
#include "TClonesArray.h"
#include "TObject.h"
 
#include <map>
 
class FairHit;
 
class CbmStsTrack : public TObject
{
 
  public:
    CbmStsTrack();
 
    virtual ~CbmStsTrack();
 
    void AddStsHit(Int_t hitIndex, FairHit* hit);
 
    // SM
    void AddPixelHit(Int_t hitIndex, CbmBaseHit* hit);
 
    void AddMvdHit(Int_t hitIndex, FairHit* mvdHit);
 
    void Print();
 
    void SortHits();
 
    void SortMvdHits();
 
    Int_t GetNStsHits() const { return fStsHits.GetSize(); }
    Int_t GetNMvdHits() const { return fMvdHits.GetSize(); }
    Int_t GetStsHitIndex(Int_t iHit) const { return fStsHits.At(iHit); }
    Int_t GetMvdHitIndex(Int_t iHit) const { return fMvdHits.At(iHit); }
    Int_t GetPidHypo() const { return fPidHypo; }
    Int_t GetFlag() const { return fFlag; }
    Double_t GetChi2() const { return fChi2; }
    Int_t GetNDF() const { return fNDF; }
    Double_t GetB() const { return fB; }
    FairTrackParam* GetParamFirst() { return &fParamFirst; }
    FairTrackParam* GetParamLast() { return &fParamLast; }
    TArrayI* GetStsHits() { return &fStsHits; }
    Int_t GetStsEv() const { return fnEv; }
    TClonesArray* GetStsHitArr() { return &fHitsArr; }
    Double_t GetPt() const;                        // AZ-260124
    Double_t GetEta() const;                       // AZ-260124
    int FromVertex() const { return fFlag & 2; }   // AZ-200825 - is from primary vertex?
 
    void SetTrkID(int id) { ststrk = id; };
    int GetTrkID() { return ststrk; };
 
    void SetPidHypo(Int_t pid) { fPidHypo = pid; }
    void SetParamFirst(FairTrackParam& par) { fParamFirst = par; }
    void SetParamLast(FairTrackParam& par) { fParamLast = par; }
    void SetFlag(Int_t flag) { fFlag = flag; }
    void SetChi2(Double_t chi2) { fChi2 = chi2; }
    void SetNDF(Int_t ndf) { fNDF = ndf; }
    void SetB(Double_t b) { fB = b; }
    void SetStsHits(TArrayI& hitsArr) { fStsHits = hitsArr; }
    void SetStsEv(Int_t evNo) { fnEv = evNo; }
    void SetStsHitArr(TClonesArray& hitar) { fHitsArr = hitar; }
 
  private:
    int ststrk;
    TArrayI fStsHits;
 
    TArrayI fMvdHits;
 
    Int_t fPidHypo;
 
    FairTrackParam fParamFirst;
    FairTrackParam fParamLast;
 
    Int_t fFlag;
 
    Double32_t fChi2;
    Int_t fNDF;
 
    Double32_t fB;
 
    std::map<Double_t, Int_t> fStsHitMap;   
    std::map<Double_t, Int_t> fMvdHitMap;   
 
    Int_t fnEv;
    TClonesArray fHitsArr;
 
    ClassDef(CbmStsTrack, 6);
};
 
#endif