CMS 3D CMS Logo

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

#include <Pythia6JetGun.h>

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

Public Member Functions

 Pythia6JetGun (const edm::ParameterSet &)
 
virtual ~Pythia6JetGun ()
 
- Public Member Functions inherited from gen::Pythia6ParticleGun
 Pythia6ParticleGun (const edm::ParameterSet &)
 
virtual ~Pythia6ParticleGun ()
 
- 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 &)
 
virtual ~Pythia6Gun ()
 
- Public Member Functions inherited from edm::EDProducer
 EDProducer ()
 
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)
 
virtual ~ProducerBase () noexcept(false)
 
- 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
 
void updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
 
virtual ~EDConsumerBase () noexcept(false)
 

Protected Member Functions

void generateEvent (CLHEP::HepRandomEngine *)
 
- 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 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

double fMaxE
 
double fMaxEta
 
double fMaxP
 
double fMinE
 
double fMinEta
 
double fMinP
 

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)
 
- Protected Attributes inherited from gen::Pythia6ParticleGun
std::vector< int > fPartIDs
 
- Protected Attributes inherited from gen::Pythia6Gun
HepMC::GenEventfEvt
 
bool fHepMCVerbosity
 
int fMaxEventsToPrint
 
double fMaxPhi
 
double fMinPhi
 
std::vector< int > fPartIDs
 
Pythia6ServicefPy6Service
 
int fPylistVerbosity
 

Detailed Description

Definition at line 12 of file Pythia6JetGun.h.

Constructor & Destructor Documentation

Pythia6JetGun::Pythia6JetGun ( const edm::ParameterSet pset)

Definition at line 18 of file Pythia6JetGun.cc.

References fMaxE, fMaxEta, fMaxP, fMinE, fMinEta, fMinP, and edm::ParameterSet::getParameter().

18  :
19  Pythia6ParticleGun(pset),
20  fMinEta(0.), fMaxEta(0.),
21  fMinE(0.), fMaxE(0.),
22  fMinP(0.), fMaxP(0.)
23 {
24 
25  ParameterSet pgun_params =
26  pset.getParameter<ParameterSet>("PGunParameters");
27  fMinEta = pgun_params.getParameter<double>("MinEta");
28  fMaxEta = pgun_params.getParameter<double>("MaxEta");
29  fMinE = pgun_params.getParameter<double>("MinE");
30  fMaxE = pgun_params.getParameter<double>("MaxE");
31  fMinP = pgun_params.getParameter<double>("MinP");
32  fMaxP = pgun_params.getParameter<double>("MaxP");
33 
34 }
T getParameter(std::string const &) const
Pythia6ParticleGun(const edm::ParameterSet &)
Pythia6JetGun::~Pythia6JetGun ( )
virtual

Definition at line 36 of file Pythia6JetGun.cc.

37 {
38 }

Member Function Documentation

void Pythia6JetGun::generateEvent ( CLHEP::HepRandomEngine *  )
protectedvirtual

Implements gen::Pythia6Gun.

Definition at line 40 of file Pythia6JetGun.cc.

