TpcSimMCEvent.cxx

Go to the documentation of this file.

#include "TParticlePDG.h"
#include "TDatabasePDG.h"
#include "TClonesArray.h"
#include "TMath.h"

#include "TpcSimMCEvent.h"
//#include "TpcSimTrack.h"

ClassImp(TpcSimMCEvent)


TpcSimMCTrack::TpcSimMCTrack() :
TObject()
{
  fX0 = 0;
  fY0 = 0;
  fZ0 = 0;
  fT0 = 0;

  fPx = 0;
  fPy = 0;
  fPz = 0;

  fPolax = 0;
  fPolay = 0;
  fPolaz = 0;

  fE = 0;
  fMass = 0;
  fPdgCode = 0;
}

TpcSimMCTrack::TpcSimMCTrack(const TpcSimMCTrack &tr) :
  TObject(tr), fX0(tr.fX0), fY0(tr.fY0), fZ0(tr.fZ0), fT0(tr.fT0), fPx(tr.fPx), fPy(tr.fPy), fPz(tr.fPz), fE(tr.fE), fPolax(tr.fPolax), fPolay(tr.fPolay), fPolaz(tr.fPolaz), fMass(tr.fMass), fPdgCode(tr.fPdgCode)
{
  //  fX0 = tr.GetX0();
  //  fY0 = tr.GetY0();
  // fZ0 = tr.GetZ0();
  // fT0 = tr.GetT0();

  // fPx = tr.GetPx();
  // fPy = tr.GetPy();
  // fPz = tr.GetPz();

  // fE = tr.GetE();
  // fMass = tr.GetMass();
  // fPdgCode = tr.GetPdgCode();
}

TpcSimMCTrack::TpcSimMCTrack(Double_t x0, Double_t y0, Double_t z0, Double_t t0,
                                 Double_t px, Double_t py, Double_t pz, 
                                 Int_t pid) :
TObject()
{
  fX0 = x0;
  fY0 = y0;
  fZ0 = z0;
  fT0 = t0;

  fPx = px;
  fPy = py;
  fPz = pz;

  fPolax = 0;
  fPolay = 0;
  fPolaz = 0;

  fPdgCode = pid;
  //  fParticlePDG 
  TParticlePDG* pdg = TDatabasePDG::Instance()->GetParticle(fPdgCode);
  fMass = pdg->Mass();

  fE = TMath::Sqrt(fPx*fPx+fPy*fPy+fPz*fPz+fMass*fMass);
}


TpcSimMCEvent::TpcSimMCEvent(Int_t event)
{
  fTracks = new TClonesArray("TpcSimMCTrack",fkMaxNtracks);
  fNtracks = 0;
  fEvent = event;

}


TpcSimMCEvent::~TpcSimMCEvent()
{
  // for(Int_t i = 0; i<fNtracks; i++)
  //   fTracks[i]->Delete();
  fTracks->Delete();
}

void TpcSimMCEvent::AddTrack(TpcSimMCTrack* track)
{
  if(fNtracks>=fkMaxNtracks){
    Warning("AddTrack","Too many tracks in event");
    return;
  }
  new((*fTracks)[fNtracks]) TpcSimMCTrack(*track);
  //fTracks->AddAt(tr,fNtracks);
  //fTracks[fNtracks] = tr;
  fNtracks++;

}


void TpcSimMCEvent::AddTrack(TpcSimIonisationTrack* track)
{
  if(fNtracks>=fkMaxNtracks){
    Warning("AddTrack","Too many tracks in event");
    return;
  }

  TpcSimMCTrack* tr = new TpcSimMCTrack(track->fXorig, track->fYorig, track->fZorig, track->fTorig,
                                        track->fPx, track->fPy, track->fPz, track->fParticleId);

  new((*fTracks)[fNtracks]) TpcSimMCTrack(*tr);
  //fTracks->AddAt(tr,fNtracks);
  //fTracks[fNtracks] = tr;
  fNtracks++;

}

// void TpcSimMCEvent::GenerateEvent(Int_t seed)
// {

//   Int_t ntracks = 3;
//   while(fNtracks<ntracks){
//     if(fNtracks>=fkMaxNtracks) break;
//     //TpcSimTrack* track = 
//     new((*fTracks)[fNtracks]) TpcSimMCTrack(0,0,10,1,1,1,0,1,fEvent,fNtracks);
//     fNtracks++;
//   }
// }