mvaMEtUtilities Class Reference

#include <mvaMEtUtilities.h>

Classes
struct	JetInfo
struct	leptonInfo
struct	pfCandInfo

Public Member Functions
std::vector< JetInfo >	cleanJets (const std::vector< JetInfo > &, const std::vector< leptonInfo > &, double, double)
std::vector< pfCandInfo >	cleanPFCands (const std::vector< pfCandInfo > &, const std::vector< leptonInfo > &, double, bool)
CommonMETData	computeJetSum_neutral (const std::vector< JetInfo > &, bool)
CommonMETData	computeNegNoPURecoil (const CommonMETData &, const std::vector< pfCandInfo > &, const std::vector< JetInfo > &, double)
CommonMETData	computeNegPFRecoil (const CommonMETData &, const std::vector< pfCandInfo > &, double)
CommonMETData	computeNegPUCRecoil (const CommonMETData &, const std::vector< pfCandInfo > &, const std::vector< JetInfo > &, double)
CommonMETData	computeNegTrackRecoil (const CommonMETData &, const std::vector< pfCandInfo > &, double)
CommonMETData	computePFCandSum (const std::vector< pfCandInfo > &, double, int)
CommonMETData	computePUMEt (const std::vector< pfCandInfo > &, const std::vector< JetInfo > &, double)
CommonMETData	computeSumLeptons (const std::vector< leptonInfo > &leptons, bool iCharged)
void	finalize (CommonMETData &metData)
reco::Candidate::LorentzVector	leadJetP4 (const std::vector< JetInfo > &)
	mvaMEtUtilities (const edm::ParameterSet &cfg)
unsigned	numJetsAboveThreshold (const std::vector< JetInfo > &, double)
bool	passesMVA (const reco::Candidate::LorentzVector &, double)
reco::Candidate::LorentzVector	subleadJetP4 (const std::vector< JetInfo > &)
virtual	~mvaMEtUtilities ()

Protected Member Functions
reco::Candidate::LorentzVector	jetP4 (const std::vector< JetInfo > &, unsigned)

Protected Attributes
double	mvaCut_ [3][4][4]

Friends
bool	operator< (const JetInfo &, const JetInfo &)

Detailed Description

Definition at line 12 of file mvaMEtUtilities.h.

Constructor & Destructor Documentation

mvaMEtUtilities::mvaMEtUtilities

(

const edm::ParameterSet &

cfg

)

Definition at line 8 of file mvaMEtUtilities.cc.

References mvaCut_.

