CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
edm::FlatRandomPtGunProducer Class Reference

#include <FlatRandomPtGunProducer.h>

Inheritance diagram for edm::FlatRandomPtGunProducer:
edm::BaseFlatGunProducer edm::one::EDProducer< one::WatchRuns, EndRunProducer > edm::one::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 FlatRandomPtGunProducer (const ParameterSet &pset)
 
virtual void produce (Event &e, const EventSetup &es) override
 
virtual ~FlatRandomPtGunProducer ()
 
- Public Member Functions inherited from edm::BaseFlatGunProducer
 BaseFlatGunProducer (const ParameterSet &)
 
void beginRun (const edm::Run &r, const edm::EventSetup &) override
 
void endRun (edm::Run const &r, const edm::EventSetup &) override
 
void endRunProduce (edm::Run &r, const edm::EventSetup &) override
 
virtual ~BaseFlatGunProducer ()
 
- Public Member Functions inherited from edm::one::EDProducer< one::WatchRuns, EndRunProducer >
 EDProducer ()=default
 
- Public Member Functions inherited from edm::one::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
virtual ~EDProducerBase ()
 
- Public Member Functions inherited from edm::ProducerBase
void callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
 
std::vector< edm::ProductResolverIndex > const & indiciesForPutProducts (BranchType iBranchType) const
 
 ProducerBase ()
 
void registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
 
std::function< void(BranchDescription const &)> registrationCallback () const
 used by the fwk to register list of products More...
 
void resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)
 
virtual ~ProducerBase () noexcept(false)
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ProductResolverIndexAndSkipBit indexFrom (EDGetToken, BranchType, TypeID const &) const
 
void itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
void itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
 
std::vector< ProductResolverIndexAndSkipBit > const & itemsToGetFrom (BranchType iType) const
 
void labelsForToken (EDGetToken iToken, Labels &oLabels) const
 
void modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
 
EDConsumerBase const & operator= (EDConsumerBase const &)=delete
 
EDConsumerBaseoperator= (EDConsumerBase &&)=default
 
bool registeredToConsume (ProductResolverIndex, bool, BranchType) const
 
bool registeredToConsumeMany (TypeID const &, BranchType) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Private Attributes

double fMaxPt
 
double fMinPt
 

Additional Inherited Members

- Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList TypeLabelList
 
- Public Types inherited from edm::EDConsumerBase
typedef ProductLabels Labels
 
- Static Public Member Functions inherited from edm::one::EDProducerBase
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
- Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > consumes (edm::InputTag const &tag)
 
