CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Types | Private Member Functions
GenSpecificAlgo Class Reference

#include <RecoMET/METAlgorithms/interface/GenSpecificAlgo.h>

Public Member Functions

reco::GenMET addInfo (edm::Handle< edm::View< reco::Candidate > > particles, CommonMETData *met, double globalThreshold=0, bool onlyFiducial=false, bool applyFiducialThresholdForFractions=false, bool usePt=false)
 

Private Types

typedef math::XYZTLorentzVector LorentzVector
 
typedef math::XYZPoint Point
 

Private Member Functions

void fillCommonMETData (CommonMETData *met, edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool usePt)
 
SpecificGenMETData mkSpecificGenMETData (edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool applyFiducialThresholdForFractions, bool usePt)
 

Detailed Description

Description: Adds generator level HEPMC specific information to MET

Implementation: [Notes on implementation]

Definition at line 29 of file GenSpecificAlgo.h.

Member Typedef Documentation

Definition at line 36 of file GenSpecificAlgo.h.

Definition at line 37 of file GenSpecificAlgo.h.

Member Function Documentation

reco::GenMET GenSpecificAlgo::addInfo ( edm::Handle< edm::View< reco::Candidate > >  particles,
CommonMETData met,
double  globalThreshold = 0,
bool  onlyFiducial = false,
bool  applyFiducialThresholdForFractions = false,
bool  usePt = false 
)

Definition at line 17 of file GenSpecificAlgo.cc.

References fillCommonMETData(), HepMCValidationHelper::genMet(), CommonMETData::met, CommonMETData::mex, CommonMETData::mey, CommonMETData::mez, mkSpecificGenMETData(), p4, timingPdfMaker::specific, and CommonMETData::sumet.

Referenced by cms::GenMETProducer::produce().

18 {
19  fillCommonMETData(met, particles, globalThreshold, onlyFiducial, usePt);
20 
21  SpecificGenMETData specific = mkSpecificGenMETData(particles, globalThreshold, onlyFiducial, applyFiducialThresholdForFractions, usePt);
22 
23  const LorentzVector p4( met->mex, met->mey, met->mez, met->met );
24  const Point vtx( 0.0, 0.0, 0.0 );
25 
26  reco::GenMET genMet(specific, met->sumet, p4, vtx );
27  return genMet;
28 }
dictionary specific
math::XYZTLorentzVector LorentzVector
double p4[4]
Definition: TauolaWrapper.h:92
void fillCommonMETData(CommonMETData *met, edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool usePt)
math::XYZPoint Point
TLorentzVector genMet(const HepMC::GenEvent *all, double etamin=-9999., double etamax=9999.)
SpecificGenMETData mkSpecificGenMETData(edm::Handle< edm::View< reco::Candidate > > &particles, double globalThreshold, bool onlyFiducial, bool applyFiducialThresholdForFractions, bool usePt)
void GenSpecificAlgo::fillCommonMETData ( CommonMETData met,
edm::Handle< edm::View< reco::Candidate > > &  particles,
double  globalThreshold,
bool  onlyFiducial,
bool  usePt 
)
private

Definition at line 31 of file GenSpecificAlgo.cc.

References Abs(), begin, funct::cos(), alignCSCRings::e, end, HLT_25ns14e33_v1_cff::globalThreshold, CommonMETData::met, CommonMETData::mex, CommonMETData::mey, CommonMETData::mez, phi, funct::sin(), mathSSE::sqrt(), CommonMETData::sumet, and theta().

Referenced by addInfo().

32 {
33  double sum_et = 0.0;
34  double sum_ex = 0.0;
35  double sum_ey = 0.0;
36  double sum_ez = 0.0;
37 
38  for( edm::View<reco::Candidate>::const_iterator iParticle = (particles.product())->begin(); iParticle != (particles.product())->end(); ++iParticle)
39  {
40  if( (onlyFiducial && TMath::Abs(iParticle->eta()) >= 5.0)) continue;
41 
42  if( iParticle->et() <= globalThreshold ) continue;
43 
44  if(usePt)
45  {
46  double phi = iParticle->phi();
47  double et = iParticle->pt();
48  sum_ez += iParticle->pz();
49  sum_et += et;
50  sum_ex += et*cos(phi);
51  sum_ey += et*sin(phi);
52  }
53  else
54  {
55  double phi = iParticle->phi();
56  double theta = iParticle->theta();
57  double e = iParticle->energy();
58  double et = e*sin(theta);
59  sum_ez += e*cos(theta);
60  sum_et += et;
61  sum_ex += et*cos(phi);
62  sum_ey += et*sin(phi);
63  }
64  }
65 
66  met->mex = -sum_ex;
67  met->mey = -sum_ey;
68  met->mez = -sum_ez;
69  met->met = sqrt( sum_ex*sum_ex + sum_ey*sum_ey );
70  met->sumet = sum_et;
71 }
Sin< T >::type sin(const T &t)
Definition: Sin.h:22
Geom::Theta< T > theta() const
T sqrt(T t)
Definition: SSEVec.h:48
Cos< T >::type cos(const T &t)
Definition: Cos.h:22
T Abs(T a)
Definition: MathUtil.h:49
#define end
Definition: vmac.h:37
T const * product() const
Definition: Handle.h:81
#define begin
Definition: vmac.h:30
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:85
SpecificGenMETData GenSpecificAlgo::mkSpecificGenMETData ( edm::Handle< edm::View< reco::Candidate > > &  particles,
double  globalThreshold,
bool  onlyFiducial,
bool  applyFiducialThresholdForFractions,
bool  usePt 
)
private

