An abstract base class to provide generic beam halo generator functionality. More...

#include <BeamHaloGenerator.h>

Inheritance diagram for BeamHaloGenerator:

Collaboration diagram for BeamHaloGenerator:

Public Member Functions
	BeamHaloGenerator (const HepPDT::ParticleDataTable *particleTable, const std::string &inputFile, const std::vector< std::string > &generatorSettings)

virtual	~BeamHaloGenerator ()

virtual int	genInitialize ()
	A function to initialise the generator. More...

virtual int	genFinalize ()
	A function to finalise the generator. More...

virtual int	fillEvt (HepMC::GenEvent evt, CLHEP::HepRandomEngine engine)=0
	A function to create one event in HepMC format. More...

void	setInterfacePlane (double interfacePlane)
	Set the position of the interface plane in mm. More...

void	setEnableFlip (bool enableFlip)
	Turn on or off the event flipping code. More...

void	setFlipProbability (float flipProbability)
	Set probability of flipping an event about Z=0. More...

void	setEnableSampling (bool enableSampling)
	Turn on or off sampling from the input ASCII file. More...

void	setBufferFileName (const std::string &bufferFileName)
	Set the name of the binary buffer file, needed for sampling from a converted file. More...

void	setDebugEnable (bool debug)
	A function to turn on or off debug print out. More...

Protected Types
enum	enumGeneratorProcessIds { MARS_READ, MARS_SINGLE, MARS_SHOWER, FLUKA_READ, FLUKA_SINGLE, FLUKA_SHOWER }
	An enum of generator process identifiers. More...

enum	enumCounters { TOT_READ, TOT_AFTER, TOT_GEN, NUM_COUNTERS }
	An enum of particle counter indices. More...

Protected Member Functions
virtual int	readEvent (std::vector< BeamHaloParticle > beamHaloEvent, CLHEP::HepRandomEngine engine)=0
	A function to read one event in a simplified format. More...

virtual int	readParticle (BeamHaloParticle *beamHaloParticle)=0
	A function to read one particle from the input ASCII file. More...

bool	flipEvent (CLHEP::HepRandomEngine *engine)
	A member function to check if the event should be flipped. More...

int	convertEvent (std::vector< BeamHaloParticle > beamHaloEvent, HepMC::GenEvent evt, CLHEP::HepRandomEngine *engine)
	A member function to convert a vector of beam halo particles into a GenEvent. More...

Protected Attributes
const HepPDT::ParticleDataTable *	m_particleTable
	A pointer to the particle data table. More...

std::string	m_inputFile
	Input file name. More...

double	m_interfacePlane
	The position of the interface plane in mm. More...

bool	m_enableFlip
	Flag for flipping event. More...

float	m_flipProbability
	Flip probability. More...

bool	m_enableSampling
	Flag to enable or disable sampling. More...

std::string	m_bufferFileName
	The name of the binary buffer file, needed for sampling from a converted file. More...

BeamHaloParticleBuffer *	m_beamHaloParticleBuffer
	Binary particle buffer for caching converted events. More...

AsciiInput *	m_asciiInput
	A pointer to an AsciiInput object, used to read data from the Ascii input file. More...

BeamHaloGeneratorSettings *	m_beamHaloGeneratorSettings
	A pointer to a BeamHaloGeneratorSettings object used to filter particles. More...

long	m_eventNumber
	A data member to count the event number. More...

long	m_counters [NUM_COUNTERS]
	Particle or event counters of dimension enumCounters. More...

double	m_wsums [NUM_COUNTERS]
	Sum of weights of particles or events of dimension enumCounters. More...

bool	m_debug
	A flat to turn on or off debug print out. More...

Private Attributes
std::vector< std::string >	m_generatorSettings
	A vector of generator settings in string command form. More...

Detailed Description

An abstract base class to provide generic beam halo generator functionality.

Author: W. H. Bell W.Bel.nosp@m.l@ce.nosp@m.rn.ch

Definition at line 31 of file BeamHaloGenerator.h.

Member Enumeration Documentation

◆ enumCounters

enum BeamHaloGenerator::enumCounters

protected

An enum of particle counter indices.

Enumerator
TOT_READ
TOT_AFTER
TOT_GEN
NUM_COUNTERS

Definition at line 141 of file BeamHaloGenerator.h.

                     {
     TOT_READ, 
     TOT_AFTER,
     TOT_GEN,
     NUM_COUNTERS};

◆ enumGeneratorProcessIds

enum BeamHaloGenerator::enumGeneratorProcessIds

protected

An enum of generator process identifiers.

