CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
HTXSRivetProducer Class Reference
Inheritance diagram for HTXSRivetProducer:
edm::one::EDProducer< edm::one::WatchRuns > edm::one::EDProducerBase edm::ProducerBase edm::EDConsumerBase edm::ProductRegistryHelper

Public Member Functions

 HTXSRivetProducer (const edm::ParameterSet &cfg)
 
 ~HTXSRivetProducer ()
 
- Public Member Functions inherited from edm::one::EDProducer< edm::one::WatchRuns >
 EDProducer ()=default
 
SerialTaskQueueglobalLuminosityBlocksQueue () final
 
SerialTaskQueueglobalRunsQueue () final
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () 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
 
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)
 

Private Member Functions

void beginRun (edm::Run const &iRun, edm::EventSetup const &es) override
 
void endRun (edm::Run const &iRun, edm::EventSetup const &es) override
 
void produce (edm::Event &, const edm::EventSetup &)
 

Private Attributes

Rivet::AnalysisHandler _analysisHandler
 
edm::EDGetTokenT< edm::HepMCProduct_hepmcCollection
 
Rivet::HiggsTemplateCrossSections_HTXS
 
bool _isFirstEvent
 
edm::EDGetTokenT< LHERunInfoProduct_lheRunInfo
 
std::string _prodMode
 
HTXS::HiggsClassification cat_
 
HTXS::HiggsProdMode m_HiggsProdMode
 

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::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)
 

Detailed Description

Definition at line 31 of file HTXSRivetProducer.cc.

Constructor & Destructor Documentation

HTXSRivetProducer::HTXSRivetProducer ( const edm::ParameterSet cfg)
inlineexplicit

Definition at line 34 of file HTXSRivetProducer.cc.

References Ecal2004TBTDCRanges_v1_cff::endRun, edm::ParameterSet::getParameter(), and HTXS::UNKNOWN.

34  :
35  _hepmcCollection(consumes<HepMCProduct>(cfg.getParameter<edm::InputTag>("HepMCCollection"))),
36  _lheRunInfo(consumes<LHERunInfoProduct,edm::InRun>(cfg.getParameter<edm::InputTag>("LHERunInfo")))
37  {
39 
40  _isFirstEvent = true;
41  _prodMode = cfg.getParameter<string>("ProductionMode");
43 
44  produces<HTXS::HiggsClassification>("HiggsClassification").setBranchAlias("HiggsClassification");
45 
46  }
T getParameter(std::string const &) const
HTXS::HiggsProdMode m_HiggsProdMode
Rivet::HiggsTemplateCrossSections * _HTXS
Rivet routine for classifying MC events according to the Higgs template cross section categories...
edm::EDGetTokenT< edm::HepMCProduct > _hepmcCollection
edm::EDGetTokenT< LHERunInfoProduct > _lheRunInfo
HTXSRivetProducer::~HTXSRivetProducer ( )

Definition at line 70 of file HTXSRivetProducer.cc.

70  {
71 }

Member Function Documentation

void HTXSRivetProducer::beginRun ( edm::Run const &  iRun,
edm::EventSetup const &  es 
)
overrideprivate

Definition at line 173 of file HTXSRivetProducer.cc.

References DEFINE_FWK_MODULE, edm::Run::getByLabel(), HTXS::GG2ZH, HTXS::GGF, LHERunInfoProduct::headers_begin(), LHERunInfoProduct::headers_end(), geometryCSVtoXML::line, groupFilesInBlocks::lines, edm::Handle< T >::product(), HTXS::QQ2ZH, findQualityFiles::run, AlCaHLTBitMon_QueryRunRegistry::string, HTXS::UNKNOWN, and HTXS::VBF.

