HTXSRivetProducer Class Reference

Inheritance diagram for HTXSRivetProducer:

Public Member Functions
	HTXSRivetProducer (const edm::ParameterSet &cfg)
	~HTXSRivetProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Private Member Functions
void	beginJob ()
void	beginRun (edm::Run const &iRun, edm::EventSetup const &es) override
void	endJob ()
void	endRun (edm::Run const &iRun, edm::EventSetup const &es) override
void	produce (edm::Event &, const edm::EventSetup &) override

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::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

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 bk::beginJob(), Ecal2004TBTDCRanges_v1_cff::endRun, edm::ParameterSet::getParameter(), and HTXS::UNKNOWN.

                                                           : 
        _hepmcCollection(consumes<HepMCProduct>(cfg.getParameter<edm::InputTag>("HepMCCollection"))),
        _lheRunInfo(consumes<LHERunInfoProduct,edm::InRun>(cfg.getParameter<edm::InputTag>("LHERunInfo")))
    {
        _HTXS = new Rivet::HiggsTemplateCrossSections();
        
        _isFirstEvent = true;
        _prodMode = cfg.getParameter<string>("ProductionMode");
        m_HiggsProdMode = HTXS::UNKNOWN;
        
        produces<HTXS::HiggsClassification>("HiggsClassification").setBranchAlias("HiggsClassification");

    }

HTXSRivetProducer::~HTXSRivetProducer ( )

override

Definition at line 72 of file HTXSRivetProducer.cc.

                                     {
}

Member Function Documentation

void HTXSRivetProducer::beginJob ( void  )

private

Definition at line 75 of file HTXSRivetProducer.cc.

                                {
    _analysisHandler.addAnalysis(_HTXS);
}

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

overrideprivate

Definition at line 180 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.

                                                                            {

    if (_prodMode == "AUTO") {    

        edm::Handle<LHERunInfoProduct> run;
        bool product_exists = iRun.getByLabel( edm::InputTag("externalLHEProducer"), run );
        if( product_exists ){
           
          typedef std::vector<LHERunInfoProduct::Header>::const_iterator headers_const_iterator;
          LHERunInfoProduct myLHERunInfoProduct = *(run.product());
          for (headers_const_iterator iter=myLHERunInfoProduct.headers_begin(); iter!=myLHERunInfoProduct.headers_end(); iter++){
              std::vector<std::string> lines = iter->lines();
              for (unsigned int iLine = 0; iLine<lines.size(); iLine++) {
                  std::string line=lines.at(iLine);
                  // POWHEG
                  if (line.find("gg_H_quark-mass-effects") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GGF;
                      break;
                  }
                  if (line.find("Process: HJ") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GGF;
                      break;
                  }
                  if (line.find("Process: HJJ") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GGF;
                      break;
                  }
                  if (line.find("VBF_H") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::VBF;
                      break;
                  }
                  if (line.find("HZJ") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::QQ2ZH;
                      break;
                  }
                  if (line.find("ggHZ") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GG2ZH;
                      break;
                  }
                  // MC@NLO
                  if (line.find("ggh012j") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GGF;
                      break;
                  }
                  if (line.find("vbfh") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::VBF;
                      break;
                  }
                  if (line.find("zh012j") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::QQ2ZH;
                      break;
                  }
                  if (line.find("ggzh01j") != std::string::npos) {
                      edm::LogInfo ("HTXSRivetProducer") <<iLine<< " "<<line<<std::endl;
                      m_HiggsProdMode = HTXS::GG2ZH;
                      break;
                  }
              }
                
              if ( m_HiggsProdMode != HTXS::UNKNOWN) break;
        }
      }
    }
}

void HTXSRivetProducer::endJob ( void  )

private

Definition at line 172 of file HTXSRivetProducer.cc.

                              {
    _HTXS->printClassificationSummary();
}

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

overrideprivate

Definition at line 176 of file HTXSRivetProducer.cc.

{
}

void HTXSRivetProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &    )

overrideprivate

