StPicoMcTrack.h

#ifndef StPicoMcTrack_h
#define StPicoMcTrack_h

// C++ headers
#include <vector>

// ROOT headers
#include "TObject.h"
#include "TVector3.h"
#include "TLorentzVector.h"

// PicoDst headers
#include "StPicoHelix.h"
#include "StPicoPhysicalHelix.h"

#if defined (_VANILLA_ROOT_)
#include "SystemOfUnits.h"
#include "PhysicalConstants.h"
#else
#include "StarClassLibrary/SystemOfUnits.h"
#include "StarClassLibrary/PhysicalConstants.h"
#endif

//_________________
class StPicoMcTrack : public TObject {

 public:

  StPicoMcTrack();
  StPicoMcTrack(const StPicoMcTrack &track);
  virtual ~StPicoMcTrack();
  virtual void Print(const Char_t *option = "") const;

  enum EHIT {ktpc, ksvt, kssd, kctb, keem, kemc, kesm, 
             kftp, kgem, khpd, kist, kigt, kfst, kfgt, 
             kfpd, kmwc, kpgc, kpmd, ksmd, kpix, ktof, 
             kvpd, ketr, khca, kfts, keto, kstg, kwca, 
             kpre, kepd, ktot};

  //
  // Getters
  //

  Int_t id() const                    { return mId; }
  Int_t pdgId() const;
  Int_t geantId() const               { return (Int_t)mGePid; }
  Int_t charge() const                { return (Int_t)mCharge; }
  TVector3 p() const                  { return TVector3(mPx, mPy, mPz); }
  Float_t pt() const                  { return p().Perp(); }
  Float_t ptot() const                { return p().Mag(); }
  Float_t eta() const                 { return p().PseudoRapidity(); }  
  TLorentzVector fourMomentum() const { return TLorentzVector(mPx,mPy,mPz,mE); }
  Float_t energy() const              { return mE; }
  Float_t rapidity() const            { return fourMomentum().Rapidity(); }
  Bool_t isFromShower() const         { return mIsFromShower; }
  Int_t idVtxStart() const            { return mIdVtxStart; }
  Int_t idVtxStop() const             { return mIdVtxStop; }
  Int_t idVtxItrmd() const            { return mIdVtxItrmd; }
  Int_t nHits() const               
  { Int_t n=0; for (Int_t i=ktpc; i<ktot; i++) n+=nHits(i); return n; }
  UChar_t nHits(Int_t k) const        { return mHits[k]; }
  UChar_t nHitsCtb() const            { return mHits[kctb]; }
  UChar_t nHitsEEmc() const           { return mHits[keem]; }
  UChar_t nHitsBEmc() const           { return mHits[kemc]; }
  UChar_t nHitsEsm() const            { return mHits[kesm]; }
  UChar_t nHitsFtpc() const           { return mHits[kftp]; }
  UChar_t nHitsGem() const            { return mHits[kgem]; }
  UChar_t nHitsHpd() const            { return mHits[khpd]; }
  UChar_t nHitsIst() const            { return mHits[kist]; }
  UChar_t nHitsIgt() const            { return mHits[kigt]; }
  UChar_t nHitsFst() const            { return mHits[kfst]; }
  UChar_t nHitsFgt() const            { return mHits[kfgt]; }
  UChar_t nHitsFpd() const            { return mHits[kfpd]; }
  UChar_t nHitsMwc() const            { return mHits[kmwc]; }
  UChar_t nHitsPgc() const            { return mHits[kpgc]; }
  UChar_t nHitsPmd() const            { return mHits[kpmd]; }
  UChar_t nHitsBsmd() const           { return mHits[ksmd]; }
  UChar_t nHitsSsd() const            { return mHits[kssd]; }
  UChar_t nHitsSvt() const            { return mHits[ksvt]; }
  UChar_t nHitsPxl() const            { return mHits[kpix]; }
  UChar_t nHitsTof() const            { return mHits[ktof]; }
  UChar_t nHitsTpc() const            { return mHits[ktpc]; }
  UChar_t nHitsVpd() const            { return mHits[kvpd]; }
  UChar_t nHitsEtr() const            { return mHits[ketr]; }
  UChar_t nHitsHca() const            { return mHits[khca]; }
  UChar_t nHitsFts() const            { return mHits[kfts]; }
  UChar_t nHitsEto() const            { return mHits[keto]; }
  UChar_t nHitsStg() const            { return mHits[kstg]; }
  UChar_t nHitsWca() const            { return mHits[kwca]; }
  UChar_t nHitsPre() const            { return mHits[kpre]; }
  UChar_t nHitsEpd() const            { return mHits[kepd]; }
  const Char_t *geName();
  Int_t correctGePid(Int_t id);

  //
  // Setters
  //

  void setId(Int_t id);
  void setGeantId(Int_t id)           { mGePid = id; }
  void setCharge(Int_t charge)        { mCharge = (Char_t)charge; }
  void setNHits(Int_t k, UChar_t n)     { mHits[k] = n; }
  void setP(Float_t px, Float_t py, Float_t pz) { mPx = px; mPy = py; mPz = pz; }
  void setE(Float_t e)                { mE = e; }
  void setIsFromShower(Bool_t isFrom) { mIsFromShower = isFrom; }
  void setIdVtxStart(Int_t id)        { mIdVtxStart = (Short_t)id; }
  void setIdVtxStop(Int_t id)         { mIdVtxStop = (Short_t)id; }
  void setIdVtxItrmd(Int_t id)        { mIdVtxItrmd = (Short_t)id; }


 private:
  
  UShort_t mId;
  Int_t mGePid;
  Char_t mCharge;
  UChar_t mHits[ktot];
  Float_t mPx;
  Float_t mPy;
  Float_t mPz;
  Float_t mE;
  Bool_t mIsFromShower;
  Short_t mIdVtxStart;
  Short_t mIdVtxStop;
  Short_t mIdVtxItrmd;

  ClassDef(StPicoMcTrack, 1)
};

#endif // #define StPicoMcTrack_h