CMS 3D CMS Logo

PFMEtSignInterfaceBase.h
Go to the documentation of this file.
1 #ifndef RecoMET_METPUSubtraction_PFMEtSignInterfaceBase_h
2 #define RecoMET_METPUSubtraction_PFMEtSignInterfaceBase_h
3 
17 
19 
35 
36 
37 #include <TFile.h>
38 #include <TH2.h>
39 
40 
42 {
43  public:
44 
47 
48  template <typename T>
49  metsig::SigInputObj compResolution(const T* particle) const
50  {
51  double pt = particle->pt();
52  double eta = particle->eta();
53  double phi = particle->phi();
54 
55  if ( dynamic_cast<const reco::GsfElectron*>(particle) != nullptr ||
56  dynamic_cast<const pat::Electron*>(particle) != nullptr ) {
57  std::string particleType = "electron";
58  // WARNING: SignAlgoResolutions::PFtype2 needs to be kept in sync with reco::PFCandidate::e !!
59  double dpt = pfMEtResolution_->eval(metsig::PFtype2, metsig::ET, pt, phi, eta);
60  double dphi = pfMEtResolution_->eval(metsig::PFtype2, metsig::PHI, pt, phi, eta);
61  //std::cout << "electron: pt = " << pt << ", eta = " << eta << ", phi = " << phi
62  // << " --> dpt = " << dpt << ", dphi = " << dphi << std::endl;
63  return metsig::SigInputObj(particleType, pt, phi, dpt, dphi);
64  } else if ( dynamic_cast<const reco::Photon*>(particle) != nullptr ||
65  dynamic_cast<const pat::Photon*>(particle) != nullptr ) {
66  // CV: assume resolutions for photons to be identical to electron resolutions
67  std::string particleType = "electron";
68  // WARNING: SignAlgoResolutions::PFtype2 needs to be kept in sync with reco::PFCandidate::e !!
69  double dpt = pfMEtResolution_->eval(metsig::PFtype2, metsig::ET, pt, phi, eta);
70  double dphi = pfMEtResolution_->eval(metsig::PFtype2, metsig::PHI, pt, phi, eta);
71  //std::cout << "photon: pt = " << pt << ", eta = " << eta << ", phi = " << phi
72  // << " --> dpt = " << dpt << ", dphi = " << dphi << std::endl;
73  return metsig::SigInputObj(particleType, pt, phi, dpt, dphi);
74  } else if ( dynamic_cast<const reco::Muon*>(particle) != nullptr ||
75  dynamic_cast<const pat::Muon*>(particle) != nullptr ) {
76  std::string particleType = "muon";
77  double dpt, dphi;
78  const reco::Track* muonTrack = nullptr;
79  if ( dynamic_cast<const pat::Muon*>(particle) != nullptr ) {
80  const pat::Muon* muon = dynamic_cast<const pat::Muon*>(particle);
81  if ( muon->track().isNonnull() && muon->track().isAvailable() ) muonTrack = muon->track().get();
82  } else if ( dynamic_cast<const reco::Muon*>(particle) != nullptr ) {
83  const reco::Muon* muon = dynamic_cast<const reco::Muon*>(particle);
84  if ( muon->track().isNonnull() && muon->track().isAvailable() ) muonTrack = muon->track().get();
85  } else assert(0);
86  if ( muonTrack ) {
87  dpt = muonTrack->ptError();
88  dphi = pt*muonTrack->phiError(); // CV: pt*dphi is indeed correct
89  } else {
90  // WARNING: SignAlgoResolutions::PFtype3 needs to be kept in sync with reco::PFCandidate::mu !!
91  dpt = pfMEtResolution_->eval(metsig::PFtype3, metsig::ET, pt, phi, eta);
92  dphi = pfMEtResolution_->eval(metsig::PFtype3, metsig::PHI, pt, phi, eta);
93  }
94  //std::cout << "muon: pt = " << pt << ", eta = " << eta << ", phi = " << phi
95  // << " --> dpt = " << dpt << ", dphi = " << dphi << std::endl;
96  return metsig::SigInputObj(particleType, pt, phi, dpt, dphi);
97  } else if ( dynamic_cast<const reco::PFTau*>(particle) != nullptr ||
98  dynamic_cast<const pat::Tau*>(particle) != nullptr ) {
99  // CV: use PFJet resolutions for PFTaus for now...
100  // (until PFTau specific resolutions are available)
101  if ( dynamic_cast<const pat::Tau*>(particle) != nullptr ) {
102  const pat::Tau* pfTau = dynamic_cast<const pat::Tau*>(particle);
103  //std::cout << "tau: pt = " << pt << ", eta = " << eta << ", phi = " << phi << std::endl;
104  return pfMEtResolution_->evalPFJet(pfTau->pfJetRef().get());
105  } else if ( dynamic_cast<const reco::PFTau*>(particle) != nullptr ) {
106  const reco::PFTau* pfTau = dynamic_cast<const reco::PFTau*>(particle);
107  //std::cout << "tau: pt = " << pt << ", eta = " << eta << ", phi = " << phi << std::endl;
108  return pfMEtResolution_->evalPFJet(pfTau->jetRef().get());
109  } else assert(0);
110  } else if ( dynamic_cast<const reco::PFJet*>(particle) != nullptr ||
111  dynamic_cast<const pat::Jet*>(particle) != nullptr ) {
112  metsig::SigInputObj pfJetResolution;
113  if ( dynamic_cast<const reco::PFJet*>(particle) != nullptr ) {
114  const reco::PFJet* pfJet = dynamic_cast<const reco::PFJet*>(particle);
115  pfJetResolution = pfMEtResolution_->evalPFJet(pfJet);
116  } else if ( dynamic_cast<const pat::Jet*>(particle) != nullptr ) {
117  const pat::Jet* jet = dynamic_cast<const pat::Jet*>(particle);
118  if ( jet->isPFJet() ) {
119  reco::PFJet pfJet(jet->p4(), jet->vertex(), jet->pfSpecific(), jet->getJetConstituents());
120  pfJetResolution = pfMEtResolution_->evalPFJet(&pfJet);
121  } else throw cms::Exception("addPFMEtSignObjects")
122  << "PAT jet not of PF-type !!\n";
123  } else assert(0);
124  //std::cout << "pfJet: pt = " << pt << ", eta = " << eta << ", phi = " << phi << std::endl;
125  // CV: apply additional jet energy resolution corrections
126  // not included in (PF)MEt significance algorithm yet
127  // (cf. CMS AN-11/400 vs. CMS AN-11/330)
128  if ( lut_ && pt > 10. ) {
129  double x = std::abs(eta);
130  double y = pt;
131  if ( x > lut_->GetXaxis()->GetXmin() && x < lut_->GetXaxis()->GetXmax() &&
132  y > lut_->GetYaxis()->GetXmin() && y < lut_->GetYaxis()->GetXmax() ) {
133  int binIndex = lut_->FindBin(x, y);
134  double addJERcorrFactor = lut_->GetBinContent(binIndex);
135  //std::cout << " addJERcorrFactor = " << addJERcorrFactor << std::endl;
136  pfJetResolution.set(pfJetResolution.get_type(),
137  pfJetResolution.get_energy(),
138  pfJetResolution.get_phi(),
139  addJERcorrFactor*pfJetResolution.get_sigma_e(),
140  pfJetResolution.get_sigma_tan());
141  }
142  }
143  return pfJetResolution;
144  } else if ( dynamic_cast<const reco::PFCandidate*>(particle) != nullptr ) {
145  const reco::PFCandidate* pfCandidate = dynamic_cast<const reco::PFCandidate*>(particle);
146  //std::cout << "pfCandidate: pt = " << pt << ", eta = " << eta << ", phi = " << phi << std::endl;
147  return pfMEtResolution_->evalPF(pfCandidate);
148  } else throw cms::Exception("addPFMEtSignObjects")
149  << "Invalid type of particle:"
150  << " valid types = { reco::GsfElectron/pat::Electron, reco::Photon/pat::Photon, reco::Muon/pat::Muon, reco::PFTau/pat::Tau,"
151  << " reco::PFJet/pat::Jet, reco::PFCandidate } !!\n";
152  }
153 
154  template <typename T>
155  std::vector<metsig::SigInputObj> compResolution(const std::list<T*>& particles) const
156  {
157  LogDebug("compResolution")
158  << " particles: entries = " << particles.size() << std::endl;
159 
160  std::vector<metsig::SigInputObj> pfMEtSignObjects;
161  addPFMEtSignObjects(pfMEtSignObjects, particles);
162 
163  return pfMEtSignObjects;
164  }
165 
166  reco::METCovMatrix operator()(const std::vector<metsig::SigInputObj>&) const;
167 
168  template <typename T>
169  reco::METCovMatrix operator()(const std::list<T*>& particles) const
170  {
171  std::vector<metsig::SigInputObj> pfMEtSignObjects = compResolution(particles);
172  return this->operator()(pfMEtSignObjects);
173  }
174 
175  protected:
176 
177  template<typename T>
178  void addPFMEtSignObjects(std::vector<metsig::SigInputObj>& metSignObjects,
179  const std::list<T*>& particles) const
180  {
181  for ( typename std::list<T*>::const_iterator particle = particles.begin();
182  particle != particles.end(); ++particle ) {
183  metSignObjects.push_back(this->compResolution(*particle));
184  }
185  }
186 
187  private:
188 
190 
191  // CV: look-up table for additional jet energy resolution corrections
192  // not included in (PF)MEt significance algorithm yet
193  // (cf. CMS AN-11/400 vs. CMS AN-11/330)
194  TFile* inputFile_;
195  TH2* lut_;
196 
198 };
199 
200 #endif
#define LogDebug(id)
bool isAvailable() const
Definition: Ref.h:577
double get_sigma_tan() const
Definition: SigInputObj.h:46
metsig::SignAlgoResolutions * pfMEtResolution_
TrackRef track() const override
reference to a Track
Definition: Muon.h:49
bool isNonnull() const
Checks for non-null.
Definition: Ref.h:253
const PFJetRef & jetRef() const
Definition: PFTau.cc:58
PFMEtSignInterfaceBase(const edm::ParameterSet &)
double get_phi() const
Definition: SigInputObj.h:44
const PFSpecific & pfSpecific() const
retrieve the pf specific part of the jet
Definition: Jet.h:271
metsig::SigInputObj evalPF(const reco::PFCandidate *candidate) const
ROOT::Math::SMatrix< double, 2 > METCovMatrix
Definition: MET.h:40
void addPFMEtSignObjects(std::vector< metsig::SigInputObj > &metSignObjects, const std::list< T * > &particles) const
virtual Constituents getJetConstituents() const
list of constituents
reco::TrackRef track() const override
reference to Track reconstructed in the tracker only (reimplemented from reco::Muon) ...
#define nullptr
double eval(const resolutionType &type, const resolutionFunc &func, const double &et, const double &phi, const double &eta, const double &p) const
Jets made from PFObjects.
Definition: PFJet.h:21
void set(const std::string &m_type, const double &m_energy, const double &m_phi, const double &m_sigma_e, const double &m_sigma_tan)
Definition: SigInputObj.h:48
std::vector< metsig::SigInputObj > compResolution(const std::list< T * > &particles) const
const Point & vertex() const override
vertex position (overwritten by PF...)
double ptError() const
error on Pt (set to 1000 TeV if charge==0 for safety)
Definition: TrackBase.h:808
double phiError() const
error on phi
Definition: TrackBase.h:835
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
const LorentzVector & p4() const final
four-momentum Lorentz vector
Definition: LeafCandidate.h:99
T const * get() const
Returns C++ pointer to the item.
Definition: Ref.h:245
Analysis-level tau class.
Definition: Tau.h:55
bool isPFJet() const
check to see if the jet is a reco::PFJet
Definition: Jet.h:255
const reco::PFJetRef & pfJetRef() const
Definition: Tau.h:163
double get_energy() const
Definition: SigInputObj.h:43
Analysis-level calorimeter jet class.
Definition: Jet.h:80
reco::METCovMatrix operator()(const std::list< T * > &particles) const
metsig::SigInputObj evalPFJet(const reco::PFJet *jet) const
Particle reconstructed by the particle flow algorithm.
Definition: PFCandidate.h:40
reco::METCovMatrix operator()(const std::vector< metsig::SigInputObj > &) const
std::string get_type() const
Definition: SigInputObj.h:42
metsig::SigInputObj compResolution(const T *particle) const
long double T
Analysis-level muon class.
Definition: Muon.h:50
double get_sigma_e() const
Definition: SigInputObj.h:45