#include <ExpoRandomPtGunSource.h>

Inheritance diagram for edm::ExpoRandomPtGunSource:

Public Member Functions
	ExpoRandomPtGunSource (const ParameterSet &, const InputSourceDescription &)

virtual	~ExpoRandomPtGunSource ()

Public Member Functions inherited from edm::BaseFlatGunSource
	BaseFlatGunSource (const ParameterSet &, const InputSourceDescription &)

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

virtual	~BaseFlatGunSource ()

Public Member Functions inherited from edm::GeneratedInputSource
	GeneratedInputSource (ParameterSet const &pset, InputSourceDescription const &desc)

virtual	~GeneratedInputSource ()

Public Member Functions inherited from edm::ConfigurableInputSource
	ConfigurableInputSource (ParameterSet const &pset, InputSourceDescription const &desc, bool realData=true)

EventNumber_t	event () const

unsigned int	eventCreationDelay () const

LuminosityBlockNumber_t	luminosityBlock () const

unsigned int	numberEventsInLumi () const

unsigned int	numberEventsInRun () const

unsigned int	numberEventsInThisLumi () const

unsigned int	numberEventsInThisRun () const

TimeValue_t	presentTime () const

RunNumber_t	run () const

unsigned int	timeBetweenEvents () const

virtual	~ConfigurableInputSource ()

Public Member Functions inherited from edm::InputSource
boost::shared_ptr < ActivityRegistry >	actReg () const
	Accessor for Activity Registry. More...

void	closeFile (boost::shared_ptr< FileBlock >, bool cleaningUpAfterException)
	close current file More...

void	doBeginJob ()
	Called by framework at beginning of job. More...

void	doBeginLumi (LuminosityBlockPrincipal &lbp)
	Called by framework at beginning of lumi block. More...

void	doBeginRun (RunPrincipal &rp)
	Called by framework at beginning of run. More...

void	doEndJob ()
	Called by framework at end of job. More...

void	doEndLumi (LuminosityBlockPrincipal &lbp, bool cleaningUpAfterException)
	Called by framework at end of lumi block. More...

void	doEndRun (RunPrincipal &rp, bool cleaningUpAfterException)
	Called by framework at end of run. More...

void	doPostForkReacquireResources (boost::shared_ptr< multicore::MessageReceiverForSource >)

void	doPreForkReleaseResources ()
	Called by the framework before forking the process. More...

ProcessingController::ForwardState	forwardState () const

bool	goToEvent (EventID const &eventID)

	InputSource (ParameterSet const &, InputSourceDescription const &)
	Constructor. More...

void	issueReports (EventID const &eventID)
	issue an event report More...

LuminosityBlockNumber_t	luminosityBlock () const
	Accessor for current luminosity block number. More...

boost::shared_ptr < LuminosityBlockAuxiliary >	luminosityBlockAuxiliary () const
	Called by the framework to merge or insert lumi in principal cache. More...

int	markLumi ()
	Mark lumi as read. More...

int	markRun ()
	Mark run as read. More...

int	maxEvents () const

int	maxLuminosityBlocks () const

ModuleDescription const &	moduleDescription () const
	Accessor for 'module' description. More...

ItemType	nextItemType ()

bool	primary () const
	Accessor for primary input source flag. More...

ProcessConfiguration const &	processConfiguration () const
	Accessor for Process Configuration. More...

std::string const &	processGUID () const
	Accessor for global process identifier. More...

ProcessingMode	processingMode () const
	RunsLumisAndEvents (default), RunsAndLumis, or Runs. More...

boost::shared_ptr < ProductRegistry const >	productRegistry () const
	Accessor for product registry. More...

bool	randomAccess () const

void	readAndCacheLumi (bool merge, HistoryAppender &historyAppender)
	Read next luminosity block. More...

void	readAndCacheRun (bool merge, HistoryAppender &historyAppender)
	Read next run. More...

EventPrincipal *	readEvent (boost::shared_ptr< LuminosityBlockPrincipal > lbCache)

EventPrincipal *	readEvent (EventID const &)
	Read a specific event. More...

boost::shared_ptr< FileBlock >	readFile ()
	Read next file. More...

boost::shared_ptr < LuminosityBlockAuxiliary >	readLuminosityBlockAuxiliary ()
	Read next luminosity block Auxilary. More...

boost::shared_ptr< RunAuxiliary >	readRunAuxiliary ()
	Read next run Auxiliary. More...

ProcessHistoryID const &	reducedProcessHistoryID () const

void	registerProducts ()
	Register any produced products. More...

int	remainingEvents () const

int	remainingLuminosityBlocks () const

void	repeat ()
	Reset the remaining number of events/lumis to the maximum number. More...

ProcessingController::ReverseState	reverseState () const

