#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 (edm::Run &r, const edm::EventSetup &)

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

virtual	~BaseFlatGunProducer ()

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

virtual	~EDProducer ()

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

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

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

virtual	~ProducerBase ()

Protected Attributes
double	fMaxPt

double	fMeanPt

double	fMinPt

CLHEP::RandExponential *	fRandomExpoGenerator

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

CLHEP::HepRandomEngine &	fRandomEngine

CLHEP::RandFlat *	fRandomGenerator

int	fVerbosity

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

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

typedef WorkerT< EDProducer >	WorkerType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDProducer
CurrentProcessingContext const *	currentContext () const

Protected Member Functions inherited from edm::ProducerBase
template<class TProducer , class TMethod >
void	callWhenNewProductsRegistered (TProducer *iProd, TMethod iMethod)

Detailed Description

Definition at line 15 of file ExpoRandomPtGunProducer.h.

Constructor & Destructor Documentation

ExpoRandomPtGunProducer::ExpoRandomPtGunProducer ( const ParameterSet & pset )

Definition at line 24 of file ExpoRandomPtGunProducer.cc.

References fMaxPt, fMeanPt, fMinPt, edm::BaseFlatGunProducer::fRandomEngine, fRandomExpoGenerator, 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>();
 
    //the explonential generator
    fRandomExpoGenerator = new CLHEP::RandExponential(fRandomEngine,fMeanPt);
    
 }

ExpoRandomPtGunProducer::~ExpoRandomPtGunProducer ( )

virtual

Definition at line 46 of file ExpoRandomPtGunProducer.cc.

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

Member Function Documentation

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

privatevirtual

Implements edm::EDProducer.

Definition at line 51 of file ExpoRandomPtGunProducer.cc.

 {
 
    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)+fRandomExpoGenerator->fire(fMeanPt);
      //shoot until in the designated range
      while (pt<fMinPt || pt>fMaxPt)
        {pt = std::max(0.00001,0.90*fMinPt) + fRandomExpoGenerator->fire(fMeanPt);}
      
      double eta    = fRandomGenerator->fire(fMinEta, fMaxEta) ;
        double phi    = fRandomGenerator->fire(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;
    }
 }