#include <FlatRandomEGunSource.h>
Public Member Functions | |
FlatRandomEGunSource (const ParameterSet &, const InputSourceDescription &) | |
virtual | ~FlatRandomEGunSource () |
Protected Attributes | |
double | fMaxE |
double | fMinE |
Private Member Functions | |
virtual bool | produce (Event &e) |
Definition at line 15 of file FlatRandomEGunSource.h.
FlatRandomEGunSource::FlatRandomEGunSource | ( | const ParameterSet & | pset, |
const InputSourceDescription & | desc | ||
) |
Definition at line 19 of file FlatRandomEGunSource.cc.
References gather_cfg::cout, fMaxE, fMinE, edm::ParameterSet::getUntrackedParameter(), and edm::InputSource::remainingEvents().
: BaseFlatGunSource(pset, desc) { ParameterSet defpset ; // ParameterSet pgun_params = pset.getParameter<ParameterSet>("PGunParameters") ; ParameterSet pgun_params = pset.getUntrackedParameter<ParameterSet>("PGunParameters",defpset) ; // doesn't seem necessary to check if pset is empty - if this // is the case, default values will be taken for params fMinE = pgun_params.getUntrackedParameter<double>("MinE",0.99); fMaxE = pgun_params.getUntrackedParameter<double>("MaxE",1.01); produces<HepMCProduct>(); cout << "Internal FlatRandomEGun is initialzed" << endl ; cout << "It is going to generate " << remainingEvents() << "events" << endl ; }
FlatRandomEGunSource::~FlatRandomEGunSource | ( | ) | [virtual] |
Definition at line 41 of file FlatRandomEGunSource.cc.
{
// no need to cleanup fEvt since it's done in HepMCProduct
}
bool FlatRandomEGunSource::produce | ( | Event & | e | ) | [private, virtual] |
Implements edm::ConfigurableInputSource.
Definition at line 46 of file FlatRandomEGunSource.cc.
References abs, funct::cos(), gather_cfg::cout, relval_parameters_module::energy, eta(), edm::ConfigurableInputSource::event(), funct::exp(), edm::BaseFlatGunSource::fAddAntiParticle, edm::BaseFlatGunSource::fEvt, fMaxE, edm::BaseFlatGunSource::fMaxEta, edm::BaseFlatGunSource::fMaxPhi, fMinE, edm::BaseFlatGunSource::fMinEta, edm::BaseFlatGunSource::fMinPhi, edm::BaseFlatGunSource::fPartIDs, edm::BaseFlatGunSource::fPDGTable, edm::BaseFlatGunSource::fRandomGenerator, edm::BaseFlatGunSource::fVerbosity, configurableAnalysis::GenParticle, scaleCards::mass, AlCaHLTBitMon_ParallelJobs::p, RecoTau_DiTaus_pt_20-420_cfg::ParticleID, phi, edm::Event::put(), funct::sin(), mathSSE::sqrt(), and theta().
{ if ( fVerbosity > 0 ) { cout << " FlatRandomEGunSource : 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 energy = fRandomGenerator->fire(fMinE, fMaxE) ; 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 mom2 = energy*energy - mass*mass ; double mom = 0. ; if (mom2 > 0.) { mom = sqrt(mom2) ; } else { mom = 0. ; } double theta = 2.*atan(exp(-eta)) ; double px = mom*sin(theta)*cos(phi) ; double py = mom*sin(theta)*sin(phi) ; double pz = mom*cos(theta) ; 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() ; // for some strange reason, it prints NO event info // after it's been put into edm::Event... cout << " FlatRandomEGunSource : Event Generation Done " << endl; } return true; }
double edm::FlatRandomEGunSource::fMaxE [protected] |
Definition at line 31 of file FlatRandomEGunSource.h.
Referenced by FlatRandomEGunSource(), and produce().
double edm::FlatRandomEGunSource::fMinE [protected] |
Definition at line 30 of file FlatRandomEGunSource.h.
Referenced by FlatRandomEGunSource(), and produce().