CMS 3D CMS Logo

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

#include <CommonTools/RecoUtils/plugins/PFCand_AssoMap.cc>

Inheritance diagram for PFCand_AssoMap:
edm::stream::EDProducer<> PFCand_AssoMapAlgos PF_PU_AssoMapAlgos

Public Member Functions

 PFCand_AssoMap (const edm::ParameterSet &)
 
 ~PFCand_AssoMap () override
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
bool hasAbilityToProduceInLumis () const final
 
bool hasAbilityToProduceInRuns () const final
 
- Public Member Functions inherited from PFCand_AssoMapAlgos
std::pair< std::unique_ptr< PFCandToVertexAssMap >, std::unique_ptr< VertexToPFCandAssMap > > createMappings (edm::Handle< reco::PFCandidateCollection > pfCandH, const edm::EventSetup &iSetup)
 
std::unique_ptr< PFCandToVertexAssMapCreatePFCandToVertexMap (edm::Handle< reco::PFCandidateCollection >, const edm::EventSetup &)
 
std::unique_ptr< VertexToPFCandAssMapCreateVertexToPFCandMap (edm::Handle< reco::PFCandidateCollection >, const edm::EventSetup &)
 
void GetInputCollections (edm::Event &, const edm::EventSetup &) override
 
 PFCand_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &&)
 
std::unique_ptr< PFCandToVertexAssMapSortPFCandAssociationMap (PFCandToVertexAssMap *, edm::EDProductGetter const *getter)
 
- Public Member Functions inherited from PF_PU_AssoMapAlgos
std::pair< std::unique_ptr< TrackToVertexAssMap >, std::unique_ptr< VertexToTrackAssMap > > createMappings (edm::Handle< reco::TrackCollection > trkcollH, const edm::EventSetup &iSetup)
 
std::unique_ptr< TrackToVertexAssMapCreateTrackToVertexMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
 
std::unique_ptr< VertexToTrackAssMapCreateVertexToTrackMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
 
 PF_PU_AssoMapAlgos (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
 
 PF_PU_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &)
 
std::unique_ptr< TrackToVertexAssMapSortAssociationMap (TrackToVertexAssMap *, edm::Handle< reco::TrackCollection >)
 
template<typename TREF >
VertexRef TrackWeightAssociation (const TREF &trackRef, const std::vector< reco::VertexRef > &vtxcollV)
 
virtual ~PF_PU_AssoMapAlgos ()(false)
 

Static Public Member Functions

static void fillDescriptions (edm::ConfigurationDescriptions &descriptions)
 

Private Member Functions

void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

edm::InputTag input_AssociationType_
 
edm::EDGetTokenT< reco::PFCandidateCollectiontoken_PFCandidates_
 

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
 
- Protected Member Functions inherited from PF_PU_AssoMapAlgos
std::vector< reco::VertexRefCreateVertexVector (edm::Handle< reco::VertexCollection >)
 
int DefineQuality (int, int, double)
 
void EraseVertex (std::vector< reco::VertexRef > &, reco::VertexRef)
 
VertexStepPair FindAssociation (const reco::TrackRef &, const std::vector< reco::VertexRef > &, edm::ESHandle< MagneticField >, const edm::EventSetup &, edm::Handle< reco::BeamSpot >, int)
 

Detailed Description

Description: Produces a map with association between pf candidates and their particular most probable vertex with a quality of this association

Definition at line 37 of file PFCand_AssoMap.h.

Constructor & Destructor Documentation

PFCand_AssoMap::PFCand_AssoMap ( const edm::ParameterSet iConfig)
explicit

Definition at line 33 of file PFCand_AssoMap.cc.

References gather_cfg::cout, edm::ParameterSet::getParameter(), input_AssociationType_, edm::InputTag::label(), and token_PFCandidates_.