173  {
174 
175  if (_prodMode == "AUTO") {
176 
178  bool product_exists = iRun.getByLabel( edm::InputTag("externalLHEProducer"), run );
179  if( product_exists ){
180 
181  typedef std::vector<LHERunInfoProduct::Header>::const_iterator headers_const_iterator;
182  LHERunInfoProduct myLHERunInfoProduct = *(run.product());
183  for (headers_const_iterator iter=myLHERunInfoProduct.headers_begin(); iter!=myLHERunInfoProduct.headers_end(); iter++){
184  std::vector<std::string> lines = iter->lines();
185  for (unsigned int iLine = 0; iLine<lines.size(); iLine++) {
186  std::string line=lines.at(iLine);
187  // POWHEG
188  if (line.find("gg_H_quark-mass-effects") != std::string::npos) {
189  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
191  break;
192  }
193  if (line.find("Process: HJ") != std::string::npos) {
194  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
196  break;
197  }
198  if (line.find("Process: HJJ") != std::string::npos) {
199  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
201  break;
202  }
203  if (line.find("VBF_H") != std::string::npos) {
204  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
206  break;
207  }
208  if (line.find("HZJ") != std::string::npos) {
209  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
211  break;
212  }
213  if (line.find("ggHZ") != std::string::npos) {
214  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
216  break;
217  }
218  // MC@NLO
219  if (line.find("ggh012j") != std::string::npos) {
220  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
222  break;
223  }
224  if (line.find("vbfh") != std::string::npos) {
225  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
227  break;
228  }
229  if (line.find("zh012j") != std::string::npos) {
230  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
232  break;
233  }
234  if (line.find("ggzh01j") != std::string::npos) {
235  edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
237  break;
238  }
239  }
240 
241  if ( m_HiggsProdMode != HTXS::UNKNOWN) break;
242  }
243  }
244  }
245 }
headers_const_iterator headers_end() const
HTXS::HiggsProdMode m_HiggsProdMode
headers_const_iterator headers_begin() const
T const * product() const
Definition: Handle.h:81
void HTXSRivetProducer::endRun ( edm::Run const &  iRun,
edm::EventSetup const &  es 
)
overrideprivate

Definition at line 168 of file HTXSRivetProducer.cc.

169 {
171 }
Rivet::HiggsTemplateCrossSections * _HTXS
void HTXSRivetProducer::produce ( edm::Event iEvent,
const edm::EventSetup  
)
private

Definition at line 73 of file HTXSRivetProducer.cc.

References funct::abs(), HTXS::BBH, eostools::cat(), class-composition::children, Exception, edm::Event::getByToken(), edm::HepMCProduct::GetEvent(), HTXS::GG2ZH, HTXS::GGF, eostools::move(), HadronAndPartonSelector_cfi::particles, edm::Event::put(), HTXS::QQ2ZH, HTXS::Rivet2Root(), HTXS::TH, HTXS::TTH, HTXS::UNKNOWN, HTXS::VBF, and HTXS::WH.

