G4TankSimulatorOG/G4TankPrimaryGenerator.cc

Go to the documentation of this file.

#include "G4TankPrimaryGenerator.h"
#include "G4TankSimulator.h"

#include <det/Detector.h>

#include <sevt/StationSimData.h>

#include <sdet/Station.h>

#include <utl/CoordinateSystem.h>
#include <utl/CoordinateSystemPtr.h>
#include <utl/Point.h>
#include <utl/Particle.h>
#include <utl/ReferenceEllipsoid.h>
#include <utl/AugerCoordinateSystem.h>

#include <cstddef>
#include <iostream>
#include <sstream>

#include <G4Event.hh>
#include <G4ParticleGun.hh>
#include <G4ParticleDefinition.hh>
#include <G4ParticleTable.hh>

#include <utl/AugerUnits.h>

using namespace G4TankSimulatorOG;
using namespace fwk;
using namespace det;
using namespace sevt;
using utl::CoordinateSystem;
using utl::CoordinateSystemPtr;
using utl::Point;
using utl::Particle;
using utl::ReferenceEllipsoid;
using utl::AugerCoordinateSystem;
using std::cout;
using std::endl;


G4TankPrimaryGenerator::G4TankPrimaryGenerator() : 
  fParticleGun(new G4ParticleGun(1)), 
  fParticleTable(G4ParticleTable::GetParticleTable()) 
{
}


G4TankPrimaryGenerator::~G4TankPrimaryGenerator() 
{    
  delete fParticleGun;
}


void
G4TankPrimaryGenerator::GeneratePrimaries(G4Event* theEvent) 
{
  const sdet::Station* theCurrentDetectorStation =
    G4TankSimulator::GetCurrentDetectorStation();

  sevt::StationSimData::ParticleIterator theCurrentParticleIt =
    G4TankSimulator::GetCurrentParticleIt();

//   // temp
//   sevt::StationSimData::ParticleIterator testit = G4TankSimulator::GetCurrentParticleIt(); // temp  
//   if (testit->GetWeight() > 2)
//     cout << "testit = " << testit->GetWeight() << endl;
//   // end temp  

  // Iterate over the particles in the hit station and pass their
  // properties to the G4ParticleGun:
  
  CoordinateSystemPtr siteCS = 
    det::Detector::GetInstance().GetSiteCoordinateSystem();
  
  const ReferenceEllipsoid& e = ReferenceEllipsoid::GetWGS84();
  
  Point stationPos = theCurrentDetectorStation->GetPosition();
  
  CoordinateSystemPtr localCS = 
    AugerCoordinateSystem::Create(stationPos, e, siteCS);
  
  CoordinateSystemPtr csDir = 
    theCurrentParticleIt->GetDirection().GetCoordinateSystem();
  
  // set properties and pass to particle gun            
  
  G4ParticleDefinition* particleDef = 
    fParticleTable->FindParticle(theCurrentParticleIt->GetName());

  if (!particleDef) {
    std::ostringstream msg;
    msg << "Undefined particle type: "
        << theCurrentParticleIt->GetName();
    WARNING(msg);
    return;
  }

  double x, y, z;               
  boost::tie(x, y, z) = 
    (theCurrentParticleIt->GetPosition() - stationPos).GetCoordinates(localCS);    

  // Convert from Auger units to G4 units
  G4ThreeVector position(x * CLHEP::m / utl::m, y * CLHEP::m / utl::m, z * CLHEP::m / utl::m);
  
  G4ThreeVector direction(theCurrentParticleIt->GetDirection().GetX(csDir), 
                          theCurrentParticleIt->GetDirection().GetY(csDir), 
                          theCurrentParticleIt->GetDirection().GetZ(csDir));
  
  fParticleGun->SetParticleDefinition(particleDef);
  fParticleGun->SetParticlePosition(position);
  fParticleGun->SetParticleMomentumDirection(direction);

  // Convert from Auger units to G4 units
  fParticleGun->SetParticleEnergy(theCurrentParticleIt->GetKineticEnergy() * CLHEP::eV / utl::eV);
  fParticleGun->SetParticleTime(theCurrentParticleIt->GetTime().GetInterval());
    
  fParticleGun->GeneratePrimaryVertex(theEvent);            
}