A class to provide conversion from a FLUKA format ASCII input record into HepMC format, with or without sampling. More...

#include <FlukaHaloGenerator.h>

Inheritance diagram for FlukaHaloGenerator:

Collaboration diagram for FlukaHaloGenerator:

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

virtual	~FlukaHaloGenerator ()=default

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)
	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)
	A function to read one event in a simplified format. More...

virtual int	readParticle (BeamHaloParticle *beamHaloParticle)
	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
bool	m_sameEvent

bool	m_firstEvent

FlukaParticle	m_flukaParticle

FlukaParticle	m_lastFlukaParticle

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

Detailed Description

A class to provide conversion from a FLUKA format ASCII input record into HepMC format, with or without sampling.

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

Definition at line 18 of file FlukaHaloGenerator.h.

Member Enumeration Documentation

◆ enumCounters

enum BeamHaloGenerator::enumCounters

protectedinherited

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

protectedinherited

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

◆ FlukaHaloGenerator()

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

Definition at line 15 of file FlukaHaloGenerator.cxx.

                                                                                      :
   BeamHaloGenerator(particleTable, inputFile, generatorSettings),
   m_sameEvent(true),
   m_firstEvent(true),
   m_flukaParticle(type),
   m_lastFlukaParticle() {
 }

◆ ~FlukaHaloGenerator()

virtual FlukaHaloGenerator::~FlukaHaloGenerator ( )

virtualdefault

Member Function Documentation

◆ convertEvent()

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

protectedinherited

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()

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

virtual

A function to create one event in HepMC format.

Implements BeamHaloGenerator.

Definition at line 48 of file FlukaHaloGenerator.cxx.

                                                               {
   std::vector<BeamHaloParticle> beamHaloEvent;
   int ret_val;
  
   // Read one FLUKA event which passes the generator settings.
   if((ret_val = readEvent(&beamHaloEvent, engine)) != 0) return ret_val;
  
   // Convert the particles to GenParticles and attach them to the
   // event.  Flip the event if needed.
   if((ret_val = BeamHaloGenerator::convertEvent(&beamHaloEvent, evt, engine)) != 0) return ret_val;
  
   // Set the event number
   evt->set_event_number(m_eventNumber);
  
   // Set the signal process
   HepMC::set_signal_process_id(evt,BeamHaloGenerator::FLUKA_SHOWER);
  
   // Increment the event number
   m_eventNumber++;
  
   return 0;
 }

◆ flipEvent()

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

protectedinherited

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 FlukaHaloGenerator::genFinalize ( )

virtual

A function to finalise the generator.

Reimplemented from BeamHaloGenerator.

Definition at line 74 of file FlukaHaloGenerator.cxx.

                                     {
   int ret_val;
  
   m_asciiInput->close();
  
   // Finalise base class
   if((ret_val = BeamHaloGenerator::genFinalize()) != 0) return ret_val;
  
   return 0;
 }

◆ genInitialize()

int FlukaHaloGenerator::genInitialize ( )

virtual

A function to initialise the generator.

Reimplemented from BeamHaloGenerator.

Definition at line 29 of file FlukaHaloGenerator.cxx.

                                       {
   int ret_val;
  
   // Initialise base class
   if((ret_val = BeamHaloGenerator::genInitialize()) != 0) return ret_val;
  
   if(m_asciiInput->open() != 0) {
     std::cout << "Error: Could not open ascii input file " << m_inputFile << std::endl;
     return 1;
   }
  
   std::cout << "Info: Reading ascii input file " << m_inputFile << std::endl;
  
   return 0;
 }

◆ readEvent()

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

protectedvirtual

A function to read one event in a simplified format.

Implements BeamHaloGenerator.