Definition at line 79 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.

                                                                          {
    
    //get the hepmc product from the event
    edm::Handle<HepMCProduct> evt;

    bool product_exists = iEvent.getByToken(_hepmcCollection, evt);
    if( product_exists ){
    
      // get HepMC GenEvent
      const HepMC::GenEvent *myGenEvent = evt->GetEvent();
      
      if (_prodMode == "AUTO") {
          
          // for these prod modes, don't change what is set in BeginRun
          if (m_HiggsProdMode != HTXS::GGF && m_HiggsProdMode != HTXS::VBF && m_HiggsProdMode != HTXS::GG2ZH) {
              
              unsigned nWs = 0;
              unsigned nZs = 0;
              unsigned nTs = 0;
              unsigned nBs = 0;
              unsigned nHs = 0;
              
              HepMC::GenVertex *HSvtx = myGenEvent->signal_process_vertex();
              
              if(HSvtx){
                for (auto ptcl:particles(HSvtx,HepMC::children)) {
                    if (std::abs(ptcl->pdg_id()) == 24) ++nWs;
                    if (ptcl->pdg_id() == 23) ++nZs;
                    if (abs(ptcl->pdg_id()) == 6) ++nTs;
                    if (abs(ptcl->pdg_id()) == 5) ++nBs;
                    if (ptcl->pdg_id() == 25) ++nHs;
                }
              }
                          
              if (nZs==1 && nHs==1 && (nWs+nTs)==0) {
                  m_HiggsProdMode = HTXS::QQ2ZH;
              } else if (nWs==1 && nHs==1 && (nZs+nTs)==0) {
                  m_HiggsProdMode = HTXS::WH; 
              } else if (nTs==2 && nHs==1 && nZs==0) {
                  m_HiggsProdMode = HTXS::TTH; 
              } else if (nTs==1 && nHs==1 && nZs==0) {
                  m_HiggsProdMode = HTXS::TH; 
              } else if (nBs==2 && nHs==1 && nZs==0) {
                  m_HiggsProdMode = HTXS::BBH; 
              }
              
              _HTXS->setHiggsProdMode(m_HiggsProdMode);
              
          }
      }
      

      if (_isFirstEvent){

          // set the production mode if not done already
          if      ( _prodMode == "GGF"   ) m_HiggsProdMode = HTXS::GGF;
          else if ( _prodMode == "VBF"   ) m_HiggsProdMode = HTXS::VBF;
          else if ( _prodMode == "WH"    ) m_HiggsProdMode = HTXS::WH;
          else if ( _prodMode == "ZH"    ) m_HiggsProdMode = HTXS::QQ2ZH;
          else if ( _prodMode == "QQ2ZH" ) m_HiggsProdMode = HTXS::QQ2ZH;
          else if ( _prodMode == "GG2ZH" ) m_HiggsProdMode = HTXS::GG2ZH;
          else if ( _prodMode == "TTH"   ) m_HiggsProdMode = HTXS::TTH;
          else if ( _prodMode == "BBH"   ) m_HiggsProdMode = HTXS::BBH;
          else if ( _prodMode == "TH"    ) m_HiggsProdMode = HTXS::TH;
          else if ( _prodMode == "AUTO"  ) {
              edm::LogInfo ("HTXSRivetProducer") <<"Using AUTO for HiggsProdMode, found it to be: "<<m_HiggsProdMode<< "\n";
              edm::LogInfo ("HTXSRivetProducer") <<"(UNKNOWN=0, GGF=1, VBF=2, WH=3, QQ2ZH=4, GG2ZH=5, TTH=6, BBH=7, TH=8)"<<endl;
          } else {
              throw cms::Exception("HTXSRivetProducer") << "ProductionMode must be one of: GGF,VBF,WH,ZH,QQ2ZH,GG2ZH,TTH,BBH,TH,AUTO " ; 
          }
          _HTXS->setHiggsProdMode(m_HiggsProdMode);

          // at this point the production mode must be known
          if (m_HiggsProdMode == HTXS::UNKNOWN) {
              edm::LogInfo   ("HTXSRivetProducer")  << "HTXSRivetProducer WARNING: HiggsProduction mode is UNKNOWN" << endl;
          }            

          // initialize rivet analysis
          _analysisHandler.init(*myGenEvent);
          _isFirstEvent = false;

      }

      // classify the event
      Rivet::HiggsClassification rivet_cat = _HTXS->classifyEvent(*myGenEvent,m_HiggsProdMode);
      cat_ = HTXS::Rivet2Root(rivet_cat);

      unique_ptr<HTXS::HiggsClassification> cat( new HTXS::HiggsClassification( cat_ ) ); 

      iEvent.put(std::move(cat),"HiggsClassification");
    }
}

Member Data Documentation

Rivet::AnalysisHandler HTXSRivetProducer::_analysisHandler

private

Definition at line 61 of file HTXSRivetProducer.cc.

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

private

Definition at line 58 of file HTXSRivetProducer.cc.

Rivet::HiggsTemplateCrossSections* HTXSRivetProducer::_HTXS

private

Definition at line 62 of file HTXSRivetProducer.cc.

bool HTXSRivetProducer::_isFirstEvent

private

Definition at line 64 of file HTXSRivetProducer.cc.

edm::EDGetTokenT<LHERunInfoProduct> HTXSRivetProducer::_lheRunInfo

private

Definition at line 59 of file HTXSRivetProducer.cc.

std::string HTXSRivetProducer::_prodMode

private

Definition at line 65 of file HTXSRivetProducer.cc.

HTXS::HiggsClassification HTXSRivetProducer::cat_

private

Definition at line 68 of file HTXSRivetProducer.cc.

HTXS::HiggsProdMode HTXSRivetProducer::m_HiggsProdMode

private

Definition at line 66 of file HTXSRivetProducer.cc.