CMS 3D CMS Logo

List of all members | Public Member Functions | Protected Member Functions | Private Attributes
gen::Pythia6PtYDistGun Class Reference

#include <Pythia6PtYDistGun.h>

Inheritance diagram for gen::Pythia6PtYDistGun:
gen::Pythia6ParticleGun gen::Pythia6Gun edm::EDProducer edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 Pythia6PtYDistGun (const edm::ParameterSet &)
 
 ~Pythia6PtYDistGun () override
 
- Public Member Functions inherited from gen::Pythia6ParticleGun
 Pythia6ParticleGun (const edm::ParameterSet &)
 
 ~Pythia6ParticleGun () override
 
- Public Member Functions inherited from gen::Pythia6Gun
void beginJob () override
 
void beginLuminosityBlock (edm::LuminosityBlock const &, edm::EventSetup const &) override
 
void beginRun (edm::Run const &, edm::EventSetup const &) override
 
void endRun (edm::Run const &, edm::EventSetup const &) override
 
void produce (edm::Event &, const edm::EventSetup &) override
 
 Pythia6Gun (const edm::ParameterSet &)
 
 ~Pythia6Gun () override
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
SerialTaskQueueglobalLuminosityBlocksQueue ()
 
SerialTaskQueueglobalRunsQueue ()
 
ModuleDescription const & moduleDescription () const
 
 ~EDProducer () 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
 
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)
 
virtual ~EDConsumerBase () noexcept(false)
 

Protected Member Functions

void generateEvent (CLHEP::HepRandomEngine *) override
 
- Protected Member Functions inherited from gen::Pythia6Gun
HepMC::GenParticle * addAntiParticle (int &, int &, double &, double &, double &)
 
void attachPy6DecaysToGenEvent ()
 
void loadEvent (edm::Event &)
 
- 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 ESProduct , Transition Tr = Transition::Event>
auto esConsumes (eventsetup::EventSetupRecordKey const &, 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)
 

Private Attributes

PtYDistributorfPtYGenerator
 

Additional Inherited Members

- Public Types inherited from edm::EDProducer
typedef EDProducer 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::EDProducer
static const std::string & baseType ()
 
static void fillDescriptions (ConfigurationDescriptions &descriptions)
 
static void prevalidate (ConfigurationDescriptions &descriptions)
 
static bool wantsGlobalLuminosityBlocks ()
 
static bool wantsGlobalRuns ()
 
static bool wantsStreamLuminosityBlocks ()
 
static bool wantsStreamRuns ()
 
- Protected Attributes inherited from gen::Pythia6ParticleGun
std::vector< int > fPartIDs
 
- Protected Attributes inherited from gen::Pythia6Gun
HepMC::GenEvent * fEvt
 
bool fHepMCVerbosity
 
int fMaxEventsToPrint
 
double fMaxPhi
 
double fMinPhi
 
std::vector< int > fPartIDs
 
Pythia6ServicefPy6Service
 
int fPylistVerbosity
 

Detailed Description

Definition at line 14 of file Pythia6PtYDistGun.h.

Constructor & Destructor Documentation

Pythia6PtYDistGun::Pythia6PtYDistGun ( const edm::ParameterSet pset)

Definition at line 19 of file Pythia6PtYDistGun.cc.

References fPtYGenerator, and edm::ParameterSet::getParameter().

19  :
20  Pythia6ParticleGun(pset), fPtYGenerator(nullptr)
21 {
22 
23  ParameterSet pgun_params =
24  pset.getParameter<ParameterSet>("PGunParameters");
25 
26  fPtYGenerator = new PtYDistributor( pgun_params.getParameter<FileInPath>("kinematicsFile"),
27  pgun_params.getParameter<double>("MaxPt"),
28  pgun_params.getParameter<double>("MinPt"),
29  pgun_params.getParameter<double>("MaxY"),
30  pgun_params.getParameter<double>("MinY"),
31  pgun_params.getParameter<int>("PtBinning"),
32  pgun_params.getParameter<int>("YBinning") );
33 }
T getParameter(std::string const &) const
Pythia6ParticleGun(const edm::ParameterSet &)
PtYDistributor * fPtYGenerator
Pythia6PtYDistGun::~Pythia6PtYDistGun ( )
override

Definition at line 35 of file Pythia6PtYDistGun.cc.

References fPtYGenerator.

36 {
37  if ( fPtYGenerator ) delete fPtYGenerator;
38 }
PtYDistributor * fPtYGenerator

Member Function Documentation

void Pythia6PtYDistGun::generateEvent ( CLHEP::HepRandomEngine *  engine)
overrideprotectedvirtual

