#include <TtSemiLepSignalSelMVATrainer.h>

Inheritance diagram for TtSemiLepSignalSelMVATrainer:

Classes
struct	JetwithHigherPt

Public Member Functions
	TtSemiLepSignalSelMVATrainer (const edm::ParameterSet &)

	~TtSemiLepSignalSelMVATrainer ()

Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzer ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

virtual	~EDAnalyzer ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	analyze (const edm::Event &evt, const edm::EventSetup &setup)

virtual void	beginJob ()

double	DeltaPhi (const math::XYZTLorentzVector &v1, const math::XYZTLorentzVector &v2)

double	DeltaR (const math::XYZTLorentzVector &v1, const math::XYZTLorentzVector &v2)

Private Attributes
edm::EDGetTokenT< edm::View < pat::Electron > >	electronsToken_

edm::EDGetTokenT< TtGenEvent >	genEvtToken_

edm::EDGetTokenT< std::vector < pat::Jet > >	jetsToken_

int	lepChannel_

int	maxEv_

edm::EDGetTokenT< edm::View < pat::MET > >	METsToken_

edm::EDGetTokenT< edm::View < pat::Muon > >	muonsToken_

PhysicsTools::MVAComputerCache	mvaComputer

int	selEv

int	whatData_

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

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 28 of file TtSemiLepSignalSelMVATrainer.h.

Constructor & Destructor Documentation

TtSemiLepSignalSelMVATrainer::TtSemiLepSignalSelMVATrainer ( const edm::ParameterSet & cfg )

explicit

Definition at line 13 of file TtSemiLepSignalSelMVATrainer.cc.

                                                                                     :
   muonsToken_     (consumes< edm::View<pat::Muon> >(cfg.getParameter<edm::InputTag>("muons"))),
   electronsToken_ (consumes< edm::View<pat::Electron> >(cfg.getParameter<edm::InputTag>("elecs"))),
   jetsToken_      (consumes< std::vector<pat::Jet> >(cfg.getParameter<edm::InputTag>("jets"))),
   METsToken_      (consumes<edm::View<pat::MET> >(cfg.getParameter<edm::InputTag>("mets"))),
   genEvtToken_      (consumes<TtGenEvent>(edm::InputTag("genEvt"))),
   lepChannel_(cfg.getParameter<int>("lepChannel")),
   whatData_  (cfg.getParameter<int>("whatData")),
   maxEv_     (cfg.getParameter<int>("maxEv"))
 {
 }

TtSemiLepSignalSelMVATrainer::~TtSemiLepSignalSelMVATrainer ( )

Definition at line 25 of file TtSemiLepSignalSelMVATrainer.cc.

26 {

27 }

Member Function Documentation

void TtSemiLepSignalSelMVATrainer::analyze	(	const edm::Event &	evt,
		const edm::EventSetup &	setup
	)

privatevirtual

Implements edm::EDAnalyzer.

Definition at line 30 of file TtSemiLepSignalSelMVATrainer.cc.

