CMS 3D CMS Logo

TtSemiLepSignalSelMVATrainer Class Reference

#include <TopQuarkAnalysis/TopEventSelection/plugins/TtSemiLepSignalSelMVATrainer.h>

Inheritance diagram for TtSemiLepSignalSelMVATrainer:

edm::EDAnalyzer

List of all members.

Public Member Functions

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

Private Member Functions

virtual void analyze (const edm::Event &evt, const edm::EventSetup &setup)
virtual void beginJob (const edm::EventSetup &)
double DeltaPhi (math::XYZTLorentzVector v1, math::XYZTLorentzVector v2)
double DeltaR (math::XYZTLorentzVector v1, math::XYZTLorentzVector v2)

Private Attributes

edm::InputTag electrons_
edm::InputTag jets_
int lepChannel_
int maxEv_
edm::InputTag METs_
edm::InputTag muons_
PhysicsTools::MVAComputerCache mvaComputer
int selEv
int whatData_

Classes

struct  JetwithHigherPt

Detailed Description

Definition at line 24 of file TtSemiLepSignalSelMVATrainer.h.

Constructor & Destructor Documentation

TtSemiLepSignalSelMVATrainer::TtSemiLepSignalSelMVATrainer ( const edm::ParameterSet &  cfg  )  [explicit]

Definition at line 19 of file TtSemiLepSignalSelMVATrainer.cc.

00019                                                                                     :
00020   muons_     (cfg.getParameter<edm::InputTag>("muons")),
00021   electrons_ (cfg.getParameter<edm::InputTag>("electrons")),
00022   jets_      (cfg.getParameter<edm::InputTag>("jets")),
00023   METs_      (cfg.getParameter<edm::InputTag>("METs")),
00024   lepChannel_(cfg.getParameter<int>("lepChannel")),
00025   whatData_  (cfg.getParameter<int>("whatData")),
00026   maxEv_     (cfg.getParameter<int>("maxEv"))
00027 {
00028 }

TtSemiLepSignalSelMVATrainer::~TtSemiLepSignalSelMVATrainer (  ) 

Definition at line 30 of file TtSemiLepSignalSelMVATrainer.cc.

00031 {
00032 }

Member Function Documentation

void TtSemiLepSignalSelMVATrainer::analyze ( const edm::Event &  evt,
const edm::EventSetup &  setup 
) [private, virtual]

Implements edm::EDAnalyzer.

Definition at line 35 of file TtSemiLepSignalSelMVATrainer.cc.

References edm::View< T >::begin(), GenMuonPlsPt100GeV_cfg::cout, DeltaR(), pat::Flags::Overlap::Electrons, electrons_, edm::View< T >::end(), evaluateTtSemiLepSignalSel(), TtGenEvtProducer_cfi::genEvt, edm::Event::getByLabel(), edm::Ref< C, T, F >::isNull(), edm::Handle< T >::isValid(), it, metsig::jet, pfTauBenchmarkGeneric_cfi::jets, jets_, lepChannel_, maxEv_, METs_, metsig::muon, muons_cfi::muons, muons_, mvaComputer, selEv, python::multivaluedict::sort(), pat::Flags::test(), PhysicsTools::MVAComputerCache::update(), weight, and whatData_.

