BmnTrack.h

Go to the documentation of this file.

/*
 * Base class to store BMN tracks
 * Author: Sergei Merts
 */
 
#ifndef BMNTRACK_H
#define BMNTRACK_H
 
#include "FairTrackParam.h"
#include "TArrayI.h"
#include "TObject.h"
#include "FairHit.h"
#include <TClonesArray.h>
#include <map>
 
using namespace std;
 
class BmnTrack : public TObject {
public:
 
    BmnTrack();
 
    virtual ~BmnTrack();
 
    void AddHit(Int_t hitIndex, FairHit* Hit);
 
    void Print();
 
    void SortHits();
    void DeleteHits() {fHitMap.clear(); fNhits=0;};
 
    Int_t GetNHits() const {
        return fNhits;
    }
 
    Int_t GetHitIndex(Int_t iHit) const {
        return fHits.At(iHit);
    }
 
    Int_t GetFlag() const {
        return fFlag;
    }
 
    Float_t GetChi2() const {
        return fChi2;
    }
 
    Int_t GetNDF() const {
        return fNDF;
    }
 
    Float_t GetB() const {
        return fB;
    }
 
    Float_t GetLength() const {
        return fLength;
    }
 
    FairTrackParam* GetParamFirst() {
        return &fParamFirst;
    }
 
    FairTrackParam* GetParamLast() {
        return &fParamLast;
    }
    
    Double_t GetP() {
        return 1.0 / fParamFirst.GetQp();
    }
 
    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 SetNHits(Int_t n) {
        fNhits = n;
    }
    
    void SetB(Double_t b) {
        fB = b;
    }
 
    void SetLength(Double_t length) {
        fLength = length;
    }
    
    Bool_t IsUsed() {
        return fUsing;
    }
    
    void SetUsing(Bool_t flag) {
        fUsing = flag;
    }
 
private:
 
    TArrayI fHits;
 
    FairTrackParam fParamFirst;
    FairTrackParam fParamLast;
 
    Int_t fFlag; // 13 for tracks not used in vertex finder
 
    Float_t fChi2;
    Int_t fNDF;
 
    Float_t fB;
 
    Float_t fLength;
    
    Int_t fNhits;
 
    map<Float_t, Int_t> fHitMap; 
    
    Bool_t fUsing;
 
    ClassDef(BmnTrack, 2);
 
};
 
#endif /* BMNTRACK_H */