{
  // jet id
  /* ===> this code parses an xml it uses parameter set 
  edm::ParameterSet jetConfig = iConfig.getParameter<edm::ParameterSet>("JetIdParams");
  for(int i0 = 0; i0 < 3; i0++) { 
    std::string lCutType                            = "Tight";
    if(i0 == PileupJetIdentifier::kMedium) lCutType = "Medium";
    if(i0 == PileupJetIdentifier::kLoose)  lCutType = "Loose";
    std::vector<double> pt010  = jetConfig.getParameter<std::vector<double> >(("Pt010_" +lCutType).c_str());
    std::vector<double> pt1020 = jetConfig.getParameter<std::vector<double> >(("Pt1020_"+lCutType).c_str());
    std::vector<double> pt2030 = jetConfig.getParameter<std::vector<double> >(("Pt2030_"+lCutType).c_str());
    std::vector<double> pt3050 = jetConfig.getParameter<std::vector<double> >(("Pt3050_"+lCutType).c_str());
    for(int i1 = 0; i1 < 4; i1++) mvacut_[i0][0][i1] = pt010 [i1];
    for(int i1 = 0; i1 < 4; i1++) mvacut_[i0][1][i1] = pt1020[i1];
    for(int i1 = 0; i1 < 4; i1++) mvacut_[i0][2][i1] = pt2030[i1];
    for(int i1 = 0; i1 < 4; i1++) mvacut_[i0][3][i1] = pt3050[i1];
    }
  */
  //Tight Id => not used
  mvaCut_[0][0][0] =  0.5; mvaCut_[0][0][1] = 0.6; mvaCut_[0][0][2] = 0.6; mvaCut_[0][0][3] = 0.9;
  mvaCut_[0][1][0] = -0.2; mvaCut_[0][1][1] = 0.2; mvaCut_[0][1][2] = 0.2; mvaCut_[0][1][3] = 0.6;
  mvaCut_[0][2][0] =  0.3; mvaCut_[0][2][1] = 0.4; mvaCut_[0][2][2] = 0.7; mvaCut_[0][2][3] = 0.8;
  mvaCut_[0][3][0] =  0.5; mvaCut_[0][3][1] = 0.4; mvaCut_[0][3][2] = 0.8; mvaCut_[0][3][3] = 0.9;
  //Medium id => not used
  mvaCut_[1][0][0] =  0.2; mvaCut_[1][0][1] = 0.4; mvaCut_[1][0][2] = 0.2; mvaCut_[1][0][3] = 0.6;
  mvaCut_[1][1][0] = -0.3; mvaCut_[1][1][1] = 0. ; mvaCut_[1][1][2] = 0. ; mvaCut_[1][1][3] = 0.5;
  mvaCut_[1][2][0] =  0.2; mvaCut_[1][2][1] = 0.2; mvaCut_[1][2][2] = 0.5; mvaCut_[1][2][3] = 0.7;
  mvaCut_[1][3][0] =  0.3; mvaCut_[1][3][1] = 0.2; mvaCut_[1][3][2] = 0.7; mvaCut_[1][3][3] = 0.8;
  //Met Id => used
  mvaCut_[2][0][0] = -0.2; mvaCut_[2][0][1] = -0.3; mvaCut_[2][0][2] = -0.5; mvaCut_[2][0][3] = -0.5;
  mvaCut_[2][1][0] = -0.2; mvaCut_[2][1][1] = -0.2; mvaCut_[2][1][2] = -0.5; mvaCut_[2][1][3] = -0.3;
  mvaCut_[2][2][0] = -0.2; mvaCut_[2][2][1] = -0.2; mvaCut_[2][2][2] = -0.2; mvaCut_[2][2][3] =  0.1;
  mvaCut_[2][3][0] = -0.2; mvaCut_[2][3][1] = -0.2; mvaCut_[2][3][2] =  0. ; mvaCut_[2][3][3] =  0.2;
}

mvaMEtUtilities::~mvaMEtUtilities ( )

virtual

Definition at line 44 of file mvaMEtUtilities.cc.

{
// nothing to be done yet...
}

Member Function Documentation

std::vector< mvaMEtUtilities::JetInfo > mvaMEtUtilities::cleanJets	(	const std::vector< JetInfo > &	jets,
		const std::vector< leptonInfo > &	leptons,
		double	ptThreshold,
		double	dRmatch
	)

Definition at line 98 of file mvaMEtUtilities.cc.

References reco::deltaR2(), and metsig::jet.

Referenced by PFMETAlgorithmMVA::setInput().

{

  double dR2match = dRmatch*dRmatch;
  std::vector<JetInfo> retVal;
  for ( std::vector<JetInfo>::const_iterator jet = jets.begin();
    jet != jets.end(); ++jet ) {
    bool isOverlap = false;
    for ( std::vector<leptonInfo>::const_iterator lepton = leptons.begin();
      lepton != leptons.end(); ++lepton ) {
      if ( deltaR2(jet->p4_, lepton->p4_) < dR2match ) isOverlap = true;    
    }
    if ( jet->p4_.pt() > ptThreshold && !isOverlap ) retVal.push_back(*jet);
  }
  return retVal;
}

std::vector< mvaMEtUtilities::pfCandInfo > mvaMEtUtilities::cleanPFCands	(	const std::vector< pfCandInfo > &	pfCandidates,
		const std::vector< leptonInfo > &	leptons,
		double	dRmatch,
		bool	invert
	)

Definition at line 117 of file mvaMEtUtilities.cc.

References reco::deltaR2().

