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 std::string &inputFile, const std::vector< std::string > &generatorSettings)
virtual	~BeamHaloGenerator ()
virtual int	genInitialize ()
	A function to initialise the generator.
virtual int	genFinalize ()
	A function to finalise the generator.
virtual int	fillEvt (HepMC::GenEvent evt, CLHEP::HepRandomEngine engine)=0
	A function to create one event in HepMC format.
void	setInterfacePlane (double interfacePlane)
	Set the position of the interface plane in mm.
void	setEnableFlip (bool enableFlip)
	Turn on or off the event flipping code.
void	setFlipProbability (float flipProbability)
	Set probability of flipping an event about Z=0.
void	setEnableSampling (bool enableSampling)
	Turn on or off sampling from the input ASCII file.
void	setBufferFileName (const std::string &bufferFileName)
	Set the name of the binary buffer file, needed for sampling from a converted file.
void	setDebugEnable (bool debug)
	A function to turn on or off debug print out.

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.
virtual int	readParticle (BeamHaloParticle *beamHaloParticle)=0
	A function to read one particle from the input ASCII file.
bool	flipEvent (CLHEP::HepRandomEngine *engine)
	A member function to check if the event should be flipped.
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.

Protected Attributes
std::shared_ptr< GenData >	m_gendata
	A shared pointer to the GenData helper object.
std::string	m_inputFile
	Input file name.
double	m_interfacePlane
	The position of the interface plane in mm.
bool	m_enableFlip
	Flag for flipping event.
float	m_flipProbability
	Flip probability.
bool	m_enableSampling
	Flag to enable or disable sampling.
std::string	m_bufferFileName
	The name of the binary buffer file, needed for sampling from a converted file.
std::unique_ptr< BeamHaloParticleBuffer >	m_beamHaloParticleBuffer
	Binary particle buffer for caching converted events.
std::unique_ptr< AsciiInput >	m_asciiInput
	A pointer to an AsciiInput object, used to read data from the Ascii input file.
std::unique_ptr< BeamHaloGeneratorSettings >	m_beamHaloGeneratorSettings
	A pointer to a BeamHaloGeneratorSettings object used to filter particles.
long	m_eventNumber
	A data member to count the event number.
long	m_counters [NUM_COUNTERS]
	Particle or event counters of dimension enumCounters.
double	m_wsums [NUM_COUNTERS]
	Sum of weights of particles or events of dimension enumCounters.
bool	m_debug
	A flat to turn on or off debug print out.

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

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 30 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 140 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 85 of file BeamHaloGenerator.h.

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

Constructor & Destructor Documentation

◆ BeamHaloGenerator()

BeamHaloGenerator::BeamHaloGenerator	(	const std::string &	inputFile,
		const std::vector< std::string > &	generatorSettings )

Definition at line 17 of file BeamHaloGenerator.cxx.

                                                                   :
  m_gendata(std::make_shared<GenData>()),
  m_inputFile(inputFile),
  m_interfacePlane(0.),
  m_enableFlip(false),
  m_flipProbability(0.),
  m_enableSampling(false),
  m_bufferFileName("buffer.bin"),
  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 38 of file BeamHaloGenerator.cxx.

38 {

39}

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 89 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 75 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 52 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 43 of file BeamHaloGenerator.cxx.

                                     {
  m_asciiInput = std::make_unique<AsciiInput>(m_inputFile);
  m_beamHaloGeneratorSettings = std::make_unique< 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 75 of file BeamHaloGenerator.h.

75{ m_bufferFileName = bufferFileName; }

◆ setDebugEnable()

void BeamHaloGenerator::setDebugEnable ( bool debug )

inline

A function to turn on or off debug print out.

Definition at line 78 of file BeamHaloGenerator.h.

78{ m_debug = debug; }

debug

const bool debug

Definition MakeUncertaintyPlots.cxx:53

◆ setEnableFlip()

void BeamHaloGenerator::setEnableFlip ( bool enableFlip )

inline

Turn on or off the event flipping code.

Definition at line 65 of file BeamHaloGenerator.h.

65{ m_enableFlip = enableFlip; }

◆ setEnableSampling()

void BeamHaloGenerator::setEnableSampling ( bool enableSampling )

inline

Turn on or off sampling from the input ASCII file.

Definition at line 71 of file BeamHaloGenerator.h.

71{ m_enableSampling = enableSampling; }

◆ setFlipProbability()

void BeamHaloGenerator::setFlipProbability ( float flipProbability )

inline

Set probability of flipping an event about Z=0.

Definition at line 68 of file BeamHaloGenerator.h.

68{ m_flipProbability = flipProbability; }

◆ setInterfacePlane()

void BeamHaloGenerator::setInterfacePlane ( double interfacePlane )

inline

Set the position of the interface plane in mm.

Definition at line 62 of file BeamHaloGenerator.h.

62{ m_interfacePlane = interfacePlane; }

Member Data Documentation

◆ m_asciiInput

std::unique_ptr<AsciiInput> BeamHaloGenerator::m_asciiInput

protected

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

Definition at line 130 of file BeamHaloGenerator.h.

◆ m_beamHaloGeneratorSettings

std::unique_ptr<BeamHaloGeneratorSettings> BeamHaloGenerator::m_beamHaloGeneratorSettings

protected

A pointer to a BeamHaloGeneratorSettings object used to filter particles.

Definition at line 134 of file BeamHaloGenerator.h.

◆ m_beamHaloParticleBuffer

std::unique_ptr<BeamHaloParticleBuffer> BeamHaloGenerator::m_beamHaloParticleBuffer

protected

Binary particle buffer for caching converted events.

Definition at line 126 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 123 of file BeamHaloGenerator.h.

◆ m_counters

long BeamHaloGenerator::m_counters[NUM_COUNTERS]

protected

Particle or event counters of dimension enumCounters.

Definition at line 147 of file BeamHaloGenerator.h.

◆ m_debug

bool BeamHaloGenerator::m_debug

protected

A flat to turn on or off debug print out.

Definition at line 153 of file BeamHaloGenerator.h.

◆ m_enableFlip

bool BeamHaloGenerator::m_enableFlip

protected

Flag for flipping event.

Definition at line 113 of file BeamHaloGenerator.h.

◆ m_enableSampling

bool BeamHaloGenerator::m_enableSampling

protected

Flag to enable or disable sampling.

Definition at line 119 of file BeamHaloGenerator.h.

◆ m_eventNumber

long BeamHaloGenerator::m_eventNumber

protected

A data member to count the event number.

Definition at line 137 of file BeamHaloGenerator.h.

◆ m_flipProbability

float BeamHaloGenerator::m_flipProbability

protected

Flip probability.

Definition at line 116 of file BeamHaloGenerator.h.

◆ m_gendata

std::shared_ptr<GenData> BeamHaloGenerator::m_gendata

protected

A shared pointer to the GenData helper object.

Definition at line 104 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 158 of file BeamHaloGenerator.h.

◆ m_inputFile

std::string BeamHaloGenerator::m_inputFile

protected

Input file name.

Definition at line 107 of file BeamHaloGenerator.h.

◆ m_interfacePlane

double BeamHaloGenerator::m_interfacePlane

protected

The position of the interface plane in mm.

Definition at line 110 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 150 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_gendata

◆ m_generatorSettings

◆ m_inputFile

◆ m_interfacePlane

◆ m_wsums