References edm::View< T >::begin(), gather_cfg::cout, DeltaR(), pat::Flags::Overlap::Electrons, HI_PhotonSkim_cff::electrons, electronsToken_, edm::View< T >::end(), evaluateTtSemiLepSignalSel(), TtGenEvtProducer_cfi::genEvt, genEvtToken_, edm::Event::getByToken(), edm::Ref< C, T, F >::isNull(), edm::HandleBase::isValid(), metsig::jet, fwrapper::jets, jetsToken_, lepChannel_, maxEv_, METsToken_, metsig::muon, patZpeak::muons, muonsToken_, mvaComputer, corrVsCorr::selection, selEv, HcalObjRepresent::setup(), python.multivaluedict::sort(), pat::Flags::test(), PhysicsTools::MVAComputerCache::update(), puppiForMET_cff::weight, and whatData_.

 {
   //communication with CMSSW CondDB
   mvaComputer.update<TtSemiLepSignalSelMVARcd>("trainer", setup, "traintreeSaver");
 
   // can occur in the last iteration when the
   // MVATrainer is about to save the result
   if(!mvaComputer) return;
 
 
   //make your preselection here!!
   //the following code is for the default example
   edm::Handle<edm::View<pat::MET> > MET_handle;
   evt.getByToken(METsToken_,MET_handle);
   if(!MET_handle.isValid()) return;
   const edm::View<pat::MET> MET = *MET_handle;
 
   edm::Handle< std::vector<pat::Jet> > jet_handle;
   evt.getByToken(jetsToken_, jet_handle);
   if(!jet_handle.isValid()) return;
   const std::vector<pat::Jet> jets = *jet_handle;
   unsigned int nJets = 0;
   std::vector<pat::Jet> seljets;
   //std::cout<<"number of jets: "<<jets.size()<<std::endl;
   for(std::vector<pat::Jet>::const_iterator it = jets.begin(); it != jets.end(); it++) {
     //std::cout<<"Jet Et: "<<it->et()<<" Eta: "<<fabs(it->eta())<<std::endl;
     if(!(pat::Flags::test(*it, pat::Flags::Overlap::Electrons))) continue;
     if(it->et()>30. && fabs(it->eta())<2.4) {
       seljets.push_back(*it);
       nJets++;
     }
   }
   //std::cout<<"selected Jets: "<<nJets<<std::endl;
   if(nJets<4) return;
 
   //sort by Pt
   sort(seljets.begin(),seljets.end(),JetwithHigherPt());
 
   edm::Handle< edm::View<pat::Muon> > muon_handle;
   evt.getByToken(muonsToken_, muon_handle);
   if(!muon_handle.isValid()) return;
   const edm::View<pat::Muon> muons = *muon_handle;
   int nmuons = 0;
   std::vector<pat::Muon> selMuons;
   for(edm::View<pat::Muon>::const_iterator it = muons.begin(); it!=muons.end(); it++) {
     reco::TrackRef gltr = it->track(); // global track
     reco::TrackRef trtr = it->innerTrack(); // tracker track
     if(it->pt()>30 && fabs(it->eta())<2.1 && (it->pt()/(it->pt()+it->trackIso()+it->caloIso()))>0.95 && it->isGlobalMuon()){
       if(gltr.isNull()) continue;  //temporary problems with dead trackrefs
       if((gltr->chi2()/gltr->ndof())<10 && trtr->numberOfValidHits()>=11) {
 
         double dRmin = 9999.;
         for(std::vector<pat::Jet>::const_iterator ajet = seljets.begin(); ajet != seljets.end(); ajet++) {
           math::XYZTLorentzVector jet = ajet->p4();
           math::XYZTLorentzVector muon = it->p4();
           double tmpdR = DeltaR(muon,jet);
           if(tmpdR<dRmin) dRmin = tmpdR;
         }
         if(dRmin>0.3) {   //temporary problems with muon isolation
           nmuons++;
           selMuons.push_back(*it);
         }
       }
     }
   }
   //std::cout<<"selected Muons: "<<nleptons<<std::endl;
   if(nmuons!=1) return;
 
   edm::Handle< edm::View<pat::Electron> > electron_handle;
   evt.getByToken(electronsToken_, electron_handle);
   if(!electron_handle.isValid()) return;
   const edm::View<pat::Electron> electrons = *electron_handle;
   int nelectrons = 0;
   for(edm::View<pat::Electron>::const_iterator it = electrons.begin(); it!=electrons.end(); it++) {
     if(it->pt()>30 && fabs(it->eta())<2.4 && (it->pt()/(it->pt()+it->trackIso()+it->caloIso()))>0.95 && it->isElectronIDAvailable("eidTight"))
     {
       if(it->electronID("eidTight")==1) nelectrons++;
     }
   }
   if(nelectrons>0) return;
   //end of the preselection
 
 
   math::XYZTLorentzVector muon = selMuons.begin()->p4();
 
   //count the number of selected events
   selEv++;
   //skip event if enough events are already selected
   if(selEv>maxEv_ && maxEv_!=-1) return;
 
   //calculation of InputVariables
   //see TopQuarkAnalysis/TopTools/interface/TtSemiLepSignalSel.h
   //                             /src/TtSemiLepSignalSel.cc
   //all objects i.e. jets, muons, electrons... which are needed for the calculation
   //of the input-variables have to be passed to this class
 
   TtSemiLepSignalSel selection(seljets,muon,MET);
 
   //this is only needed for the default example
   edm::Handle<TtGenEvent> genEvt;
   evt.getByToken(genEvtToken_, genEvt);
 
   double weight = 1.0; //standard no weight, i.e. weight=1.0, set this to the corresponding weight if
                        //different weights for different events are available
   if(whatData_==-1) {  //your training-file contains both, signal and background events
     bool isSignal;
     isSignal = true;//true for signal, false for background this has to be derived in some way
     evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, isSignal);
   }
   else {
 
     if(whatData_==1){ //your tree contains only signal events
       //if needed do a special signal selection here
       //the following code is for the default example
       if(genEvt->isSemiLeptonic() && genEvt->semiLeptonicChannel() == lepChannel_) {
         //std::cout<<"a tt_semlep muon event"<<std::endl;
         evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, true);
       }
       else selEv--;
     }
     else if(whatData_==0){
       //std::cout<<"a Wjets event"<<std::endl;
       evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, false);
     }
     else std::cout<<"Config File Error!! Please check <whatData> in TtSemiLepSignalSelMVATrainer.cfi";
   }
 }

