MatchJet Class Reference

#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 22 of file MatchJet.h.

{}

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 28 of file MatchJet.cc.

References edm::RefToBaseVector< T >::begin(), EnergyCorrector::c, edm::RefToBaseVector< T >::clear(), edm::RefToBaseVector< T >::end(), reco::LeafCandidate::energy(), metsig::jet, MultipleCompare::Match(), patRefSel_triggerMatching_cfi::matches, PFTauMatching_cfi::matching, 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(), funct::sqr(), threshold, and x.

Referenced by BTagPerformanceAnalyzerMC::getJetWithFlavour(), and setThreshold().

{
        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);
    }

    this->refJets = refJets;
    refToRec.clear();
    refToRec.resize(refJets.size(), -1);

    this->recJets = recJets;
    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 97 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().

Referenced by setThreshold().

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

void MatchJet::setThreshold ( const double &  energy )

inline

Definition at line 25 of file MatchJet.h.

References mitigatedMETSequence_cff::corrector, matchCollections(), operator()(), recJets, refJets, and threshold.

Referenced by BTagPerformanceAnalyzerMC::BTagPerformanceAnalyzerMC().

{ threshold = energy; }

Member Data Documentation

double MatchJet::maxChi2

private

Definition at line 45 of file MatchJet.h.

Referenced by matchCollections().

CorrectJet MatchJet::recJetCorrector

private

Definition at line 43 of file MatchJet.h.

Referenced by matchCollections().

edm::RefToBaseVector<reco::Jet> MatchJet::recJets

private

Definition at line 40 of file MatchJet.h.

Referenced by matchCollections(), operator()(), and setThreshold().

std::vector<int> MatchJet::recToRef

private

Definition at line 38 of file MatchJet.h.

Referenced by matchCollections(), and operator()().

CorrectJet MatchJet::refJetCorrector

private

Definition at line 42 of file MatchJet.h.

Referenced by matchCollections().

edm::RefToBaseVector<reco::Jet> MatchJet::refJets

private

Definition at line 39 of file MatchJet.h.

Referenced by matchCollections(), operator()(), and setThreshold().

std::vector<int> MatchJet::refToRec

private

Definition at line 38 of file MatchJet.h.

Referenced by matchCollections().

double MatchJet::sigmaDeltaE2

private

Definition at line 47 of file MatchJet.h.

Referenced by matchCollections().

double MatchJet::sigmaDeltaR2

private

Definition at line 46 of file MatchJet.h.

Referenced by matchCollections().

double MatchJet::threshold

private

Definition at line 48 of file MatchJet.h.

Referenced by utils.StatisticalTest::get_status(), matchCollections(), and setThreshold().