33  :PFCand_AssoMapAlgos(iConfig, consumesCollector())
34 {
35 
36  //now do what ever other initialization is needed
37 
38  input_AssociationType_ = iConfig.getParameter<edm::InputTag>("AssociationType");
39 
40  token_PFCandidates_ = consumes<reco::PFCandidateCollection>(iConfig.getParameter<edm::InputTag>("PFCandidateCollection"));
41 
42  //register your products
43 
44  if ( input_AssociationType_.label() == "PFCandsToVertex" ) {
45  produces<PFCandToVertexAssMap>();
46  } else {
47  if ( input_AssociationType_.label() == "VertexToPFCands" ) {
48  produces<VertexToPFCandAssMap>();
49  } else {
50  if ( input_AssociationType_.label() == "Both" ) {
51  produces<PFCandToVertexAssMap>();
52  produces<VertexToPFCandAssMap>();
53  } else {
54  std::cout << "No correct InputTag for AssociationType!" << std::endl;
55  std::cout << "Won't produce any AssociationMap!" << std::endl;
56  }
57  }
58  }
59 
60 }
T getParameter(std::string const &) const
edm::InputTag input_AssociationType_
edm::EDGetTokenT< reco::PFCandidateCollection > token_PFCandidates_
std::string const & label() const
Definition: InputTag.h:36
PFCand_AssoMapAlgos(const edm::ParameterSet &, edm::ConsumesCollector &&)
PFCand_AssoMap::~PFCand_AssoMap ( )
override

Definition at line 63 of file PFCand_AssoMap.cc.

64 {
65 }

Member Function Documentation

void PFCand_AssoMap::fillDescriptions ( edm::ConfigurationDescriptions descriptions)
static

Definition at line 103 of file PFCand_AssoMap.cc.

References edm::ConfigurationDescriptions::addDefault(), DEFINE_FWK_MODULE, and edm::ParameterSetDescription::setUnknown().

103  {
104  //The following says we do not know what parameters are allowed so do no validation
105  // Please change this to state exactly what you do use, even if it is no parameters
107  desc.setUnknown();
108  descriptions.addDefault(desc);
109 }
void addDefault(ParameterSetDescription const &psetDescription)
void PFCand_AssoMap::produce ( edm::Event iEvent,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 74 of file PFCand_AssoMap.cc.

References PFCand_AssoMapAlgos::createMappings(), edm::Event::getByToken(), PFCand_AssoMapAlgos::GetInputCollections(), input_AssociationType_, edm::InputTag::label(), eostools::move(), edm::Event::productGetter(), edm::Event::put(), PFCand_AssoMapAlgos::SortPFCandAssociationMap(), and token_PFCandidates_.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

75 {
76 
77  using namespace edm;
78  using namespace std;
79  using namespace reco;
80 
81  //get the input pfCandidateCollection
83  iEvent.getByToken(token_PFCandidates_,pfCandH);
84 
85  string asstype = input_AssociationType_.label();
86 
88 
89  if (asstype == "PFCandsToVertex" || asstype == "VertexToPFCands" || asstype == "Both") {
90  auto mappings = createMappings(pfCandH, iSetup);
91  if (asstype == "PFCandsToVertex" || asstype == "Both") {
92  iEvent.put( SortPFCandAssociationMap( &(*mappings.first), &iEvent.productGetter() ) );
93  }
94  if (asstype == "VertexToPFCands" || asstype == "Both") {
95  iEvent.put( std::move(mappings.second));
96  }
97  }
98 
99 }
void GetInputCollections(edm::Event &, const edm::EventSetup &) override
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:125
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:517
EDProductGetter const & productGetter() const
Definition: Event.cc:93
std::pair< std::unique_ptr< PFCandToVertexAssMap >, std::unique_ptr< VertexToPFCandAssMap > > createMappings(edm::Handle< reco::PFCandidateCollection > pfCandH, const edm::EventSetup &iSetup)
edm::InputTag input_AssociationType_
edm::EDGetTokenT< reco::PFCandidateCollection > token_PFCandidates_
std::string const & label() const
Definition: InputTag.h:36
fixed size matrix
HLT enums.
std::unique_ptr< PFCandToVertexAssMap > SortPFCandAssociationMap(PFCandToVertexAssMap *, edm::EDProductGetter const *getter)
def move(src, dest)
Definition: eostools.py:511

Member Data Documentation

edm::InputTag PFCand_AssoMap::input_AssociationType_
private

Definition at line 49 of file PFCand_AssoMap.h.

Referenced by PFCand_AssoMap(), and produce().

edm::EDGetTokenT<reco::PFCandidateCollection> PFCand_AssoMap::token_PFCandidates_
private

Definition at line 51 of file PFCand_AssoMap.h.

Referenced by PFCand_AssoMap(), and produce().