|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// -*- C++ -*-
//
// Package: Utilities
// Class: SimpleJetFilter
//
/**\class SimpleJetFilter SimpleJetFilter.cc DPGAnalysis/Skims/src/SimpleJetFilter.cc
Description:
Implementation:
<Notes on implementation>
*/
//
// Original Author: Andrea Venturi
// Created: Tue Oct 21 20:55:22 CEST 2008
//
//
// system include files
#include <memory>
#include <string>
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/JetReco/interface/JetID.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "PhysicsTools/SelectorUtils/interface/JetIDSelectionFunctor.h"
#include "PhysicsTools/SelectorUtils/interface/strbitset.h"
//
// class declaration
//
class SimpleJetFilter : public edm::stream::EDFilter<> {
public:
explicit SimpleJetFilter(const edm::ParameterSet&);
~SimpleJetFilter() override;
private:
bool filter(edm::Event&, const edm::EventSetup&) override;
// ----------member data ---------------------------
edm::InputTag m_jetCollection;
edm::InputTag m_jetIDMap;
const double m_ptcut;
const double m_etamaxcut;
const double m_njetmin;
JetIDSelectionFunctor m_jetIDfunc;
};
//
// constructors and destructor
//
SimpleJetFilter::SimpleJetFilter(const edm::ParameterSet& iConfig)
: m_jetCollection(iConfig.getParameter<edm::InputTag>("jetCollection")),
m_jetIDMap(iConfig.getParameter<edm::InputTag>("jetIDMap")),
m_ptcut(iConfig.getParameter<double>("ptCut")),
m_etamaxcut(iConfig.getParameter<double>("maxRapidityCut")),
m_njetmin(iConfig.getParameter<unsigned int>("nJetMin")),
m_jetIDfunc(JetIDSelectionFunctor::PURE09, JetIDSelectionFunctor::LOOSE) {
//now do what ever initialization is needed
}
SimpleJetFilter::~SimpleJetFilter() {
// do anything here that needs to be done at desctruction time
// (e.g. close files, deallocate resources etc.)
}
//
// member functions
//
// ------------ method called on each new Event ------------
bool SimpleJetFilter::filter(edm::Event& iEvent, const edm::EventSetup& iSetup) {
using namespace edm;
bool selected = false;
Handle<reco::CaloJetCollection> jetcoll;
iEvent.getByLabel(m_jetCollection, jetcoll);
Handle<reco::JetIDValueMap> jetIDmap;
iEvent.getByLabel(m_jetIDMap, jetIDmap);
unsigned int goodjets = 0;
for (unsigned int ijet = 0; ijet < jetcoll->size(); ++ijet) {
const reco::CaloJetRef jet(jetcoll, ijet);
LogDebug("JetUnderTest") << "Jet with eta = " << jet->eta() << " and pt = " << jet->pt() << " under test";
if (!(std::abs(jet->eta()) < m_etamaxcut && jet->pt() > m_ptcut))
continue;
LogDebug("JetUnderTest") << "kincut passed";
if (jetIDmap->contains(jet.id())) {
const reco::JetID& jetid = (*jetIDmap)[jet];
pat::strbitset ret = m_jetIDfunc.getBitTemplate();
ret.set(false);
bool goodjet = m_jetIDfunc((*jetcoll)[ijet], jetid, ret);
if (goodjet) {
++goodjets;
LogDebug("JetUnderTest") << "JetID passed";
}
if (goodjets >= m_njetmin)
return true;
} else {
edm::LogWarning("JetIDNotFound") << "JetID not found ";
}
}
return selected;
}
//define this as a plug-in
DEFINE_FWK_MODULE(SimpleJetFilter);
|