Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Convert JetConstituentSelector template from legacy to stream filter #17215

Merged
merged 4 commits into from Jan 27, 2017
Merged

Convert JetConstituentSelector template from legacy to stream filter #17215

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
648c6bd
Tidying up of legacy version.
knoepfel Jan 19, 2017
801e5b4
Switch to stream filter.
knoepfel Jan 19, 2017
055b352
Change plugins from filters to producers + related changed in configu…
knoepfel Jan 20, 2017
c91e20e
Add fillDescriptions static member functions.
knoepfel Jan 23, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
115 changes: 50 additions & 65 deletions CommonTools/RecoAlgos/plugins/JetConstituentSelector.cc
View file
Open in desktop
Expand Up @@ -12,92 +12,77 @@
*
* https://twiki.cern.ch/twiki/bin/view/CMS/SWGuidePhysicsCutParser
*
*
*/


#include "FWCore/Framework/interface/EDFilter.h"

#include "CommonTools/UtilAlgos/interface/StringCutObjectSelector.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/PatCandidates/interface/Jet.h"
#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
#include "DataFormats/JetReco/interface/CaloJet.h"

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "CommonTools/UtilAlgos/interface/StringCutObjectSelector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

template < class T, typename C = std::vector<typename T::ConstituentTypeFwdPtr> >
class JetConstituentSelector : public edm::EDFilter {

template <class T, typename C = std::vector<typename T::ConstituentTypeFwdPtr>>
class JetConstituentSelector : public edm::stream::EDProducer<> {
public:

typedef std::vector<T> JetsOutput;
typedef C ConstituentsOutput;
using JetsOutput = std::vector<T>;
using ConstituentsOutput = C;

JetConstituentSelector ( edm::ParameterSet const & params ) :
srcToken_( consumes< typename edm::View<T> >( params.getParameter<edm::InputTag>("src") ) ),
cut_( params.getParameter<std::string>("cut") ),
filter_(false),
selector_( cut_ )
JetConstituentSelector(edm::ParameterSet const& params) :
srcToken_{consumes<edm::View<T>>(params.getParameter<edm::InputTag>("src"))},
selector_{params.getParameter<std::string>("cut")}
{
produces< JetsOutput >();
produces< ConstituentsOutput > ("constituents");
produces<JetsOutput>();
produces<ConstituentsOutput>("constituents");
}

virtual ~JetConstituentSelector() {}

virtual void beginJob() override {}
virtual void endJob() override {}

virtual bool filter(edm::Event& iEvent, const edm::EventSetup& iSetup) override {

std::unique_ptr< JetsOutput > jets ( new std::vector<T>() );
std::unique_ptr< ConstituentsOutput > candsOut( new ConstituentsOutput );

edm::Handle< typename edm::View<T> > h_jets;
iEvent.getByToken( srcToken_, h_jets );

// Now set the Ptrs with the orphan handles.
for ( typename edm::View<T>::const_iterator ibegin = h_jets->begin(),
iend = h_jets->end(), ijet = ibegin;
ijet != iend; ++ijet ) {
static void fillDescriptions(edm::ConfigurationDescriptions& descriptions)
{
edm::ParameterSetDescription desc;
desc.add<edm::InputTag>("src")->setComment("InputTag used for retrieving jets in event.");
desc.add<std::string>("cut")->setComment("Cut used by which to select jets. For example:\n"
" \"pt > 100.0 && abs(rapidity()) < 2.4\".");
descriptions.add("JetConsituentSelector", desc);
}

// Check the selection
if ( selector_(*ijet) ) {
// Add the jets that pass to the output collection
jets->push_back( *ijet );
for ( unsigned int ida = 0; ida < ijet->numberOfDaughters(); ++ida ) {
candsOut->push_back( typename ConstituentsOutput::value_type( ijet->daughterPtr(ida), ijet->daughterPtr(ida) ) );
}
}
void produce(edm::Event& iEvent, edm::EventSetup const& iSetup) override
{
auto jets = std::make_unique<JetsOutput>();
auto candsOut = std::make_unique<ConstituentsOutput>();

edm::Handle<edm::View<T>> h_jets;
iEvent.getByToken(srcToken_, h_jets);

// Now set the Ptrs with the orphan handles.
for (auto const& jet : *h_jets) {
// Check the selection
if (selector_(jet)) {
// Add the jets that pass to the output collection
jets->push_back(jet);
for (unsigned int ida {}; ida < jet.numberOfDaughters(); ++ida) {
candsOut->emplace_back(jet.daughterPtr(ida), jet.daughterPtr(ida));
}
}

// put in Event
bool pass = jets->size() > 0;
iEvent.put(std::move(jets));
iEvent.put(std::move(candsOut), "constituents");

if ( filter_ )
return pass;
else
return true;

}

protected:
edm::EDGetTokenT< typename edm::View<T> > srcToken_;
std::string cut_;
bool filter_;
StringCutObjectSelector<T> selector_;
iEvent.put(std::move(jets));
iEvent.put(std::move(candsOut), "constituents");
}

private:
edm::EDGetTokenT<edm::View<T>> const srcToken_;
StringCutObjectSelector<T> const selector_;
};

typedef JetConstituentSelector<reco::PFJet> PFJetConstituentSelector;
typedef JetConstituentSelector<pat::Jet, std::vector< edm::FwdPtr<pat::PackedCandidate> > > PatJetConstituentSelector;
typedef JetConstituentSelector<reco::PFJet, std::vector< edm::FwdPtr<pat::PackedCandidate> > > MiniAODJetConstituentSelector;
using PFJetConstituentSelector = JetConstituentSelector<reco::PFJet>;
using PatJetConstituentSelector = JetConstituentSelector<pat::Jet, std::vector<edm::FwdPtr<pat::PackedCandidate>>>;
using MiniAODJetConstituentSelector = JetConstituentSelector<reco::PFJet, std::vector<edm::FwdPtr<pat::PackedCandidate>>>;

DEFINE_FWK_MODULE( PFJetConstituentSelector );
DEFINE_FWK_MODULE( PatJetConstituentSelector );
DEFINE_FWK_MODULE( MiniAODJetConstituentSelector );
DEFINE_FWK_MODULE(PFJetConstituentSelector);
DEFINE_FWK_MODULE(PatJetConstituentSelector);
DEFINE_FWK_MODULE(MiniAODJetConstituentSelector);
8 changes: 4 additions & 4 deletions RecoJets/Configuration/python/RecoPFJets_cff.py
View file
Open in desktop
Expand Up @@ -79,10 +79,10 @@



ak8PFJetsCHSConstituents = cms.EDFilter("PFJetConstituentSelector",
src = cms.InputTag("ak8PFJetsCHS"),
cut = cms.string("pt > 100.0 && abs(rapidity()) < 2.4")
)
ak8PFJetsCHSConstituents = cms.EDProducer("PFJetConstituentSelector",
src = cms.InputTag("ak8PFJetsCHS"),
cut = cms.string("pt > 100.0 && abs(rapidity()) < 2.4")
)


# Advanced Algorithms for AK4, AK5, AK8 and CA8 :
Expand Down
6 changes: 3 additions & 3 deletions RecoJets/JetProducers/python/ak8PFJetsCS_cfi.py
View file
Open in desktop
Expand Up @@ -20,9 +20,9 @@



ak8PFJetsCSConstituents = cms.EDFilter("PFJetConstituentSelector",
src = cms.InputTag("ak8PFJetsCS"),
cut = cms.string("pt > 100.0")
ak8PFJetsCSConstituents = cms.EDProducer("PFJetConstituentSelector",
src = cms.InputTag("ak8PFJetsCS"),
cut = cms.string("pt > 100.0")
)


Expand Down