void	rewind ()
	Begin again at the first event. More...

RunNumber_t	run () const
	Accessor for current run number. More...

boost::shared_ptr< RunAuxiliary >	runAuxiliary () const
	Called by the framework to merge or insert run in principal cache. More...

void	setLuminosityBlockNumber_t (LuminosityBlockNumber_t lb)
	Set the luminosity block ID. More...

void	setRunNumber (RunNumber_t r)
	Set the run number. More...

void	skipEvents (int offset)

Timestamp const &	timestamp () const
	Accessor for the current time, as seen by the input source. More...

virtual	~InputSource ()
	Destructor. More...

Protected Attributes
double	fMaxPt

double	fMeanPt

double	fMinPt

CLHEP::RandExponential *	fRandomExpoGenerator

Protected Attributes inherited from edm::BaseFlatGunSource
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 bool	produce (Event &e)

Additional Inherited Members
Public Types inherited from edm::InputSource
enum	ItemType { IsInvalid, IsStop, IsFile, IsRun, IsLumi, IsEvent, IsRepeat }

enum	ProcessingMode { Runs, RunsAndLumis, RunsLumisAndEvents }

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::ConfigurableInputSource
static void	fillDescription (ParameterSetDescription &desc)

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

static void	fillDescription (ParameterSetDescription &desc)

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

Protected Member Functions inherited from edm::ConfigurableInputSource
void	reallyReadEvent ()

void	setEventNumber (EventNumber_t e)

void	setTime (TimeValue_t t)

Protected Member Functions inherited from edm::InputSource
void	decreaseRemainingEventsBy (int iSkipped)

EventPrincipal *	eventPrincipalCache ()

boost::shared_ptr < LuminosityBlockPrincipal > const	luminosityBlockPrincipal () const

PrincipalCache const &	principalCache () const

PrincipalCache &	principalCache ()

ProductRegistry &	productRegistryUpdate () const

void	reset () const

void	resetLuminosityBlockAuxiliary () const

void	resetRunAuxiliary () const

boost::shared_ptr < RunPrincipal > const	runPrincipal () const

void	setLuminosityBlockAuxiliary (LuminosityBlockAuxiliary *lbp)

void	setRunAuxiliary (RunAuxiliary *rp)

void	setTimestamp (Timestamp const &theTime)
	To set the current time, as seen by the input source. More...

ItemType	state () const

Detailed Description

Definition at line 15 of file ExpoRandomPtGunSource.h.

Constructor & Destructor Documentation

ExpoRandomPtGunSource::ExpoRandomPtGunSource	(	const ParameterSet &	pset,
		const InputSourceDescription &	desc
	)

Definition at line 23 of file ExpoRandomPtGunSource.cc.

References fMaxPt, fMeanPt, fMinPt, edm::BaseFlatGunSource::fRandomEngine, fRandomExpoGenerator, and edm::ParameterSet::getUntrackedParameter().

                                                                                  : 
    BaseFlatGunSource(pset, desc)
 {
 
 
    ParameterSet defpset ;
    ParameterSet pgun_params = 
       pset.getUntrackedParameter<ParameterSet>("PGunParameters",defpset) ;
   
    fMinPt = pgun_params.getUntrackedParameter<double>("MinPt",0.99);
    fMaxPt = pgun_params.getUntrackedParameter<double>("MaxPt",1.01);
 
    fMeanPt = pgun_params.getUntrackedParameter<double>("MeanPt",-1.);
 
    produces<HepMCProduct>();
 
    //the explonential generator
    fRandomExpoGenerator = new CLHEP::RandExponential(fRandomEngine,fMeanPt);
    
 }

ExpoRandomPtGunSource::~ExpoRandomPtGunSource ( )

virtual

Definition at line 45 of file ExpoRandomPtGunSource.cc.

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

Member Function Documentation

bool ExpoRandomPtGunSource::produce ( Event & e )

privatevirtual

Implements edm::ConfigurableInputSource.

Definition at line 50 of file ExpoRandomPtGunSource.cc.

 {
 
    if ( fVerbosity > 0 )
    {
       cout << " ExpoRandomPtGunSource : 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(event()) ;
    fEvt->set_signal_process_id(20) ; 
         
    if ( fVerbosity > 0 )
    {
       fEvt->print() ;  
    }
 
    auto_ptr<HepMCProduct> BProduct(new HepMCProduct()) ;
    BProduct->addHepMCData( fEvt );
    e.put(BProduct);
     
    if ( fVerbosity > 0 )
    {
       // for testing purpose only
       // fEvt->print() ; // prints empty info after it's made into edm::Event
       cout << " FlatRandomPtGunSource : Event Generation Done " << endl;
    }
 
    return true;
 }