File indexing completed on 2023-03-17 11:16:30
0001 #include "CommonTools/UtilAlgos/interface/TFileService.h"
0002 #include "DataFormats/PatCandidates/interface/Tau.h"
0003 #include "FWCore/Framework/interface/one/EDAnalyzer.h"
0004 #include "FWCore/Framework/interface/Event.h"
0005 #include "FWCore/Framework/interface/EventSetup.h"
0006 #include "FWCore/Framework/interface/Frameworkfwd.h"
0007 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0008 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0009 #include "FWCore/ServiceRegistry/interface/Service.h"
0010 #include "FWCore/Utilities/interface/InputTag.h"
0011
0012 #include <TH1.h>
0013 #include <TMath.h>
0014
0015 #include <string>
0016
0017 class PatTauAnalyzer : public edm::one::EDAnalyzer<edm::one::SharedResources> {
0018 public:
0019 explicit PatTauAnalyzer(const edm::ParameterSet&);
0020 ~PatTauAnalyzer() override;
0021
0022
0023 void beginJob() override;
0024 void analyze(const edm::Event&, const edm::EventSetup&) override;
0025 void endJob() override;
0026
0027 private:
0028
0029 edm::InputTag src_;
0030 edm::EDGetTokenT<pat::TauCollection> srcToken_;
0031
0032 bool requireGenTauMatch_;
0033
0034 std::string discrByLeadTrack_;
0035 std::string discrByIso_;
0036 std::string discrByTaNC_;
0037
0038
0039 TH1* hGenTauEnergy_;
0040 TH1* hGenTauPt_;
0041 TH1* hGenTauEta_;
0042 TH1* hGenTauPhi_;
0043
0044
0045 TH1* hTauJetEnergy_;
0046 TH1* hTauJetPt_;
0047 TH1* hTauJetEta_;
0048 TH1* hTauJetPhi_;
0049
0050 TH1* hNumTauJets_;
0051
0052 TH1* hTauLeadTrackPt_;
0053
0054 TH1* hTauNumSigConeTracks_;
0055 TH1* hTauNumIsoConeTracks_;
0056
0057 TH1* hTauDiscrByIso_;
0058 TH1* hTauDiscrByTaNC_;
0059 TH1* hTauDiscrAgainstElectrons_;
0060 TH1* hTauDiscrAgainstMuons_;
0061
0062 TH1* hTauJetEnergyIsoPassed_;
0063 TH1* hTauJetPtIsoPassed_;
0064 TH1* hTauJetEtaIsoPassed_;
0065 TH1* hTauJetPhiIsoPassed_;
0066
0067 TH1* hTauJetEnergyTaNCpassed_;
0068 TH1* hTauJetPtTaNCpassed_;
0069 TH1* hTauJetEtaTaNCpassed_;
0070 TH1* hTauJetPhiTaNCpassed_;
0071 };
0072
0073 const reco::GenParticle* getGenTau(const pat::Tau& patTau) {
0074 std::vector<reco::GenParticleRef> associatedGenParticles = patTau.genParticleRefs();
0075 for (std::vector<reco::GenParticleRef>::const_iterator it = associatedGenParticles.begin();
0076 it != associatedGenParticles.end();
0077 ++it) {
0078 if (it->isAvailable()) {
0079 const reco::GenParticleRef& genParticle = (*it);
0080 if (genParticle->pdgId() == -15 || genParticle->pdgId() == +15)
0081 return genParticle.get();
0082 }
0083 }
0084
0085 return nullptr;
0086 }
0087
0088 PatTauAnalyzer::PatTauAnalyzer(const edm::ParameterSet& cfg) {
0089
0090 usesResource(TFileService::kSharedResource);
0091
0092
0093 src_ = cfg.getParameter<edm::InputTag>("src");
0094
0095 srcToken_ = consumes<pat::TauCollection>(src_);
0096
0097
0098 requireGenTauMatch_ = cfg.getParameter<bool>("requireGenTauMatch");
0099
0100
0101
0102 discrByLeadTrack_ = cfg.getParameter<std::string>("discrByLeadTrack");
0103
0104
0105 discrByIso_ = cfg.getParameter<std::string>("discrByIso");
0106
0107
0108 discrByTaNC_ = cfg.getParameter<std::string>("discrByTaNC");
0109
0110 }
0111
0112 PatTauAnalyzer::~PatTauAnalyzer() {
0113
0114
0115
0116
0117
0118
0119
0120
0121
0122
0123
0124
0125
0126
0127
0128
0129
0130
0131
0132
0133
0134
0135
0136
0137
0138
0139
0140
0141
0142
0143
0144
0145
0146
0147 }
0148
0149 void PatTauAnalyzer::beginJob() {
0150
0151
0152 edm::Service<TFileService> fs;
0153
0154
0155 hGenTauEnergy_ = fs->make<TH1F>("GenTauEnergy", "GenTauEnergy", 30, 0., 150.);
0156 hGenTauPt_ = fs->make<TH1F>("GenTauPt", "GenTauPt", 30, 0., 150.);
0157 hGenTauEta_ = fs->make<TH1F>("GenTauEta", "GenTauEta", 24, -3., +3.);
0158 hGenTauPhi_ = fs->make<TH1F>("GenTauPhi", "GenTauPhi", 18, -TMath::Pi(), +TMath::Pi());
0159
0160
0161
0162 hTauJetEnergy_ = fs->make<TH1F>("TauJetEnergy", "TauJetEnergy", 30, 0., 150.);
0163 hTauJetPt_ = fs->make<TH1F>("TauJetPt", "TauJetPt", 30, 0., 150.);
0164
0165
0166
0167
0168
0169
0170
0171
0172
0173
0174
0175
0176
0177
0178
0179
0180
0181
0182
0183 hNumTauJets_ = fs->make<TH1F>("NumTauJets", "NumTauJets", 10, -0.5, 9.5);
0184
0185
0186 hTauLeadTrackPt_ = fs->make<TH1F>("TauLeadTrackPt", "TauLeadTrackPt", 40, 0., 100.);
0187
0188
0189 hTauNumSigConeTracks_ = fs->make<TH1F>("TauNumSigConeTracks", "TauNumSigConeTracks", 10, -0.5, 9.5);
0190 hTauNumIsoConeTracks_ = fs->make<TH1F>("TauNumIsoConeTracks", "TauNumIsoConeTracks", 20, -0.5, 19.5);
0191
0192
0193
0194 hTauDiscrByIso_ = fs->make<TH1F>("TauDiscrByIso", "TauDiscrByIso", 103, -0.015, 1.015);
0195 hTauDiscrByTaNC_ = fs->make<TH1F>("TauDiscrByTaNC", "TauDiscrByTaNC", 103, -0.015, 1.015);
0196
0197
0198 hTauDiscrAgainstElectrons_ =
0199 fs->make<TH1F>("TauDiscrAgainstElectrons", "TauDiscrAgainstElectrons", 103, -0.015, 1.015);
0200 hTauDiscrAgainstMuons_ = fs->make<TH1F>("TauDiscrAgainstMuons", "TauDiscrAgainstMuons", 103, -0.015, 1.015);
0201
0202
0203 hTauJetEnergyIsoPassed_ = fs->make<TH1F>("TauJetEnergyIsoPassed", "TauJetEnergyIsoPassed", 30, 0., 150.);
0204 hTauJetPtIsoPassed_ = fs->make<TH1F>("TauJetPtIsoPassed", "TauJetPtIsoPassed", 30, 0., 150.);
0205 hTauJetEtaIsoPassed_ = fs->make<TH1F>("TauJetEtaIsoPassed", "TauJetEtaIsoPassed", 24, -3., +3.);
0206 hTauJetPhiIsoPassed_ = fs->make<TH1F>("TauJetPhiIsoPassed", "TauJetPhiIsoPassed", 18, -TMath::Pi(), +TMath::Pi());
0207
0208
0209 hTauJetEnergyTaNCpassed_ = fs->make<TH1F>("TauJetEnergyTaNCpassed", "TauJetEnergyTaNCpassed", 30, 0., 150.);
0210 hTauJetPtTaNCpassed_ = fs->make<TH1F>("TauJetPtTaNCpassed", "TauJetPtTaNCpassed", 30, 0., 150.);
0211 hTauJetEtaTaNCpassed_ = fs->make<TH1F>("TauJetEtaTaNCpassed", "TauJetEtaTaNCpassed", 24, -3., +3.);
0212 hTauJetPhiTaNCpassed_ = fs->make<TH1F>("TauJetPhiTaNCpassed", "TauJetPhiTaNCpassed", 18, -TMath::Pi(), +TMath::Pi());
0213 }
0214
0215 void PatTauAnalyzer::analyze(const edm::Event& evt, const edm::EventSetup& es) {
0216
0217
0218 edm::Handle<pat::TauCollection> patTaus;
0219 evt.getByToken(srcToken_, patTaus);
0220
0221 hNumTauJets_->Fill(patTaus->size());
0222
0223 for (pat::TauCollection::const_iterator patTau = patTaus->begin(); patTau != patTaus->end(); ++patTau) {
0224
0225 const reco::GenParticle* genTau = getGenTau(*patTau);
0226 if (requireGenTauMatch_ && !genTau)
0227 continue;
0228
0229
0230 if (genTau) {
0231 hGenTauEnergy_->Fill(genTau->energy());
0232 hGenTauPt_->Fill(genTau->pt());
0233 hGenTauEta_->Fill(genTau->eta());
0234 hGenTauPhi_->Fill(genTau->phi());
0235 }
0236
0237
0238
0239 hTauJetEnergy_->Fill(patTau->energy());
0240 hTauJetPt_->Fill(patTau->pt());
0241
0242
0243
0244
0245
0246
0247
0248
0249
0250
0251
0252
0253
0254
0255
0256
0257
0258
0259
0260
0261
0262
0263
0264
0265
0266
0267
0268
0269
0270 hTauNumSigConeTracks_->Fill(patTau->signalTracks().size());
0271 hTauNumIsoConeTracks_->Fill(patTau->isolationTracks().size());
0272
0273
0274
0275
0276
0277 float discrByIso = (patTau->tauID(discrByLeadTrack_.data()) > 0.5) ? patTau->tauID(discrByIso_.data()) : 0.;
0278 hTauDiscrByIso_->Fill(discrByIso);
0279 float discrByTaNC = (patTau->tauID(discrByLeadTrack_.data()) > 0.5) ? patTau->tauID(discrByTaNC_.data()) : 0.;
0280 hTauDiscrByTaNC_->Fill(discrByTaNC);
0281
0282
0283
0284
0285
0286
0287
0288
0289
0290
0291
0292
0293
0294
0295
0296
0297
0298
0299 hTauDiscrAgainstMuons_->Fill(patTau->tauID("againstMuonLoose"));
0300
0301
0302 if (discrByIso > 0.5) {
0303 hTauJetEnergyIsoPassed_->Fill(patTau->energy());
0304 hTauJetPtIsoPassed_->Fill(patTau->pt());
0305 hTauJetEtaIsoPassed_->Fill(patTau->eta());
0306 hTauJetPhiIsoPassed_->Fill(patTau->phi());
0307 }
0308
0309
0310 if (discrByTaNC > 0.5) {
0311 hTauJetEnergyTaNCpassed_->Fill(patTau->energy());
0312 hTauJetPtTaNCpassed_->Fill(patTau->pt());
0313 hTauJetEtaTaNCpassed_->Fill(patTau->eta());
0314 hTauJetPhiTaNCpassed_->Fill(patTau->phi());
0315 }
0316 }
0317 }
0318
0319 void PatTauAnalyzer::endJob() {
0320
0321 }
0322
0323 #include "FWCore/Framework/interface/MakerMacros.h"
0324
0325 DEFINE_FWK_MODULE(PatTauAnalyzer);