#include <FlatRandomMultiParticlePGunProducer.h>

Inheritance diagram for edm::FlatRandomMultiParticlePGunProducer:

Public Member Functions
	FlatRandomMultiParticlePGunProducer (const ParameterSet &pset)

void	produce (Event &e, const EventSetup &es) override

	~FlatRandomMultiParticlePGunProducer () override

Public Member Functions inherited from edm::BaseFlatGunProducer
	BaseFlatGunProducer (const ParameterSet &)

void	beginRun (const edm::Run &r, const edm::EventSetup &) override

void	endRun (edm::Run const &r, const edm::EventSetup &) override

void	endRunProduce (edm::Run &r, const edm::EventSetup &) override

	~BaseFlatGunProducer () override

Public Member Functions inherited from edm::one::EDProducer< one::WatchRuns, EndRunProducer >
	EDProducer ()=default

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

Public Member Functions inherited from edm::one::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDProducerBase () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

	~ProducerBase () noexcept(false) override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Private Attributes
double	fMaxP_

double	fMinP_

std::vector< double >	fProbParticle_

Additional Inherited Members
Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex >>

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::one::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Protected Attributes inherited from edm::BaseFlatGunProducer
bool	fAddAntiParticle

HepMC::GenEvent *	fEvt

double	fMaxEta

double	fMaxPhi

double	fMinEta

double	fMinPhi

std::vector< int >	fPartIDs

ESHandle< HepPDT::ParticleDataTable >	fPDGTable

int	fVerbosity

Detailed Description

Definition at line 9 of file FlatRandomMultiParticlePGunProducer.h.

Constructor & Destructor Documentation

FlatRandomMultiParticlePGunProducer::FlatRandomMultiParticlePGunProducer ( const ParameterSet & pset )

Definition at line 18 of file FlatRandomMultiParticlePGunProducer.cc.

References gather_cfg::cout, fMaxP_, fMinP_, edm::BaseFlatGunProducer::fPartIDs, fProbParticle_, edm::ParameterSet::getParameter(), and gen::k.

                                                                                                  : BaseFlatGunProducer(pset) {
 
   ParameterSet pgunParams = pset.getParameter<ParameterSet>("PGunParameters");
   fProbParticle_ = pgunParams.getParameter<std::vector<double> >("ProbParts");
   fMinP_         = pgunParams.getParameter<double>("MinP");
   fMaxP_         = pgunParams.getParameter<double>("MaxP"); 
   if (fProbParticle_.size() != fPartIDs.size()) throw cms::Exception("Configuration") << "Not all probabilities given for all particle types " << fProbParticle_.size() << ":" << fPartIDs.size() << " need them to match\n";
 
   produces<HepMCProduct>("unsmeared");
   produces<GenEventInfoProduct>();
 #ifdef DebugLog
   std::cout << "Internal FlatRandomPGun is initialzed for " << fPartIDs.size()
         << " particles in momentum range " << fMinP_ << ":" << fMaxP_
         << std::endl ;
   for (unsigned int k=0; k<fPartIDs.size(); ++k)
     std::cout << " [" << k << "] " << fPartIDs[k] << ":" << fProbParticle_[k];
   std::cout << std::endl;
 #endif
   for (unsigned int k=1; k<fProbParticle_.size(); ++k)
     fProbParticle_[k] += fProbParticle_[k-1];
   for (unsigned int k=0; k<fProbParticle_.size(); ++k)
     fProbParticle_[k] /= fProbParticle_[fProbParticle_.size()-1];
 #ifdef DebugLog
   std::cout << "Corrected probabilities:";
   for (unsigned int k=0; k<fProbParticle_.size(); ++k)
     std::cout << "  " << fProbParticle_[k];
   std::cout << std::endl;
 #endif
 }

FlatRandomMultiParticlePGunProducer::~FlatRandomMultiParticlePGunProducer ( )

override

Definition at line 48 of file FlatRandomMultiParticlePGunProducer.cc.

48 {}

Member Function Documentation

void FlatRandomMultiParticlePGunProducer::produce	(	edm::Event &	e,
		const EventSetup &	es
	)

override

Definition at line 50 of file FlatRandomMultiParticlePGunProducer.cc.

                                                    {
 
   edm::Service<edm::RandomNumberGenerator> rng;
   CLHEP::HepRandomEngine* engine = &rng->getEngine(e.streamID());
 
 #ifdef DebugLog
   if (fVerbosity > 0)
     std::cout << "FlatRandomMultiParticlePGunProducer: Begin New Event Generation" << std::endl ; 
 #endif
    
   // event loop (well, another step in it...)
   // no need to clean up GenEvent memory - done in HepMCProduct
   // here re-create fEvt (memory)
   //
   fEvt = new HepMC::GenEvent() ;
    
   // now actualy, cook up the event from PDGTable and gun parameters
   //
 
   // 1st, primary vertex
   //
   HepMC::GenVertex* Vtx = new HepMC::GenVertex( HepMC::FourVector(0.,0.,0.));
   
   // loop over particles
   //
   int    barcode(0), PartID(fPartIDs[0]);
   double r1      = CLHEP::RandFlat::shoot(engine, 0., 1.);
   for (unsigned int ip=0; ip<fPartIDs.size(); ip++) {
     if (r1 <= fProbParticle_[ip]) {
       PartID = fPartIDs[ip];
       break;
     }
   }
 #ifdef DebugLog
   if (fVerbosity > 0) std::cout << "Random " << r1 << " PartID " << PartID
                 << std::endl;
 #endif
   double mom    = CLHEP::RandFlat::shoot(engine, fMinP_,  fMaxP_);
   double eta    = CLHEP::RandFlat::shoot(engine, fMinEta, fMaxEta);
   double phi    = CLHEP::RandFlat::shoot(engine, fMinPhi, fMaxPhi);
   const HepPDT::ParticleData* 
     PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID))) ;
   double mass   = PData->mass().value() ;
   double energy = sqrt(mom*mom + mass*mass);
   double theta  = 2.*atan(exp(-eta));
   double px     = mom*sin(theta)*cos(phi);
   double py     = mom*sin(theta)*sin(phi);
   double pz     = mom*cos(theta);
 
   HepMC::FourVector p(px,py,pz,energy) ;
   HepMC::GenParticle* Part = new HepMC::GenParticle(p,PartID,1);
   barcode++ ;
   Part->suggest_barcode(barcode);
   Vtx->add_particle_out(Part);
 
   if (fAddAntiParticle) {
     HepMC::FourVector ap(-px,-py,-pz,energy) ;
     int APartID = (PartID == 22 || PartID == 23) ? PartID : -PartID;
     HepMC::GenParticle* APart = new HepMC::GenParticle(ap,APartID,1);
     barcode++ ;
     APart->suggest_barcode(barcode);
     Vtx->add_particle_out(APart);
   }
 
   fEvt->add_vertex(Vtx) ;
   fEvt->set_event_number(e.id().event()) ;
   fEvt->set_signal_process_id(20) ;  
    
 #ifdef DebugLog 
   if (fVerbosity > 0) fEvt->print() ;  
 #endif
 
   std::unique_ptr<HepMCProduct> BProduct(new HepMCProduct()) ;
   BProduct->addHepMCData( fEvt );
   e.put(std::move(BProduct), "unsmeared");
 
   std::unique_ptr<GenEventInfoProduct> genEventInfo(new GenEventInfoProduct(fEvt));
   e.put(std::move(genEventInfo));
 #ifdef DebugLog    
    if (fVerbosity > 0) 
      std::cout << " FlatRandomMultiParticlePGunProducer : Event Generation Done " << std::endl;
 #endif
 }