73  {
74 
75  //get the hepmc product from the event
77 
78  bool product_exists = iEvent.getByToken(_hepmcCollection, evt);
79  if( product_exists ){
80 
81  // get HepMC GenEvent
82  const HepMC::GenEvent *myGenEvent = evt->GetEvent();
83 
84  if (_prodMode == "AUTO") {
85 
86  // for these prod modes, don't change what is set in BeginRun
88 
89  unsigned nWs = 0;
90  unsigned nZs = 0;
91  unsigned nTs = 0;
92  unsigned nBs = 0;
93  unsigned nHs = 0;
94 
95  HepMC::GenVertex *HSvtx = myGenEvent->signal_process_vertex();
96 
97  if(HSvtx){
98  for (auto ptcl:particles(HSvtx,HepMC::children)) {
99  if (std::abs(ptcl->pdg_id()) == 24) ++nWs;
100  if (ptcl->pdg_id() == 23) ++nZs;
101  if (abs(ptcl->pdg_id()) == 6) ++nTs;
102  if (abs(ptcl->pdg_id()) == 5) ++nBs;
103  if (ptcl->pdg_id() == 25) ++nHs;
104  }
105  }
106 
107  if (nZs==1 && nHs==1 && (nWs+nTs)==0) {
109  } else if (nWs==1 && nHs==1 && (nZs+nTs)==0) {
111  } else if (nTs==2 && nHs==1 && nZs==0) {
113  } else if (nTs==1 && nHs==1 && nZs==0) {
115  } else if (nBs==2 && nHs==1 && nZs==0) {
117  }
118 
120 
121  }
122  }
123 
124 
125  if (_isFirstEvent){
126 
127  _analysisHandler.addAnalysis(_HTXS);
128 
129  // set the production mode if not done already
130  if ( _prodMode == "GGF" ) m_HiggsProdMode = HTXS::GGF;
131  else if ( _prodMode == "VBF" ) m_HiggsProdMode = HTXS::VBF;
132  else if ( _prodMode == "WH" ) m_HiggsProdMode = HTXS::WH;
133  else if ( _prodMode == "ZH" ) m_HiggsProdMode = HTXS::QQ2ZH;
134  else if ( _prodMode == "QQ2ZH" ) m_HiggsProdMode = HTXS::QQ2ZH;
135  else if ( _prodMode == "GG2ZH" ) m_HiggsProdMode = HTXS::GG2ZH;
136  else if ( _prodMode == "TTH" ) m_HiggsProdMode = HTXS::TTH;
137  else if ( _prodMode == "BBH" ) m_HiggsProdMode = HTXS::BBH;
138  else if ( _prodMode == "TH" ) m_HiggsProdMode = HTXS::TH;
139  else if ( _prodMode == "AUTO" ) {
140  edm::LogInfo ("HTXSRivetProducer") <<"Using AUTO for HiggsProdMode, found it to be: "<<m_HiggsProdMode<< "\n";
141  edm::LogInfo ("HTXSRivetProducer") <<"(UNKNOWN=0, GGF=1, VBF=2, WH=3, QQ2ZH=4, GG2ZH=5, TTH=6, BBH=7, TH=8)"<<endl;
142  } else {
143  throw cms::Exception("HTXSRivetProducer") << "ProductionMode must be one of: GGF,VBF,WH,ZH,QQ2ZH,GG2ZH,TTH,BBH,TH,AUTO " ;
144  }
146 
147  // at this point the production mode must be known
149  edm::LogInfo ("HTXSRivetProducer") << "HTXSRivetProducer WARNING: HiggsProduction mode is UNKNOWN" << endl;
150  }
151 
152  // initialize rivet analysis
153  _analysisHandler.init(*myGenEvent);
154  _isFirstEvent = false;
155 
156  }
157 
158  // classify the event
159  Rivet::HiggsClassification rivet_cat = _HTXS->classifyEvent(*myGenEvent,m_HiggsProdMode);
160  cat_ = HTXS::Rivet2Root(rivet_cat);
161 
162  unique_ptr<HTXS::HiggsClassification> cat( new HTXS::HiggsClassification( cat_ ) );
163 
164  iEvent.put(std::move(cat),"HiggsClassification");
165  }
166 }
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:137
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:579
HTXS::HiggsProdMode m_HiggsProdMode
Rivet::AnalysisHandler _analysisHandler
Rivet::HiggsTemplateCrossSections * _HTXS
def cat(path)
Definition: eostools.py:400
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
void setHiggsProdMode(HTXS::HiggsProdMode prodMode)
Sets the Higgs production mode.
HiggsClassification classifyEvent(const Event &event, const HTXS::HiggsProdMode prodMode)
Main classificaion method.
const HepMC::GenEvent * GetEvent() const
Definition: HepMCProduct.h:38
edm::EDGetTokenT< edm::HepMCProduct > _hepmcCollection
HTXS::HiggsClassification Rivet2Root(category const &htxs_cat_rivet)
HTXS::HiggsClassification cat_
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

Rivet::AnalysisHandler HTXSRivetProducer::_analysisHandler
private

Definition at line 59 of file HTXSRivetProducer.cc.

edm::EDGetTokenT<edm::HepMCProduct> HTXSRivetProducer::_hepmcCollection
private

Definition at line 56 of file HTXSRivetProducer.cc.

Rivet::HiggsTemplateCrossSections* HTXSRivetProducer::_HTXS
private

Definition at line 60 of file HTXSRivetProducer.cc.

bool HTXSRivetProducer::_isFirstEvent
private

Definition at line 62 of file HTXSRivetProducer.cc.

edm::EDGetTokenT<LHERunInfoProduct> HTXSRivetProducer::_lheRunInfo
private

Definition at line 57 of file HTXSRivetProducer.cc.

std::string HTXSRivetProducer::_prodMode
private

Definition at line 63 of file HTXSRivetProducer.cc.

HTXS::HiggsClassification HTXSRivetProducer::cat_
private

Definition at line 66 of file HTXSRivetProducer.cc.

HTXS::HiggsProdMode HTXSRivetProducer::m_HiggsProdMode
private

Definition at line 64 of file HTXSRivetProducer.cc.