Definition at line 95 of file FlukaHaloGenerator.cxx.

                                                                 {
   BeamHaloParticle beamHaloParticle;
  
   // Clear the event
   beamHaloEvent->clear();
  
   // If there was a last event.
   if(!m_firstEvent) {
  
     // If the last event caused the same event flag to be set to false
     // copy the last particle into the vector of those in this event.
     if(!m_sameEvent) {
       // Fill the BeamHaloParticle with the data in the FlukaParticle
       if(beamHaloParticle.fill(m_particleTable, &m_lastFlukaParticle)) {
         std::cout << "Error: Conversion from FlukaParticle to BeamHaloParticle failed." << std::endl;
         return 1;
       }
       // Append the BeamHalo particle to the event.
       beamHaloEvent->push_back(beamHaloParticle);
  
       // Set the same event flag to enter the while loop to read the
       // rest of this event.
       m_sameEvent = true;
     }
   }
  
   // Loop over the ascii input and read each particle until a new
   // event is found or there are no more particles.
   std::vector<std::string> row;
   bool endOfFile = false;
   while(m_sameEvent && !endOfFile) {
     row = m_asciiInput->readRow(); // Read one line of the ascii file.11
  
     if(row.size() == 0) {
       endOfFile = true;
       continue;
     }
  
     if(m_flukaParticle.read(&row)) { // Fill the particle from the string vector
       continue;
     }
  
     // Check if there was a last particle.
     if(!m_firstEvent) {
  
       m_sameEvent = false;
  
       // Check if the event id of the last particle is the same as this particle.
       if((m_lastFlukaParticle.type() == FlukaParticle::BRUCE)
          && (m_lastFlukaParticle.eventId() == m_flukaParticle.eventId())
          && (m_lastFlukaParticle.partGenNum() == m_flukaParticle.partGenNum())) m_sameEvent = true;
  
       if((m_lastFlukaParticle.type() == FlukaParticle::TALANOV)
          && (m_lastFlukaParticle.eventId() == m_flukaParticle.eventId())) m_sameEvent = true;
     }
     else {
       // For the first event.
       m_firstEvent = false;
       m_sameEvent = true;
     }
  
     // If this is the same event copy the particle into the vector for
     // this event.
     if(m_sameEvent) {
  
       // Fill the BeamHaloParticle with the data in the FlukaParticle
       if(beamHaloParticle.fill(m_particleTable, &m_flukaParticle)) {
         std::cout << "Error: Conversion from FlukaParticle to BeamHaloParticle failed." << std::endl;
         return 1;
       }
  
       // Append the BeamHalo particle to the event.
       beamHaloEvent->push_back(beamHaloParticle);
     }
  
     // Copy this particle into the last particle.
     m_lastFlukaParticle = m_flukaParticle;
  
   }
  
   if(beamHaloEvent->size() == 0) {
     std::cout << "Error: No particles read from " << m_inputFile << std::endl;
     return 1;
   }
  
   m_counters[TOT_READ]++;
   m_wsums[TOT_READ]+= 1.0;
  
   // Check if one of the particles in the event passes the generator settings.
   std::vector<BeamHaloParticle>::iterator itr = beamHaloEvent->begin();
   std::vector<BeamHaloParticle>::iterator itr_end = beamHaloEvent->end();
   bool passed = false;
   for(;itr!=itr_end;++itr) {
     beamHaloParticle = *itr;
     // Check the generator settings for this particle.
     if(m_beamHaloGeneratorSettings->checkParticle(&beamHaloParticle)) {
       passed = true;
     }
     else {
       if(m_debug) std::cout << "Debug: Particle fails generator settings cuts." << std::endl;
     }
   }
  
   m_counters[TOT_AFTER]++;
   m_wsums[TOT_AFTER]+= 1.0;
   m_counters[TOT_GEN]++;
   m_wsums[TOT_GEN]+= 1.0;
  
   // If all of the particles from this event fail read another event.
   // If there are no more events this function will exit with a
   // WARNING.
   if(!passed) {
     int ret_val;
     if((ret_val = readEvent(beamHaloEvent, engine)) != 0) return ret_val;
   }
  
   return 0;
 }

◆ readParticle()

int FlukaHaloGenerator::readParticle ( BeamHaloParticle * beamHaloParticle )