References funct::cos(), DEFINE_FWK_MODULE, PVValHelper::eta, JetChargeProducer_cfi::exp, gen::Pythia6Gun::fEvt, plotBeamSpotDB::first, fMaxE, fMaxEta, fMaxP, gen::Pythia6Gun::fMaxPhi, fMinE, fMinEta, fMinP, gen::Pythia6Gun::fMinPhi, gen::Pythia6ParticleGun::fPartIDs, gen::Pythia6Gun::fPy6Service, GenParticle::GenParticle, mps_fire::i, plotBeamSpotDB::last, M_PI, ResonanceBuilder::mass, gen::p, EgammaObjectsElectrons_cfi::particleID, createTree::pp, gen::py1ent_(), gen::pyexec_(), gen::pymass_(), gen::pyr_(), gen::pyrobo_(), funct::sin(), 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  double totPx = 0.;
57  double totPy = 0.;
58  double totPz = 0.;
59  double totE = 0.;
60  double totM = 0.;
61  double phi, eta, the, ee, pp;
62  int dum = 0;
63  for ( size_t i=0; i<fPartIDs.size(); i++ )
64  {
65  int particleID = fPartIDs[i]; // this is PDG - need to convert to Py6 !!!
66  int py6PID = HepPID::translatePDTtoPythia( particleID );
67 
68  // internal numbers
69  //
70  phi = 2. * M_PI * pyr_(&dum);
71  the = std::acos( -1. + 2.*pyr_(&dum) );
72 
73  // from input
74  //
75  ee = (fMaxE-fMinE)*pyr_(&dum)+fMinE;
76 
77  // fill p(ip,5) (in PYJETS) with mass value right now,
78  // because the (hardcoded) mstu(10)=1 will make py1ent
79  // pick the mass from there
80  double mass = pymass_(py6PID);
81  pyjets.p[4][ip-1]=mass;
82 
83  // add entry to py6
84  //
85  py1ent_(ip, py6PID, ee, the, phi);
86 
87  // values for computing total mass
88  //
89  totPx += pyjets.p[0][ip-1];
90  totPy += pyjets.p[1][ip-1];
91  totPz += pyjets.p[2][ip-1];
92  totE += pyjets.p[3][ip-1];
93 
94  ip++;
95 
96  } // end forming up py6 record of the jet
97 
98 
99  // compute total mass
100  //
101  totM = std::sqrt( totE*totE - (totPx*totPx+totPy*totPy+totPz*totPz) );
102 
103  //now the boost (from input params)
104  //
105  pp = (fMaxP-fMinP)*pyr_(&dum)+fMinP;
106  ee = std::sqrt( totM*totM + pp*pp );
107 
108  //the boost direction (from input params)
109  //
110  phi = (fMaxPhi-fMinPhi)*pyr_(&dum)+fMinPhi;
111  eta = (fMaxEta-fMinEta)*pyr_(&dum)+fMinEta;
112  the = 2.*atan(exp(-eta));
113 
114  double betaX = pp/ee * std::sin(the) * std::cos(phi);
115  double betaY = pp/ee * std::sin(the) * std::sin(phi);
116  double betaZ = pp/ee * std::cos(the);
117 
118  // boost all particles
119  // the first 2 params (-1) tell to boost all (fisrt-to-last),
120  // and the next 2 params (0.) tell no rotation
121  //
122  int first=-1, last=-1;
123  double rothe=0, rophi=0.;
124 
125  pyrobo_( first, last, rothe, rophi, betaX, betaY, betaZ );
126 
127  // event should be formed from boosted record !!!
128  // that's why additional loop
129  //
130  for ( int i=0; i<pyjets.n; i++ )
131  {
132  HepMC::FourVector p(pyjets.p[0][i],pyjets.p[1][i],pyjets.p[2][i],pyjets.p[3][i]) ;
133  HepMC::GenParticle* Part =
134  new HepMC::GenParticle(p,
135  HepPID::translatePythiatoPDT( pyjets.k[1][i] ),
136  1);
137  Part->suggest_barcode( i+1 ) ;
138  Vtx->add_particle_out(Part);
139  }
140  fEvt->add_vertex(Vtx);
141 
142  // run pythia
143  pyexec_();
144 
145  return;
146 }
Sin< T >::type sin(const T &t)
Definition: Sin.h:22
HepMC::GenEvent * fEvt
Definition: Pythia6Gun.h:71
double p[5][pyjets_maxn]
double fMinPhi
Definition: Pythia6Gun.h:66
T sqrt(T t)
Definition: SSEVec.h:18
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
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 pyrobo_(int &, int &, double &, double &, double &, double &, double &)
void pyexec_()

Member Data Documentation

double gen::Pythia6JetGun::fMaxE
private

Definition at line 29 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().

double gen::Pythia6JetGun::fMaxEta
private

Definition at line 27 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().

double gen::Pythia6JetGun::fMaxP
private

Definition at line 31 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().

double gen::Pythia6JetGun::fMinE
private

Definition at line 28 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().

double gen::Pythia6JetGun::fMinEta
private

Definition at line 26 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().

double gen::Pythia6JetGun::fMinP
private

Definition at line 30 of file Pythia6JetGun.h.

Referenced by generateEvent(), and Pythia6JetGun().