Implements gen::Pythia6Gun.

Definition at line 40 of file Pythia6PtYDistGun.cc.

References DEFINE_FWK_MODULE, JetChargeProducer_cfi::exp, gen::Pythia6Gun::fEvt, gen::PtYDistributor::firePt(), gen::PtYDistributor::fireY(), gen::Pythia6Gun::fMaxPhi, gen::Pythia6Gun::fMinPhi, gen::Pythia6ParticleGun::fPartIDs, fPtYGenerator, gen::Pythia6Gun::fPy6Service, GenParticle::GenParticle, mps_fire::i, M_PI, ResonanceBuilder::mass, gen::p, EgammaObjectsElectrons_cfi::particleID, EnergyCorrector::pt, gen::py1ent_(), gen::pyexec_(), gen::pymass_(), gen::pyr_(), and mathSSE::sqrt().

41 {
42  Pythia6Service::InstanceWrapper guard(fPy6Service); // grab Py6 instance
43 
44  // now actualy, start cooking up the event gun
45  //
46 
47  // 1st, primary vertex
48  //
49  HepMC::GenVertex* Vtx = new HepMC::GenVertex( HepMC::FourVector(0.,0.,0.) );
50 
51  // here re-create fEvt (memory)
52  //
53  fEvt = new HepMC::GenEvent() ;
54 
55  int ip=1;
56 
57  for ( size_t i=0; i<fPartIDs.size(); i++ )
58  {
59 
60  int particleID = fPartIDs[i]; // this is PDG - need to convert to Py6 !!!
61  int py6PID = HepPID::translatePDTtoPythia( particleID );
62 
63  int dum = 0;
64  double pt=0, y=0, u=0, ee=0, the=0;
65 
66  pt = fPtYGenerator->firePt(engine);
67  y = fPtYGenerator->fireY(engine);
68  u = exp(y);
69 
70  double mass = pymass_(py6PID);
71 
72  // fill p(ip,5) (in PYJETS) with mass value right now,
73  // because the (hardcoded) mstu(10)=1 will make py1ent
74  // pick the mass from there
75  pyjets.p[4][ip-1]=mass;
76 
77  ee = 0.5*std::sqrt(mass*mass+pt*pt)*(u*u+1)/u;
78 
79  double pz = std::sqrt(ee*ee-pt*pt-mass*mass);
80  if ( y < 0. ) pz *= -1;
81 
82  double phi = (fMaxPhi-fMinPhi)*pyr_(&dum)+fMinPhi;
83 
84  the = atan(pt/pz);
85  if ( pz < 0. ) the += M_PI ;
86 
87  py1ent_(ip, py6PID, ee, the, phi);
88 
89 /*
90  double px = pt*cos(phi) ;
91  double py = pt*sin(phi) ;
92 */
93  double px = pyjets.p[0][ip-1]; // pt*cos(phi) ;
94  double py = pyjets.p[1][ip-1]; // pt*sin(phi) ;
95  pz = pyjets.p[2][ip-1]; // mom*cos(the) ;
96 
97  HepMC::FourVector p(px,py,pz,ee) ;
98  HepMC::GenParticle* Part =
99  new HepMC::GenParticle(p,particleID,1);
100  Part->suggest_barcode( ip ) ;
101  Vtx->add_particle_out(Part);
102 
103  ip++;
104 
105  }
106 
107  fEvt->add_vertex(Vtx);
108 
109  // run pythia
110  pyexec_();
111 
112  return;
113 }
HepMC::GenEvent * fEvt
Definition: Pythia6Gun.h:71
double firePt(CLHEP::HepRandomEngine *)
double fireY(CLHEP::HepRandomEngine *)
double p[5][pyjets_maxn]
PtYDistributor * fPtYGenerator
double fMinPhi
Definition: Pythia6Gun.h:66
T sqrt(T t)
Definition: SSEVec.h:18
double pymass_(int &)
std::vector< int > fPartIDs
#define M_PI
double fMaxPhi
Definition: Pythia6Gun.h:67
void py1ent_(int &ip, int &kf, double &pe, double &the, double &phi)
Pythia6Service * fPy6Service
Definition: Pythia6Gun.h:61
double pyr_(int *idummy)
void pyexec_()

Member Data Documentation

PtYDistributor* gen::Pythia6PtYDistGun::fPtYGenerator
private

Definition at line 27 of file Pythia6PtYDistGun.h.

Referenced by generateEvent(), Pythia6PtYDistGun(), and ~Pythia6PtYDistGun().