test
CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
Py8InterfaceBase.cc
Go to the documentation of this file.
2 
5 
6 // EvtGen plugin
7 //
8 //#include "Pythia8Plugins/EvtGen.h"
9 
10 using namespace Pythia8;
11 
12 namespace gen {
13 
14 Py8InterfaceBase::Py8InterfaceBase( edm::ParameterSet const& ps ) :
15 BaseHadronizer(ps),
16 useEvtGen(false), evtgenDecays(0)
17 {
18  fParameters = ps;
19 
20  pythiaPylistVerbosity = ps.getUntrackedParameter<int>("pythiaPylistVerbosity", 0);
21  pythiaHepMCVerbosity = ps.getUntrackedParameter<bool>("pythiaHepMCVerbosity", false);
22  pythiaHepMCVerbosityParticles = ps.getUntrackedParameter<bool>("pythiaHepMCVerbosityParticles", false);
23  maxEventsToPrint = ps.getUntrackedParameter<int>("maxEventsToPrint", 0);
24 
25  if(pythiaHepMCVerbosityParticles)
26  ascii_io = new HepMC::IO_AsciiParticles("cout", std::ios::out);
27 
28  if ( ps.exists("useEvtGenPlugin") ) {
29 
30  useEvtGen = true;
31 
32  string evtgenpath(getenv("EVTGENDATA"));
33  evtgenDecFile = evtgenpath + string("/DECAY_2010.DEC");
34  evtgenPdlFile = evtgenpath + string("/evt.pdl");
35 
36  if ( ps.exists( "evtgenDecFile" ) )
37  evtgenDecFile = ps.getParameter<string>("evtgenDecFile");
38 
39  if ( ps.exists( "evtgenPdlFile" ) )
40  evtgenPdlFile = ps.getParameter<string>("evtgenPdlFile");
41 
42  if ( ps.exists( "evtgenUserFile" ) )
43  evtgenUserFiles = ps.getParameter< std::vector<std::string> >("evtgenUserFile");
44 
45  }
46 
47 }
48 
50 {
51 
52  fMasterGen.reset(new Pythia);
53  fDecayer.reset(new Pythia);
54 
55  //add settings for resonance decay filter
56  fMasterGen->settings.addFlag("ResonanceDecayFilter:filter",false);
57  fMasterGen->settings.addFlag("ResonanceDecayFilter:exclusive",false);
58  fMasterGen->settings.addFlag("ResonanceDecayFilter:eMuAsEquivalent",false);
59  fMasterGen->settings.addFlag("ResonanceDecayFilter:eMuTauAsEquivalent",false);
60  fMasterGen->settings.addFlag("ResonanceDecayFilter:allNuAsEquivalent",false);
61  fMasterGen->settings.addFlag("ResonanceDecayFilter:udscAsEquivalent",false);
62  fMasterGen->settings.addFlag("ResonanceDecayFilter:udscbAsEquivalent",false);
63  fMasterGen->settings.addMVec("ResonanceDecayFilter:mothers",std::vector<int>(),false,false,0,0);
64  fMasterGen->settings.addMVec("ResonanceDecayFilter:daughters",std::vector<int>(),false,false,0,0);
65 
66  fMasterGen->setRndmEnginePtr( &p8RndmEngine_ );
67  fDecayer->setRndmEnginePtr( &p8RndmEngine_ );
68 
69  fMasterGen->readString("Next:numberShowEvent = 0");
70  fDecayer->readString("Next:numberShowEvent = 0");
71 
72  edm::ParameterSet currentParameters;
73  if (randomIndex()>=0) {
74  std::vector<edm::ParameterSet> randomizedParameters = fParameters.getParameter<std::vector<edm::ParameterSet> >("RandomizedParameters");
75  currentParameters = randomizedParameters[randomIndex()];
76  }
77  else {
78  currentParameters = fParameters;
79  }
80 
81  ParameterCollector pCollector = currentParameters.getParameter<edm::ParameterSet>("PythiaParameters");
82 
83  for ( ParameterCollector::const_iterator line = pCollector.begin();
84  line != pCollector.end(); ++line )
85  {
86  if (line->find("Random:") != std::string::npos)
87  throw cms::Exception("PythiaError") << "Attempted to set random number "
88  "using Pythia commands. Please use " "the RandomNumberGeneratorService."
89  << std::endl;
90 
91  if (!fMasterGen->readString(*line)) throw cms::Exception("PythiaError")
92  << "Pythia 8 did not accept \""
93  << *line << "\"." << std::endl;
94 
95  if (line->find("ParticleDecays:") != std::string::npos) {
96 
97  if (!fDecayer->readString(*line)) throw cms::Exception("PythiaError")
98  << "Pythia 8 Decayer did not accept \""
99  << *line << "\"." << std::endl;
100  }
101 
102  }
103 
104  slhafile_.clear();
105 
106  if( currentParameters.exists( "SLHAFileForPythia8" ) ) {
107  std::string slhafilenameshort = currentParameters.getParameter<std::string>("SLHAFileForPythia8");
108  edm::FileInPath f1( slhafilenameshort );
109 
110  fMasterGen->settings.mode("SLHA:readFrom", 2);
111  fMasterGen->settings.word("SLHA:file", f1.fullPath());
112  }
113  else if( currentParameters.exists( "SLHATableForPythia8" ) ) {
114  std::string slhatable = currentParameters.getParameter<std::string>("SLHATableForPythia8");
115 
116  char tempslhaname[] = "pythia8SLHAtableXXXXXX";
117  int fd = mkstemp(tempslhaname);
118  write(fd,slhatable.c_str(),slhatable.size());
119  close(fd);
120 
121  slhafile_ = tempslhaname;
122 
123  fMasterGen->settings.mode("SLHA:readFrom", 2);
124  fMasterGen->settings.word("SLHA:file", slhafile_);
125  }
126 
127  return true;
128 
129 }
130 
131 bool Py8InterfaceBase::declareStableParticles( const std::vector<int>& pdgIds )
132 {
133 
134  for ( size_t i=0; i<pdgIds.size(); i++ )
135  {
136  // FIXME: need to double check if PID's are the same in Py6 & Py8,
137  // because the HepPDT translation tool is actually for **Py6**
138  //
139  // well, actually it looks like Py8 operates in PDT id's rather than Py6's
140  //
141 // int PyID = HepPID::translatePDTtoPythia( pdgIds[i] );
142  int PyID = pdgIds[i];
143  std::ostringstream pyCard ;
144  pyCard << PyID <<":mayDecay=false";
145 
146  if ( fMasterGen->particleData.isParticle( PyID ) ) {
147  fMasterGen->readString( pyCard.str() );
148  } else {
149 
150  edm::LogWarning("DataNotUnderstood") << "Pythia8 does not "
151  << "recognize particle id = "
152  << PyID << std::endl;
153  }
154  // alternative:
155  // set the 2nd input argument warn=false
156  // - this way Py8 will NOT print warnings about unknown particle code(s)
157  // fMasterPtr->readString( pyCard.str(), false )
158  }
159 
160  return true;
161 
162 }
163 
164 bool Py8InterfaceBase:: declareSpecialSettings( const std::vector<std::string>& settings ){
165  for ( unsigned int iss=0; iss<settings.size(); iss++ ){
166  if ( settings[iss].find("QED-brem-off") != std::string::npos ){
167  fMasterGen->readString( "TimeShower:QEDshowerByL=off" );
168  }
169  else{
170  size_t fnd1 = settings[iss].find("Pythia8:");
171  if ( fnd1 != std::string::npos ){
172  std::string value = settings[iss].substr (fnd1+8);
173  fDecayer->readString(value);
174  }
175  }
176  }
177  return true;
178 }
179 
180 
182 {
183 
184  fMasterGen->stat();
185  return;
186 
187 }
188 
189 }
bool declareSpecialSettings(const std::vector< std::string > &)
T getParameter(std::string const &) const
T getUntrackedParameter(std::string const &, T const &) const
int i
Definition: DBlmapReader.cc:9
std::auto_ptr< Pythia8::Pythia > fMasterGen
HepMC::IO_AsciiParticles * ascii_io
bool exists(std::string const &parameterName) const
checks if a parameter exists
void find(edm::Handle< EcalRecHitCollection > &hits, DetId thisDet, std::vector< EcalRecHitCollection::const_iterator > &hit, bool debug=false)
Definition: FindCaloHit.cc:7
std::vector< std::string > evtgenUserFiles
def gen
run2 Cosmic #### Run 256259 @ 0T 2015C### Run 272133 @ 3.8T 2016B###
P8RndmEngine p8RndmEngine_
tuple fd
Definition: ztee.py:136
unsigned int pythiaPylistVerbosity
bool declareStableParticles(const std::vector< int > &)
unsigned int maxEventsToPrint
std::auto_ptr< Pythia8::Pythia > fDecayer
int randomIndex() const
const_iterator end() const
const_iterator begin() const
volatile std::atomic< bool > shutdown_flag false
edm::ParameterSet fParameters