void TtSemiLepSignalSelMVATrainer::beginJob ( void )

privatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 158 of file TtSemiLepSignalSelMVATrainer.cc.

References gather_cfg::cout, selEv, and whatData_.

                                            {
   selEv = 0;
   if(whatData_!=-1 && whatData_!=0 && whatData_!=1){
     std::cout<<"Config File Error!! Please check <whatData> in TtSemiLepSignalSelMVATrainer.cfi"<<std::endl;;
     return;
   }
 }

double TtSemiLepSignalSelMVATrainer::DeltaPhi	(	const math::XYZTLorentzVector &	v1,
		const math::XYZTLorentzVector &	v2
	)

private

Definition at line 166 of file TtSemiLepSignalSelMVATrainer.cc.

References dPhi(), and Pi.

Referenced by DeltaR().

 {
   double dPhi = fabs(v1.Phi() - v2.Phi());
   if (dPhi > TMath::Pi()) dPhi =  2*TMath::Pi() - dPhi;
   return dPhi;
 }

double TtSemiLepSignalSelMVATrainer::DeltaR	(	const math::XYZTLorentzVector &	v1,
		const math::XYZTLorentzVector &	v2
	)

private

Definition at line 173 of file TtSemiLepSignalSelMVATrainer.cc.

References DeltaPhi(), dPhi(), and PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR.

Referenced by analyze().

 {
   double dPhi = DeltaPhi(v1,v2);
   double dR = TMath::Sqrt((v1.Eta()-v2.Eta())*(v1.Eta()-v2.Eta())+dPhi*dPhi);
   return dR;
 }

Member Data Documentation

edm::EDGetTokenT< edm::View<pat::Electron> > TtSemiLepSignalSelMVATrainer::electronsToken_

private

Definition at line 51 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::EDGetTokenT<TtGenEvent> TtSemiLepSignalSelMVATrainer::genEvtToken_

private

Definition at line 54 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::EDGetTokenT< std::vector<pat::Jet> > TtSemiLepSignalSelMVATrainer::jetsToken_

private

Definition at line 52 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::lepChannel_

private

Definition at line 56 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::maxEv_

private

Definition at line 58 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::EDGetTokenT<edm::View<pat::MET> > TtSemiLepSignalSelMVATrainer::METsToken_

private

Definition at line 53 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::EDGetTokenT< edm::View<pat::Muon> > TtSemiLepSignalSelMVATrainer::muonsToken_

private

Definition at line 50 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

PhysicsTools::MVAComputerCache TtSemiLepSignalSelMVATrainer::mvaComputer

private

Definition at line 61 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::selEv

private

Definition at line 59 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze(), and beginJob().

int TtSemiLepSignalSelMVATrainer::whatData_

private

Definition at line 57 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze(), and beginJob().

Classes

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation