#include <MultiParticleInConeGunSource.h>

Inheritance diagram for edm::MultiParticleInConeGunSource:

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

virtual	~MultiParticleInConeGunSource ()

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
std::vector< int >	fInConeIds

double	fInConeMaxEta

double	fInConeMaxPhi

unsigned int	fInConeMaxTry

double	fInConeMinEta

double	fInConeMinPhi

double	fMaxDeltaR

double	fMaxMomRatio

double	fMaxPt

double	fMinDeltaR

double	fMinMomRatio

double	fMinPt

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 MultiParticleInConeGunSource.h.

Constructor & Destructor Documentation

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

Definition at line 23 of file MultiParticleInConeGunSource.cc.

References fInConeIds, fInConeMaxEta, fInConeMaxPhi, fInConeMaxTry, fInConeMinEta, fInConeMinPhi, fMaxDeltaR, fMaxMomRatio, fMaxPt, fMinDeltaR, fMinMomRatio, fMinPt, 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);
 
    fInConeIds = pgun_params.getUntrackedParameter< vector<int> >("InConeID",vector<int>());
    fMinDeltaR = pgun_params.getUntrackedParameter<double>("MinDeltaR",0.0);
    fMaxDeltaR = pgun_params.getUntrackedParameter<double>("MaxDeltaR",0.1);
    fMinMomRatio = pgun_params.getUntrackedParameter<double>("MinMomRatio",0.99);
    fMaxMomRatio = pgun_params.getUntrackedParameter<double>("MaxMomRatio",1.01);
 
    fInConeMinEta = pgun_params.getUntrackedParameter<double>("InConeMinEta",-5.5);
    fInConeMaxEta = pgun_params.getUntrackedParameter<double>("InConeMaxEta",5.5);
    fInConeMinPhi = pgun_params.getUntrackedParameter<double>("InConeMinPhi",-3.14159265358979323846);
    fInConeMaxPhi = pgun_params.getUntrackedParameter<double>("InConeMaxPhi",3.14159265358979323846);
    fInConeMaxTry = pgun_params.getUntrackedParameter<unsigned int>("InConeMaxTry",100);
    
    produces<HepMCProduct>();
 }

MultiParticleInConeGunSource::~MultiParticleInConeGunSource ( )

virtual

Definition at line 51 of file MultiParticleInConeGunSource.cc.

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

Member Function Documentation

bool MultiParticleInConeGunSource::produce ( Event & e )

privatevirtual

Implements edm::ConfigurableInputSource.

Definition at line 56 of file MultiParticleInConeGunSource.cc.

 {
 
    if ( fVerbosity > 0 )
    {
       cout << " MultiParticleInConeGunSource : 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)
    {
 
        double pt     = fRandomGenerator->fire(fMinPt, fMaxPt) ;
        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) ; 
 
        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 ){}
 
        // now add the particles in the cone
        for (unsigned iPic=0; iPic!=fInConeIds.size();iPic++){
      unsigned int nTry=0;
      while(true){
        //shoot flat Deltar
        double dR = fRandomGenerator->fire(fMinDeltaR, fMaxDeltaR);
        //shoot flat eta/phi mixing
        double alpha = fRandomGenerator->fire(-3.14159265358979323846, 3.14159265358979323846);
        double dEta = dR*cos(alpha);
        double dPhi = dR*sin(alpha);
        
        /*
        //shoot Energy of associated particle     
        double energyIc = fRandomGenerator->fire(fMinEInCone, fMaxEInCone);
        if (mom2Ic>0){ momIC = sqrt(mom2Ic);}
        */
        //    get kinematics
        double etaIc = eta+dEta;
        double phiIc = phi+dPhi;
        //put it back in -Pi:Pi if necessary. multiple time might be necessary if dR > 3
        const unsigned int maxL=100;
        unsigned int iL=0;
        while(iL++<maxL){
          if (phiIc > 3.14159265358979323846) phiIc-=2*3.14159265358979323846;
          else if(phiIc <-3.14159265358979323846) phiIc+=2*3.14159265358979323846;
          
          if (abs(phiIc)<3.14159265358979323846) break;
        }
          
 
        //allow to skip it if you did not run out of possible drawing
        if (nTry++<=fInConeMaxTry){
          //draw another one if this one is not in acceptance
          if (etaIc<fInConeMinEta || etaIc > fInConeMaxEta) continue;
          if (phiIc<fInConeMinPhi || phiIc > fInConeMaxPhi) continue;
        }
        else{
          if ( fVerbosity > 0 )
            {
          cout << " MultiParticleInConeGunSource : could not produce a particle "
               <<  fInConeIds[iPic]<<" in cone "
               <<  fMinDeltaR<<" to "<<fMaxDeltaR<<" within the "<<fInConeMaxTry<<" allowed tries."<<endl;
            }
          /*      edm::LogError("MultiParticleInConeGunSource")<< " MultiParticleInConeGunSource : could not produce a particle "<<
            fInConeIds[iPic]<<" in cone "<<
            fMinDeltaR<<" to "<<fMaxDeltaR<<" within the "<<fInConeMaxTry<<" allowed tries.";*/
        }
        int PartIDIc=fInConeIds[iPic];
        const HepPDT::ParticleData* 
          PDataIc = fPDGTable->particle(HepPDT::ParticleID(abs(PartIDIc)));
        
        //shoot momentum ratio
        double momR = fRandomGenerator->fire(fMinMomRatio, fMaxMomRatio);
        double massIc= PDataIc->mass().value() ;
        double momIc = momR * mom;
        double energyIc = sqrt(momIc*momIc + massIc*massIc);
 
        double thetaIc = 2.*atan(exp(-etaIc)) ;
        double pxIc = momIc*sin(thetaIc)*cos(phiIc);
        double pyIc = momIc*sin(thetaIc)*sin(phiIc);
        double pzIc = momIc*cos(thetaIc);
 
        HepMC::FourVector pIc(pxIc,pyIc,pzIc,energyIc) ;
        HepMC::GenParticle* PartIc = new HepMC::GenParticle(pIc, PartIDIc, 1);
        PartIc->suggest_barcode( barcode ) ;
        barcode++ ;
        Vtx->add_particle_out(PartIc);
        break;
      }//try many times while not in acceptance
        }//loop over the particle Ids in the cone
    }
 
    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 )
      {
        cout << " MultiParticleInConeGunSource : Event Generation Done " << endl;
      }
 
    return true;
 }