00036 {
00037   //communication with CMSSW CondDB
00038   mvaComputer.update<TtSemiLepSignalSelMVARcd>("trainer", setup, "traintreeSaver");
00039 
00040   // can occur in the last iteration when the 
00041   // MVATrainer is about to save the result
00042   if(!mvaComputer) return;
00043 
00044 
00045   //make your preselection here!!
00046   //the following code is for the default example  
00047   edm::Handle<edm::View<pat::MET> > MET_handle;
00048   evt.getByLabel(METs_,MET_handle);
00049   if(!MET_handle.isValid()) return;
00050   const edm::View<pat::MET> MET = *MET_handle;
00051 
00052   edm::Handle< std::vector<pat::Jet> > jet_handle;
00053   evt.getByLabel(jets_, jet_handle);
00054   if(!jet_handle.isValid()) return;
00055   const std::vector<pat::Jet> jets = *jet_handle;
00056   unsigned int nJets = 0;
00057   std::vector<pat::Jet> seljets;
00058   //std::cout<<"number of jets: "<<jets.size()<<std::endl;
00059   for(std::vector<pat::Jet>::const_iterator it = jets.begin(); it != jets.end(); it++) {
00060     //std::cout<<"Jet Et: "<<it->et()<<" Eta: "<<fabs(it->eta())<<std::endl;
00061     if(!(pat::Flags::test(*it, pat::Flags::Overlap::Electrons))) continue;
00062     if(it->et()>30. && fabs(it->eta())<2.4) {
00063       seljets.push_back(*it);
00064       nJets++;
00065     }
00066   }
00067   //std::cout<<"selected Jets: "<<nJets<<std::endl;
00068   if(nJets<4) return;
00069 
00070   //sort by Pt
00071   sort(seljets.begin(),seljets.end(),JetwithHigherPt());
00072  
00073   edm::Handle< edm::View<pat::Muon> > muon_handle; 
00074   evt.getByLabel(muons_, muon_handle);
00075   if(!muon_handle.isValid()) return;
00076   const edm::View<pat::Muon> muons = *muon_handle;
00077   int nmuons = 0;
00078   std::vector<pat::Muon> selMuons;
00079   for(edm::View<pat::Muon>::const_iterator it = muons.begin(); it!=muons.end(); it++) {
00080     reco::TrackRef gltr = it->track(); // global track
00081     reco::TrackRef trtr = it->innerTrack(); // tracker track
00082     if(it->pt()>30 && fabs(it->eta())<2.1 && (it->pt()/(it->pt()+it->trackIso()+it->caloIso()))>0.95 && it->isGlobalMuon()){      
00083       if(gltr.isNull()) continue;  //temporary problems with dead trackrefs
00084       if((gltr->chi2()/gltr->ndof())<10 && trtr->numberOfValidHits()>=11) {
00085      
00086         double dRmin = 9999.;
00087         for(std::vector<pat::Jet>::const_iterator ajet = seljets.begin(); ajet != seljets.end(); ajet++) {
00088           math::XYZTLorentzVector jet = ajet->p4();
00089           math::XYZTLorentzVector muon = it->p4();
00090           double tmpdR = DeltaR(muon,jet);
00091           if(tmpdR<dRmin) dRmin = tmpdR;
00092         }
00093         reco::TrackRef trtr = it->track(); // tracker track
00094         if(dRmin>0.3) {   //temporary problems with muon isolation
00095           nmuons++;
00096           selMuons.push_back(*it);
00097         }
00098       }
00099     }
00100   }
00101   //std::cout<<"selected Muons: "<<nleptons<<std::endl;
00102   if(nmuons!=1) return;
00103   
00104   edm::Handle< edm::View<pat::Electron> > electron_handle; 
00105   evt.getByLabel(electrons_, electron_handle);
00106   if(!electron_handle.isValid()) return;
00107   const edm::View<pat::Electron> electrons = *electron_handle;
00108   int nelectrons = 0;
00109   for(edm::View<pat::Electron>::const_iterator it = electrons.begin(); it!=electrons.end(); it++) {
00110     if(it->pt()>30 && fabs(it->eta())<2.4 && (it->pt()/(it->pt()+it->trackIso()+it->caloIso()))>0.95 && it->isElectronIDAvailable("eidTight"))
00111     { 
00112       if(it->electronID("eidTight")==1) nelectrons++;
00113     }
00114   }
00115   if(nelectrons>0) return;
00116   //end of the preselection
00117 
00118   
00119   math::XYZTLorentzVector muon = selMuons.begin()->p4();
00120 
00121   //count the number of selected events
00122   selEv++;
00123   //skip event if enough events are already selected
00124   if(selEv>maxEv_ && maxEv_!=-1) return;
00125 
00126   //calculation of InputVariables
00127   //see TopQuarkAnalysis/TopTools/interface/TtSemiLepSignalSel.h
00128   //                             /src/TtSemiLepSignalSel.cc
00129   //all objects i.e. jets, muons, electrons... which are needed for the calculation
00130   //of the input-variables have to be passed to this class
00131 
00132   TtSemiLepSignalSel selection(seljets,muon,MET);
00133 
00134   //this is only needed for the default example
00135   edm::Handle<TtGenEvent> genEvt;
00136   evt.getByLabel("genEvt", genEvt);
00137 
00138   double weight = 1.0; //standard no weight, i.e. weight=1.0, set this to the corresponding weight if 
00139                        //different weights for different events are available  
00140   if(whatData_==-1) {  //your training-file contains both, signal and background events
00141     bool isSignal;
00142     isSignal = true;//true for signal, false for background this has to be derived in some way
00143     evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, isSignal);
00144   }
00145   else {
00146    
00147     if(whatData_==1){ //your tree contains only signal events
00148       //if needed do a special signal selection here
00149       //the following code is for the default example
00150       if(genEvt->isSemiLeptonic() && genEvt->semiLeptonicChannel() == lepChannel_) {
00151         //std::cout<<"a tt_semlep muon event"<<std::endl;
00152         evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, true);
00153       }
00154       else selEv--;
00155     }
00156     else if(whatData_==0){
00157       //std::cout<<"a Wjets event"<<std::endl;
00158       evaluateTtSemiLepSignalSel(mvaComputer, selection, weight, true, false);
00159     }
00160     else std::cout<<"Config File Error!! Please check <whatData> in TtSemiLepSignalSelMVATrainer.cfi";
00161   }
00162 }

