InttSeedTrackPerformance-Jingyu/src/SPHTracKuma.h

0001 /*
0002   Kuma track for RHIC-sPHENIX
0003 */
0004 #ifndef  SPHTRACKUMA_H
0005 #define  SPHTRACKUMA_H
0006
0007 #include <Rtypes.h>
0008 class tracKuma
0009 {
0010   public:
0011     tracKuma() :
0012       m_electronIs(false),
0013       m_charge(0),
0014       m_phi(0),
0015       m_theta(0),
0016       m_energy(0),
0017       m_p(0),
0018       m_pt(0),
0019       m_saggitaR(0),
0020       m_saggitaX(0),
0021       m_saggitaY(0),
0022       m_refPhi(0)
0023   {
0024     // 0: vertex, 1-3: MVTX, 4-5: INTT, 6: Cal
0025     for(unsigned int layId = 0; layId < 7; layId++){
0026       m_hitIs[layId] = false;
0027       m_hitR[layId] = 0.;
0028       m_hitZ[layId] = 0.;
0029       m_hitPhi[layId] = 0.;
0030       m_hitTheta[layId] = 0.;
0031     }
0032   };
0033   ~tracKuma() {};
0034     void setHitIs(Int_t layId, Bool_t is)
0035     {
0036         m_hitIs[layId] = is;
0037     }
0038     void setHitR(Int_t layId, Double_t r)
0039     {
0040         m_hitR[layId] = r;
0041     }
0042     void setHitZ(Int_t layId, Double_t z)
0043     {
0044         m_hitZ[layId] = z;
0045     }
0046     void setHitPhi(Int_t layId, Double_t phi)
0047     {
0048         m_hitPhi[layId] = phi;
0049     }
0050     void setHitTheta(Int_t layId, Double_t theta)
0051     {
0052         m_hitTheta[layId] = theta;
0053     }
0054
0055     void setTrackElectronIs(bool electronIs)
0056     {
0057         m_electronIs = electronIs;
0058     }
0059     void setTrackCharge(Int_t pCharge)
0060     {
0061         m_charge = pCharge;
0062     }
0063     void setTrackPhi(Double_t phi)
0064     {
0065         m_phi = phi;
0066     }
0067     void setTrackTheta(Double_t theta)
0068     {
0069         m_theta = theta;
0070     }
0071     void setTrackE(Double_t energy)
0072     {
0073         m_energy = energy;
0074     }
0075     void setTrackP(Double_t p)
0076     {
0077         m_p = p;
0078     }
0079     void setTrackPt(Double_t pt)
0080     {
0081         m_pt = pt;
0082     }
0083     void setTrackSagR(Double_t sagittaR)
0084     {
0085         m_saggitaR = sagittaR;
0086     }
0087     void setTrackSagX(Double_t sagittaX)
0088     {
0089         m_saggitaX= sagittaX;
0090     }
0091     void setTrackSagY(Double_t sagittaY)
0092     {
0093         m_saggitaY = sagittaY;
0094     }
0095     void setRefPhi(Double_t refPhi)
0096     {
0097         m_refPhi = refPhi;
0098     }
0099
0100     Bool_t getHitIs(Int_t layId){
0101         return m_hitIs[layId];
0102     }
0103     Double_t getHitR(Int_t layId){
0104         return m_hitR[layId];
0105     }
0106     Double_t getHitZ(Int_t layId){
0107         return m_hitZ[layId];
0108     }
0109     Double_t getHitPhi(Int_t layId){
0110         return m_hitPhi[layId];
0111     }
0112     Double_t getHitTheta(Int_t layId)
0113     {
0114         return m_hitTheta[layId];
0115     }
0116
0117     bool getTrackElectronIs(){
0118         return m_electronIs;
0119     }
0120     Int_t getTrackCharge(){
0121         return m_charge;
0122     }
0123     Double_t getTrackPhi(){
0124         return m_phi;
0125     }
0126     Double_t getTrackTheta(){
0127         return m_theta;
0128     }
0129     Double_t getTrackE(){
0130         return m_energy;
0131     }
0132     Double_t getTrackPt(){
0133         return m_pt;
0134     }
0135     Double_t getTrackP(){
0136         return m_p;
0137     }
0138     Double_t getTrackSagR(){
0139         return m_saggitaR;
0140     }
0141     Double_t getTrackSagX(){
0142         return m_saggitaX;
0143     }
0144     Double_t getTrackSagY(){
0145         return m_saggitaY;
0146     }
0147     Double_t getTrackRefPhi(){
0148         return m_refPhi;
0149     }
0150
0151 private:
0152     Bool_t m_hitIs[7];
0153     Double_t m_hitR[7];
0154     Double_t m_hitZ[7];
0155     Double_t m_hitPhi[7];
0156     Double_t m_hitTheta[7];
0157     bool     m_electronIs;
0158     Int_t    m_charge;
0159     Double_t m_phi;
0160     Double_t m_theta;
0161     Double_t m_energy;
0162     Double_t m_p;
0163     Double_t m_pt;
0164     Double_t m_saggitaR;
0165     Double_t m_saggitaX;
0166     Double_t m_saggitaY;
0167     Double_t m_refPhi;
0168 };
0169
0170 struct hitStruct{
0171     Double_t r;
0172     Double_t z;
0173     Double_t phi;
0174     Double_t eta;
0175     Double_t pt;
0176     Double_t energy;
0177 };
0178
0179 // -----------------------------------------------------------------------
0180 // typedef std::vector<tracKuma>  KumaTracks;
0181 // ----------------------------------------------------------------------
0182
0183 #endif  // SPHTRACKUMA_H
0184