{

  double dR2match = dRmatch*dRmatch;
  std::vector<pfCandInfo> retVal;
  for ( std::vector<pfCandInfo>::const_iterator pfCandidate = pfCandidates.begin();
    pfCandidate != pfCandidates.end(); ++pfCandidate ) {
    bool isOverlap = false;
    for ( std::vector<leptonInfo>::const_iterator lepton = leptons.begin();
      lepton != leptons.end(); ++lepton ) {
      if ( deltaR2(pfCandidate->p4_, lepton->p4_) < dR2match ) isOverlap = true;
    }
    if ( (!isOverlap && !invert) || (isOverlap && invert) ) retVal.push_back(*pfCandidate);
  }
  return retVal;
}

CommonMETData mvaMEtUtilities::computeJetSum_neutral	(	const std::vector< JetInfo > &	jets,
		bool	mvaPassFlag
	)

Definition at line 186 of file mvaMEtUtilities.cc.

References finalize(), metsig::jet, CommonMETData::mex, CommonMETData::mey, passesMVA(), and CommonMETData::sumet.

Referenced by computeNegNoPURecoil(), computeNegPUCRecoil(), and computePUMEt().

{
  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  for ( std::vector<JetInfo>::const_iterator jet = jets.begin();
    jet != jets.end(); ++jet ) {
    bool passesMVAjetId = passesMVA(jet->p4_, jet->mva_);
    if (  passesMVAjetId && !mvaPassFlag ) continue;
    if ( !passesMVAjetId &&  mvaPassFlag ) continue;
    retVal.mex   += jet->p4_.px()*jet->neutralEnFrac_;
    retVal.mey   += jet->p4_.py()*jet->neutralEnFrac_;
    retVal.sumet += jet->p4_.pt()*jet->neutralEnFrac_;
  }
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computeNegNoPURecoil	(	const CommonMETData &	leptons,
		const std::vector< pfCandInfo > &	pfCandidates,
		const std::vector< JetInfo > &	jets,
		double	dZcut
	)

Definition at line 245 of file mvaMEtUtilities.cc.

References computeJetSum_neutral(), computePFCandSum(), finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  CommonMETData trackSumNoPU = computePFCandSum(pfCandidates, dZcut, 0);
  CommonMETData jetSumNoPU_neutral = computeJetSum_neutral(jets, true);
  retVal.mex   = -(trackSumNoPU.mex + jetSumNoPU_neutral.mex)  + leptons.mex;
  retVal.mey   = -(trackSumNoPU.mey + jetSumNoPU_neutral.mey)  + leptons.mey;
  retVal.sumet = trackSumNoPU.sumet + jetSumNoPU_neutral.sumet - leptons.sumet;
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computeNegPFRecoil	(	const CommonMETData &	leptons,
		const std::vector< pfCandInfo > &	pfCandidates,
		double	dZcut
	)

Definition at line 221 of file mvaMEtUtilities.cc.

References computePFCandSum(), finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  CommonMETData retVal;
  CommonMETData pfCandSum = computePFCandSum(pfCandidates, dZcut, 2);
  retVal.mex   = -pfCandSum.mex + leptons.mex; 
  retVal.mey   = -pfCandSum.mey + leptons.mey;
  retVal.sumet = pfCandSum.sumet - leptons.sumet;
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computeNegPUCRecoil	(	const CommonMETData &	leptons,
		const std::vector< pfCandInfo > &	pfCandidates,
		const std::vector< JetInfo > &	jets,
		double	dZcut
	)

Definition at line 262 of file mvaMEtUtilities.cc.

References computeJetSum_neutral(), computePFCandSum(), finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
   CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  CommonMETData pfCandSum = computePFCandSum(pfCandidates, dZcut, 2);
  CommonMETData trackSumNoPU = computePFCandSum(pfCandidates, dZcut, 1);
  CommonMETData jetSumPU_neutral = computeJetSum_neutral(jets, false);
  retVal.mex   = -(pfCandSum.mex - (trackSumNoPU.mex + jetSumPU_neutral.mex))    + leptons.mex;
  retVal.mey   = -(pfCandSum.mey - (trackSumNoPU.mey + jetSumPU_neutral.mey))    + leptons.mey;
  retVal.sumet = pfCandSum.sumet - (trackSumNoPU.sumet + jetSumPU_neutral.sumet) - leptons.sumet;
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computeNegTrackRecoil	(	const CommonMETData &	leptons,
		const std::vector< pfCandInfo > &	pfCandidates,
		double	dZcut
	)

Definition at line 233 of file mvaMEtUtilities.cc.

References computePFCandSum(), finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  CommonMETData retVal;
  CommonMETData trackSum = computePFCandSum(pfCandidates, dZcut, 0);
  retVal.mex   = -trackSum.mex  + leptons.mex; 
  retVal.mey   = -trackSum.mey  + leptons.mey;
  retVal.sumet = trackSum.sumet - leptons.sumet;
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computePFCandSum	(	const std::vector< pfCandInfo > &	pfCandidates,
		double	dZmax,
		int	dZflag
	)

Definition at line 142 of file mvaMEtUtilities.cc.

References finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by computeNegNoPURecoil(), computeNegPFRecoil(), computeNegPUCRecoil(), computeNegTrackRecoil(), and computePUMEt().

{
  // dZcut
  //   maximum distance within which tracks are considered to be associated to hard scatter vertex
  // dZflag 
  //   0 : select charged PFCandidates originating from hard scatter vertex
  //   1 : select charged PFCandidates originating from pile-up vertices
  //   2 : select all PFCandidates
  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  for ( std::vector<pfCandInfo>::const_iterator pfCandidate = pfCandidates.begin();
    pfCandidate != pfCandidates.end(); ++pfCandidate ) {
    if ( pfCandidate->dZ_ < 0.    && dZflag != 2 ) continue;
    if ( pfCandidate->dZ_ > dZmax && dZflag == 0 ) continue;
    if ( pfCandidate->dZ_ < dZmax && dZflag == 1 ) continue;
    retVal.mex   += pfCandidate->p4_.px();
    retVal.mey   += pfCandidate->p4_.py();
    retVal.sumet += pfCandidate->p4_.pt();
  }
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computePUMEt	(	const std::vector< pfCandInfo > &	pfCandidates,
		const std::vector< JetInfo > &	jets,
		double	dZcut
	)

Definition at line 205 of file mvaMEtUtilities.cc.

References computeJetSum_neutral(), computePFCandSum(), finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  CommonMETData trackSumPU = computePFCandSum(pfCandidates, dZcut, 1);
  CommonMETData jetSumPU_neutral = computeJetSum_neutral(jets, false);
  retVal.mex   = -(trackSumPU.mex + jetSumPU_neutral.mex);
  retVal.mey   = -(trackSumPU.mey + jetSumPU_neutral.mey);
  retVal.sumet = trackSumPU.sumet + jetSumPU_neutral.sumet;
  finalize(retVal);
  return retVal;
}

CommonMETData mvaMEtUtilities::computeSumLeptons	(	const std::vector< leptonInfo > &	leptons,
		bool	iCharged
	)

Definition at line 167 of file mvaMEtUtilities.cc.

References finalize(), CommonMETData::mex, CommonMETData::mey, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  //Computes vectorial sum of charged(iCharged == true) or all(iCharged == false) components
  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;
  for ( std::vector<leptonInfo>::const_iterator lepton = leptons.begin();
    lepton != leptons.end(); ++lepton ) {
    double pChargedFrac = 1;
    if(iCharged) pChargedFrac = lepton->chargedFrac_;
    retVal.mex   += lepton->p4_.px()*pChargedFrac;
    retVal.mey   += lepton->p4_.py()*pChargedFrac;
    retVal.sumet += lepton->p4_.pt()*pChargedFrac;
  }
  finalize(retVal);
  return retVal;
}