Definition at line 74 of file GenSpecificAlgo.cc.

References Abs(), begin, SpecificGenMETData::ChargedEMEtFraction, SpecificGenMETData::ChargedHadEtFraction, end, HLT_25ns14e33_v1_cff::globalThreshold, SpecificGenMETData::InvisibleEtFraction, SpecificGenMETData::MuonEtFraction, SpecificGenMETData::NeutralEMEtFraction, SpecificGenMETData::NeutralHadEtFraction, benchmark_cfg::pdgId, EnergyCorrector::pt, timingPdfMaker::specific, and HLT_25ns14e33_v1_cff::usePt.

Referenced by addInfo().

75 {
76  const static int neutralEMpdgId[] = { 22 /* photon */ };
77  const static std::set<int> neutralEMpdgIdSet(neutralEMpdgId, neutralEMpdgId + sizeof(neutralEMpdgId)/sizeof(int));
78 
79  const static int chargedEMpdgId[] = { 11 /* e */ };
80  const static std::set<int> chargedEMpdgIdSet(chargedEMpdgId, chargedEMpdgId + sizeof(chargedEMpdgId)/sizeof(int));
81 
82  const static int muonpdgId[] = { 13 /* muon */ };
83  const static std::set<int> muonpdgIdSet(muonpdgId, muonpdgId + sizeof(muonpdgId)/sizeof(int));
84 
85  const static int neutralHADpdgId[] = {
86  130 /* K_long */,
87  310 /* K_short */,
88  3122 /* Lambda */,
89  2112 /* n */,
90  3222 /* Neutral Cascade */
91  };
92  const static std::set<int> neutralHADpdgIdSet(neutralHADpdgId, neutralHADpdgId + sizeof(neutralHADpdgId)/sizeof(int));
93 
94  const static int chargedHADpdgId[] = {
95  211 /* pi */,
96  321 /* K+/K- */,
97  2212 /* p */,
98  3312 /* Cascade - */,
99  3112 /* Sigma - */,
100  3322 /* Sigma + */,
101  3334 /* Omega - */
102  };
103  const static std::set<int> chargedHADpdgIdSet(chargedHADpdgId, chargedHADpdgId + sizeof(chargedHADpdgId)/sizeof(int));
104 
105  const static int invisiblepdgId[] = {
106  12 /* e_nu */,
107  14 /* mu_nu */,
108  16 /* tau_nu */,
109  1000022 /* Neutral ~Chi_0 */,
110  1000012 /* LH ~e_nu */,
111  1000014 /* LH ~mu_nu */,
112  1000016 /* LH ~tau_nu */,
113  2000012 /* RH ~e_nu */,
114  2000014 /* RH ~mu_nu */,
115  2000016 /* RH ~tau_nu */,
116  39 /* G */,
117  1000039 /* ~G */,
118  5100039 /* KK G */,
119  4000012 /* excited e_nu */,
120  4000014 /* excited mu_nu */,
121  4000016 /* excited tau_nu */,
122  9900012 /* Maj e_nu */,
123  9900014 /* Maj mu_nu */,
124  9900016 /* Maj tau_nu */,
125  };
126  const static std::set<int> invisiblepdgIdSet(invisiblepdgId, invisiblepdgId + sizeof(invisiblepdgId)/sizeof(int));
127 
129  double Et_unclassified = 0.0;
130 
131  for(edm::View<reco::Candidate>::const_iterator iParticle = (particles.product())->begin(); iParticle != (particles.product())->end(); ++iParticle)
132  {
133  if(applyFiducialThresholdForFractions) if( onlyFiducial && (TMath::Abs(iParticle->eta()) >= 5.0) ) continue;
134  if(applyFiducialThresholdForFractions) if( iParticle->et() <= globalThreshold ) continue;
135 
136  int pdgId = TMath::Abs( iParticle->pdgId() ) ;
137  double pt = (usePt) ? iParticle->pt() : iParticle->et();
138  if(neutralEMpdgIdSet.count(pdgId)) specific.NeutralEMEtFraction += pt;
139  else if(chargedEMpdgIdSet.count(pdgId)) specific.ChargedEMEtFraction += pt;
140  else if(muonpdgIdSet.count(pdgId)) specific.MuonEtFraction += pt;
141  else if(neutralHADpdgIdSet.count(pdgId)) specific.NeutralHadEtFraction += pt;
142  else if(chargedHADpdgIdSet.count(pdgId)) specific.ChargedHadEtFraction += pt;
143  else if(invisiblepdgIdSet.count(pdgId)) specific.InvisibleEtFraction += pt;
144  else Et_unclassified += pt;
145  }
146 
147  double Et_Total = specific.NeutralEMEtFraction + specific.NeutralHadEtFraction + specific.ChargedEMEtFraction +
148  specific.ChargedHadEtFraction + specific.MuonEtFraction + specific.InvisibleEtFraction + Et_unclassified;
149 
150  if(Et_Total)
151  {
152  specific.NeutralEMEtFraction /= Et_Total;
153  specific.NeutralHadEtFraction /= Et_Total;
154  specific.ChargedEMEtFraction /= Et_Total;
155  specific.ChargedHadEtFraction /= Et_Total;
156  specific.MuonEtFraction /= Et_Total;
157  specific.InvisibleEtFraction /= Et_Total;
158  }
159 
160  return specific;
161 }
dictionary specific
T Abs(T a)
Definition: MathUtil.h:49
#define end
Definition: vmac.h:37
T const * product() const
Definition: Handle.h:81
#define begin
Definition: vmac.h:30
boost::indirect_iterator< typename seq_t::const_iterator > const_iterator
Definition: View.h:85