_READ corresponds to no sampling and therefore simple conversion from input file to HepMC record.

Enumerator
MARS_READ
MARS_SINGLE
MARS_SHOWER
FLUKA_READ
FLUKA_SINGLE
FLUKA_SHOWER

Definition at line 86 of file BeamHaloGenerator.h.

                                {
     MARS_READ,
     MARS_SINGLE,
     MARS_SHOWER,
     FLUKA_READ,
     FLUKA_SINGLE,
     FLUKA_SHOWER };

Constructor & Destructor Documentation

◆ BeamHaloGenerator()

BeamHaloGenerator::BeamHaloGenerator	(	const HepPDT::ParticleDataTable *	particleTable,
		const std::string &	inputFile,
		const std::vector< std::string > &	generatorSettings
	)

Definition at line 16 of file BeamHaloGenerator.cxx.

                                                                    :
   m_particleTable(particleTable),
   m_inputFile(inputFile),
   m_interfacePlane(0.),
   m_enableFlip(false),
   m_flipProbability(0.),
   m_enableSampling(false),
   m_bufferFileName("buffer.bin"),
   m_beamHaloParticleBuffer(0),
   m_asciiInput(0),
   m_beamHaloGeneratorSettings(0),
   m_eventNumber(0),
   m_debug(false),
   m_generatorSettings(generatorSettings) {
   for(int i=0;i<NUM_COUNTERS;i++) {
     m_counters[i] = 0;
     m_wsums[i] = 0.;
   }
 }

◆ ~BeamHaloGenerator()

BeamHaloGenerator::~BeamHaloGenerator ( )

virtual

Definition at line 40 of file BeamHaloGenerator.cxx.

                                       {
   if(m_asciiInput) delete m_asciiInput;
   if(m_beamHaloGeneratorSettings) delete m_beamHaloGeneratorSettings;
   if(m_beamHaloParticleBuffer) delete m_beamHaloParticleBuffer;
 }

Member Function Documentation

◆ convertEvent()

int BeamHaloGenerator::convertEvent	(	std::vector< BeamHaloParticle > *	beamHaloEvent,
		HepMC::GenEvent *	evt,
		CLHEP::HepRandomEngine *	engine
	)

protected

A member function to convert a vector of beam halo particles into a GenEvent.

Definition at line 94 of file BeamHaloGenerator.cxx.

                                                                   {
   double pz;
   bool flipFlag = flipEvent(engine);
   HepMC::GenParticlePtr genParticle;
   HepMC::GenVertexPtr genVertex;
   const double c = 2.99792458E+11; // speed of light in mm/s
  
   std::vector<BeamHaloParticle>::iterator itr = beamHaloEvent->begin();
   std::vector<BeamHaloParticle>::iterator itr_end = beamHaloEvent->end();
  
   if(itr != itr_end) {
     // Take the primary information from the first particle in the event.  If the event is more than one
     // particle, the particles come from the same primary proton.
     evt->weights().push_back((*itr).weight()); // event weight
     HepMC::FourVector primaryPosition = (*itr).positionAtPrimary();
     evt->weights().push_back(primaryPosition.x()); // Starting position of primary particle: x position (cm)
     evt->weights().push_back(primaryPosition.y()); // Starting position of primary particle: y position (cm)
     evt->weights().push_back(primaryPosition.z()); // Starting position of primary particle: z position (cm)
     evt->weights().push_back(primaryPosition.t()); // Starting position of primary particle: t (s)
   } 
  
   // Append each particle to the GenEvent
   for(;itr!=itr_end;++itr) {
     HepMC::FourVector position = (*itr).positionAtScoringPlane();
     HepMC::FourVector fourVector = (*itr).fourVector();
  
     if(!flipFlag) {
       genParticle = HepMC::newGenParticlePtr(fourVector,
                        (*itr).pdgId(),
                        1);
  
       genVertex = HepMC::newGenVertexPtr(HepMC::FourVector(position.x(),
                              position.y(),
                              (position.z() + m_interfacePlane),
                              c*position.t() -1.0*std::fabs(m_interfacePlane)),
                        1);
     }
     else {
       pz = fourVector.pz();
       fourVector.setPz(-pz);
  
       genParticle = HepMC::newGenParticlePtr(fourVector,
                        (*itr).pdgId(),
                        1);
  
       genVertex = HepMC::newGenVertexPtr(HepMC::FourVector(position.x(),
                              position.y(),
                              -1.0*(position.z() + m_interfacePlane),
                              c*position.t() -1.0*std::fabs(m_interfacePlane)),
                        1);
     }
  
     genVertex->add_particle_out(genParticle);
     evt->add_vertex(genVertex);
   }
  
   return 0;
 }

◆ fillEvt()

virtual int BeamHaloGenerator::fillEvt	(	HepMC::GenEvent *	evt,
		CLHEP::HepRandomEngine *	engine
	)

pure virtual

A function to create one event in HepMC format.

Implemented in FlukaHaloGenerator, and MarsHaloGenerator.

◆ flipEvent()

bool BeamHaloGenerator::flipEvent ( CLHEP::HepRandomEngine * engine )

protected

A member function to check if the event should be flipped.

Definition at line 80 of file BeamHaloGenerator.cxx.

                                                               {
   if(!m_enableFlip) return false;
  
   // Check to see if the event should be flipped or not
   if(CLHEP::RandFlat::shoot(engine) <= m_flipProbability) {
     if(m_debug) std::cout << "Debug: Flipping event" << std::endl;
     return true;
   }
  
   return false;
 }

◆ genFinalize()

int BeamHaloGenerator::genFinalize ( )

virtual

A function to finalise the generator.

Reimplemented in FlukaHaloGenerator, and MarsHaloGenerator.

Definition at line 57 of file BeamHaloGenerator.cxx.

                                    {
  
   std::cout << "=================================================================" << std::endl;
   std::cout << "|                                 | Total Number | Total Weight |" << std::endl;
   std::cout << "|---------------------------------------------------------------|" << std::endl;
   std::cout << "| Particles read from input file  | ";
   std::cout << std::setw(12) << m_counters[TOT_READ] << " | ";
   std::cout << std::setw(12) << std::setprecision(8) << m_wsums[TOT_READ] << " |" << std::endl;
   std::cout << "| Particles after cuts            | ";
   std::cout << std::setw(12) << m_counters[TOT_AFTER] << " | ";
   std::cout << std::setw(12) << std::setprecision(8) << m_wsums[TOT_AFTER] << " |" << std::endl;
   std::cout << "| Particles generated             | ";
   std::cout << std::setw(12) << m_counters[TOT_GEN] << " | ";
   std::cout << std::setw(12) << std::setprecision(8) << m_wsums[TOT_GEN] << " |" << std::endl;
   std::cout << "=================================================================" << std::endl;
   
   std::cout << std::setprecision(6);
  
   return 0;
 }

◆ genInitialize()

int BeamHaloGenerator::genInitialize ( )

virtual

A function to initialise the generator.

Reimplemented in FlukaHaloGenerator, and MarsHaloGenerator.

Definition at line 48 of file BeamHaloGenerator.cxx.

                                      {
   m_asciiInput = new AsciiInput(m_inputFile);
   m_beamHaloGeneratorSettings = new BeamHaloGeneratorSettings(m_generatorSettings);
  
   return 0;
 }

◆ readEvent()

virtual int BeamHaloGenerator::readEvent	(	std::vector< BeamHaloParticle > *	beamHaloEvent,
		CLHEP::HepRandomEngine *	engine
	)

protectedpure virtual

A function to read one event in a simplified format.

Implemented in FlukaHaloGenerator, and MarsHaloGenerator.

◆ readParticle()

virtual int BeamHaloGenerator::readParticle ( BeamHaloParticle * beamHaloParticle )

protectedpure virtual

A function to read one particle from the input ASCII file.

Implemented in FlukaHaloGenerator, and MarsHaloGenerator.

◆ setBufferFileName()

void BeamHaloGenerator::setBufferFileName ( const std::string & bufferFileName )

inline

Set the name of the binary buffer file, needed for sampling from a converted file.

Definition at line 76 of file BeamHaloGenerator.h.

76 { m_bufferFileName = bufferFileName; }

◆ setDebugEnable()

void BeamHaloGenerator::setDebugEnable ( bool debug )

inline

A function to turn on or off debug print out.

Definition at line 79 of file BeamHaloGenerator.h.

79 { m_debug = debug; }

◆ setEnableFlip()

void BeamHaloGenerator::setEnableFlip ( bool enableFlip )

inline

Turn on or off the event flipping code.

Definition at line 66 of file BeamHaloGenerator.h.

66 { m_enableFlip = enableFlip; }

◆ setEnableSampling()

void BeamHaloGenerator::setEnableSampling ( bool enableSampling )

inline

Turn on or off sampling from the input ASCII file.

Definition at line 72 of file BeamHaloGenerator.h.

72 { m_enableSampling = enableSampling; }

◆ setFlipProbability()

void BeamHaloGenerator::setFlipProbability ( float flipProbability )

inline

Set probability of flipping an event about Z=0.

Definition at line 69 of file BeamHaloGenerator.h.