EDGetToken consumes (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken consumes (TypeToGet const &id, edm::InputTag const &tag)
 
ConsumesCollector consumesCollector ()
 Use a ConsumesCollector to gather consumes information from helper functions. More...
 
template<typename ProductType , BranchType B = InEvent>
void consumesMany ()
 
void consumesMany (const TypeToGet &id)
 
template<BranchType B>
void consumesMany (const TypeToGet &id)
 
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType > mayConsume (edm::InputTag const &tag)
 
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
template<BranchType B>
EDGetToken mayConsume (const TypeToGet &id, edm::InputTag const &tag)
 
- 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::ParticleDataTablefPDGTable
 
int fVerbosity
 

Detailed Description

Definition at line 15 of file FlatRandomPtGunProducer.h.

Constructor & Destructor Documentation

FlatRandomPtGunProducer::FlatRandomPtGunProducer ( const ParameterSet pset)

Definition at line 22 of file FlatRandomPtGunProducer.cc.

References fMaxPt, fMinPt, and edm::ParameterSet::getParameter().

22  :
24 {
25 
26 
27  ParameterSet defpset ;
28  ParameterSet pgun_params =
29  pset.getParameter<ParameterSet>("PGunParameters") ;
30 
31  fMinPt = pgun_params.getParameter<double>("MinPt");
32  fMaxPt = pgun_params.getParameter<double>("MaxPt");
33 
34  produces<HepMCProduct>("unsmeared");
35  produces<GenEventInfoProduct>();
36 }
T getParameter(std::string const &) const
BaseFlatGunProducer(const ParameterSet &)
FlatRandomPtGunProducer::~FlatRandomPtGunProducer ( )
virtual

Definition at line 38 of file FlatRandomPtGunProducer.cc.

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

Member Function Documentation

void FlatRandomPtGunProducer::produce ( Event e,
const EventSetup es 
)
overridevirtual

Definition at line 43 of file FlatRandomPtGunProducer.cc.

References funct::abs(), funct::cos(), gather_cfg::cout, stringResolutionProvider_cfi::eta, edm::EventID::event(), JetChargeProducer_cfi::exp, edm::BaseFlatGunProducer::fAddAntiParticle, edm::BaseFlatGunProducer::fEvt, edm::BaseFlatGunProducer::fMaxEta, edm::BaseFlatGunProducer::fMaxPhi, fMaxPt, edm::BaseFlatGunProducer::fMinEta, edm::BaseFlatGunProducer::fMinPhi, fMinPt, edm::BaseFlatGunProducer::fPartIDs, edm::BaseFlatGunProducer::fPDGTable, edm::BaseFlatGunProducer::fVerbosity, GenParticle::GenParticle, edm::RandomNumberGenerator::getEngine(), edm::EventBase::id(), ResonanceBuilder::mass, eostools::move(), AlCaHLTBitMon_ParallelJobs::p, source_particleGun_cfi::ParticleID, CastorDigiValidation::PartID, phi, EnergyCorrector::pt, edm::Event::put(), funct::sin(), mathSSE::sqrt(), edm::Event::streamID(), and theta().

44 {
46  CLHEP::HepRandomEngine* engine = &rng->getEngine(e.streamID());
47 
48  if ( fVerbosity > 0 )
49  {
50  cout << " FlatRandomPtGunProducer : Begin New Event Generation" << endl ;
51  }
52  // event loop (well, another step in it...)
53 
54  // no need to clean up GenEvent memory - done in HepMCProduct
55  //
56 
57  // here re-create fEvt (memory)
58  //
59  fEvt = new HepMC::GenEvent() ;
60 
61  // now actualy, cook up the event from PDGTable and gun parameters
62  //
63  // 1st, primary vertex
64  //
65  HepMC::GenVertex* Vtx = new HepMC::GenVertex(HepMC::FourVector(0.,0.,0.));
66 
67  // loop over particles
68  //
69  int barcode = 1 ;
70  for (unsigned int ip=0; ip<fPartIDs.size(); ++ip)
71  {
72  double pt = CLHEP::RandFlat::shoot(engine, fMinPt, fMaxPt) ;
73  double eta = CLHEP::RandFlat::shoot(engine, fMinEta, fMaxEta) ;
74  double phi = CLHEP::RandFlat::shoot(engine, fMinPhi, fMaxPhi) ;
75  int PartID = fPartIDs[ip] ;
76  const HepPDT::ParticleData*
77  PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID))) ;
78  double mass = PData->mass().value() ;
79  double theta = 2.*atan(exp(-eta)) ;
80  double mom = pt/sin(theta) ;
81  double px = pt*cos(phi) ;
82  double py = pt*sin(phi) ;
83  double pz = mom*cos(theta) ;
84  double energy2= mom*mom + mass*mass ;
85  double energy = sqrt(energy2) ;
86  HepMC::FourVector p(px,py,pz,energy) ;
87  HepMC::GenParticle* Part =
88  new HepMC::GenParticle(p,PartID,1);
89  Part->suggest_barcode( barcode ) ;
90  barcode++ ;
91  Vtx->add_particle_out(Part);
92 
93  if ( fAddAntiParticle )
94  {
95  HepMC::FourVector ap(-px,-py,-pz,energy) ;
96  int APartID = -PartID ;
97  if ( PartID == 22 || PartID == 23 )
98  {
99  APartID = PartID ;
100  }
101  HepMC::GenParticle* APart =
102  new HepMC::GenParticle(ap,APartID,1);
103  APart->suggest_barcode( barcode ) ;
104  barcode++ ;
105  Vtx->add_particle_out(APart) ;
106  }
107 
108  }
109 
110  fEvt->add_vertex(Vtx) ;
111  fEvt->set_event_number(e.id().event()) ;
112  fEvt->set_signal_process_id(20) ;
113 
114  if ( fVerbosity > 0 )
115  {
116  fEvt->print() ;
117  }
118 
119  unique_ptr<HepMCProduct> BProduct(new HepMCProduct()) ;
120  BProduct->addHepMCData( fEvt );
121  e.put(std::move(BProduct), "unsmeared");
122 
123  unique_ptr<GenEventInfoProduct> genEventInfo(new GenEventInfoProduct(fEvt));
124  e.put(std::move(genEventInfo));
125 
126  if ( fVerbosity > 0 )
127  {
128  // for testing purpose only
129  // fEvt->print() ; // prints empty info after it's made into edm::Event
130  cout << " FlatRandomPtGunProducer : Event Generation Done " << endl;
131  }
132 }
EventNumber_t event() const
Definition: EventID.h:41
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:122
Sin< T >::type sin(const T &t)
Definition: Sin.h:22
Geom::Theta< T > theta() const
virtual CLHEP::HepRandomEngine & getEngine(StreamID const &)=0
Use this engine in event methods.
ESHandle< HepPDT::ParticleDataTable > fPDGTable
T sqrt(T t)
Definition: SSEVec.h:18
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
std::vector< int > fPartIDs
HepPDT::ParticleData ParticleData
edm::EventID id() const
Definition: EventBase.h:60
StreamID streamID() const
Definition: Event.h:81
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

double edm::FlatRandomPtGunProducer::fMaxPt
private

Definition at line 29 of file FlatRandomPtGunProducer.h.

Referenced by FlatRandomPtGunProducer(), and produce().

double edm::FlatRandomPtGunProducer::fMinPt
private

Definition at line 28 of file FlatRandomPtGunProducer.h.

Referenced by FlatRandomPtGunProducer(), and produce().