|
|
|
#include <PATJetSelector.h>
Public Member Functions | |
| virtual void | beginJob () |
| virtual void | endJob () |
| virtual bool | filter (edm::Event &iEvent, const edm::EventSetup &iSetup) |
| PATJetSelector (edm::ParameterSet const ¶ms) | |
| virtual | ~PATJetSelector () |
Protected Attributes | |
| std::string | cut_ |
| bool | filter_ |
| StringCutObjectSelector< Jet > | selector_ |
| edm::InputTag | src_ |
Definition at line 25 of file PATJetSelector.h.
| pat::PATJetSelector::PATJetSelector | ( | edm::ParameterSet const & | params | ) | [inline] |
Definition at line 29 of file PATJetSelector.h.
References edm::ParameterSet::exists(), filter_, and edm::ParameterSet::getParameter().
:
edm::EDFilter( ),
src_( params.getParameter<edm::InputTag>("src") ),
cut_( params.getParameter<std::string>("cut") ),
filter_(false),
selector_( cut_ )
{
produces< std::vector<pat::Jet> >();
produces<reco::GenJetCollection> ("genJets");
produces<std::vector<CaloTower> > ("caloTowers");
produces<reco::PFCandidateCollection > ("pfCandidates");
produces<edm::OwnVector<reco::BaseTagInfo> > ("tagInfos");
if ( params.exists("filter") ) {
filter_ = params.getParameter<bool>("filter");
}
}
| virtual pat::PATJetSelector::~PATJetSelector | ( | ) | [inline, virtual] |
Definition at line 47 of file PATJetSelector.h.
{}
| virtual void pat::PATJetSelector::beginJob | ( | void | ) | [inline, virtual] |
| virtual void pat::PATJetSelector::endJob | ( | void | ) | [inline, virtual] |
| virtual bool pat::PATJetSelector::filter | ( | edm::Event & | iEvent, |
| const edm::EventSetup & | iSetup | ||
| ) | [inline, virtual] |
Implements edm::EDFilter.
Definition at line 52 of file PATJetSelector.h.
References edm::OwnVector< T, P >::begin(), edm::OwnVector< T, P >::end(), filter_, edm::Event::getByLabel(), edm::Event::getRefBeforePut(), jetProducer_cfi::patJets, edm::Event::put(), selector_, and src_.
{
std::auto_ptr< std::vector<Jet> > patJets ( new std::vector<Jet>() );
std::auto_ptr<reco::GenJetCollection > genJetsOut ( new reco::GenJetCollection() );
std::auto_ptr<std::vector<CaloTower> > caloTowersOut( new std::vector<CaloTower> () );
std::auto_ptr<reco::PFCandidateCollection > pfCandidatesOut( new reco::PFCandidateCollection() );
std::auto_ptr<edm::OwnVector<reco::BaseTagInfo> > tagInfosOut ( new edm::OwnVector<reco::BaseTagInfo>() );
edm::RefProd<reco::GenJetCollection > h_genJetsOut = iEvent.getRefBeforePut<reco::GenJetCollection >( "genJets" );
edm::RefProd<std::vector<CaloTower> > h_caloTowersOut = iEvent.getRefBeforePut<std::vector<CaloTower> > ( "caloTowers" );
edm::RefProd<reco::PFCandidateCollection > h_pfCandidatesOut = iEvent.getRefBeforePut<reco::PFCandidateCollection > ( "pfCandidates" );
edm::RefProd<edm::OwnVector<reco::BaseTagInfo> > h_tagInfosOut = iEvent.getRefBeforePut<edm::OwnVector<reco::BaseTagInfo> > ( "tagInfos" );
edm::Handle< edm::View<pat::Jet> > h_jets;
iEvent.getByLabel( src_, h_jets );
// First loop over the products and make the secondary output collections
for ( edm::View<pat::Jet>::const_iterator ibegin = h_jets->begin(),
iend = h_jets->end(), ijet = ibegin;
ijet != iend; ++ijet ) {
// Check the selection
if ( selector_(*ijet) ) {
// Copy over the calo towers
for ( CaloTowerFwdPtrVector::const_iterator itowerBegin = ijet->caloTowersFwdPtr().begin(),
itowerEnd = ijet->caloTowersFwdPtr().end(), itower = itowerBegin;
itower != itowerEnd; ++itower ) {
// Add to global calo tower list
caloTowersOut->push_back( **itower );
}
// Copy over the pf candidates
for ( reco::PFCandidateFwdPtrVector::const_iterator icandBegin = ijet->pfCandidatesFwdPtr().begin(),
icandEnd = ijet->pfCandidatesFwdPtr().end(), icand = icandBegin;
icand != icandEnd; ++icand ) {
// Add to global pf candidate list
pfCandidatesOut->push_back( **icand );
}
// Copy the tag infos
for ( TagInfoFwdPtrCollection::const_iterator iinfoBegin = ijet->tagInfosFwdPtr().begin(),
iinfoEnd = ijet->tagInfosFwdPtr().end(), iinfo = iinfoBegin;
iinfo != iinfoEnd; ++iinfo ) {
// Add to global calo tower list
tagInfosOut->push_back( **iinfo );
}
// Copy the gen jet
if ( ijet->genJet() != 0 ) {
genJetsOut->push_back( *(ijet->genJet()) );
}
}
}
// Output the secondary collections.
edm::OrphanHandle<reco::GenJetCollection> oh_genJetsOut = iEvent.put( genJetsOut, "genJets" );
edm::OrphanHandle<std::vector<CaloTower> > oh_caloTowersOut = iEvent.put( caloTowersOut, "caloTowers" );
edm::OrphanHandle<reco::PFCandidateCollection> oh_pfCandidatesOut = iEvent.put( pfCandidatesOut, "pfCandidates" );
edm::OrphanHandle<edm::OwnVector<reco::BaseTagInfo> > oh_tagInfosOut = iEvent.put( tagInfosOut, "tagInfos" );
unsigned int caloTowerIndex = 0;
unsigned int pfCandidateIndex = 0;
unsigned int tagInfoIndex = 0;
unsigned int genJetIndex = 0;
// Now set the Ptrs with the orphan handles.
for ( edm::View<pat::Jet>::const_iterator ibegin = h_jets->begin(),
iend = h_jets->end(), ijet = ibegin;
ijet != iend; ++ijet ) {
// Check the selection
if ( selector_(*ijet) ) {
// Add the jets that pass to the output collection
patJets->push_back( *ijet );
// Copy over the calo towers
for ( CaloTowerFwdPtrVector::const_iterator itowerBegin = ijet->caloTowersFwdPtr().begin(),
itowerEnd = ijet->caloTowersFwdPtr().end(), itower = itowerBegin;
itower != itowerEnd; ++itower ) {
// Update the "forward" bit of the FwdPtr to point at the new tower collection.
// ptr to "this" tower in the global list
edm::Ptr<CaloTower> outPtr( oh_caloTowersOut, caloTowerIndex);
patJets->back().updateFwdCaloTowerFwdPtr( itower - itowerBegin,// index of "this" tower in the jet
outPtr
);
++caloTowerIndex;
}
// Copy over the pf candidates
for ( reco::PFCandidateFwdPtrVector::const_iterator icandBegin = ijet->pfCandidatesFwdPtr().begin(),
icandEnd = ijet->pfCandidatesFwdPtr().end(), icand = icandBegin;
icand != icandEnd; ++icand ) {
// Update the "forward" bit of the FwdPtr to point at the new tower collection.
// ptr to "this" cand in the global list
edm::Ptr<reco::PFCandidate> outPtr( oh_pfCandidatesOut, pfCandidateIndex );
patJets->back().updateFwdPFCandidateFwdPtr( icand - icandBegin,// index of "this" tower in the jet
outPtr
);
++pfCandidateIndex;
}
// Copy the tag infos
for ( TagInfoFwdPtrCollection::const_iterator iinfoBegin = ijet->tagInfosFwdPtr().begin(),
iinfoEnd = ijet->tagInfosFwdPtr().end(), iinfo = iinfoBegin;
iinfo != iinfoEnd; ++iinfo ) {
// Update the "forward" bit of the FwdPtr to point at the new tower collection.
// ptr to "this" info in the global list
edm::Ptr<reco::BaseTagInfo > outPtr( oh_tagInfosOut, tagInfoIndex );
patJets->back().updateFwdTagInfoFwdPtr( iinfo - iinfoBegin,// index of "this" tower in the jet
outPtr
);
++tagInfoIndex;
}
// Copy the gen jet
if ( ijet->genJet() != 0 ) {
patJets->back().updateFwdGenJetFwdRef( edm::Ref<reco::GenJetCollection>( oh_genJetsOut, genJetIndex) // ref to "this" genjet in the global list
);
++genJetIndex;
}
}
}
// put genEvt in Event
bool pass = patJets->size() > 0;
iEvent.put(patJets);
if ( filter_ )
return pass;
else
return true;
}
std::string pat::PATJetSelector::cut_ [protected] |
Definition at line 201 of file PATJetSelector.h.
bool pat::PATJetSelector::filter_ [protected] |
Definition at line 202 of file PATJetSelector.h.
Referenced by filter(), and PATJetSelector().
StringCutObjectSelector<Jet> pat::PATJetSelector::selector_ [protected] |
Definition at line 203 of file PATJetSelector.h.
Referenced by filter().
edm::InputTag pat::PATJetSelector::src_ [protected] |
Definition at line 200 of file PATJetSelector.h.
Referenced by filter().
1.7.3