MvaMEtUtilities Class Reference

#include <MvaMEtUtilities.h>

Public Types
enum	{ kPFCands =0, kLeptons, kJets }
enum	{   kPF =0, kChHS, kHS, kPU,   kHSMinusNeutralPU }

Public Member Functions
void	computeAllSums (const std::vector< reco::PUSubMETCandInfo > &jets, const std::vector< reco::PUSubMETCandInfo > &leptons, const std::vector< reco::PUSubMETCandInfo > &pfCandidates)
CommonMETData	computeRecoil (int metType)
const std::vector < reco::PUSubMETCandInfo > &	getCleanedJets () const
double	getLeptonsChSumMEX () const
double	getLeptonsChSumMEY () const
double	getLeptonsSumMEX () const
double	getLeptonsSumMEY () const
reco::Candidate::LorentzVector	leadJetP4 (const std::vector< reco::PUSubMETCandInfo > &)
	MvaMEtUtilities (const edm::ParameterSet &cfg)
unsigned	numJetsAboveThreshold (const std::vector< reco::PUSubMETCandInfo > &, double)
reco::Candidate::LorentzVector	subleadJetP4 (const std::vector< reco::PUSubMETCandInfo > &)
virtual	~MvaMEtUtilities ()

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

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

Private Member Functions
std::vector < reco::PUSubMETCandInfo >	cleanJets (const std::vector< reco::PUSubMETCandInfo > &, const std::vector< reco::PUSubMETCandInfo > &, double, double)
std::vector < reco::PUSubMETCandInfo >	cleanPFCands (const std::vector< reco::PUSubMETCandInfo > &, const std::vector< reco::PUSubMETCandInfo > &, double, bool)
CommonMETData	computeCandSum (int compKey, double dZmax, int dZflag, bool iCharged, bool mvaPassFlag, const std::vector< reco::PUSubMETCandInfo > &objects)
void	finalize (CommonMETData &metData)
bool	passesMVA (const reco::Candidate::LorentzVector &, double)

Private Attributes
std::vector < reco::PUSubMETCandInfo >	cleanedJets_
double	dzCut_
CommonMETData	leptonsChSum_
CommonMETData	leptonsSum_
CommonMETData	neutralJetHSSum_
CommonMETData	neutralJetPUSum_
CommonMETData	pfCandChHSSum_
CommonMETData	pfCandChPUSum_
CommonMETData	pfCandSum_
double	ptThreshold_

Detailed Description

Definition at line 13 of file MvaMEtUtilities.h.

Member Enumeration Documentation

anonymous enum

Enumerator
kPFCands
kLeptons
kJets

Definition at line 17 of file MvaMEtUtilities.h.

{kPFCands=0,kLeptons,kJets};

anonymous enum

Enumerator
kPF
kChHS
kHS
kPU
kHSMinusNeutralPU

Definition at line 18 of file MvaMEtUtilities.h.

{kPF=0, kChHS, kHS, kPU, kHSMinusNeutralPU};

Constructor & Destructor Documentation

MvaMEtUtilities::MvaMEtUtilities

(

const edm::ParameterSet &

cfg

)

Definition at line 8 of file MvaMEtUtilities.cc.

References dzCut_, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), mvaCut_, and ptThreshold_.

{
  // 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;

  dzCut_ = cfg.getParameter<double>("dZcut");
  ptThreshold_ = ( cfg.exists("ptThreshold") ) ?
    cfg.getParameter<int>("ptThreshold") : -1000;
}

MvaMEtUtilities::~MvaMEtUtilities ( )

virtual

Definition at line 48 of file MvaMEtUtilities.cc.

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

Member Function Documentation

std::vector< reco::PUSubMETCandInfo > MvaMEtUtilities::cleanJets ( const std::vector< reco::PUSubMETCandInfo > &  jets, const std::vector< reco::PUSubMETCandInfo > &  leptons, double  ptThreshold, double  dRmatch  )

private

Definition at line 97 of file MvaMEtUtilities.cc.

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

Referenced by computeAllSums().

