CMS 3D CMS Logo

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

#include <CloseByParticleGunProducer.h>

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

Public Member Functions

 CloseByParticleGunProducer (const ParameterSet &)
 
 ~CloseByParticleGunProducer () override
 
- 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
 
 ~BaseFlatGunProducer () override
 
- Public Member Functions inherited from edm::one::EDProducer< one::WatchRuns, EndRunProducer >
 EDProducer ()=default
 
SerialTaskQueueglobalLuminosityBlocksQueue () final
 
SerialTaskQueueglobalRunsQueue () final
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndRuns () const final
 
bool wantsGlobalLuminosityBlocks () const final
 
bool wantsGlobalRuns () const final
 
- Public Member Functions inherited from edm::one::EDProducerBase
 EDProducerBase ()
 
ModuleDescription const & moduleDescription () const
 
bool wantsStreamLuminosityBlocks () const
 
bool wantsStreamRuns () const
 
 ~EDProducerBase () override
 
- 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 ()
 
std::vector< edm::ProductResolverIndex > const & putTokenIndexToProductResolverIndex () const
 
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, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
 
 ~ProducerBase () noexcept(false) override
 
- Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfoconsumesInfo () const
 
void convertCurrentProcessAlias (std::string const &processName)
 Convert "@currentProcess" in InputTag process names to the actual current process name. More...
 
 EDConsumerBase ()
 
 EDConsumerBase (EDConsumerBase const &)=delete
 
 EDConsumerBase (EDConsumerBase &&)=default
 
ESProxyIndex const * esGetTokenIndices (edm::Transition iTrans) const
 
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
 
ProductResolverIndexAndSkipBit uncheckedIndexFrom (EDGetToken) const
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
void updateLookup (eventsetup::ESRecordsToProxyIndices const &)
 
virtual ~EDConsumerBase () noexcept(false)
 

Protected Attributes

double fDelta
 
double fEnMax
 
double fEnMin
 
int fNParticles
 
bool fOverlapping = false
 
std::vector< int > fPartIDs
 
double fPhiMax
 
double fPhiMin
 
bool fPointing = false
 
bool fRandomShoot = false
 
double fRMax
 
double fRMin
 
double fZMax
 
double fZMin
 
- Protected Attributes inherited from edm::BaseFlatGunProducer
bool fAddAntiParticle
 
HepMC::GenEventfEvt
 
double fMaxEta
 
double fMaxPhi
 
double fMinEta
 
double fMinPhi
 
std::vector< int > fPartIDs
 
ESHandle< HepPDT::ParticleDataTablefPDGTable
 
int fVerbosity
 

Private Member Functions

void produce (Event &e, const EventSetup &es) override
 

Additional Inherited Members

- Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase ModuleType
 
- Public Types inherited from edm::ProducerBase
using ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
 
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::ProducerBase
ProducesCollector producesCollector ()
 
- 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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes ()
 
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto esConsumes (ESInputTag const &tag)
 
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)
 

Detailed Description

Definition at line 8 of file CloseByParticleGunProducer.h.

Constructor & Destructor Documentation

CloseByParticleGunProducer::CloseByParticleGunProducer ( const ParameterSet pset)

Definition at line 24 of file CloseByParticleGunProducer.cc.

References fDelta, fEnMax, fEnMin, fNParticles, fOverlapping, fPartIDs, fPhiMax, fPhiMin, fPointing, fRandomShoot, fRMax, fRMin, fZMax, fZMin, and edm::ParameterSet::getParameter().

