MvaMEtUtilities.h

Go to the documentation of this file.

#ifndef RecoMET_METPUSubtraction_mvaMEtUtilities_h
#define RecoMET_METPUSubtraction_mvaMEtUtilities_h
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/METReco/interface/CommonMETData.h"
#include "DataFormats/METReco/interface/PUSubMETData.h"
 
#include <vector>
#include <utility>
 
class MvaMEtUtilities {
public:
  enum { kPFCands = 0, kLeptons, kJets };
  enum { kPF = 0, kChHS, kHS, kPU, kHSMinusNeutralPU };
 
private:
  CommonMETData leptonsSum_;
  CommonMETData leptonsChSum_;
  CommonMETData pfCandSum_;
  CommonMETData pfCandChHSSum_;
  CommonMETData pfCandChPUSum_;
  CommonMETData neutralJetHSSum_;
  CommonMETData neutralJetPUSum_;
 
  std::vector<reco::PUSubMETCandInfo> cleanedJets_;
 
  double dzCut_;
  double ptThreshold_;
 
public:
  MvaMEtUtilities(const edm::ParameterSet& cfg);
  virtual ~MvaMEtUtilities();
 
  reco::Candidate::LorentzVector leadJetP4(const std::vector<reco::PUSubMETCandInfo>&);
  reco::Candidate::LorentzVector subleadJetP4(const std::vector<reco::PUSubMETCandInfo>&);
  unsigned numJetsAboveThreshold(const std::vector<reco::PUSubMETCandInfo>&, double);
 
  const std::vector<reco::PUSubMETCandInfo>& getCleanedJets() const;
 
  //access functions for lepton suns ============
  double getLeptonsSumMEX() const;
  double getLeptonsSumMEY() const;
 
  double getLeptonsChSumMEX() const;
  double getLeptonsChSumMEY() const;
 
  //recoil and sum computing 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);
 
protected:
  reco::Candidate::LorentzVector jetP4(const std::vector<reco::PUSubMETCandInfo>&, unsigned);
 
  // cuts on jet Id. MVA output in bins of jet Pt and eta
  double mvaCut_[3][4][4];
 
private:
  //utilities functions for jets ===============
  bool passesMVA(const reco::Candidate::LorentzVector&, double);
 
  std::vector<reco::PUSubMETCandInfo> cleanJets(const std::vector<reco::PUSubMETCandInfo>&,
                                                const std::vector<reco::PUSubMETCandInfo>&,
                                                double,
                                                double);
 
  //utilities functions for pf candidate ======
  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);
};
 
#endif