FDsimG4PrimaryGeneratorAction.cc

Go to the documentation of this file.

//
// Description: PrimaryGeneratorAction for FDsimG4 application
//              

#include <math.h>

#include "FDsimG4PrimaryGeneratorAction.hh"
#include "FDsimG4DetectorConstruction.hh"

#include "globals.hh"
#include "G4RunManager.hh"
#include "G4Event.hh"
#include "G4GeneralParticleSource.hh"
#include "G4SPSAngDistribution.hh"
#include "G4SPSEneDistribution.hh"
#include "G4SPSPosDistribution.hh"
#include "G4ParticleTable.hh"
#include "G4ParticleDefinition.hh"
#include "TelescopeSimulator.h"
#include <fwk/RunController.h>
#include <fwk/VModule.h>


using namespace TelescopeSimulatorLX;

FDsimG4PrimaryGeneratorAction::FDsimG4PrimaryGeneratorAction()
{
  fUseGenPartSource = true;

  particleGun = new G4GeneralParticleSource();
  pParticleTable = G4ParticleTable::GetParticleTable();

  TelescopeSimulator &FDSimulator =
    dynamic_cast<TelescopeSimulator&>(
      fwk::RunController::GetInstance().GetModule("TelescopeSimulatorLX")
      );

  theFDSimulator = &FDSimulator;
}


FDsimG4PrimaryGeneratorAction::~FDsimG4PrimaryGeneratorAction()
{
  delete particleGun;
}


void FDsimG4PrimaryGeneratorAction::GeneratePrimaries(G4Event* anEvent)
{
  if (fUseGenPartSource)
    GenerateGPSPrimaries(anEvent);
  else
    ReadFDSimulatorEvent(anEvent);
}


void FDsimG4PrimaryGeneratorAction::ReadFDSimulatorEvent(G4Event* pEvent)
{

  // Get position and thickness of the filter

  static G4ThreeVector FilterPosition = 
    (dynamic_cast<const FDsimG4DetectorConstruction *>
     ((G4RunManager::GetRunManager())->GetUserDetectorConstruction()))->fFilterPosition;

  static G4double FilterThickness = 
    (dynamic_cast<const FDsimG4DetectorConstruction *>((G4RunManager::GetRunManager())->GetUserDetectorConstruction()))->fFilterThickness;
  
  G4PrimaryParticle* particle = new G4PrimaryParticle();

  FDsimG4StoreOpticalHit* G4Photon = theFDSimulator->GetG4Photon(); 
 
  G4ThreeVector position = 
    G4ThreeVector(G4Photon->GetPosX()*m,G4Photon->GetPosY()*m,G4Photon->GetPosZ()*m);
  position.setZ(FilterPosition.z() + FilterThickness/2.0 + 0.5*mm);

  G4ThreeVector momentum = 
    G4ThreeVector(G4Photon->GetDirX(),G4Photon->GetDirY(),G4Photon->GetDirZ());
  momentum=momentum*(G4Photon->GetEnergy()*eV);

  G4double time = G4Photon->GetTime()*ns;
  G4double w = G4Photon->GetWeight();

  particle->SetMomentum(momentum.x(),momentum.y(),momentum.z());
  particle->SetPolarization(1.0,0.0,0.0);
  particle->SetCharge(0.0);
  particle->SetG4code(pParticleTable->FindParticle("opticalphoton"));

  G4PrimaryVertex * vertex    = new G4PrimaryVertex(position,time);
  vertex->SetPrimary( particle );
  vertex->SetWeight(w);

  pEvent->AddPrimaryVertex( vertex );
}


void FDsimG4PrimaryGeneratorAction::GenerateGPSPrimaries(G4Event* anEvent)
{
  particleGun->GeneratePrimaryVertex(anEvent);
}