PF_PU_AssoMap Class Reference

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

Inheritance diagram for PF_PU_AssoMap:

Public Member Functions
	PF_PU_AssoMap (const edm::ParameterSet &)
	~PF_PU_AssoMap () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndRuns () const final
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< TrackToVertexAssMap >	CreateTrackToVertexMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
std::unique_ptr< VertexToTrackAssMap >	CreateVertexToTrackMap (edm::Handle< reco::TrackCollection >, const edm::EventSetup &)
virtual void	GetInputCollections (edm::Event &, const edm::EventSetup &)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC)
	PF_PU_AssoMapAlgos (const edm::ParameterSet &, edm::ConsumesCollector &)
std::unique_ptr< TrackToVertexAssMap >	SortAssociationMap (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::TrackCollection >	token_TrackCollection_

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, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache
Protected Member Functions inherited from PF_PU_AssoMapAlgos
std::vector< reco::VertexRef >	CreateVertexVector (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 tracks and their particular most probable vertex with a quality of this association

Description: Produces collection of tracks associated to the first vertex based on the pf_pu Association Map

Description: Produces a collection of PFCandidates associated to the first vertex based on the association map

Definition at line 40 of file PF_PU_AssoMap.h.

Constructor & Destructor Documentation

PF_PU_AssoMap::PF_PU_AssoMap ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 26 of file PF_PU_AssoMap.cc.

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

                                                           : PF_PU_AssoMapAlgos(iConfig, consumesCollector()) {
  //now do what ever other initialization is needed

  input_AssociationType_ = iConfig.getParameter<edm::InputTag>("AssociationType");

  token_TrackCollection_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("TrackCollection"));

  //register your products

  if (input_AssociationType_.label() == "TracksToVertex") {
    produces<TrackToVertexAssMap>();
  } else {
    if (input_AssociationType_.label() == "VertexToTracks") {
      produces<VertexToTrackAssMap>();
    } else {
      if (input_AssociationType_.label() == "Both") {
        produces<TrackToVertexAssMap>();
        produces<VertexToTrackAssMap>();
      } else {
        std::cout << "No correct InputTag for AssociationType!" << std::endl;
        std::cout << "Won't produce any AssociationMap!" << std::endl;
      }
    }
  }
}

PF_PU_AssoMap::~PF_PU_AssoMap ( )

override

Definition at line 52 of file PF_PU_AssoMap.cc.

                              {
  // do anything here that needs to be done at destruction time
  // (e.g. close files, deallocate resources etc.)
}

Member Function Documentation

void PF_PU_AssoMap::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 87 of file PF_PU_AssoMap.cc.

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

                                                                               {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

void PF_PU_AssoMap::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivate

Definition at line 62 of file PF_PU_AssoMap.cc.

References PF_PU_AssoMapAlgos::createMappings(), edm::Event::getByToken(), PF_PU_AssoMapAlgos::GetInputCollections(), input_AssociationType_, edm::InputTag::label(), eostools::move(), edm::Event::put(), PF_PU_AssoMapAlgos::SortAssociationMap(), and token_TrackCollection_.

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

                                                                         {
  using namespace edm;
  using namespace std;
  using namespace reco;

  //get the input track collection
  Handle<TrackCollection> trkcollH;
  iEvent.getByToken(token_TrackCollection_, trkcollH);

  string asstype = input_AssociationType_.label();

  PF_PU_AssoMapAlgos::GetInputCollections(iEvent, iSetup);

  if (asstype == "TracksToVertex" || asstype == "VertexToTracks" || asstype == "Both") {
    auto mappings = createMappings(trkcollH, iSetup);
    if (asstype == "TracksToVertex" || asstype == "Both") {
      iEvent.put(SortAssociationMap(&(*mappings.first), trkcollH));
    }
    if (asstype == "VertexToTracks" || asstype == "Both") {
      iEvent.put(std::move(mappings.second));
    }
  }
}

Member Data Documentation

edm::InputTag PF_PU_AssoMap::input_AssociationType_

private

Definition at line 52 of file PF_PU_AssoMap.h.

Referenced by PF_PU_AssoMap(), and produce().

edm::EDGetTokenT<reco::TrackCollection> PF_PU_AssoMap::token_TrackCollection_

private

Definition at line 54 of file PF_PU_AssoMap.h.

Referenced by PF_PU_AssoMap(), and produce().