TpcSimMCEvent.h

Go to the documentation of this file.

#ifndef TPCSIMMCEVENT_H
#define TPCSIMMCEVENT_H

#include "TObject.h"
#include "TVector3.h"
#include "TLorentzVector.h"
//#include "TParticle.h"
#include "TClonesArray.h"
#include "HarpoDebug.h"

#include "TpcSimIonisationTrack.h"

class TpcSimMCTrack : public TObject {

 public:
  TpcSimMCTrack();
  TpcSimMCTrack(const TpcSimMCTrack &tr);
  TpcSimMCTrack(Double_t x0, Double_t y0, Double_t z0, Double_t t0,
                  Double_t px, Double_t py, Double_t pz, 
                  Int_t fPid);
  ~TpcSimMCTrack(){}

  Double_t GetX0() {return fX0;} // X vertex
  Double_t GetY0() {return fY0;} // Y vertex
  Double_t GetZ0() {return fZ0;} // Z vertex
  Double_t GetT0() {return fT0;} // T vertex
  TVector3 GetOrigin() {return TVector3(fX0,fY0,fZ0);} // T vertex
  TLorentzVector GetOrigin4() {return TLorentzVector(fX0,fY0,fZ0,fT0);} // T vertex
  
  Double_t GetPx() {return fPx;} // 
  Double_t GetPy() {return fPy;} //
  Double_t GetPz() {return fPz;} //
  Double_t GetE() {return fE;}  //
  TVector3 GetP() {return TVector3(fPx,fPy,fPz);}  //
  TLorentzVector Get4Vector() {return TLorentzVector(fPx,fPy,fPz,fE);}  //

  Double_t GetPolax() {return fPolax;} // 
  Double_t GetPolay() {return fPolay;} //
  Double_t GetPolaz() {return fPolaz;} //
  TVector3 GetPola() {return TVector3(fPolax,fPolay,fPolaz);} //

  Double_t GetMass() {return fMass;}
  Int_t GetPdgCode() {return fPdgCode;}

  void SetX0(Double_t value) {fX0 = value;} // X vertex
  void SetY0(Double_t value) {fY0 = value;} // Y vertex
  void SetZ0(Double_t value) {fZ0 = value;} // Z vertex
  void SetT0(Double_t value) {fT0 = value;} // T vertex
  
  void SetPx(Double_t value) {fPx = value;} // 
  void SetPy(Double_t value) {fPy = value;} //
  void SetPz(Double_t value) {fPz = value;} //
  void SetE(Double_t value) {fE = value;}  //
  void SetPolax(Double_t value) {fPolax = value;} // 
  void SetPolay(Double_t value) {fPolay = value;} //
  void SetPolaz(Double_t value) {fPolaz = value;} //
  
  void SetMass(Double_t value) {fMass = value;}
  void SetPdgCode(Int_t value) {fPdgCode = value;}

 private:

  Double_t fX0; // X vertex
  Double_t fY0; // Y vertex
  Double_t fZ0; // Z vertex
  Double_t fT0; // T vertex
  
  Double_t fPx; // 
  Double_t fPy; //
  Double_t fPz; //
  Double_t fE;  //
  Double_t fPolax; // 
  Double_t fPolay; //
  Double_t fPolaz; //

  Double_t fMass;
  Int_t fPdgCode; // PDG Code: e-: 11, mu-: 13, pi0: 111, pi+: 211, 
                  //           proton: 2212, gamma: 22

  ClassDef(TpcSimMCTrack,1);
};


class TpcSimMCEvent : public TObject {

 public:
  TpcSimMCEvent(Int_t event = -1);
  virtual ~TpcSimMCEvent();

  /* void GenerateEvent(Int_t seed = 0); */
  TpcSimMCTrack* GetTrack(Int_t i) {
    if(gHarpoDebug>1) 
      Info("GetTrack","%f",((TpcSimMCTrack*)fTracks->At(i))->GetT0());
    if(i<0 || i>=fkMaxNtracks) return 0;
    return (TpcSimMCTrack*)fTracks->At(i);
  }
  void AddTrack(TpcSimMCTrack* tr);
  void AddTrack(TpcSimIonisationTrack* tr);

  Int_t GetNtracks() {return fNtracks;}

 private:

  Int_t fNtracks;
  TClonesArray* fTracks;
  Int_t fEvent;

  static const Int_t fkMaxNtracks = 100;

  ClassDef(TpcSimMCEvent,1);
};

#endif