void mvaMEtUtilities::finalize

(

CommonMETData &

metData

)

Definition at line 135 of file mvaMEtUtilities.cc.

References CommonMETData::met, CommonMETData::mex, CommonMETData::mey, CommonMETData::mez, CommonMETData::phi, and mathSSE::sqrt().

Referenced by computeJetSum_neutral(), computeNegNoPURecoil(), computeNegPFRecoil(), computeNegPUCRecoil(), computeNegTrackRecoil(), computePFCandSum(), computePUMEt(), and computeSumLeptons().

{
  metData.met = sqrt(metData.mex*metData.mex + metData.mey*metData.mey);
  metData.mez = 0.;
  metData.phi = atan2(metData.mey, metData.mex);
}

reco::Candidate::LorentzVector mvaMEtUtilities::jetP4 ( const std::vector< JetInfo > &  jets, unsigned  idx  )

protected

Definition at line 79 of file mvaMEtUtilities.cc.

References customizeTrackingMonitorSeedNumber::idx, fwrapper::jets, and python.multivaluedict::sort().

Referenced by leadJetP4(), and subleadJetP4().

{
  reco::Candidate::LorentzVector retVal(0.,0.,0.,0.);
  if ( idx < jets.size() ) {
    std::vector<JetInfo> jets_sorted = jets;
    std::sort(jets_sorted.begin(), jets_sorted.end()); 
    retVal = jets_sorted[idx].p4_;
  }
  return retVal;
}

