#include <CloseByParticleGunProducer.h>

Inheritance diagram for edm::CloseByParticleGunProducer:

Public Member Functions
	CloseByParticleGunProducer (const ParameterSet &)

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

Protected Attributes
double	fDelta

double	fEnMax

double	fEnMin

int	fNParticles

bool	fOverlapping = false

std::vector< int >	fPartIDs

double	fPhiMax

double	fPhiMin

bool	fPointing = false

bool	fRandomShoot = false

double	fRMax

double	fRMin

double	fZMax

double	fZMin

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

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

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)

Detailed Description

Definition at line 9 of file CloseByParticleGunProducer.h.

Constructor & Destructor Documentation

CloseByParticleGunProducer::CloseByParticleGunProducer ( const ParameterSet & pset )

Definition at line 24 of file CloseByParticleGunProducer.cc.

References fDelta, fEnMax, fEnMin, fNParticles, fOverlapping, fPartIDs, fPhiMax, fPhiMin, fPointing, fRandomShoot, fRMax, fRMin, fZMax, fZMin, and edm::ParameterSet::getParameter().

                                                                                :
    BaseFlatGunProducer(pset)
 {
 
   ParameterSet defpset ;
   ParameterSet pgun_params =
     pset.getParameter<ParameterSet>("PGunParameters") ;
 
   fEnMax = pgun_params.getParameter<double>("EnMax");
   fEnMin = pgun_params.getParameter<double>("EnMin");
   fRMax = pgun_params.getParameter<double>("RMax");
   fRMin = pgun_params.getParameter<double>("RMin");
   fZMax = pgun_params.getParameter<double>("ZMax");
   fZMin = pgun_params.getParameter<double>("ZMin");
   fDelta = pgun_params.getParameter<double>("Delta");
   fPhiMin = pgun_params.getParameter<double>("MinPhi");
   fPhiMax = pgun_params.getParameter<double>("MaxPhi");
   fPointing = pgun_params.getParameter<bool>("Pointing");
   fOverlapping = pgun_params.getParameter<bool>("Overlapping");
   fRandomShoot = pgun_params.getParameter<bool>("RandomShoot");
   fNParticles = pgun_params.getParameter<int>("NParticles");
   fPartIDs = pgun_params.getParameter< vector<int> >("PartID");
 
   produces<HepMCProduct>("unsmeared");
   produces<GenEventInfoProduct>();
 
 }

CloseByParticleGunProducer::~CloseByParticleGunProducer ( )

override

Definition at line 52 of file CloseByParticleGunProducer.cc.

 {
    // no need to cleanup GenEvent memory - done in HepMCProduct
 }

Member Function Documentation

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

overrideprivate

Definition at line 57 of file CloseByParticleGunProducer.cc.

References funct::abs(), EnergyCorrector::c, constexpr, funct::cos(), edm::EventID::event(), fDelta, fEnMax, fEnMin, edm::BaseFlatGunProducer::fEvt, fNParticles, fOverlapping, fPartIDs, edm::BaseFlatGunProducer::fPDGTable, fPhiMax, fPhiMin, fPointing, fRandomShoot, fRMax, fRMin, edm::BaseFlatGunProducer::fVerbosity, fZMax, fZMin, particlelevel_cff::genEventInfo, GenParticle::GenParticle, edm::RandomNumberGenerator::getEngine(), mps_fire::i, edm::EventBase::id(), LogDebug, ResonanceBuilder::mass, eostools::move(), AlCaHLTBitMon_ParallelJobs::p, source_particleGun_cfi::ParticleID, HadronAndPartonSelector_cfi::particles, CastorDigiValidation::PartID, partIdx(), edm::Event::put(), funct::sin(), mathSSE::sqrt(), edm::Event::streamID(), and hgcalLayerClusters_cff::timeOffset.

 {
    edm::Service<edm::RandomNumberGenerator> rng;
    CLHEP::HepRandomEngine* engine = &rng->getEngine(e.streamID());
 
    if ( fVerbosity > 0 )
      {
        LogDebug("CloseByParticleGunProducer") << " CloseByParticleGunProducer : Begin New Event Generation" << endl ;
      }
    fEvt = new HepMC::GenEvent() ;
 
    // loop over particles
    //
    int barcode = 1 ;
    int numParticles = fRandomShoot ? CLHEP::RandFlat::shoot(engine, 1, fNParticles) : fNParticles;
    std::vector<int> particles;
 
    for(int i=0; i<numParticles; i++){
      int partIdx = CLHEP::RandFlat::shoot(engine, 0, fPartIDs.size());
      particles.push_back(fPartIDs[partIdx]);
      }
 
    double phi = CLHEP::RandFlat::shoot(engine, fPhiMin, fPhiMax);
    double fR = CLHEP::RandFlat::shoot(engine,fRMin,fRMax);
    double fZ = CLHEP::RandFlat::shoot(engine,fZMin,fZMax);
    double tmpPhi = phi;
    double tmpR = fR;
 
    for (unsigned int ip=0; ip<particles.size(); ++ip)
    {
      if(fOverlapping)
        {
         fR = CLHEP::RandFlat::shoot(engine,tmpR-fDelta,tmpR+fDelta);
         phi = CLHEP::RandFlat::shoot(engine, tmpPhi-fDelta/fR, tmpPhi+fDelta/fR);
        }
      else
        phi += fDelta/fR;
        
      double fEn = CLHEP::RandFlat::shoot(engine,fEnMin,fEnMax);
      int PartID = particles[ip] ;
      const HepPDT::ParticleData *PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID))) ;
      double mass   = PData->mass().value() ;
      double mom    = sqrt(fEn*fEn-mass*mass);
      double px     = 0.;
      double py     = 0.;
      double pz     = mom;
      double energy = fEn;
 
      // Compute Vertex Position
      double x=fR*cos(phi);
      double y=fR*sin(phi);
      constexpr double c= 2.99792458e+1; // cm/ns
      double timeOffset = sqrt(x*x + y*y + fZ*fZ)/c*ns*c_light;
      HepMC::GenVertex* Vtx = new HepMC::GenVertex(HepMC::FourVector(x*cm,y*cm,fZ*cm,timeOffset));
 
      HepMC::FourVector p(px,py,pz,energy) ;
      // If we are requested to be pointing to (0,0,0), correct the momentum direction
      if (fPointing) {
        math::XYZVector direction(x,y,fZ);
        math::XYZVector momentum = direction.unit() * mom;
        p.setX(momentum.x());
        p.setY(momentum.y());
        p.setZ(momentum.z());
      }
      HepMC::GenParticle* Part = new HepMC::GenParticle(p,PartID,1);
      Part->suggest_barcode( barcode );
      barcode++;
 
      Vtx->add_particle_out(Part);
 
      if (fVerbosity > 0) {
        Vtx->print();
        Part->print();
      }
      fEvt->add_vertex(Vtx);
    }
 
 
    fEvt->set_event_number(e.id().event());
    fEvt->set_signal_process_id(20);
 
    if ( fVerbosity > 0 )
    {
       fEvt->print();
    }
 
    unique_ptr<HepMCProduct> BProduct(new HepMCProduct());
    BProduct->addHepMCData( fEvt );
    e.put(std::move(BProduct), "unsmeared");
 
    unique_ptr<GenEventInfoProduct> genEventInfo(new GenEventInfoProduct(fEvt));
    e.put(std::move(genEventInfo));
 
    if ( fVerbosity > 0 )
      {
        LogDebug("CloseByParticleGunProducer") << " CloseByParticleGunProducer : Event Generation Done " << endl;
      }
 
    particles.clear();
 }