#include <ExpoRandomPtGunProducer.h>

Inheritance diagram for edm::ExpoRandomPtGunProducer:

Public Member Functions
	ExpoRandomPtGunProducer (const ParameterSet &pset)

virtual	~ExpoRandomPtGunProducer ()

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

virtual	~BaseFlatGunProducer ()

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

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

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Protected Attributes
double	fMaxPt

double	fMeanPt

double	fMinPt

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
virtual 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
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

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::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

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)

Detailed Description

Definition at line 15 of file ExpoRandomPtGunProducer.h.

Constructor & Destructor Documentation

ExpoRandomPtGunProducer::ExpoRandomPtGunProducer ( const ParameterSet & pset )

Definition at line 23 of file ExpoRandomPtGunProducer.cc.

References fMaxPt, fMeanPt, fMinPt, and edm::ParameterSet::getParameter().

                                                                          : 
    BaseFlatGunProducer(pset)
 {
 
 
    ParameterSet defpset ;
    ParameterSet pgun_params = 
       pset.getParameter<ParameterSet>("PGunParameters") ;
   
    fMinPt = pgun_params.getParameter<double>("MinPt");
    fMaxPt = pgun_params.getParameter<double>("MaxPt");
 
    fMeanPt = pgun_params.getParameter<double>("MeanPt");
 
    produces<HepMCProduct>();
    produces<GenEventInfoProduct>();
 }

ExpoRandomPtGunProducer::~ExpoRandomPtGunProducer ( )

virtual

Definition at line 41 of file ExpoRandomPtGunProducer.cc.

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

Member Function Documentation

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

overrideprivatevirtual

Implements edm::one::EDProducerBase.

Definition at line 46 of file ExpoRandomPtGunProducer.cc.

 {
    edm::Service<edm::RandomNumberGenerator> rng;
    CLHEP::HepRandomEngine* engine = &rng->getEngine(e.streamID());
 
    if ( fVerbosity > 0 )
    {
       cout << " ExpoRandomPtGunProducer : Begin New Event Generation" << endl ; 
    }
    // 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(CLHEP::HepLorentzVector(0.,0.,0.));
    HepMC::GenVertex* Vtx = new HepMC::GenVertex(HepMC::FourVector(0.,0.,0.));
 
    // loop over particles
    //
    int barcode = 1 ;
    for (unsigned int ip=0; ip<fPartIDs.size(); ++ip)
    {
 
      //the max is to ensure you don't generate at 0
      //the 90% is to get rid of edge effect
      
      double pt     =  std::max(0.00001,0.90*fMinPt)+CLHEP::RandExponential::shoot(engine, fMeanPt);
      //shoot until in the designated range
      while (pt<fMinPt || pt>fMaxPt)
        {pt = std::max(0.00001,0.90*fMinPt) + CLHEP::RandExponential::shoot(engine, fMeanPt);}
      
        double eta    = CLHEP::RandFlat::shoot(engine, fMinEta, fMaxEta) ;
        double phi    = CLHEP::RandFlat::shoot(engine, fMinPhi, fMaxPhi) ;
        int PartID = fPartIDs[ip] ;
        const HepPDT::ParticleData* 
           PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID))) ;
        double mass   = PData->mass().value() ;
        double theta  = 2.*atan(exp(-eta)) ;
        double mom    = pt/sin(theta) ;
        double px     = pt*cos(phi) ;
        double py     = pt*sin(phi) ;
        double pz     = mom*cos(theta) ;
        double energy2= mom*mom + mass*mass ;
        double energy = sqrt(energy2) ; 
        //CLHEP::Hep3Vector p(px,py,pz) ;
        //HepMC::GenParticle* Part = 
        //    new HepMC::GenParticle(CLHEP::HepLorentzVector(p,energy),PartID,1);
        HepMC::FourVector p(px,py,pz,energy) ;
        HepMC::GenParticle* Part = 
            new HepMC::GenParticle(p,PartID,1);
        Part->suggest_barcode( barcode ) ;
        barcode++ ;
        Vtx->add_particle_out(Part);
 
        if ( fAddAntiParticle )
        {
           //CLHEP::Hep3Vector ap(-px,-py,-pz) ;
           HepMC::FourVector ap(-px,-py,-pz,energy) ;
       int APartID = -PartID ;
       if ( PartID == 22 || PartID == 23 )
       {
          APartID = PartID ;
       }   
       //HepMC::GenParticle* APart =
       //   new HepMC::GenParticle(CLHEP::HepLorentzVector(ap,energy),APartID,1);
       HepMC::GenParticle* APart =
          new HepMC::GenParticle(ap,APartID,1);
       APart->suggest_barcode( barcode ) ;
       barcode++ ;
       Vtx->add_particle_out(APart) ;
        }
 
    }
 
    fEvt->add_vertex(Vtx) ;
    fEvt->set_event_number(e.id().event()) ;
    fEvt->set_signal_process_id(20) ; 
         
    if ( fVerbosity > 0 )
    {
       fEvt->print() ;  
    }
 
    auto_ptr<HepMCProduct> BProduct(new HepMCProduct()) ;
    BProduct->addHepMCData( fEvt );
    e.put(BProduct);
 
    auto_ptr<GenEventInfoProduct> genEventInfo(new GenEventInfoProduct(fEvt));
    e.put(genEventInfo);
     
    if ( fVerbosity > 0 )
    {
       // for testing purpose only
       // fEvt->print() ; // prints empty info after it's made into edm::Event
       cout << " FlatRandomPtGunProducer : Event Generation Done " << endl;
    }
 }