24  : BaseFlatGunProducer(pset) {
25  ParameterSet defpset;
26  ParameterSet pgun_params = pset.getParameter<ParameterSet>("PGunParameters");
27 
28  fEnMax = pgun_params.getParameter<double>("EnMax");
29  fEnMin = pgun_params.getParameter<double>("EnMin");
30  fRMax = pgun_params.getParameter<double>("RMax");
31  fRMin = pgun_params.getParameter<double>("RMin");
32  fZMax = pgun_params.getParameter<double>("ZMax");
33  fZMin = pgun_params.getParameter<double>("ZMin");
34  fDelta = pgun_params.getParameter<double>("Delta");
35  fPhiMin = pgun_params.getParameter<double>("MinPhi");
36  fPhiMax = pgun_params.getParameter<double>("MaxPhi");
37  fPointing = pgun_params.getParameter<bool>("Pointing");
38  fOverlapping = pgun_params.getParameter<bool>("Overlapping");
39  fRandomShoot = pgun_params.getParameter<bool>("RandomShoot");
40  fNParticles = pgun_params.getParameter<int>("NParticles");
41  fPartIDs = pgun_params.getParameter<vector<int> >("PartID");
42 
43  produces<HepMCProduct>("unsmeared");
44  produces<GenEventInfoProduct>();
45 }
T getParameter(std::string const &) const
BaseFlatGunProducer(const ParameterSet &)
CloseByParticleGunProducer::~CloseByParticleGunProducer ( )
override

Definition at line 47 of file CloseByParticleGunProducer.cc.

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

Member Function Documentation

void CloseByParticleGunProducer::produce ( Event e,
const EventSetup es 
)
overrideprivate

Definition at line 51 of file CloseByParticleGunProducer.cc.

References funct::abs(), HltBtagPostValidation_cff::c, constexpr, funct::cos(), HCALHighEnergyHPDFilter_cfi::energy, edm::EventID::event(), fDelta, fEnMax, fEnMin, edm::BaseFlatGunProducer::fEvt, fNParticles, fOverlapping, fPartIDs, edm::BaseFlatGunProducer::fPDGTable, fPhiMax, fPhiMin, fPointing, hydjet2DefaultParameters_cff::fR, fRandomShoot, fRMax, fRMin, edm::BaseFlatGunProducer::fVerbosity, fZMax, fZMin, genParticles2HepMC_cfi::genEventInfo, GenParticle::GenParticle, edm::RandomNumberGenerator::getEngine(), GenLeadTrackFilter_cfi::HepMCProduct, mps_fire::i, edm::EventBase::id(), LogDebug, EgHLTOffHistBins_cfi::mass, eostools::move(), AlCaHLTBitMon_ParallelJobs::p, LHEGenericFilter_cfi::ParticleID, ecalTrigSettings_cff::particles, genfragment_ptgun_cfg::PartID, partIdx(), edm::Event::put(), multPhiCorr_741_25nsDY_cfi::px, multPhiCorr_741_25nsDY_cfi::py, funct::sin(), mathSSE::sqrt(), edm::Event::streamID(), and hgcalLayerClusters_cff::timeOffset.