reco::Candidate::LorentzVector mvaMEtUtilities::leadJetP4

(

const std::vector< JetInfo > &

jets

)

Definition at line 64 of file mvaMEtUtilities.cc.

References jetP4().

Referenced by PFMETAlgorithmMVA::setInput().

{
  return jetP4(jets, 0);
}

unsigned mvaMEtUtilities::numJetsAboveThreshold	(	const std::vector< JetInfo > &	jets,
		double	ptThreshold
	)

Definition at line 89 of file mvaMEtUtilities.cc.

References metsig::jet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  unsigned retVal = 0;
  for ( std::vector<JetInfo>::const_iterator jet = jets.begin();
    jet != jets.end(); ++jet ) {
    if ( jet->p4_.pt() > ptThreshold ) ++retVal;
  }
  return retVal;
}

bool mvaMEtUtilities::passesMVA	(	const reco::Candidate::LorentzVector &	jetP4,
		double	mvaJetId
	)

Definition at line 49 of file mvaMEtUtilities.cc.

References funct::abs(), and mvaCut_.

Referenced by computeJetSum_neutral().

{ 
  int ptBin = 0; 
  if ( jetP4.pt() > 10. && jetP4.pt() < 20. ) ptBin = 1;
  if ( jetP4.pt() > 20. && jetP4.pt() < 30. ) ptBin = 2;
  if ( jetP4.pt() > 30.                     ) ptBin = 3;
  
  int etaBin = 0;
  if ( std::abs(jetP4.eta()) > 2.5  && std::abs(jetP4.eta()) < 2.75) etaBin = 1; 
  if ( std::abs(jetP4.eta()) > 2.75 && std::abs(jetP4.eta()) < 3.0 ) etaBin = 2; 
  if ( std::abs(jetP4.eta()) > 3.0  && std::abs(jetP4.eta()) < 5.0 ) etaBin = 3; 

  return ( mvaJetId > mvaCut_[2][ptBin][etaBin] );
}

reco::Candidate::LorentzVector mvaMEtUtilities::subleadJetP4

(

const std::vector< JetInfo > &

jets

)

Definition at line 69 of file mvaMEtUtilities.cc.

References jetP4().

Referenced by PFMETAlgorithmMVA::setInput().

{
  return jetP4(jets, 1);
}

Friends And Related Function Documentation

bool operator< ( const JetInfo &  jet1, const JetInfo &  jet2  )

friend

Definition at line 74 of file mvaMEtUtilities.cc.

{
  return jet1.p4_.pt() > jet2.p4_.pt();
} 

Member Data Documentation

double mvaMEtUtilities::mvaCut_[3][4][4]

protected

Definition at line 84 of file mvaMEtUtilities.h.

Referenced by mvaMEtUtilities(), and passesMVA().