StChargedPionTrack.h

#ifndef ST_CHARGED_PION_TRACK_HH
#define ST_CHARGED_PION_TRACK_HH

// $Id: StChargedPionTrack.h,v 1.6 2012/11/09 03:31:34 perev Exp $

#include "StPhysicalHelixD.hh"
#include "StThreeVectorF.hh"

#include "TObject.h"
#include "TLorentzVector.h"

class StChargedPionTrack : public TLorentzVector
{
public:
    StChargedPionTrack();
    virtual ~StChargedPionTrack();
    StChargedPionTrack(const StChargedPionTrack& t);
    
    short id() const; 
    short flag() const; 
    
    short vertexIndex() const; 
    StThreeVectorF vertex() const; 
    
    unsigned short nHits() const;     
    unsigned short nHitsPoss() const; 
    unsigned short nHitsDedx() const; 
    unsigned short nHitsFit() const;  
    
    float nSigmaElectron() const;  
    float nSigmaPion() const;      
    float nSigmaKaon() const;      
    float nSigmaProton() const;    
    float dEdx() const; 
    
    float chi2() const;     
    float chi2prob() const; 
    
    float pt() const;   
    float phi() const;  
    float eta() const;  
    
    short charge() const;  
    double B() const; 
    
    //some random stuff
    double length() const; 
    double lengthMeasured() const;  
    
    //global track properties
    StThreeVectorF          globalLastPoint() const; 
    StPhysicalHelixD        globalHelix() const; 
    StThreeVector<double>   globalFirstPoint() const; 
    
    //these are calculated using first measured point
    double                  globalPt() const;
    double                  globalPhi() const;
    double                  globalEta() const;
    TLorentzVector          globalP() const;
    
    //can choose a different point if you like ... specify vertex to recover StMuTrack global kinematics
    double                  globalPt(StThreeVectorF position) const;
    double                  globalPhi(StThreeVectorF position) const;
    double                  globalEta(StThreeVectorF position) const;
    TLorentzVector          globalP(StThreeVectorF position) const;
    
    StThreeVectorF          globalDca() const; 
    StThreeVectorF          globalDca(StThreeVectorF position) const; 
    
    //setters
    void setId(short aId) {mId = aId;}
    void setFlag(short aFlag) {mFlag = aFlag;}
    
    void setVertexIndex(int aVertexIndex) {mVertexIndex = aVertexIndex;}
    void setVertex(StThreeVectorF aPosition) {mVertex = aPosition;}
    
    void setNHits(unsigned short aNHits) {mNHits = aNHits;}
    void setNHitsPoss(unsigned short aNHitsPoss) {mNHitsPoss = aNHitsPoss;}
    void setNHitsDedx(unsigned short aNHitsDedx) {mNHitsDedx = aNHitsDedx;}
    void setNHitsFit(unsigned short aNHitsFit) {mNHitsFit = aNHitsFit;}
    
    void setNSigmaElectron(float aNSigmaElectron) {mNSigmaElectron = aNSigmaElectron;}
    void setNSigmaPion(float aNSigmaPion) {mNSigmaPion = aNSigmaPion;}
    void setNSigmaKaon(float aNSigmaKaon) {mNSigmaKaon = aNSigmaKaon;}
    void setNSigmaProton(float aNSigmaProton) {mNSigmaProton = aNSigmaProton;}
    void setDedx(float aDedx) {mdEdx = aDedx;}
    
    void setChi2(float aChi2) {mChi2 = aChi2;}
    void setChi2prob(float aChi2Prob) {mChi2Prob = aChi2Prob;}
    
    void setPtEtaPhi(float aPt, float aEta, float aPhi);
    void setPt(float aPt);
    void setPhi(float aPhi);
    void setEta(float aEta);
    
    void setCharge(short aCharge) {mCharge = aCharge;}
    void setB(double aMagneticField) {mB = aMagneticField;}
    
    void setGlobalLastPoint(StThreeVectorF aLastPoint) {mGlobalLastPoint = aLastPoint;}
    void setGlobalHelix(StPhysicalHelixD aHelix) {mGlobalHelix = aHelix;}
    
protected:    
    //basic QA -- could also use fBits to store online sanity checks
    Short_t mId;
    Short_t mFlag;
    
    Char_t mVertexIndex;
    StThreeVectorF mVertex; //stored for convenience of DCA calculation

    UChar_t mNHits;           
    UChar_t mNHitsPoss;      
    UChar_t mNHitsDedx;       
    UChar_t mNHitsFit;
    
    Float_t mNSigmaElectron;
    Float_t mNSigmaPion;
    Float_t mNSigmaKaon;
    Float_t mNSigmaProton;
    Float_t mdEdx;
    
    Float_t mChi2;
    Float_t mChi2Prob;
    
    Char_t mCharge;
    Double_t mB;
        
    //global track properties -- most can be recalculated from helix
    StThreeVectorF      mGlobalLastPoint;
    StPhysicalHelixD    mGlobalHelix;
    
    ClassDef(StChargedPionTrack,2)
};

inline short StChargedPionTrack::id() const {return mId;}
inline short StChargedPionTrack::flag() const {return mFlag;}

inline short StChargedPionTrack::vertexIndex() const {return mVertexIndex;}
inline StThreeVectorF StChargedPionTrack::vertex() const {return mVertex;}

inline unsigned short StChargedPionTrack::nHits() const {return mNHits;}
inline unsigned short StChargedPionTrack::nHitsPoss() const {return mNHitsPoss;}
inline unsigned short StChargedPionTrack::nHitsDedx() const {return mNHitsDedx;}
inline unsigned short StChargedPionTrack::nHitsFit() const {return mNHitsFit;}

inline float StChargedPionTrack::nSigmaElectron() const {return mNSigmaElectron;}
inline float StChargedPionTrack::nSigmaPion() const {return mNSigmaPion;}
inline float StChargedPionTrack::nSigmaKaon() const {return mNSigmaKaon;}
inline float StChargedPionTrack::nSigmaProton() const {return mNSigmaProton;}
inline float StChargedPionTrack::dEdx() const {return mdEdx;}

inline float StChargedPionTrack::chi2() const {return mChi2;}
inline float StChargedPionTrack::chi2prob() const {return mChi2Prob;}

inline float StChargedPionTrack::pt() const {return this->Pt();}
inline float StChargedPionTrack::phi() const {return this->Phi();}
inline float StChargedPionTrack::eta() const {return this->Eta();}

inline short StChargedPionTrack::charge() const {return mCharge;}
inline double StChargedPionTrack::B() const {return mB;}

inline StThreeVectorF StChargedPionTrack::globalLastPoint() const {return mGlobalLastPoint;}
inline StPhysicalHelixD StChargedPionTrack::globalHelix() const {return mGlobalHelix;}
inline StThreeVector<double> StChargedPionTrack::globalFirstPoint() const {return mGlobalHelix.origin();}

#endif

/*****************************************************************************
 * $Log: StChargedPionTrack.h,v $
 * Revision 1.6  2012/11/09 03:31:34  perev
 * Cleanup
 *
 * Revision 1.5  2008/12/29 15:58:31  kocolosk
 * removed commented code and added $Id: StChargedPionTrack.h,v 1.6 2012/11/09 03:31:34 perev Exp $/$Log: StChargedPionTrack.h,v $
 * removed commented code and added $Id$/Revision 1.6  2012/11/09 03:31:34  perev
 * removed commented code and added $Id$/Cleanup
 * removed commented code and added $Id$/ as needed
 *
 *****************************************************************************/