#include <MVAJetIdMaker.h>

Public Member Functions
	MVAJetIdMaker (const edm::ParameterSet &, edm::ConsumesCollector &&)

void	SetVars (HWW &, const edm::Event &, const edm::EventSetup &)

Private Attributes
PileupJetIdAlgo *	fPUJetIdAlgo

edm::EDGetTokenT < reco::JetCorrector >	jetCorrectorToken_

edm::EDGetTokenT < reco::PFJetCollection >	PFJetCollection_

edm::EDGetTokenT < reco::VertexCollection >	thePVCollection_

edm::EDGetTokenT< double >	theRhoCollection_

Detailed Description

Definition at line 16 of file MVAJetIdMaker.h.

Constructor & Destructor Documentation

MVAJetIdMaker::MVAJetIdMaker	(	const edm::ParameterSet &	iConfig,
		edm::ConsumesCollector &&	iCollector
	)

Definition at line 10 of file MVAJetIdMaker.cc.

References fPUJetIdAlgo, edm::ParameterSet::getParameter(), jetCorrectorToken_, PFJetCollection_, thePVCollection_, and theRhoCollection_.

                                                                                               {
 
   PFJetCollection_     = iCollector.consumes<reco::PFJetCollection> (iConfig.getParameter<edm::InputTag>("pfJetsInputTag"));
   thePVCollection_     = iCollector.consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("primaryVertexInputTag"));
   theRhoCollection_    = iCollector.consumes<double>(iConfig.getParameter<edm::InputTag>("rhoInputTag"));
   jetCorrectorToken_   = iCollector.consumes<reco::JetCorrector>(iConfig.getParameter<edm::InputTag>("jetCorrector"));
 
   fPUJetIdAlgo = new PileupJetIdAlgo(iConfig.getParameter<edm::ParameterSet>("puJetIDParams"),true);
 
 }

Member Function Documentation

void MVAJetIdMaker::SetVars	(	HWW &	hww,
		const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

Definition at line 32 of file MVAJetIdMaker.cc.

Referenced by HWWAnalyzer::analyze().

                                                                                         {
 
   using namespace std;
   using namespace edm;
   using namespace reco;
 
   hww.Load_pfjets_corr_p4();
   hww.Load_pfjets_mvavalue();
   hww.Load_pfjets_JEC();
 
   bool validToken;
 
   //Uncorrected Jets
   Handle<PFJetCollection>       lHUCJets;
   validToken = iEvent.getByToken(PFJetCollection_, lHUCJets);
   if(!validToken) return;
   PFJetCollection               lUCJets = *lHUCJets;
 
   // vertices    
   Handle<reco::VertexCollection> lHVertices;
   validToken = iEvent.getByToken(thePVCollection_, lHVertices); 
   if(!validToken) return;
   VertexCollection lVertices = *lHVertices;
 
   Handle<double> lHrho;
   validToken = iEvent.getByToken(theRhoCollection_, lHrho); 
   if(!validToken) return;
   double lrho = *lHrho;
 
   edm::Handle<reco::JetCorrector> jetCorrector;
   iEvent.getByToken( jetCorrectorToken_, jetCorrector );
   std::vector<reco::PFJet> lCJets;
   for(reco::PFJetCollection::const_iterator jet=lUCJets.begin(); jet!=lUCJets.end(); ++jet){
 
     reco::PFJet tempJet = *jet; 
     tempJet.scaleEnergy(jetCorrector->correction(*jet));
     lCJets.push_back(tempJet);
 
   }
 
 
   // select good vertices 
   // make new collection to put into computeIdVariables(...)
   VertexCollection lGoodVertices;
   for(int ivtx    = 0; ivtx < (int)lVertices.size(); ivtx++) {
 
       const Vertex       *vtx = &(lVertices.at(ivtx));
       if( vtx->isFake()                     )  continue;
       if( vtx->ndof()<=4                    )  continue;
       if( vtx->position().Rho()>2.0         )  continue;
       if( fabs(vtx->position().Z())>24.0  )  continue;
       lGoodVertices.push_back(*vtx);
   }
 
   // loop over jets 
   for(int i0   = 0; i0 < (int) lUCJets.size(); i0++) {   // uncorrected jets collection
       const PFJet       *pUCJet = &(lUCJets.at(i0));
       for(int i1 = 0; i1 < (int) lCJets.size(); i1++) {   // corrected jets collection
           const PFJet     *pCJet  = &(lCJets.at(i1));
           if( pUCJet->jetArea() != pCJet->jetArea()                     ) continue;
           if( fabs(pUCJet->eta() - pCJet->eta())         > 0.001         ) continue;
           if( pUCJet->pt()                               < 0.0    ) continue;
           double lJec = pCJet ->pt()/pUCJet->pt();
 
           // calculate mva value only when there are good vertices 
           // otherwise store -999
           if( lGoodVertices.size()>0 ) {
             PileupJetIdentifier lPUJetId =  fPUJetIdAlgo->computeIdVariables(pCJet,lJec,&lGoodVertices[0],lGoodVertices,lrho);
             hww.pfjets_mvavalue() .push_back( lPUJetId.mva()              );
         hww.pfjets_JEC() .push_back( lJec ); 
           
         // print out MVA inputs 
         if(0) {
 
           LogDebug("MVAJetIdMaker")
             << "Debug Jet MVA : "
             << iEvent.id()          << " : "
             << lPUJetId.nvtx()      << " "
             << pCJet->pt()          << " "
             << lPUJetId.jetEta()    << " "
             << lPUJetId.jetPhi()    << " "
             << lPUJetId.d0()        << " "
             << lPUJetId.dZ()        << " "
             << lPUJetId.beta()      << " "
             << lPUJetId.betaStar()  << " "
             << lPUJetId.nCharged()  << " "
             << lPUJetId.nNeutrals() << " "
             << lPUJetId.dRMean()    << " "
             << lPUJetId.frac01()    << " "
             << lPUJetId.frac02()    << " "
             << lPUJetId.frac03()    << " "
             << lPUJetId.frac04()    << " "
             << lPUJetId.frac05()
             << " === : === "
             << lPUJetId.mva();
         }
           }
           else             
 
             hww.pfjets_mvavalue() .push_back( -999. );
 
           break;
 
       } // lCJets
   } // lUCJets
 }