CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
PATMETSlimmer.cc
Go to the documentation of this file.
1 
10 
13 
15 
16 namespace pat {
17 
19  public:
20  explicit PATMETSlimmer(const edm::ParameterSet &iConfig);
21  ~PATMETSlimmer() override {}
22 
23  void produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override;
24 
25  private:
26  class OneMETShift {
27  public:
29  : shift(pat::MET::NoShift),
30  level(pat::MET::None),
32  corShift(false),
33  uncShift(false),
34  isSmeared(false) {}
37  const edm::InputTag &baseTag,
39  bool t0FromMiniAOD_,
40  bool corShift_,
41  bool uncShift_,
42  bool isSmeared_ = false);
43  void readAndSet(const edm::Event &ev, pat::MET &met) const;
44 
45  private:
49  const bool t0FromMiniAOD;
50  const bool corShift;
51  const bool uncShift;
52  const bool isSmeared;
53  };
54  void maybeReadShifts(const edm::ParameterSet &basePSet,
55  const std::string &name,
57  bool readFromMiniAOD = false);
58 
60  std::vector<OneMETShift> shifts_;
61 
62  const bool onMiniAOD_;
63  };
64 
65 } // namespace pat
66 
68  : src_(consumes<pat::METCollection>(iConfig.getParameter<edm::InputTag>("src"))),
69  onMiniAOD_(iConfig.existsAs<bool>("runningOnMiniAOD") ? iConfig.getParameter<bool>("runningOnMiniAOD") : false) {
70  maybeReadShifts(iConfig, "rawVariation", pat::MET::None);
71  maybeReadShifts(iConfig, "t1Uncertainties", pat::MET::T1);
72  maybeReadShifts(iConfig, "t01Variation", pat::MET::T0, onMiniAOD_);
73  maybeReadShifts(iConfig, "t1SmearedVarsAndUncs", pat::MET::Smear);
74 
75  maybeReadShifts(iConfig, "tXYUncForRaw", pat::MET::TXYForRaw);
76  maybeReadShifts(iConfig, "tXYUncForT1", pat::MET::TXY);
77  maybeReadShifts(iConfig, "tXYUncForT01", pat::MET::TXYForT01);
78  maybeReadShifts(iConfig, "tXYUncForT1Smear", pat::MET::TXYForT1Smear);
79  maybeReadShifts(iConfig, "tXYUncForT01Smear", pat::MET::TXYForT01Smear);
80  maybeReadShifts(iConfig, "caloMET", pat::MET::Calo);
81  maybeReadShifts(iConfig, "chsMET", pat::MET::Chs);
82  maybeReadShifts(iConfig, "trkMET", pat::MET::Trk);
83 
84  produces<std::vector<pat::MET>>();
85 }
86 
88  const std::string &name,
90  bool readFromMiniAOD) {
91  if (basePSet.existsAs<edm::ParameterSet>(name)) {
92  throw cms::Exception("Unsupported", "Reading PSets not supported, for now just use input tag");
93  } else if (basePSet.existsAs<edm::InputTag>(name)) {
94  const edm::InputTag &baseTag = basePSet.getParameter<edm::InputTag>(name);
95 
96  if (level == pat::MET::T1) {
97  shifts_.push_back(
98  OneMETShift(pat::MET::NoShift, level, baseTag, consumesCollector(), readFromMiniAOD, true, false, false));
99  shifts_.push_back(
100  OneMETShift(pat::MET::NoShift, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
101  shifts_.push_back(
102  OneMETShift(pat::MET::JetResUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
103  shifts_.push_back(
104  OneMETShift(pat::MET::JetResDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
105  shifts_.push_back(
106  OneMETShift(pat::MET::JetEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
107  shifts_.push_back(
108  OneMETShift(pat::MET::JetEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
109  shifts_.push_back(
110  OneMETShift(pat::MET::MuonEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
111  shifts_.push_back(
112  OneMETShift(pat::MET::MuonEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
113  shifts_.push_back(
114  OneMETShift(pat::MET::ElectronEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
115  shifts_.push_back(
116  OneMETShift(pat::MET::ElectronEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
117  shifts_.push_back(
118  OneMETShift(pat::MET::PhotonEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
119  shifts_.push_back(
120  OneMETShift(pat::MET::PhotonEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
121  shifts_.push_back(
122  OneMETShift(pat::MET::TauEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
123  shifts_.push_back(
124  OneMETShift(pat::MET::TauEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
125  shifts_.push_back(
126  OneMETShift(pat::MET::UnclusteredEnUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
127  shifts_.push_back(
128  OneMETShift(pat::MET::UnclusteredEnDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true));
129  } else if (level == pat::MET::Smear) {
130  shifts_.push_back(
131  OneMETShift(pat::MET::NoShift, level, baseTag, consumesCollector(), readFromMiniAOD, true, false, true));
132  shifts_.push_back(
133  OneMETShift(pat::MET::JetResUp, level, baseTag, consumesCollector(), readFromMiniAOD, false, true, true));
134  shifts_.push_back(
135  OneMETShift(pat::MET::JetResDown, level, baseTag, consumesCollector(), readFromMiniAOD, false, true, true));
136  } else {
137  shifts_.push_back(
138  OneMETShift(pat::MET::NoShift, level, baseTag, consumesCollector(), readFromMiniAOD, true, false));
139  }
140  }
141 }
142 
145  const edm::InputTag &baseTag,
147  bool t0FromMiniAOD_,
148  bool corShift_,
149  bool uncShift_,
150  bool isSmeared)
151  : shift(shift_),
152  level(level_),
153  t0FromMiniAOD(t0FromMiniAOD_),
154  corShift(corShift_),
155  uncShift(uncShift_),
156  isSmeared(isSmeared) {
157  std::string baseTagStr = baseTag.encode();
158  char buff[1024];
159  switch (shift) {
160  case pat::MET::NoShift:
161  snprintf(buff, 1023, baseTagStr.c_str(), "");
162  break;
163  case pat::MET::JetEnUp:
164  snprintf(buff, 1023, baseTagStr.c_str(), "JetEnUp");
165  break;
166  case pat::MET::JetEnDown:
167  snprintf(buff, 1023, baseTagStr.c_str(), "JetEnDown");
168  break;
169  case pat::MET::JetResUp:
170  snprintf(buff, 1023, baseTagStr.c_str(), "JetResUp");
171  break;
173  snprintf(buff, 1023, baseTagStr.c_str(), "JetResDown");
174  break;
175  case pat::MET::MuonEnUp:
176  snprintf(buff, 1023, baseTagStr.c_str(), "MuonEnUp");
177  break;
179  snprintf(buff, 1023, baseTagStr.c_str(), "MuonEnDown");
180  break;
182  snprintf(buff, 1023, baseTagStr.c_str(), "ElectronEnUp");
183  break;
185  snprintf(buff, 1023, baseTagStr.c_str(), "ElectronEnDown");
186  break;
188  snprintf(buff, 1023, baseTagStr.c_str(), "PhotonEnUp");
189  break;
191  snprintf(buff, 1023, baseTagStr.c_str(), "PhotonEnDown");
192  break;
193  case pat::MET::TauEnUp:
194  snprintf(buff, 1023, baseTagStr.c_str(), "TauEnUp");
195  break;
196  case pat::MET::TauEnDown:
197  snprintf(buff, 1023, baseTagStr.c_str(), "TauEnDown");
198  break;
200  snprintf(buff, 1023, baseTagStr.c_str(), "UnclusteredEnUp");
201  break;
203  snprintf(buff, 1023, baseTagStr.c_str(), "UnclusteredEnDown");
204  break;
205  default:
206  throw cms::Exception("LogicError", "OneMETShift constructor called with bogus shift");
207  }
208  token = cc.consumes<pat::METCollection>(edm::InputTag(buff));
209 }
210 
212  using namespace edm;
213  using namespace std;
214 
216  iEvent.getByToken(src_, src);
217  if (src->size() != 1)
218  throw cms::Exception("CorruptData", "More than one MET in the collection");
219 
220  auto out = std::make_unique<std::vector<pat::MET>>(1, src->front());
221  pat::MET &met = out->back();
222 
223  for (const OneMETShift &shift : shifts_) {
224  shift.readAndSet(iEvent, met);
225  }
226 
227  iEvent.put(std::move(out));
228 }
229 
230 void
231 
234  ev.getByToken(token, src);
235 
236  if (src->size() != 1)
237  throw cms::Exception("CorruptData", "More than one MET in the shifted collection");
238  const pat::MET &met2 = src->front();
239 
240  if (t0FromMiniAOD) {
241  if (uncShift)
245  shift,
246  isSmeared);
247  if (corShift)
248  met.setCorShift(
250  } else {
251  if (uncShift)
252  met.setUncShift(met2.px(), met2.py(), met2.sumEt(), shift, isSmeared);
253  if (corShift)
254  met.setCorShift(met2.px(), met2.py(), met2.sumEt(), level);
255  }
256 }
257 
259 using namespace pat;
Analysis-level MET class.
Definition: MET.h:40
T getParameter(std::string const &) const
double shiftedPx(METUncertainty shift, METCorrectionLevel level=Type1) const
Definition: MET.h:214
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:131
bool existsAs(std::string const &parameterName, bool trackiness=true) const
checks if a parameter exists as a given type
Definition: ParameterSet.h:160
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:525
const pat::MET::METCorrectionType level
~PATMETSlimmer() override
std::vector< MET > METCollection
Definition: MET.h:31
double px() const final
x coordinate of momentum vector
bool ev
void setUncShift(double px, double py, double sumEt, METUncertainty shift, bool isSmeared=false)
Definition: MET.cc:326
METUncertainty
Definition: MET.h:152
double corPy(METCorrectionLevel level=Type1) const
Definition: MET.h:232
std::string encode() const
Definition: InputTag.cc:159
EDGetTokenT< ProductType > consumes(edm::InputTag const &tag)
Definition: HeavyIon.h:7
double corSumEt(METCorrectionLevel level=Type1) const
Definition: MET.cc:319
double shiftedPy(METUncertainty shift, METCorrectionLevel level=Type1) const
Definition: MET.h:217
void maybeReadShifts(const edm::ParameterSet &basePSet, const std::string &name, pat::MET::METCorrectionType level, bool readFromMiniAOD=false)
edm::EDGetTokenT< pat::METCollection > token
int iEvent
Definition: GenABIO.cc:224
double sumEt() const
Definition: MET.h:52
#define DEFINE_FWK_MODULE(type)
Definition: MakerMacros.h:16
void produce(edm::StreamID, edm::Event &iEvent, const edm::EventSetup &iSetup) const override
const pat::MET::METUncertainty shift
ConsumesCollector consumesCollector()
Use a ConsumesCollector to gather consumes information from helper functions.
const edm::EDGetTokenT< pat::METCollection > src_
double shiftedSumEt(METUncertainty shift, METCorrectionLevel level=Type1) const
Definition: MET.cc:296
Collection of MET.
double corPx(METCorrectionLevel level=Type1) const
Definition: MET.h:231
Slimmer of PAT METs.
double py() const final
y coordinate of momentum vector
HLT enums.
METCorrectionType
Definition: MET.h:189
PATMETSlimmer(const edm::ParameterSet &iConfig)
static unsigned int const shift
std::vector< OneMETShift > shifts_
void setCorShift(double px, double py, double sumEt, METCorrectionType level)
Definition: MET.cc:342
void readAndSet(const edm::Event &ev, pat::MET &met) const
const bool onMiniAOD_
def move(src, dest)
Definition: eostools.py:511