edm::FlatRandomMultiParticlePGunProducer Class Reference

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

{}

Member Function Documentation

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

override

Definition at line 50 of file FlatRandomMultiParticlePGunProducer.cc.

References funct::abs(), funct::cos(), gather_cfg::cout, PVValHelper::eta, edm::EventID::event(), JetChargeProducer_cfi::exp, edm::BaseFlatGunProducer::fAddAntiParticle, edm::BaseFlatGunProducer::fEvt, edm::BaseFlatGunProducer::fMaxEta, fMaxP_, edm::BaseFlatGunProducer::fMaxPhi, edm::BaseFlatGunProducer::fMinEta, fMinP_, edm::BaseFlatGunProducer::fMinPhi, edm::BaseFlatGunProducer::fPartIDs, edm::BaseFlatGunProducer::fPDGTable, fProbParticle_, edm::BaseFlatGunProducer::fVerbosity, particlelevel_cff::genEventInfo, GenParticle::GenParticle, edm::RandomNumberGenerator::getEngine(), edm::EventBase::id(), ResonanceBuilder::mass, eostools::move(), AlCaHLTBitMon_ParallelJobs::p, source_particleGun_cfi::ParticleID, CastorDigiValidation::PartID, phi, edm::Event::put(), diffTwoXMLs::r1, funct::sin(), mathSSE::sqrt(), edm::Event::streamID(), and theta().

                                                   {

  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
}

Member Data Documentation

double edm::FlatRandomMultiParticlePGunProducer::fMaxP_

private

Definition at line 22 of file FlatRandomMultiParticlePGunProducer.h.

Referenced by FlatRandomMultiParticlePGunProducer(), and produce().

double edm::FlatRandomMultiParticlePGunProducer::fMinP_

private

Definition at line 21 of file FlatRandomMultiParticlePGunProducer.h.

Referenced by FlatRandomMultiParticlePGunProducer(), and produce().

std::vector<double> edm::FlatRandomMultiParticlePGunProducer::fProbParticle_

private

Definition at line 20 of file FlatRandomMultiParticlePGunProducer.h.

Referenced by FlatRandomMultiParticlePGunProducer(), and produce().