69 { m_flipProbability = flipProbability; }

◆ setInterfacePlane()

void BeamHaloGenerator::setInterfacePlane ( double interfacePlane )

inline

Set the position of the interface plane in mm.

Definition at line 63 of file BeamHaloGenerator.h.

63 { m_interfacePlane = interfacePlane; }

Member Data Documentation

◆ m_asciiInput

AsciiInput* BeamHaloGenerator::m_asciiInput

protected

A pointer to an AsciiInput object, used to read data from the Ascii input file.

Definition at line 131 of file BeamHaloGenerator.h.

◆ m_beamHaloGeneratorSettings

BeamHaloGeneratorSettings* BeamHaloGenerator::m_beamHaloGeneratorSettings

protected

A pointer to a BeamHaloGeneratorSettings object used to filter particles.

Definition at line 135 of file BeamHaloGenerator.h.

◆ m_beamHaloParticleBuffer

BeamHaloParticleBuffer* BeamHaloGenerator::m_beamHaloParticleBuffer

protected

Binary particle buffer for caching converted events.

Definition at line 127 of file BeamHaloGenerator.h.

◆ m_bufferFileName

std::string BeamHaloGenerator::m_bufferFileName

protected

The name of the binary buffer file, needed for sampling from a converted file.

Definition at line 124 of file BeamHaloGenerator.h.

◆ m_counters

long BeamHaloGenerator::m_counters[NUM_COUNTERS]

protected

Particle or event counters of dimension enumCounters.

Definition at line 148 of file BeamHaloGenerator.h.

◆ m_debug

bool BeamHaloGenerator::m_debug

protected

A flat to turn on or off debug print out.

Definition at line 154 of file BeamHaloGenerator.h.

◆ m_enableFlip

bool BeamHaloGenerator::m_enableFlip

protected

Flag for flipping event.

Definition at line 114 of file BeamHaloGenerator.h.

◆ m_enableSampling

bool BeamHaloGenerator::m_enableSampling

protected

Flag to enable or disable sampling.

Definition at line 120 of file BeamHaloGenerator.h.

◆ m_eventNumber

long BeamHaloGenerator::m_eventNumber

protected

A data member to count the event number.

Definition at line 138 of file BeamHaloGenerator.h.

◆ m_flipProbability

float BeamHaloGenerator::m_flipProbability

protected

Flip probability.

Definition at line 117 of file BeamHaloGenerator.h.

◆ m_generatorSettings

std::vector<std::string> BeamHaloGenerator::m_generatorSettings

private

A vector of generator settings in string command form.

Definition at line 159 of file BeamHaloGenerator.h.

◆ m_inputFile

std::string BeamHaloGenerator::m_inputFile

protected

Input file name.

Definition at line 108 of file BeamHaloGenerator.h.

◆ m_interfacePlane

double BeamHaloGenerator::m_interfacePlane

protected

The position of the interface plane in mm.

Definition at line 111 of file BeamHaloGenerator.h.

◆ m_particleTable

const HepPDT::ParticleDataTable* BeamHaloGenerator::m_particleTable

protected

A pointer to the particle data table.

Definition at line 105 of file BeamHaloGenerator.h.

◆ m_wsums

double BeamHaloGenerator::m_wsums[NUM_COUNTERS]

protected

Sum of weights of particles or events of dimension enumCounters.

Definition at line 151 of file BeamHaloGenerator.h.

The documentation for this class was generated from the following files:

Public Member Functions

Protected Types

Protected Member Functions

Protected Attributes

Private Attributes

Detailed Description

Member Enumeration Documentation

◆ enumCounters

◆ enumGeneratorProcessIds

Constructor & Destructor Documentation

◆ BeamHaloGenerator()

◆ ~BeamHaloGenerator()

Member Function Documentation

◆ convertEvent()

◆ fillEvt()

◆ flipEvent()

◆ genFinalize()

◆ genInitialize()

◆ readEvent()

◆ readParticle()

◆ setBufferFileName()

◆ setDebugEnable()

◆ setEnableFlip()

◆ setEnableSampling()

◆ setFlipProbability()

◆ setInterfacePlane()

Member Data Documentation

◆ m_asciiInput

◆ m_beamHaloGeneratorSettings

◆ m_beamHaloParticleBuffer

◆ m_bufferFileName

◆ m_counters

◆ m_debug

◆ m_enableFlip

◆ m_enableSampling

◆ m_eventNumber

◆ m_flipProbability

◆ m_generatorSettings

◆ m_inputFile

◆ m_interfacePlane

◆ m_particleTable

◆ m_wsums