#include <FlatRandomPtGunSource.h>
Public Member Functions | |
FlatRandomPtGunSource (const ParameterSet &, const InputSourceDescription &) | |
virtual | ~FlatRandomPtGunSource () |
Protected Attributes | |
double | fMaxPt |
double | fMinPt |
Private Member Functions | |
virtual bool | produce (Event &e) |
Definition at line 15 of file FlatRandomPtGunSource.h.
FlatRandomPtGunSource::FlatRandomPtGunSource | ( | const ParameterSet & | pset, |
const InputSourceDescription & | desc | ||
) |
Definition at line 20 of file FlatRandomPtGunSource.cc.
References fMaxPt, 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); produces<HepMCProduct>(); }
FlatRandomPtGunSource::~FlatRandomPtGunSource | ( | ) | [virtual] |
Definition at line 37 of file FlatRandomPtGunSource.cc.
{
// no need to cleanup GenEvent memory - done in HepMCProduct
}
bool FlatRandomPtGunSource::produce | ( | Event & | e | ) | [private, virtual] |
Implements edm::ConfigurableInputSource.
Definition at line 42 of file FlatRandomPtGunSource.cc.
References abs, funct::cos(), gather_cfg::cout, relval_parameters_module::energy, eta(), edm::ConfigurableInputSource::event(), funct::exp(), edm::BaseFlatGunSource::fAddAntiParticle, edm::BaseFlatGunSource::fEvt, edm::BaseFlatGunSource::fMaxEta, edm::BaseFlatGunSource::fMaxPhi, fMaxPt, edm::BaseFlatGunSource::fMinEta, edm::BaseFlatGunSource::fMinPhi, fMinPt, edm::BaseFlatGunSource::fPartIDs, edm::BaseFlatGunSource::fPDGTable, edm::BaseFlatGunSource::fRandomGenerator, edm::BaseFlatGunSource::fVerbosity, configurableAnalysis::GenParticle, AlCaHLTBitMon_ParallelJobs::p, RecoTau_DiTaus_pt_20-420_cfg::ParticleID, phi, edm::Event::put(), funct::sin(), mathSSE::sqrt(), and theta().
{ if ( fVerbosity > 0 ) { cout << " FlatRandomPtGunSource : 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(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 ) { HepMC::FourVector ap(-px,-py,-pz,energy) ; int APartID = -PartID ; if ( PartID == 22 || PartID == 23 ) { APartID = PartID ; } 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; }
double edm::FlatRandomPtGunSource::fMaxPt [protected] |
Definition at line 31 of file FlatRandomPtGunSource.h.
Referenced by FlatRandomPtGunSource(), and produce().
double edm::FlatRandomPtGunSource::fMinPt [protected] |
Definition at line 30 of file FlatRandomPtGunSource.h.
Referenced by FlatRandomPtGunSource(), and produce().