MatcherByPullsAlgorithm.h

Go to the documentation of this file.

#ifndef MuonAnalysis_MuonAssociators_src_MatcherByPullsAlgorithm_h
#define MuonAnalysis_MuonAssociators_src_MatcherByPullsAlgorithm_h
// -*- C++ -*-
//
// Package:    MuonAnalysis/MuonAssociators
// Class:      MatcherByPullsAlgorithm
//
//
// Original Author:  Giovanni Petrucciani (SNS Pisa and CERN PH-CMG)
//         Created:  Sun Nov 16 16:14:09 CET 2008
//
//

#include <string>
#include <algorithm>
#include <vector>
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/CLHEP/interface/AlgebraicObjects.h"

//
// class decleration

class MatcherByPullsAlgorithm {
public:
  explicit MatcherByPullsAlgorithm(const edm::ParameterSet &);
  ~MatcherByPullsAlgorithm();

  std::pair<bool, float> match(const reco::Track &tk,
                               const reco::Candidate &mc,
                               const AlgebraicSymMatrix55 &invertedCovariance) const;

  std::pair<int, float> match(const reco::RecoCandidate &src,
                              const std::vector<reco::GenParticle> &cands,
                              const std::vector<uint8_t> &good) const;

  void matchMany(const reco::RecoCandidate &src,
                 const std::vector<reco::GenParticle> &cands,
                 const std::vector<uint8_t> &good,
                 std::vector<std::pair<double, int> > &matchesToFill) const;

  std::pair<int, float> match(const reco::Track &src,
                              const std::vector<reco::GenParticle> &cands,
                              const std::vector<uint8_t> &good) const;

  void matchMany(const reco::Track &src,
                 const std::vector<reco::GenParticle> &cands,
                 const std::vector<uint8_t> &good,
                 std::vector<std::pair<double, int> > &matchesToFill) const;

  void fillInvCov(const reco::Track &tk, AlgebraicSymMatrix55 &invCov) const;

private:
  const reco::Track *track(const reco::RecoCandidate &src) const;

  enum TrackChoice { StaTrack, TrkTrack, GlbTrack };

  TrackChoice track_;

  double dr2_;

  double cut_;

  bool diagOnly_;

  bool useVertex_;
};

#endif