CMS 3D CMS Logo

List of all members | Public Member Functions | Private Attributes
JetTracksAssociatorAtVertex Class Reference

#include <JetTracksAssociatorAtVertex.h>

Inheritance diagram for JetTracksAssociatorAtVertex:
edm::stream::EDProducer<>

Public Member Functions

 JetTracksAssociatorAtVertex (const edm::ParameterSet &)
 
void produce (edm::Event &, const edm::EventSetup &) override
 
 ~JetTracksAssociatorAtVertex () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 

Private Attributes

JetTracksAssociationDRVertex mAssociator
 
JetTracksAssociationDRVertexAssigned mAssociatorAssigned
 
edm::EDGetTokenT< edm::View< reco::Jet > > mJets
 
int mTrackQuality
 
edm::EDGetTokenT< reco::TrackCollectionmTracks
 
edm::EDGetTokenT< reco::VertexCollectionpvSrc
 if true, use the track/jet association with vertex assignment to tracks More...
 
bool useAssigned
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... > CacheTypes
 
typedef CacheTypes::GlobalCache GlobalCache
 
typedef AbilityChecker< T... > HasAbility
 
typedef CacheTypes::LuminosityBlockCache LuminosityBlockCache
 
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCacheLuminosityBlockContext
 
typedef CacheTypes::LuminosityBlockSummaryCache LuminosityBlockSummaryCache
 
typedef CacheTypes::RunCache RunCache
 
typedef RunContextT< RunCache, GlobalCacheRunContext
 
typedef CacheTypes::RunSummaryCache RunSummaryCache
 

Detailed Description

Definition at line 19 of file JetTracksAssociatorAtVertex.h.

Constructor & Destructor Documentation

JetTracksAssociatorAtVertex::JetTracksAssociatorAtVertex ( const edm::ParameterSet fConfig)

Definition at line 18 of file JetTracksAssociatorAtVertex.cc.

References edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), mJets, mTracks, pvSrc, and useAssigned.

19  : mAssociator (fConfig.getParameter<double> ("coneSize")),
20  mAssociatorAssigned (fConfig.getParameter<double> ("coneSize")),
21  useAssigned(false), pvSrc()
22 {
23  mJets = consumes<edm::View <reco::Jet> >(fConfig.getParameter<edm::InputTag> ("jets"));
24  mTracks = consumes<reco::TrackCollection>(fConfig.getParameter<edm::InputTag>("tracks"));
25  if ( fConfig.exists("useAssigned") ) {
26  useAssigned = fConfig.getParameter<bool> ("useAssigned");
27  pvSrc = consumes<reco::VertexCollection>(fConfig.getParameter<edm::InputTag> ("pvSrc"));
28  }
29 
30  produces<reco::JetTracksAssociation::Container> ();
31 }
T getParameter(std::string const &) const
edm::EDGetTokenT< reco::TrackCollection > mTracks
bool exists(std::string const &parameterName) const
checks if a parameter exists
edm::EDGetTokenT< edm::View< reco::Jet > > mJets
edm::EDGetTokenT< reco::VertexCollection > pvSrc
if true, use the track/jet association with vertex assignment to tracks
JetTracksAssociationDRVertexAssigned mAssociatorAssigned
JetTracksAssociationDRVertex mAssociator
JetTracksAssociatorAtVertex::~JetTracksAssociatorAtVertex ( )
override

Definition at line 33 of file JetTracksAssociatorAtVertex.cc.

33 {}

Member Function Documentation

void JetTracksAssociatorAtVertex::produce ( edm::Event fEvent,
const edm::EventSetup fSetup 
)
override

Definition at line 35 of file JetTracksAssociatorAtVertex.cc.

References reco::JetExtendedAssociation::allJets(), allTracks_cfi::allTracks, edm::Event::getByToken(), mps_fire::i, impactParameterTagInfos_cfi::jetTracks, mAssociator, mAssociatorAssigned, mJets, eostools::move(), mTracks, JetTracksAssociationDRVertex::produce(), JetTracksAssociationDRVertexAssigned::produce(), edm::Handle< T >::product(), edm::Event::put(), pvSrc, useAssigned, and electrons_cff::vertices.