51  {
53  CLHEP::HepRandomEngine* engine = &rng->getEngine(e.streamID());
54 
55  if (fVerbosity > 0) {
56  LogDebug("CloseByParticleGunProducer") << " CloseByParticleGunProducer : Begin New Event Generation" << endl;
57  }
58  fEvt = new HepMC::GenEvent();
59 
60  // loop over particles
61  //
62  int barcode = 1;
63  int numParticles = fRandomShoot ? CLHEP::RandFlat::shoot(engine, 1, fNParticles) : fNParticles;
64  std::vector<int> particles;
65 
66  for (int i = 0; i < numParticles; i++) {
67  int partIdx = CLHEP::RandFlat::shoot(engine, 0, fPartIDs.size());
68  particles.push_back(fPartIDs[partIdx]);
69  }
70 
71  double phi = CLHEP::RandFlat::shoot(engine, fPhiMin, fPhiMax);
72  double fR = CLHEP::RandFlat::shoot(engine, fRMin, fRMax);
73  double fZ = CLHEP::RandFlat::shoot(engine, fZMin, fZMax);
74  double tmpPhi = phi;
75  double tmpR = fR;
76 
77  for (unsigned int ip = 0; ip < particles.size(); ++ip) {
78  if (fOverlapping) {
79  fR = CLHEP::RandFlat::shoot(engine, tmpR - fDelta, tmpR + fDelta);
80  phi = CLHEP::RandFlat::shoot(engine, tmpPhi - fDelta / fR, tmpPhi + fDelta / fR);
81  } else
82  phi += fDelta / fR;
83 
84  double fEn = CLHEP::RandFlat::shoot(engine, fEnMin, fEnMax);
85  int PartID = particles[ip];
86  const HepPDT::ParticleData* PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID)));
87  double mass = PData->mass().value();
88  double mom2 = fEn * fEn - mass * mass;
89  double mom = 0.;
90  if (mom2 > 0.) {
91  mom = sqrt(mom2);
92  }
93  double px = 0.;
94  double py = 0.;
95  double pz = mom;
96  double energy = fEn;
97 
98  // Compute Vertex Position
99  double x = fR * cos(phi);
100  double y = fR * sin(phi);
101  constexpr double c = 2.99792458e+1; // cm/ns
102  double timeOffset = sqrt(x * x + y * y + fZ * fZ) / c * ns * c_light;
103  HepMC::GenVertex* Vtx = new HepMC::GenVertex(HepMC::FourVector(x * cm, y * cm, fZ * cm, timeOffset));
104 
105  HepMC::FourVector p(px, py, pz, energy);
106  // If we are requested to be pointing to (0,0,0), correct the momentum direction
107  if (fPointing) {
108  math::XYZVector direction(x, y, fZ);
109  math::XYZVector momentum = direction.unit() * mom;
110  p.setX(momentum.x());
111  p.setY(momentum.y());
112  p.setZ(momentum.z());
113  }
114  HepMC::GenParticle* Part = new HepMC::GenParticle(p, PartID, 1);
115  Part->suggest_barcode(barcode);
116  barcode++;
117 
118  Vtx->add_particle_out(Part);
119 
120  if (fVerbosity > 0) {
121  Vtx->print();
122  Part->print();
123  }
124  fEvt->add_vertex(Vtx);
125  }
126 
127  fEvt->set_event_number(e.id().event());
128  fEvt->set_signal_process_id(20);
129 
130  if (fVerbosity > 0) {
131  fEvt->print();
132  }
133 
134  unique_ptr<HepMCProduct> BProduct(new HepMCProduct());
135  BProduct->addHepMCData(fEvt);
136  e.put(std::move(BProduct), "unsmeared");
137 
138  unique_ptr<GenEventInfoProduct> genEventInfo(new GenEventInfoProduct(fEvt));
140 
141  if (fVerbosity > 0) {
142  LogDebug("CloseByParticleGunProducer") << " CloseByParticleGunProducer : Event Generation Done " << endl;
143  }
144 
145  particles.clear();
146 }
#define LogDebug(id)
EventNumber_t event() const
Definition: EventID.h:40
static unsigned int partIdx(const InputGenJetsParticleSelector::ParticleVector &p, const reco::Candidate *particle)
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:131
Sin< T >::type sin(const T &t)
Definition: Sin.h:22
virtual CLHEP::HepRandomEngine & getEngine(StreamID const &)=0
Use this engine in event methods.
ESHandle< HepPDT::ParticleDataTable > fPDGTable
T sqrt(T t)
Definition: SSEVec.h:19
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
HepPDT::ParticleData ParticleData
XYZVectorD XYZVector
spatial vector with cartesian internal representation
Definition: Vector3D.h:31
edm::EventID id() const
Definition: EventBase.h:59
StreamID streamID() const
Definition: Event.h:96
def move(src, dest)
Definition: eostools.py:511
#define constexpr

Member Data Documentation

double edm::CloseByParticleGunProducer::fDelta
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fEnMax
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fEnMin
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

int edm::CloseByParticleGunProducer::fNParticles
protected

Definition at line 19 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

bool edm::CloseByParticleGunProducer::fOverlapping = false
protected

Definition at line 21 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

std::vector<int> edm::CloseByParticleGunProducer::fPartIDs
protected

Definition at line 23 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fPhiMax
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fPhiMin
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

bool edm::CloseByParticleGunProducer::fPointing = false
protected

Definition at line 20 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

bool edm::CloseByParticleGunProducer::fRandomShoot = false
protected

Definition at line 22 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fRMax
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fRMin
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fZMax
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().

double edm::CloseByParticleGunProducer::fZMin
protected

Definition at line 18 of file CloseByParticleGunProducer.h.

Referenced by CloseByParticleGunProducer(), and produce().