void TtSemiLepSignalSelMVATrainer::beginJob ( const edm::EventSetup &   )  [private, virtual]

Reimplemented from edm::EDAnalyzer.

Definition at line 164 of file TtSemiLepSignalSelMVATrainer.cc.

References GenMuonPlsPt100GeV_cfg::cout, lat::endl(), selEv, and whatData_.

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

double TtSemiLepSignalSelMVATrainer::DeltaPhi ( math::XYZTLorentzVector  v1,
math::XYZTLorentzVector  v2 
) [private]

Definition at line 172 of file TtSemiLepSignalSelMVATrainer.cc.

References dPhi(), and Pi.

Referenced by DeltaR().

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

double TtSemiLepSignalSelMVATrainer::DeltaR ( math::XYZTLorentzVector  v1,
math::XYZTLorentzVector  v2 
) [private]

Definition at line 179 of file TtSemiLepSignalSelMVATrainer.cc.

References DeltaPhi(), and dPhi().

Referenced by analyze().

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

Member Data Documentation

edm::InputTag TtSemiLepSignalSelMVATrainer::electrons_ [private]

Definition at line 47 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::InputTag TtSemiLepSignalSelMVATrainer::jets_ [private]

Definition at line 48 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::lepChannel_ [private]

Definition at line 51 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::maxEv_ [private]

Definition at line 53 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::InputTag TtSemiLepSignalSelMVATrainer::METs_ [private]

Definition at line 49 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

edm::InputTag TtSemiLepSignalSelMVATrainer::muons_ [private]

Definition at line 46 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

PhysicsTools::MVAComputerCache TtSemiLepSignalSelMVATrainer::mvaComputer [private]

Definition at line 56 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze().

int TtSemiLepSignalSelMVATrainer::selEv [private]

Definition at line 54 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze(), and beginJob().

int TtSemiLepSignalSelMVATrainer::whatData_ [private]

Definition at line 52 of file TtSemiLepSignalSelMVATrainer.h.

Referenced by analyze(), and beginJob().

The documentation for this class was generated from the following files:
Generated on Tue Jun 9 18:34:46 2009 for CMSSW by  doxygen 1.5.4