35  {
37  fEvent.getByToken (mJets, jets_h);
39  fEvent.getByToken (mTracks, tracks_h);
40 
41  auto jetTracks = std::make_unique<reco::JetTracksAssociation::Container>(reco::JetRefBaseProd(jets_h));
42 
43  // format inputs
44  std::vector <edm::RefToBase<reco::Jet> > allJets;
45  allJets.reserve (jets_h->size());
46  for (unsigned i = 0; i < jets_h->size(); ++i) allJets.push_back (jets_h->refAt(i));
47  std::vector <reco::TrackRef> allTracks;
48  allTracks.reserve (tracks_h->size());
49  // run algo
50  for (unsigned i = 0; i < tracks_h->size(); ++i) {
51  allTracks.push_back (reco::TrackRef (tracks_h, i));
52  }
53  if ( !useAssigned ) {
54  mAssociator.produce (&*jetTracks, allJets, allTracks);
55  } else {
57  fEvent.getByToken(pvSrc,pvHandle);
58  const reco::VertexCollection & vertices = *pvHandle.product();
59  mAssociatorAssigned.produce (&*jetTracks, allJets, allTracks,vertices);
60  }
61 
62 
63  // store output
64  fEvent.put(std::move(jetTracks));
65 }
std::vector< reco::JetBaseRef > allJets(const Container &)
fill list of all jets associated with values. Return # of jets in the list
edm::EDGetTokenT< reco::TrackCollection > mTracks
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:136
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:519
std::vector< Vertex > VertexCollection
collection of Vertex objects
Definition: VertexFwd.h:9
edm::EDGetTokenT< edm::View< reco::Jet > > mJets
void produce(reco::JetTracksAssociation::Container *fAssociation, const std::vector< edm::RefToBase< reco::Jet > > &fJets, const std::vector< reco::TrackRef > &fTracks, const reco::VertexCollection &vertices) const
edm::RefToBaseProd< reco::Jet > JetRefBaseProd
Definition: JetCollection.h:14
edm::EDGetTokenT< reco::VertexCollection > pvSrc
if true, use the track/jet association with vertex assignment to tracks
void produce(reco::JetTracksAssociation::Container *fAssociation, const std::vector< edm::RefToBase< reco::Jet > > &fJets, const std::vector< reco::TrackRef > &fTracks) const
JetTracksAssociationDRVertexAssigned mAssociatorAssigned
T const * product() const
Definition: Handle.h:81
JetTracksAssociationDRVertex mAssociator
def move(src, dest)
Definition: eostools.py:510

Member Data Documentation

JetTracksAssociationDRVertex JetTracksAssociatorAtVertex::mAssociator
private

Definition at line 31 of file JetTracksAssociatorAtVertex.h.

Referenced by produce().

JetTracksAssociationDRVertexAssigned JetTracksAssociatorAtVertex::mAssociatorAssigned
private

Definition at line 32 of file JetTracksAssociatorAtVertex.h.

Referenced by produce().

edm::EDGetTokenT<edm::View <reco::Jet> > JetTracksAssociatorAtVertex::mJets
private

Definition at line 27 of file JetTracksAssociatorAtVertex.h.

Referenced by JetTracksAssociatorAtVertex(), and produce().

int JetTracksAssociatorAtVertex::mTrackQuality
private

Definition at line 30 of file JetTracksAssociatorAtVertex.h.

edm::EDGetTokenT<reco::TrackCollection> JetTracksAssociatorAtVertex::mTracks
private

Definition at line 28 of file JetTracksAssociatorAtVertex.h.

Referenced by JetTracksAssociatorAtVertex(), and produce().

edm::EDGetTokenT<reco::VertexCollection> JetTracksAssociatorAtVertex::pvSrc
private

if true, use the track/jet association with vertex assignment to tracks

Definition at line 34 of file JetTracksAssociatorAtVertex.h.

Referenced by JetTracksAssociatorAtVertex(), and produce().

bool JetTracksAssociatorAtVertex::useAssigned
private

Definition at line 33 of file JetTracksAssociatorAtVertex.h.

Referenced by JetTracksAssociatorAtVertex(), and produce().