protectedvirtual

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

Implements BeamHaloGenerator.

Definition at line 87 of file FlukaHaloGenerator.cxx.

                                                                        {
  
   std::cout << "Warning: FlukaHaloGenerator::readParticle is not yet available: "<< beamHaloParticle  << std::endl;
   return 0;
 }

◆ setBufferFileName()

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

inlineinherited

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 )

inlineinherited

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 )

inlineinherited

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 )

inlineinherited

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 )

inlineinherited

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 )

inlineinherited

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

protectedinherited

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

protectedinherited

A pointer to a BeamHaloGeneratorSettings object used to filter particles.

Definition at line 135 of file BeamHaloGenerator.h.

◆ m_beamHaloParticleBuffer

BeamHaloParticleBuffer* BeamHaloGenerator::m_beamHaloParticleBuffer

protectedinherited

Binary particle buffer for caching converted events.

Definition at line 127 of file BeamHaloGenerator.h.

◆ m_bufferFileName

std::string BeamHaloGenerator::m_bufferFileName

protectedinherited

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]

protectedinherited

Particle or event counters of dimension enumCounters.

Definition at line 148 of file BeamHaloGenerator.h.

◆ m_debug

bool BeamHaloGenerator::m_debug

protectedinherited

A flat to turn on or off debug print out.

Definition at line 154 of file BeamHaloGenerator.h.

◆ m_enableFlip

bool BeamHaloGenerator::m_enableFlip

protectedinherited

Flag for flipping event.

Definition at line 114 of file BeamHaloGenerator.h.

◆ m_enableSampling

bool BeamHaloGenerator::m_enableSampling

protectedinherited

Flag to enable or disable sampling.

Definition at line 120 of file BeamHaloGenerator.h.

◆ m_eventNumber

long BeamHaloGenerator::m_eventNumber

protectedinherited

A data member to count the event number.

Definition at line 138 of file BeamHaloGenerator.h.

◆ m_firstEvent

bool FlukaHaloGenerator::m_firstEvent

private

Definition at line 50 of file FlukaHaloGenerator.h.

◆ m_flipProbability

float BeamHaloGenerator::m_flipProbability

protectedinherited

Flip probability.

Definition at line 117 of file BeamHaloGenerator.h.

◆ m_flukaParticle

FlukaParticle FlukaHaloGenerator::m_flukaParticle

private

Definition at line 51 of file FlukaHaloGenerator.h.

◆ m_generatorSettings

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

privateinherited

A vector of generator settings in string command form.

Definition at line 159 of file BeamHaloGenerator.h.

◆ m_inputFile

std::string BeamHaloGenerator::m_inputFile

protectedinherited

Input file name.

Definition at line 108 of file BeamHaloGenerator.h.

◆ m_interfacePlane

double BeamHaloGenerator::m_interfacePlane

protectedinherited

The position of the interface plane in mm.

Definition at line 111 of file BeamHaloGenerator.h.

◆ m_lastFlukaParticle

FlukaParticle FlukaHaloGenerator::m_lastFlukaParticle

private

Definition at line 52 of file FlukaHaloGenerator.h.

◆ m_particleTable

const HepPDT::ParticleDataTable* BeamHaloGenerator::m_particleTable

protectedinherited

A pointer to the particle data table.

Definition at line 105 of file BeamHaloGenerator.h.

◆ m_sameEvent

bool FlukaHaloGenerator::m_sameEvent

private

Definition at line 49 of file FlukaHaloGenerator.h.

◆ m_wsums

double BeamHaloGenerator::m_wsums[NUM_COUNTERS]

protectedinherited

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

◆ FlukaHaloGenerator()

◆ ~FlukaHaloGenerator()

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_firstEvent

◆ m_flipProbability

◆ m_flukaParticle

◆ m_generatorSettings

◆ m_inputFile

◆ m_interfacePlane

◆ m_lastFlukaParticle

◆ m_particleTable

◆ m_sameEvent

◆ m_wsums