#include <MatchJet.h>

Public Member Functions
void	matchCollections (const edm::RefToBaseVector< reco::Jet > &refJets, const edm::RefToBaseVector< reco::Jet > &recJets, const reco::JetCorrector *corrector)
	match the collections More...

	MatchJet ()

	MatchJet (const edm::ParameterSet &pSet)

edm::RefToBase< reco::Jet >	operator() (const edm::RefToBase< reco::Jet > &recJet) const
	Returns the matched "reference" jet. More...

void	setThreshold (const double &energy)

Private Attributes
double	maxChi2

CorrectJet	recJetCorrector

edm::RefToBaseVector< reco::Jet >	recJets

std::vector< int >	recToRef

CorrectJet	refJetCorrector

edm::RefToBaseVector< reco::Jet >	refJets

std::vector< int >	refToRec

double	sigmaDeltaE2

double	sigmaDeltaR2

double	threshold

Detailed Description

Match jets

Definition at line 19 of file MatchJet.h.

Constructor & Destructor Documentation

MatchJet::MatchJet ( )

inline

Definition at line 21 of file MatchJet.h.

21 {}

MatchJet::MatchJet ( const edm::ParameterSet & pSet )

Definition at line 20 of file MatchJet.cc.

     : maxChi2(pSet.getParameter<double>("maxChi2")),
       sigmaDeltaR2(sqr(pSet.getParameter<double>("sigmaDeltaR"))),
       sigmaDeltaE2(sqr(pSet.getParameter<double>("sigmaDeltaE"))),
       threshold(1.0) {}

Member Function Documentation

void MatchJet::matchCollections	(	const edm::RefToBaseVector< reco::Jet > &	refJets,
		const edm::RefToBaseVector< reco::Jet > &	recJets,
		const reco::JetCorrector *	corrector
	)

match the collections

Definition at line 26 of file MatchJet.cc.

References edm::RefToBaseVector< T >::begin(), c, edm::RefToBaseVector< T >::clear(), edm::RefToBaseVector< T >::end(), reco::LeafCandidate::energy(), metsig::jet, MultipleCompare::Match(), maxChi2, edm::RefToBaseVector< T >::push_back(), reco::LeafCandidate::px(), reco::LeafCandidate::py(), reco::LeafCandidate::pz(), recJetCorrector, recJets, recToRef, refJetCorrector, refJets, refToRec, CorrectJet::setCorrector(), sigmaDeltaE2, sigmaDeltaR2, edm::RefToBaseVector< T >::size(), sqr(), threshold, and x.

Referenced by BTagPerformanceAnalyzerMC::getJetWithFlavour().

                                                                    {
   refJetCorrector.setCorrector(corrector);
   recJetCorrector.setCorrector(corrector);
 
   typedef ROOT::Math::Cartesian3D<double> Vector;
 
   std::vector<Vector> corrRefJets;
   refJets.clear();
   for (edm::RefToBaseVector<reco::Jet>::const_iterator iter = refJets.begin(); iter != refJets_.end(); ++iter) {
     edm::RefToBase<reco::Jet> jetRef = *iter;
     reco::Jet jet = refJetCorrector(*jetRef);
     if (jet.energy() < threshold)
       continue;
 
     corrRefJets.push_back(Vector(jet.px(), jet.py(), jet.pz()));
     refJets.push_back(jetRef);
   }
 
   std::vector<Vector> corrRecJets;
   recJets.clear();
   for (edm::RefToBaseVector<reco::Jet>::const_iterator iter = recJets.begin(); iter != recJets_.end(); ++iter) {
     edm::RefToBase<reco::Jet> jetRec = *iter;
     reco::Jet jet = recJetCorrector(*jetRec);
     if (jet.energy() < threshold)
       continue;
 
     corrRecJets.push_back(Vector(jet.px(), jet.py(), jet.pz()));
     recJets.push_back(jetRec);
   }
 
   refToRec.clear();
   refToRec.resize(refJets.size(), -1);
 
   recToRef.clear();
   recToRef.resize(recJets.size(), -1);
 
   Matching<double> matching(corrRefJets, corrRecJets, [this](auto& v1, auto& v2) {
     double x = ROOT::Math::VectorUtil::DeltaR2(v1, v2) / this->sigmaDeltaR2 +
                sqr(2. * (v1.R() - v2.R()) / (v1.R() + v2.R())) / this->sigmaDeltaE2;
     // Possible debug output
     // std::cout << "xxx " << ROOT::Math::VectorUtil::DeltaPhi(v1, v2) << " " << (v1.Eta() - v2.Eta())
     // << " " << (v1.R() - v2.R()) / (v1.R() + v2.R()) << " " << x << std::endl;
     return x;
   });
 
   typedef Matching<double>::Match Match;
 
   const std::vector<Match>& matches = matching.match(std::less<double>(), [&](auto& c) { return c < this->maxChi2; });
   for (std::vector<Match>::const_iterator iter = matches.begin(); iter != matches.end(); ++iter) {
     refToRec[iter->index1] = iter->index2;
     recToRef[iter->index2] = iter->index1;
   }
 }

edm::RefToBase< reco::Jet > MatchJet::operator() ( const edm::RefToBase< reco::Jet > & recJet ) const

Returns the matched "reference" jet.

Definition at line 82 of file MatchJet.cc.

References mps_fire::i, edm::RefToBaseVector< T >::id(), edm::RefToBase< T >::id(), match(), recJets, recToRef, refJets, mps_fire::result, and edm::RefToBaseVector< T >::size().

                                                                                     {
   edm::RefToBase<reco::Jet> result;
   if (recJet.id() != recJets.id())
     return result;
 
   for (unsigned int i = 0; i != recJets.size(); ++i) {
     if (recJets[i] == recJet) {
       int match = recToRef[i];
       if (match >= 0)
         result = refJets[match];
       break;
     }
   }
 
   return result;
 }