{

  double dR2match = dRmatch*dRmatch;
  std::vector<reco::PUSubMETCandInfo> retVal;
  for ( std::vector<reco::PUSubMETCandInfo>::const_iterator jet = jets.begin();
    jet != jets.end(); ++jet ) {
    bool isOverlap = false;
    for ( std::vector<reco::PUSubMETCandInfo>::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< reco::PUSubMETCandInfo > MvaMEtUtilities::cleanPFCands ( const std::vector< reco::PUSubMETCandInfo > &  pfCandidates, const std::vector< reco::PUSubMETCandInfo > &  leptons, double  dRmatch, bool  invert  )

private

Definition at line 116 of file MvaMEtUtilities.cc.

References reco::deltaR2(), and HLT_25ns14e33_v1_cff::invert.

{

  double dR2match = dRmatch*dRmatch;
  std::vector<reco::PUSubMETCandInfo> retVal;
  for ( std::vector<reco::PUSubMETCandInfo>::const_iterator pfCandidate = pfCandidates.begin();
    pfCandidate != pfCandidates.end(); ++pfCandidate ) {
    bool isOverlap = false;
    for ( std::vector<reco::PUSubMETCandInfo>::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;
}

void MvaMEtUtilities::computeAllSums	(	const std::vector< reco::PUSubMETCandInfo > &	jets,
		const std::vector< reco::PUSubMETCandInfo > &	leptons,
		const std::vector< reco::PUSubMETCandInfo > &	pfCandidates
	)

Definition at line 244 of file MvaMEtUtilities.cc.

References cleanedJets_, cleanJets(), computeCandSum(), dzCut_, kJets, kLeptons, kPFCands, leptonsChSum_, leptonsSum_, neutralJetHSSum_, neutralJetPUSum_, pfCandChHSSum_, pfCandChPUSum_, pfCandSum_, and ptThreshold_.

Referenced by PFMETAlgorithmMVA::setInput().

                                                                         {

  cleanedJets_ = cleanJets(jets, leptons, ptThreshold_, 0.5);

  leptonsSum_ = computeCandSum( kLeptons, 0., 0, false , false, leptons );
  leptonsChSum_ = computeCandSum( kLeptons, 0., 0, true , false, leptons);
  pfCandSum_ = computeCandSum( kPFCands, dzCut_, 2, false , false, pfCandidates);
  pfCandChHSSum_ = computeCandSum( kPFCands, dzCut_, 0, false , false, pfCandidates);
  pfCandChPUSum_ = computeCandSum( kPFCands, dzCut_, 1, false , false, pfCandidates);
  neutralJetHSSum_ = computeCandSum( kJets, 0., 0, false , true, jets );
  neutralJetPUSum_ = computeCandSum( kJets, 0., 0, false , false, jets );

}

CommonMETData MvaMEtUtilities::computeCandSum ( int  compKey, double  dZmax, int  dZflag, bool  iCharged, bool  mvaPassFlag, const std::vector< reco::PUSubMETCandInfo > &  objects  )

private

Definition at line 142 of file MvaMEtUtilities.cc.

References finalize(), kJets, kLeptons, kPFCands, CommonMETData::mex, CommonMETData::mey, dbtoconf::object, passesMVA(), and CommonMETData::sumet.

Referenced by computeAllSums().

                                                                  {

  CommonMETData retVal;
  retVal.mex   = 0.;
  retVal.mey   = 0.;
  retVal.sumet = 0.;

  for ( std::vector<reco::PUSubMETCandInfo>::const_iterator object = objects.begin();
    object != objects.end(); ++object ) {

    double pFrac = 1;

    //pf candidates
    // 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
    if( compKey==MvaMEtUtilities::kPFCands ) {
      if ( object->dZ() < 0.    && dZflag != 2 ) continue;
      if ( object->dZ() > dZmax && dZflag == 0 ) continue;
      if ( object->dZ() < dZmax && dZflag == 1 ) continue;
    }

    //leptons
    if( compKey==MvaMEtUtilities::kLeptons) {
      if(iCharged) pFrac = object->chargedEnFrac();
    }

    //jets
    if( compKey==MvaMEtUtilities::kJets) {
      bool passesMVAjetId = passesMVA(object->p4(), object->mva() );
      
      if (  passesMVAjetId && !mvaPassFlag ) continue;
      if ( !passesMVAjetId &&  mvaPassFlag ) continue;
   
      pFrac = 1-object->chargedEnFrac();//neutral energy fraction
    }

    retVal.mex   += object->p4().px()*pFrac;
    retVal.mey   += object->p4().py()*pFrac;
    retVal.sumet += object->p4().pt()*pFrac;
  }

  finalize(retVal);
  return retVal;
}

CommonMETData MvaMEtUtilities::computeRecoil

(

int

metType

)

Definition at line 196 of file MvaMEtUtilities.cc.

References finalize(), kChHS, kHS, kHSMinusNeutralPU, kPF, kPU, leptonsChSum_, leptonsSum_, CommonMETData::mex, CommonMETData::mey, neutralJetHSSum_, neutralJetPUSum_, pfCandChHSSum_, pfCandChPUSum_, pfCandSum_, and CommonMETData::sumet.

Referenced by PFMETAlgorithmMVA::setInput().

                                          {

  CommonMETData retVal;

  if(metType == MvaMEtUtilities::kPF ) {
    //MET = pfMET = - all candidates 
    // MET (1) in JME-13-003
    retVal.mex  = leptonsSum_.mex - pfCandSum_.mex;
    retVal.mey  = leptonsSum_.mey - pfCandSum_.mey;
    retVal.sumet = pfCandSum_.sumet - leptonsSum_.sumet;
  }
  if(metType == MvaMEtUtilities::kChHS ) {
    //MET = - charged HS
    // MET (2) in JME-13-003
    retVal.mex  = leptonsChSum_.mex - pfCandChHSSum_.mex;
    retVal.mey  = leptonsChSum_.mey - pfCandChHSSum_.mey;
    retVal.sumet = pfCandChHSSum_.sumet - leptonsChSum_.sumet;
  }
  if(metType == MvaMEtUtilities::kHS ) { 
    //MET = - charged HS - neutral HS in jets
    // MET (3) in JME-13-003
    retVal.mex  = leptonsChSum_.mex - (pfCandChHSSum_.mex + neutralJetHSSum_.mex);
    retVal.mey  = leptonsChSum_.mey - (pfCandChHSSum_.mey + neutralJetHSSum_.mey);
    retVal.sumet = pfCandChHSSum_.sumet + neutralJetHSSum_.sumet - leptonsChSum_.sumet;
  }
  if(metType == MvaMEtUtilities::kPU ) {
    //MET = - charged PU - neutral PU in jets  
    //MET = -recoil in that particular case, - sign not useful for the MVA and then discarded
    //motivated as PU IS its own recoil
    // MET (4) in JME-13-003
    retVal.mex   = -(pfCandChPUSum_.mex + neutralJetPUSum_.mex);
    retVal.mey   = -(pfCandChPUSum_.mey + neutralJetPUSum_.mey);
    retVal.sumet = pfCandChPUSum_.sumet + neutralJetPUSum_.sumet;
  }
  if(metType == MvaMEtUtilities::kHSMinusNeutralPU ) {
    //MET = all candidates - charged PU - neutral PU in jets
    // = all charged HS + all neutrals - neutral PU in jets
    // MET (5) in JME-13-003
    retVal.mex  = leptonsSum_.mex - (pfCandSum_.mex - pfCandChPUSum_.mex - neutralJetPUSum_.mex);
    retVal.mey  = leptonsSum_.mey - (pfCandSum_.mey - pfCandChPUSum_.mey - neutralJetPUSum_.mey);
    retVal.sumet = (pfCandSum_.sumet - pfCandChPUSum_.sumet - neutralJetPUSum_.sumet) -leptonsSum_.sumet;
  }

  finalize(retVal);
  return retVal;
}

void MvaMEtUtilities::finalize ( CommonMETData &  metData )

private

Definition at line 134 of file MvaMEtUtilities.cc.

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

Referenced by computeCandSum(), and computeRecoil().

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

const std::vector< reco::PUSubMETCandInfo > & MvaMEtUtilities::getCleanedJets

(

)

const

Definition at line 282 of file MvaMEtUtilities.cc.

References cleanedJets_.

Referenced by PFMETAlgorithmMVA::setInput().

                                      {
  return cleanedJets_;
}

double MvaMEtUtilities::getLeptonsChSumMEX

(

)

const

Definition at line 271 of file MvaMEtUtilities.cc.

References leptonsChSum_, and CommonMETData::mex.

Referenced by PFMETAlgorithmMVA::setInput().

                                          {
  return leptonsChSum_.mex;
}

double MvaMEtUtilities::getLeptonsChSumMEY

(

)

const

Definition at line 276 of file MvaMEtUtilities.cc.

References leptonsChSum_, and CommonMETData::mey.

Referenced by PFMETAlgorithmMVA::setInput().

                                          {
  return leptonsChSum_.mey;
} 

double MvaMEtUtilities::getLeptonsSumMEX

(

)

const

Definition at line 261 of file MvaMEtUtilities.cc.

References leptonsSum_, and CommonMETData::mex.

Referenced by PFMETAlgorithmMVA::setInput().

                                        {
  return leptonsSum_.mex;
}

double MvaMEtUtilities::getLeptonsSumMEY

(

)

const

Definition at line 266 of file MvaMEtUtilities.cc.

References leptonsSum_, and CommonMETData::mey.

Referenced by PFMETAlgorithmMVA::setInput().

                                        {
  return leptonsSum_.mey;
}

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

protected

Definition at line 78 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<reco::PUSubMETCandInfo> jets_sorted = jets;
    std::sort(jets_sorted.rbegin(), jets_sorted.rend());
    retVal = jets_sorted[idx].p4();
  }
  return retVal;
}

reco::Candidate::LorentzVector MvaMEtUtilities::leadJetP4

(

const std::vector< reco::PUSubMETCandInfo > &

jets

)

Definition at line 68 of file MvaMEtUtilities.cc.

References jetP4().

Referenced by PFMETAlgorithmMVA::setInput().

{
  return jetP4(jets, 0);
}

unsigned MvaMEtUtilities::numJetsAboveThreshold	(	const std::vector< reco::PUSubMETCandInfo > &	jets,
		double	ptThreshold
	)

Definition at line 88 of file MvaMEtUtilities.cc.

References metsig::jet.

Referenced by PFMETAlgorithmMVA::setInput().

{
  unsigned retVal = 0;
  for ( std::vector<reco::PUSubMETCandInfo>::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  )

private

Definition at line 53 of file MvaMEtUtilities.cc.

References funct::abs(), and mvaCut_.

Referenced by computeCandSum().

{ 
  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< reco::PUSubMETCandInfo > &

jets

)

Definition at line 73 of file MvaMEtUtilities.cc.

References jetP4().

Referenced by PFMETAlgorithmMVA::setInput().

{
  return jetP4(jets, 1);
}

Member Data Documentation

std::vector<reco::PUSubMETCandInfo> MvaMEtUtilities::cleanedJets_

private

Definition at line 30 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and getCleanedJets().

double MvaMEtUtilities::dzCut_

private

Definition at line 32 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and MvaMEtUtilities().

CommonMETData MvaMEtUtilities::leptonsChSum_

private

Definition at line 23 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), computeRecoil(), getLeptonsChSumMEX(), and getLeptonsChSumMEY().

CommonMETData MvaMEtUtilities::leptonsSum_

private

Definition at line 22 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), computeRecoil(), getLeptonsSumMEX(), and getLeptonsSumMEY().

double MvaMEtUtilities::mvaCut_[3][4][4]

protected

Definition at line 66 of file MvaMEtUtilities.h.

Referenced by MvaMEtUtilities(), and passesMVA().

CommonMETData MvaMEtUtilities::neutralJetHSSum_

private

Definition at line 27 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and computeRecoil().

CommonMETData MvaMEtUtilities::neutralJetPUSum_

private

Definition at line 28 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and computeRecoil().

CommonMETData MvaMEtUtilities::pfCandChHSSum_

private

Definition at line 25 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and computeRecoil().

CommonMETData MvaMEtUtilities::pfCandChPUSum_

private

Definition at line 26 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and computeRecoil().

CommonMETData MvaMEtUtilities::pfCandSum_

private

Definition at line 24 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and computeRecoil().

double MvaMEtUtilities::ptThreshold_

private

Definition at line 33 of file MvaMEtUtilities.h.

Referenced by computeAllSums(), and MvaMEtUtilities().