PHG4IonGun.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4IonGun.cc

#include "PHG4IonGun.h"

#include "PHG4InEvent.h"
#include "PHG4Particle.h"  // for PHG4Particle
#include "PHG4Particlev3.h"

#include <fun4all/Fun4AllReturnCodes.h>

#include <phool/getClass.h>

#include <Geant4/G4IonTable.hh>
#include <Geant4/G4ParticleDefinition.hh>  // for G4ParticleDefinition
#include <Geant4/G4String.hh>              // for G4String
#include <Geant4/G4SystemOfUnits.hh>

#include <algorithm>  // for fill
#include <cmath>      // for NAN
#include <iostream>   // for operator<<, basic_ostream
#include <iterator>   // for begin, end

using namespace std;

PHG4IonGun::PHG4IonGun(const string &name)
  : PHG4ParticleGeneratorBase(name)
  , ion(new PHG4Particlev3())
{
}

void PHG4IonGun::SetCharge(const int c)
{
  ioncharge = c * eplus;
}

void PHG4IonGun::SetMom(const double px, const double py, const double pz)
{
  mom[0] = px;
  mom[1] = py;
  mom[2] = pz;
}

int PHG4IonGun::process_event(PHCompositeNode *topNode)
{
  // get pdg code for ions
  UpdateParticle();
  PHG4InEvent *ineve = findNode::getClass<PHG4InEvent>(topNode, "PHG4INEVENT");
  ReuseExistingVertex(topNode);  // checks if we should reuse existing vertex
  int vtxindex = ineve->AddVtx(get_vtx_x(), get_vtx_y(), get_vtx_z(), get_t0());
  //   G4ParticleDefinition* ion = G4IonTable::GetIonTable()->GetIon(Z, A, excitEnergy);
  //G4PrimaryParticle* g4part = new G4PrimaryParticle(ion);
  //   cout << "name: " << ion->GetParticleName() << ", pdgcode: " << ion->GetPDGEncoding() << endl;
  PHG4Particle *particle = new PHG4Particlev3(ion);
  SetParticleId(particle, ineve);
  ineve->AddParticle(vtxindex, particle);
  return Fun4AllReturnCodes::EVENT_OK;
}

void PHG4IonGun::UpdateParticle()
{
  ion->set_A(A);
  ion->set_Z(Z);
  ion->set_NumCharge(ioncharge);
  ion->set_ExcitEnergy(excitEnergy);
  ion->set_px(mom[0]);
  ion->set_py(mom[1]);
  ion->set_pz(mom[2]);
  G4ParticleDefinition *iondef = G4IonTable::GetIonTable()->GetIon(Z, A, excitEnergy);
  ion->set_name(iondef->GetParticleName());
  ion->set_pid(iondef->GetPDGEncoding());
  return;
}

void PHG4IonGun::Print(const string &/*what*/) const
{
  cout << "PHG4IonGun, using ions of" << endl;
  cout << "A: " << A << ", Z: " << Z << ", charge: " << ioncharge
       << ", excitation Energy: " << excitEnergy << endl;
  cout << "px: " << mom[0] / GeV << " GeV, py: " << mom[1] / GeV << " GeV, pz: " << mom[2] / GeV << " GeV" << endl;
}